Skip to main content

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

相关


此页面是否有帮助?


您还可以留下详细的反馈 在 GitHub 上