diff --git a/backend/open_webui/apps/retrieval/utils.py b/backend/open_webui/apps/retrieval/utils.py index 174e37495..992d46788 100644 --- a/backend/open_webui/apps/retrieval/utils.py +++ b/backend/open_webui/apps/retrieval/utils.py @@ -71,7 +71,7 @@ def query_doc( try: result = VECTOR_DB_CLIENT.search( collection_name=collection_name, - vectors=query_embedding, + vectors=[query_embedding], limit=k, )