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

API 参考：LLMChain | YandexGPT | 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.'

相关

• LLM 概念指南
• LLM 操作指南