LocalAI
信息
langchain-localai
是 LocalAI 的第三方集成包。它提供了一种在 Langchain 中使用 LocalAI 服务的简单方法。
源代码可在 Github 上找到
让我们加载 LocalAI 嵌入类。为了使用 LocalAI 嵌入类,您需要将 LocalAI 服务托管在某个地方并配置嵌入模型。请参阅 https://localai.io/basics/getting_started/index.html 和 https://localai.io/features/embeddings/index.html 上的文档。
%pip install -U langchain-localai
from langchain_localai import LocalAIEmbeddings
embeddings = LocalAIEmbeddings(
openai_api_base="https://127.0.0.1:8080", model="embedding-model-name"
)
text = "This is a test document."
query_result = embeddings.embed_query(text)
doc_result = embeddings.embed_documents([text])
让我们加载使用第一代模型(例如 text-search-ada-doc-001/text-search-ada-query-001)的 LocalAI 嵌入类。注意:这些不是推荐的模型 - 请参阅此处
from langchain_community.embeddings import LocalAIEmbeddings
API 参考:LocalAIEmbeddings
embeddings = LocalAIEmbeddings(
openai_api_base="https://127.0.0.1:8080", model="embedding-model-name"
)
text = "This is a test document."
query_result = embeddings.embed_query(text)
doc_result = embeddings.embed_documents([text])
import os
# if you are behind an explicit proxy, you can use the OPENAI_PROXY environment variable to pass through
os.environ["OPENAI_PROXY"] = "http://proxy.yourcompany.com:8080"