haystack-core-integrations
haystack-core-integrations copied to clipboard
feat: add async support for Chroma
Related Issues
- fixes #1389
Proposed Changes:
added async version of the following ChromaDocumentStore methods:
_ensure_initialized_asynccount_documents_asyncfilter_documents_asyncwrite_documents_asyncdelete_documents_asyncsearch_asyncsearch_embeddings_async
added run_async methods for ChromaQueryTextRetriever and ChromaEmbeddingRetriever
How did you test it?
- added
test_document_store_async.pythat tests methods above
Notes for the reviewer
I still need to check a couple of tests in test_document_store_async.py and write tests for retrievers, will be done with next commits
Checklist
- I have read the contributors guidelines and the code of conduct
- I have updated the related issue with new insights and changes
- I added unit tests and updated the docstrings
- I've used one of the conventional commit types for my PR title:
fix:,feat:,build:,chore:,ci:,docs:,style:,refactor:,perf:,test:.