我是 Chroma 数据库(以及相关的 python 库)的全新用户。
当我调用geta时collection,嵌入总是none,即使在将文档添加到集合时显式设置/定义了嵌入(因此生成嵌入不会成为问题 - 我不认为)。
对于以下代码(Python 3.10,chromadb 0.3.26),我希望在返回的字典中看到嵌入列表,但它是none.
import chromadb
chroma_client = chromadb.Client()
collection = chroma_client.create_collection(name="my_collection")
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"]
)
print(collection.get())
Run Code Online (Sandbox Code Playgroud)
输出:
{'ids': ['id1', 'id2'], 'embeddings': None, 'documents': ['This is a document', 'This is another document'], 'metadatas': [{'source': 'my_source'}, {'source': 'my_source'}]}
Run Code Online (Sandbox Code Playgroud)
query使用代替时不会出现同样的问题get:
print(collection.query(query_embeddings=[[1.2, 2.3, 4.4]], include=["embeddings"]))
Run Code Online (Sandbox Code Playgroud)
输出:
{'ids': …Run Code Online (Sandbox Code Playgroud)