wagtail / wagtail-vector-index

Store Wagtail pages & Django models as embeddings in vector databases
https://wagtail-vector-index.readthedocs.io/en/latest/
MIT License
20 stars 13 forks source link

Add async search and find_similar APIs #90

Open tomusher opened 1 month ago

tomusher commented 1 month ago

This PR adds async methods afind_similar and asearch to the public API exposed by a Vector Index.

This involved creating various async variations of methods further up the chain. I've also broke up the bulk_generate_documents methods in to smaller functions to both reduce the size of that function body and so we can reuse as much as possible between the async and non-async versions of the function.