[Feature Request]: query by distance

[javixeneize]

Describe the problem

ChromaDB can be configured to return n number of documents, but sometimes the documents dont relate to the query

Describe the proposed solution

It would be good to add an extra parameter to the query method to return documents only when a given similarity is reached. It is, if i set it to return 2 documents, and the distances are 1.05 and 1.78, if i set the new distance parameter to 1.5 for example, it will only return the first document

Alternatives considered

No response

Importance

would make my life easier

Additional Information

No response

[frieda-huang]

Hi @javixeneize! I agree, it’s a good feature to have. I’ve opened PR #3268 to address it and will wait for the Chroma team’s feedback.