From f80a20ebae19a243c0b55c52505da585a7ab5b6b Mon Sep 17 00:00:00 2001 From: maco Date: Sun, 9 Aug 2026 21:30:04 +0200 Subject: [PATCH] fix: send the query embedding as a vector in external pgvector retrieval Closes #26663 --- backend/open_webui/retrieval/external.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/retrieval/external.py b/backend/open_webui/retrieval/external.py index 1f0c06d8a7..65a4a845d6 100644 --- a/backend/open_webui/retrieval/external.py +++ b/backend/open_webui/retrieval/external.py @@ -208,7 +208,7 @@ async def _retrieve_milvus(connection, auth_config, knowledge, query, count, emb async def _retrieve_pgvector(connection, auth_config, knowledge, query, count, embedding_function) -> list[dict]: try: import psycopg - from pgvector.psycopg import register_vector + from pgvector.psycopg import Vector, register_vector from psycopg.rows import dict_row except ImportError as exc: raise RuntimeError('psycopg and pgvector are required for pgvector retrieval') from exc @@ -275,7 +275,7 @@ async def _retrieve_pgvector(connection, auth_config, knowledge, query, count, e table_name=table_identifier, collection=collection_identifier, ), - (vector, collection_name, count), + (Vector(vector), collection_name, count), ) return cur.fetchall()