跳至主要内容

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_idYC_FOLDER_ID 环境变量中指定的文件夹中的最新版本的 text-search-query

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]

此页面是否有帮助?


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