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.'