scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-28 05:27:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Merge f80a20ebae into 240c795efe

Browse source
This commit is contained in:
maco 2026-08-27 20:31:25 +02:00 • committed by GitHub
commit cc6ce1434b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
backend/open_webui/retrieval/external.py
View file

@ -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()