haystack-core-integrations icon indicating copy to clipboard operation
haystack-core-integrations copied to clipboard

feat: add async support for Chroma

Open Ryzhtus opened this issue 7 months ago • 2 comments

Related Issues

  • fixes #1389

Proposed Changes:

added async version of the following ChromaDocumentStore methods:

  • _ensure_initialized_async
  • count_documents_async
  • filter_documents_async
  • write_documents_async
  • delete_documents_async
  • search_async
  • search_embeddings_async

added run_async methods for ChromaQueryTextRetriever and ChromaEmbeddingRetriever

How did you test it?

  • added test_document_store_async.py that 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

Ryzhtus avatar May 03 '25 17:05 Ryzhtus