2、Chroma 快速开始

Chroma 快速开始

Chroma 是一个用于构建带有嵌入的 AI 应用程序的数据库。它内置了您开始使用所需的一切，并在您的计算机上运行。托管版本即将推出！

1. 安装 Chroma 客户端

pip install chromadb

2. 导入 Chroma

import chromadb
chroma_client = chromadb.Client()

3. 创建 Collection

集合是您存储嵌入、文档和任何其他元数据的地方。您可以创建一个具有以下名称的集合：

collection = chroma_client.create_collection(name="my_collection")

4. 添加一些文本文档到

Chroma 将存储您的文本，并自动处理标记化、嵌入和索引。

collection.add(
    documents=["This is a document", "This is another document"],
    metadatas=[{"source": "my_source"}, {"source": "my_source"}],
    ids=["id1", "id2"]
)

如果您已经自己生成了嵌入，则可以直接将它们加载到：

collection.add(
    embeddings=[[1.2, 2.3, 4.5], [6.7, 8.2, 9.2]],
    documents=["This is a document", "This is another document"],
    metadatas=[{"source": "my_source"}, {"source": "my_source"}],
    ids=["id1", "id2"]
)

5.查询

您可以使用查询文本列表来查询集合，Chroma 将返回n最相似的结果。就是这么简单！

results = collection.query(
    query_texts=["This is a query document"],
    n_results=2
)

默认情况下，Chroma 中存储的数据是短暂的，因此可以轻松创建脚本原型。让 Chroma 持久化很容易，这样您就可以重复使用您创建的每个集合，并在以后向其中添加更多文档。当您启动客户端时，它会自动加载您的数据，并在您关闭客户端时自动保存。查看使用指南了解更多信息。