Skip to main content
Open In Colab在 GitHub 上打开

YandexGPT

本笔记本介绍了如何将 Langchain 与 YandexGPT 嵌入模型结合使用。

要使用yandexcloudpython 软件包。

%pip install --upgrade --quiet  yandexcloud

首先，您应该使用ai.languageModels.user角色。

接下来，您有两个身份验证选项：

  • IAM 令牌。 您可以在 constructor 参数中指定令牌iam_token或在环境变量YC_IAM_TOKEN.
  • API 密钥您可以在 constructor 参数中指定 keyapi_key或在环境变量YC_API_KEY.

指定可以使用model_uri参数，请参阅文档了解更多详细信息。

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

from langchain_community.embeddings.yandex import YandexGPTEmbeddings
API 参考：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]