From 9192da00a44d48482945bbb34b7f5442e1f739ca Mon Sep 17 00:00:00 2001 From: MiXaiLL76 Date: Thu, 30 Apr 2026 22:38:27 +0300 Subject: [PATCH] update async client --- backend/open_webui/retrieval/vector/async_client.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/retrieval/vector/async_client.py b/backend/open_webui/retrieval/vector/async_client.py index 0bea6696a9..ddb073add7 100644 --- a/backend/open_webui/retrieval/vector/async_client.py +++ b/backend/open_webui/retrieval/vector/async_client.py @@ -100,16 +100,18 @@ class AsyncVectorDBClient: vectors: List[List[Union[float, int]]], filter: Optional[Dict] = None, limit: int = 10, + query: Optional[str] = None, ) -> Optional[SearchResult]: - return await asyncio.to_thread(self._sync.search, collection_name, vectors, filter, limit) + return await asyncio.to_thread(self._sync.search, collection_name, vectors, filter, limit, query) async def query( self, collection_name: str, filter: Dict, limit: Optional[int] = None, + query: Optional[str] = None, ) -> Optional[GetResult]: - return await asyncio.to_thread(self._sync.query, collection_name, filter, limit) + return await asyncio.to_thread(self._sync.query, collection_name, filter, limit, query) async def get(self, collection_name: str) -> Optional[GetResult]: return await asyncio.to_thread(self._sync.get, collection_name)