Skip to main content
加载 LLMRails 嵌入类。 要使用 LLMRails 嵌入,您需要通过参数传入 API 密钥,或在环境变量中设置 LLM_RAILS_API_KEY。 要获取 API 密钥,请先在 console.llmrails.com/signup 注册账号,然后前往 console.llmrails.com/api-keys,在平台上创建一个密钥后复制使用。
from langchain_community.embeddings import 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]