eliott/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-05 06:42:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-08-09 23:57:35 +04:00
parent f85aaa4ed9
commit 77189664c2
6 changed files with 46 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/retrieval/vector/dbs/pgvector.py
View File

@@ -421,7 +421,7 @@ class PgvectorClient(VectorDBBase):
documents[qid].append(row.text)
metadatas[qid].append(row.vmetadata)
self.session.rollback() # read-only transaction
self.session.rollback() # read-only transaction
return SearchResult(
ids=ids, distances=distances, documents=documents, metadatas=metadatas
)
@@ -479,7 +479,7 @@ class PgvectorClient(VectorDBBase):
documents = [[result.text for result in results]]
metadatas = [[result.vmetadata for result in results]]
self.session.rollback() # read-only transaction
self.session.rollback() # read-only transaction
return GetResult(
ids=ids,
documents=documents,
@@ -527,7 +527,7 @@ class PgvectorClient(VectorDBBase):
documents = [[result.text for result in results]]
metadatas = [[result.vmetadata for result in results]]
self.session.rollback() # read-only transaction
self.session.rollback() # read-only transaction
return GetResult(ids=ids, documents=documents, metadatas=metadatas)
except Exception as e:
self.session.rollback()
@@ -598,7 +598,7 @@ class PgvectorClient(VectorDBBase):
.first()
is not None
)
self.session.rollback() # read-only transaction
self.session.rollback() # read-only transaction
return exists
except Exception as e:
self.session.rollback()

6
backend/open_webui/retrieval/vector/dbs/qdrant.py
View File

@@ -60,7 +60,11 @@ class QdrantClient(VectorDBBase):
timeout=self.QDRANT_TIMEOUT,
)
else:
self.client = Qclient(url=self.QDRANT_URI, api_key=self.QDRANT_API_KEY, timeout=QDRANT_TIMEOUT,)
self.client = Qclient(
url=self.QDRANT_URI,
api_key=self.QDRANT_API_KEY,
timeout=QDRANT_TIMEOUT,
)
def _result_to_get_result(self, points) -> GetResult:
ids = []

6
backend/open_webui/retrieval/vector/dbs/qdrant_multitenancy.py
View File

@@ -76,7 +76,11 @@ class QdrantClient(VectorDBBase):
timeout=self.QDRANT_TIMEOUT,
)
if self.PREFER_GRPC
else Qclient(url=self.QDRANT_URI, api_key=self.QDRANT_API_KEY, timeout=self.QDRANT_TIMEOUT,)
else Qclient(
url=self.QDRANT_URI,
api_key=self.QDRANT_API_KEY,
timeout=self.QDRANT_TIMEOUT,
)
)
# Main collection types for multi-tenancy