Skip to main content

rag-redis-multi-modal-multi-vector

多模态 LLM 使得能够进行图像问答的视觉助手成为可能。

此模板创建了一个用于幻灯片的视觉助手,幻灯片通常包含图表或图形等视觉元素。

它使用 GPT-4V 为每个幻灯片创建图像摘要,嵌入摘要并将其存储在 Redis 中。

在给定问题的情况下,相关的幻灯片被检索并传递给 GPT-4V 进行答案合成。

输入

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

默认情况下,此模板包含有关 NVIDIA 最近收益的幻灯片文档。

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

1/ H100 TensorRT 可以提高 LLama2 推理性能多少?
2/ 从 2020 年到 2023 年,GPU 加速应用程序的百分比变化是多少?

要创建幻灯片文档的索引,请运行:

poetry install
poetry shell
python ingest.py

存储

这里是模板将用于创建幻灯片索引的过程(见 blog):

  • 将幻灯片提取为图像集合
  • 使用 GPT-4V 对每个图像进行摘要
  • 使用文本嵌入将图像摘要嵌入,并链接到原始图像
  • 根据图像摘要与用户输入问题之间的相似性检索相关图像
  • 将这些图像传递给 GPT-4V 进行答案合成

Redis

此模板使用 Redis 来支持 MultiVectorRetriever,包括:

  • Redis 作为 VectorStore(用于存储 + 索引图像摘要嵌入)
  • Redis 作为 ByteStore(用于存储图像)

确保在 云端(免费)或使用 docker 在本地部署一个 Redis 实例。

这将为您提供一个可访问的 Redis 端点,您可以将其用作 URL。如果在本地部署,只需使用 redis://localhost:6379

LLM

该应用将根据文本输入与图像摘要(文本)之间的相似性检索图像,并将图像传递给GPT-4V进行答案合成。

环境设置

设置 OPENAI_API_KEY 环境变量以访问 OpenAI GPT-4V。

设置 REDIS_URL 环境变量以访问您的 Redis 数据库。

使用方法

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

pip install -U langchain-cli

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

langchain app new my-app --package rag-redis-multi-modal-multi-vector

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

langchain app add rag-redis-multi-modal-multi-vector

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

from rag_redis_multi_modal_multi_vector import chain as rag_redis_multi_modal_chain_mv

add_routes(app, rag_redis_multi_modal_chain_mv, path="/rag-redis-multi-modal-multi-vector")

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

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

from langserve.client import RemoteRunnable

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

此页面是否有帮助?


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