Skip to main content

RAG自查询

该模板使用自查询检索技术执行RAG。其主要思想是让语言模型将非结构化查询转换为结构化查询。详细信息请参阅文档

环境设置

在此模板中,我们将使用OpenAI模型和Elasticsearch向量存储,但该方法可以推广到所有语言模型/聊天模型和多种向量存储

设置OPENAI_API_KEY环境变量以访问OpenAI模型。

要连接到您的Elasticsearch实例,请使用以下环境变量:

export ELASTIC_CLOUD_ID = <ClOUD_ID>
export ELASTIC_USERNAME = <ClOUD_USERNAME>
export ELASTIC_PASSWORD = <ClOUD_PASSWORD>

对于使用Docker进行本地开发,请使用:

export ES_URL = "http://localhost:9200"
docker run -p 9200:9200 -e "discovery.type=single-node" -e "xpack.security.enabled=false" -e "xpack.security.http.ssl.enabled=false" docker.elastic.co/elasticsearch/elasticsearch:8.9.0

用法

要使用此软件包,您应首先安装LangChain CLI:

pip install -U "langchain-cli[serve]"

要创建一个新的LangChain项目并将其安装为唯一软件包,您可以执行:

langchain app new my-app --package rag-self-query

如果要将其添加到现有项目中,只需运行:

langchain app add rag-self-query

并将以下代码添加到您的server.py文件中:

from rag_self_query import chain
add_routes(app, chain, path="/rag-elasticsearch")

要使用示例数据填充向量存储,请从目录的根部运行:

python ingest.py

(可选)现在让我们配置LangSmith。

LangSmith将帮助我们跟踪、监视和调试LangChain应用程序。

您可以在此处注册LangSmith。

如果您没有访问权限,可以跳过此部分。

export LANGCHAIN_TRACING_V2=true
export LANGCHAIN_API_KEY=<your-api-key>
export LANGCHAIN_PROJECT=<your-project> # 如果未指定,默认为"default"

如果您在此目录中,则可以通过以下方式直接启动LangServe实例:

langchain serve

这将启动FastAPI应用程序,服务器在本地运行,地址为

http://localhost:8000

我们可以在http://127.0.0.1:8000/docs查看所有模板。

我们可以在http://127.0.0.1:8000/rag-elasticsearch/playground访问playground。

我们可以通过以下代码从代码中访问模板:

from langserve.client import RemoteRunnable
runnable = RemoteRunnable("http://localhost:8000/rag-self-query")

Was this page helpful?


You can leave detailed feedback on GitHub.