LLMRails

讓我們載入 LLMRails Embeddings 類別。

要使用 LLMRails 嵌入，您需要透過引數傳遞 API 金鑰，或使用 LLM_RAILS_API_KEY 金鑰在環境中設定。 要取得 API 金鑰，您需要在 https://console.llmrails.com/signup 註冊，然後前往 https://console.llmrails.com/api-keys，並在平台上建立金鑰後從那裡複製金鑰。

from langchain_community.embeddings import LLMRailsEmbeddings

API 參考：LLMRailsEmbeddings

embeddings = LLMRailsEmbeddings(model="embedding-english-v1")  # or embedding-multi-v1

text = "This is a test document."

要產生嵌入，您可以查詢單個文本，也可以查詢文本列表。

query_result = embeddings.embed_query(text)
query_result[:5]

[-0.09996652603149414,
 0.015568195842206478,
 0.17670190334320068,
 0.16521021723747253,
 0.21193109452724457]

doc_result = embeddings.embed_documents([text])
doc_result[0][:5]

[-0.04242777079343796,
 0.016536075621843338,
 0.10052520781755447,
 0.18272875249385834,
 0.2079043835401535]

相關

• 嵌入模型 概念指南
• 嵌入模型 操作指南