将文档关联的嵌入添加到ChromaDB中的集合

本教程将指导您如何在ChromaDB中向集合添加文档关联的嵌入。

1. 创建集合

client = chromadb.Client()
morpheus_collection = client.create_collection(name="morpheus")
​

我们首先初始化ChromaDB客户端，并创建一个名为morpheus的集合。

2. 添加文档关联的嵌入

morpheus_collection.add(
    documents=[
        "What if I told you everything you knew was a lie.",
        "Welcome to the real world.",
    ],
    embeddings=[[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]],
    ids=["quote_1", "quote_2"],
)
​

我们使用add方法向集合中添加了来自角色墨菲斯的两句名言。每个文档都与以下内容相关联：

• 唯一的ID
• 嵌入（一组数字，用于表示文档的语义内容）

⚠️ 重要提示：
本示例中使用的嵌入（[0.1, 0.2, 0.3]和[0.4, 0.5, 0.6]）纯属假设，仅用于说明目的。在实际场景中，您应使用由适合您用例的嵌入模型或算法生成的实际嵌入。

3. 统计集合中的项目数量

item_count = morpheus_collection.count()
print(f"Count of items in collection: {item_count}")
​

我们可以使用count method. Since we have added two documents to our collection, the count returns 2方法统计集合中的项目数量。

4. 从集合中检索项目

items = morpheus_collection.get()
print(items)
​

我们可以使用get method. By default, this will return a dictionary with the following information for each item in the collection:

• IDs
• Metadatas (if included)
• Documents

Please refer to the add_doc_emb.py文件从集合中检索所有项目。以下是本教程中使用的完整Python代码。

免责声明：
本文档采用基于机器的 AI 翻译服务进行翻译。尽管我们力求准确，但请注意，自动翻译可能存在错误或不准确之处。应以原文语言版本的文档作为权威依据。如需获取关键信息，建议使用专业的人工翻译。对于因使用本翻译而产生的任何误解或误读，我们概不负责。