跳到主要内容
Open In ColabOpen on GitHub

ChatYandexGPT

本 Notebook 介绍了如何将 Langchain 与 YandexGPT 聊天模型结合使用。

要使用它,您需要安装 yandexcloud Python 包。

%pip install --upgrade --quiet  yandexcloud

首先,您应该创建服务账户并授予 ai.languageModels.user 角色。

接下来,您有两种认证方式:

  • IAM 令牌。您可以在构造函数参数 iam_token 或环境变量 YC_IAM_TOKEN 中指定令牌。

  • API 密钥。您可以在构造函数参数 api_key 或环境变量 YC_API_KEY 中指定密钥。

要指定模型,您可以使用 model_uri 参数,详情请参阅文档

默认情况下,将使用 yandexgpt-lite 的最新版本,其文件夹由参数 folder_id 或环境变量 YC_FOLDER_ID 指定。

from langchain_community.chat_models import ChatYandexGPT
from langchain_core.messages import HumanMessage, SystemMessage
chat_model = ChatYandexGPT()
answer = chat_model.invoke(
[
SystemMessage(
content="You are a helpful assistant that translates English to French."
),
HumanMessage(content="I love programming."),
]
)
answer
AIMessage(content='Je adore le programmement.')