Skip to main content
Open In ColabOpen on GitHub

YandexGPT

本指南将介绍如何在 Langchain 中使用 YandexGPT embeddings 模型。

要使用，您应该已安装 yandexcloud python 包。

%pip install --upgrade --quiet  yandexcloud

首先，您需要创建服务账户，并为其分配ai.languageModels.user角色。

接下来，您有两种认证方式：

  • IAM token。 您可以在构造函数参数 iam_token 或环境变量 YC_IAM_TOKEN 中指定 token。
  • API key 您可以在构造函数参数 api_key 或环境变量 YC_API_KEY 中指定 key。

要指定模型，您可以使用 model_uri 参数，更多详情请参阅文档。

默认情况下，将使用 folder_id 参数或 YC_FOLDER_ID 环境变量中指定的文件夹内的最新版本的 text-search-query。

from langchain_community.embeddings.yandex import YandexGPTEmbeddings
API Reference: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]