local_persistence.ipynb in /examples is giving embeddings as none

[shivankar-p]

I tried running examples/local_persistence.ipynb locally on my machine.

# Query the collection
results = collection.query(
    query_embeddings=[[1.1, 2.3, 3.2]],
    n_results=1,
)

print(results)

This cell is supposed to print

{'embeddings': [[[1.1, 2.3, 3.2]]], 'documents': [['doc5']], 'ids': [['id5']], 'metadatas': [[{'uri': 'img5.png', 'style': 'style1'}]], 'distances': [[0.0]]}' as per the given output.

But its printing

{'ids': [['id5']], 'embeddings': None, 'documents': [['doc5']], 'metadatas': [[{'uri': 'img5.png', 'style': 'style1'}]], 'distances': [[0.0]]}

Why is embeddings None here?

[jeffchuber]

@shivankar-p this is because .query by default does not return the embedding themselves, https://github.com/chroma-core/chroma/blob/main/chromadb/api/local.py#L245

This is a default for performance reasons. To get them to return, you can add an explicit include=['embeddings'] to your query.

There has been discussion elsewhere about changing this behavior to make it more intuitive. #300