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_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]