Skip to main content
Open In ColabOpen on GitHub

YandexGPT

本笔记本介绍了如何使用 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 参数，更多详细信息请参阅 文档。

默认情况下，会从参数 folder_id 或环境变量 YC_FOLDER_ID 指定的文件夹中使用 yandexgpt-lite 的最新版本。

from langchain.chains import LLMChain
from langchain_community.llms import YandexGPT
from langchain_core.prompts import PromptTemplate
template = "What is the capital of {country}?"
prompt = PromptTemplate.from_template(template)
llm = YandexGPT()
llm_chain = LLMChain(prompt=prompt, llm=llm)
country = "Russia"

llm_chain.invoke(country)
'The capital of Russia is Moscow.'