Skip to main content

rag-google-cloud-vertexai-search

此模板是一个应用程序,利用了 Google Vertex AI Search,这是一个基于机器学习的搜索服务,以及 PaLM 2 for Chat (chat-bison)。该应用程序使用检索链根据您的文档回答问题。

有关使用 Vertex AI Search 构建 RAG 应用程序的更多背景信息,请查看 这里

环境设置

在使用此模板之前,请确保您已通过 Vertex AI Search 进行身份验证。请参阅身份验证指南:这里

您还需要创建:

适合测试该模板的数据集是 Alphabet 财报,您可以在 这里 找到。数据也可以在 gs://cloud-samples-data/gen-app-builder/search/alphabet-investor-pdfs 获取。

设置以下环境变量:

  • GOOGLE_CLOUD_PROJECT_ID - 您的 Google Cloud 项目 ID。
  • DATA_STORE_ID - Vertex AI Search 中数据存储的 ID,这是一个 36 字符的字母数字值,可以在数据存储详细信息页面找到。
  • MODEL_TYPE - Vertex AI Search 的模型类型。

使用方法

要使用此包,您首先需要安装 LangChain CLI:

pip install -U langchain-cli

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

langchain app new my-app --package rag-google-cloud-vertexai-search

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

langchain app add rag-google-cloud-vertexai-search

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

from rag_google_cloud_vertexai_search.chain import chain as rag_google_cloud_vertexai_search_chain

add_routes(app, rag_google_cloud_vertexai_search_chain, path="/rag-google-cloud-vertexai-search")

(可选)现在让我们配置 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-google-cloud-vertexai-search/playground

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

from langserve.client import RemoteRunnable

runnable = RemoteRunnable("http://localhost:8000/rag-google-cloud-vertexai-search")

此页面是否有帮助?


您还可以留下详细的反馈 在 GitHub 上