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
环境变量中指定的文件夹中的最新版 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]