scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-10-05 02:41:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Merge 24bbde7277 into 6e3ad226ba

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

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

@ -208,6 +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 import Vector
from pgvector.psycopg import register_vector
from psycopg.rows import dict_row
except ImportError as exc:
@ -275,7 +276,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()