From 3391a855f0952c0dece229b3409ffba4156f882c Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Tue, 8 Oct 2024 17:03:42 -0700 Subject: [PATCH] fix --- backend/open_webui/apps/retrieval/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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, )