使用 Deep Lake 构建自查询向量库的实践

本文链接：https://blog.csdn.net/qahaj/article/details/146485032

### 技术背景介绍

Deep Lake 是一个用于构建 AI 应用程序的多模态数据库，支持存储和查询向量、图像、文本、视频等多种数据类型，特别适合与大语言模型 (LLMs) 和 LangChain 集成使用。它提供了流数据实时处理能力，适用于 PyTorch 和 TensorFlow 平台。在本次演示中，我们将创建一个 Deep Lake 自查询向量库，并利用 SelfQueryRetriever 来查询库中的数据。

### 核心原理解析

自查询向量库 (SelfQueryRetriever) 是一种创新的方法，通过利用文档的元数据和描述性信息，使用户能够使用自然语言查询来检索相关数据。Deep Lake 提供了一个灵活的数据存储结构，可以轻松存放各种类型的 AI 数据，并结合 OpenAIEmbeddings 提供强大的向量化能力。

### 代码实现演示

我们将首先设置 Deep Lake 向量库并输入一些电影摘要数据。

```python
# 安装必要的库
%pip install --upgrade --quiet lark
%pip install --upgrade --quiet libdeeplake

import getpass
import os

# 设置 API Keys
os.environ["OPENAI_API_KEY"] = getpass.getpass("OpenAI API Key:")
os.environ["ACTIVELOOP_TOKEN"] = getpass.getpass("Activeloop token:")

from langchain_community.vectorstores import DeepLake
from langchain_core.documents import Document
from langchain_openai import OpenAIEmbeddings

embeddings = OpenAIEmbeddings()

# 创建文档列表
docs = [
    Document(
        page_content="A bunch of scientists bring back dinosaurs and mayhem breaks loose",
        metadata={"year": 1993, "rating": 7.7, "genre": "science fiction"},
    ),
    Document(
        page_content="Leo DiCaprio gets lost in a dream within a dream within a dream within a ...",
        metadata={"year": 2010, "director": "Christopher Nolan", "rating": 8.2},
    ),
    # 其他文档省略...
]

username_or_org = "<USERNAME_OR_ORG>"
# 创建 Deep Lake 向量库
vectorstore = DeepLake.from_documents(
    docs,
    embeddings,
    dataset_path=f"hub://{username_or_org}/self_queery",
    overwrite=True,
)

print("Your Deep Lake dataset has been successfully created!")