跳到主要內容
Open In ColabOpen on GitHub

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 參數,詳情請參閱文件

預設情況下,會使用 text-search-query 的最新版本,該版本來自參數 folder_idYC_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]

此頁面是否對您有幫助?