github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-01 17:59:28 -05:00
Code Issues 915 Packages Projects Releases 126 Wiki Activity

feat: Initial support for pgvector

Browse Source
This commit is contained in:
Jason Kidd
2024-11-04 12:33:58 -08:00
parent 6c1d0a8e39
commit 701f40aedd
5 changed files with 350 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/apps/retrieval/vector/connector.py
View File

@@ -12,6 +12,10 @@ elif VECTOR_DB == "opensearch":
from open_webui.apps.retrieval.vector.dbs.opensearch import OpenSearchClient
VECTOR_DB_CLIENT = OpenSearchClient()
elif VECTOR_DB == "pgvector":
from open_webui.apps.retrieval.vector.dbs.pgvector import PgvectorClient
VECTOR_DB_CLIENT = PgvectorClient()
else:
from open_webui.apps.retrieval.vector.dbs.chroma import ChromaClient