Skip to content

feat: add async support for Chroma #1697

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

Ryzhtus
Copy link
Contributor

@Ryzhtus Ryzhtus commented May 3, 2025

Related Issues

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 Ryzhtus requested a review from a team as a code owner May 3, 2025 17:50
@Ryzhtus Ryzhtus requested review from anakin87 and removed request for a team May 3, 2025 17:50
@github-actions github-actions bot added integration:chroma type:documentation Improvements or additions to documentation labels May 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
integration:chroma type:documentation Improvements or additions to documentation
Projects
None yet
Development

Successfully merging this pull request may close these issues.

add run_async to chroma retrievers
1 participant