Skip to main content
本文介绍如何在 LangChain 中使用 YandexGPT 嵌入模型。 使用前,你需要安装 yandexcloud Python 包。
pip install -qU  yandexcloud
首先,你需要创建一个服务账号,并赋予其 ai.languageModels.user 角色。 接下来,你有两种身份验证方式可供选择:
  • IAM 令牌。 你可以在构造参数 iam_token 或环境变量 YC_IAM_TOKEN 中指定令牌。
  • API 密钥 你可以在构造参数 api_key 或环境变量 YC_API_KEY 中指定密钥。
要指定模型,可以使用 model_uri 参数,详情请参阅文档 默认情况下,将使用由 folder_id 参数或 YC_FOLDER_ID 环境变量所指定文件夹中 text-search-query 的最新版本。
from langchain_community.embeddings.yandex import YandexGPTEmbeddings
embeddings = YandexGPTEmbeddings()
text = "This is a test document."
query_result = embeddings.embed_query(text)
doc_result = embeddings.embed_documents([text])
query_result[:5]
[-0.021392822265625,
 0.096435546875,
 -0.046966552734375,
 -0.0183258056640625,
 -0.00555419921875]
doc_result[0][:5]
[-0.021392822265625,
 0.096435546875,
 -0.046966552734375,
 -0.0183258056640625,
 -0.00555419921875]