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。
<!--IMPORTS:[{"imported": "YandexGPTEmbeddings", "source": "langchain_community.embeddings.yandex", "docs": "https://python.langchain.com/api_reference/community/embeddings/langchain_community.embeddings.yandex.YandexGPTEmbeddings.html", "title": "YandexGPT"}]-->
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]
