Skip to content

feat(astra): add run_async to AstraEmbeddingRetriever#3016

Merged
davidsbatista merged 6 commits intodeepset-ai:mainfrom
nehaaprasad:feat/add-run-asyn
Mar 24, 2026
Merged

feat(astra): add run_async to AstraEmbeddingRetriever#3016
davidsbatista merged 6 commits intodeepset-ai:mainfrom
nehaaprasad:feat/add-run-asyn

Conversation

@nehaaprasad
Copy link
Copy Markdown
Contributor

Related Issues

fixes #1387

Proposed Changes

  • Add run_async on AstraEmbeddingRetriever with the same inputs and outputs as run, using asyncio.to_thread so sync search does not block the event loop.

  • Add a unit test that mocks search and checks run_async.

How did you test it?

  • hatch run test:unit in integrations/astra (or pytest -m "not integration" tests/test_retriever.py).

Notes for the reviewer

  • Async path delegates to the existing sync AstraDocumentStore.search; no search_async on the document store yet.

@nehaaprasad nehaaprasad requested a review from a team as a code owner March 22, 2026 15:03
@nehaaprasad nehaaprasad requested review from julian-risch and removed request for a team March 22, 2026 15:03
@CLAassistant
Copy link
Copy Markdown

CLAassistant commented Mar 22, 2026

CLA assistant check
All committers have signed the CLA.

@github-actions github-actions Bot added integration:astra type:documentation Improvements or additions to documentation labels Mar 22, 2026
@julian-risch julian-risch requested review from davidsbatista and removed request for julian-risch March 23, 2026 08:25
@davidsbatista
Copy link
Copy Markdown
Contributor

@nehaaprasad thank you for this contribution! I did a few adjustments and added two more tests.

Copy link
Copy Markdown
Contributor

@davidsbatista davidsbatista left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@davidsbatista davidsbatista merged commit 2e83f98 into deepset-ai:main Mar 24, 2026
13 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

integration:astra type:documentation Improvements or additions to documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

add run_async to Astra retrievers

3 participants