chromadb Collection.add(Inincludes=[""]) 给出错误

Question

我是 chromadb 的新手。第一次学呢。这里只是一个简单的代码：

 collection.add(
    documents=["This is a document", "This is another document"],
    metadatas=[{"source": "test_source", "page": 1},{"source": "test_source2", "page": 2}, ],
    ids=["id1", "id2"],
    include=['distances', 'metadatas', 'embeddings', 'documents'],
)

我收到错误

TypeError: Collection.add() got an unexpected keyword argument 'include'

。需要一些帮助，如何在 chromadb 中显示嵌入。我已经安装了最新版本。

Answer 1

仅在“get”或“query”方法中使用 include 参数

collection.get(
    include=['distances', 'metadatas', 'embeddings', 'documents']
)


collection.query(
    query_texts=["some document"],
    include=['distances', 'metadatas', 'embeddings', 'documents']
)