Skip to main content
Open In ColabOpen on GitHub

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 Reference: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]

这个页面有帮助吗?