Skip to main content

rag-gemini-multi-modal

多模态 LLMs 使得视觉助手能够对图像进行问答。

此模板为幻灯片创建一个视觉助手,幻灯片通常包含图表或图形等视觉内容。

它使用 OpenCLIP 嵌入将所有幻灯片图像嵌入并存储在 Chroma 中。

给定一个问题,相关的幻灯片将被检索并传递给 Google Gemini 进行答案合成。

输入

/docs 目录中提供一个 PDF 格式的幻灯片演示文稿。

默认情况下,此模板包含关于 DataDog(一家上市科技公司)第三季度收益的幻灯片演示文稿。

可以提出的示例问题包括:

Datadog 有多少客户?
Datadog 平台在 FY20、FY21 和 FY22 的同比增长百分比是多少?

要创建幻灯片演示文稿的索引,请运行:

poetry install
python ingest.py

存储

此模板将使用 OpenCLIP 多模态嵌入来嵌入图像。

您可以选择不同的嵌入模型选项(请参见结果 这里)。

第一次运行应用时,它会自动下载多模态嵌入模型。

默认情况下,LangChain 将使用具有适中性能但内存需求较低的嵌入模型 ViT-H-14

您可以在 rag_chroma_multi_modal/ingest.py 中选择其他 OpenCLIPEmbeddings 模型:

vectorstore_mmembd = Chroma(
collection_name="multi-modal-rag",
persist_directory=str(re_vectorstore_path),
embedding_function=OpenCLIPEmbeddings(
model_name="ViT-H-14", checkpoint="laion2b_s32b_b79k"
),
)

LLM

该应用将使用多模态嵌入检索图像,并将其传递给 Google Gemini。

环境设置

设置您的 GOOGLE_API_KEY 环境变量以访问 Gemini。

使用方法

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

pip install -U langchain-cli

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

langchain app new my-app --package rag-gemini-multi-modal

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

langchain app add rag-gemini-multi-modal

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

from rag_gemini_multi_modal import chain as rag_gemini_multi_modal_chain

add_routes(app, rag_gemini_multi_modal_chain, path="/rag-gemini-multi-modal")

(可选)现在让我们配置 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-gemini-multi-modal/playground 访问游乐场

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

from langserve.client import RemoteRunnable

runnable = RemoteRunnable("http://localhost:8000/rag-gemini-multi-modal")

此页面是否有帮助?


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