Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac: s3 vector

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-08-08 12:24:47 +04:00
parent 601f1af0a5
commit 8b489cb31f
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/open_webui/retrieval/vector/dbs/s3vector.py
View File

@@ -1,3 +1,4 @@
from backend.open_webui.retrieval.vector.utils import stringify_metadata
from open_webui.retrieval.vector.main import ( from open_webui.retrieval.vector.main import (
VectorDBBase, VectorDBBase,
VectorItem, VectorItem,
@@ -183,6 +184,9 @@ class S3VectorClient(VectorDBBase):
# Add the text field to metadata so it's available for retrieval # Add the text field to metadata so it's available for retrieval
metadata["text"] = item["text"] metadata["text"] = item["text"]
# Convert metadata to string format for consistency
metadata = stringify_metadata(metadata)
# Filter metadata to comply with S3 Vector API limit of 10 keys # Filter metadata to comply with S3 Vector API limit of 10 keys
metadata = self._filter_metadata(metadata, item["id"]) metadata = self._filter_metadata(metadata, item["id"])
@@ -241,6 +245,9 @@ class S3VectorClient(VectorDBBase):
# Add the text field to metadata so it's available for retrieval # Add the text field to metadata so it's available for retrieval
metadata["text"] = item["text"] metadata["text"] = item["text"]
# Convert metadata to string format for consistency
metadata = stringify_metadata(metadata)
# Filter metadata to comply with S3 Vector API limit of 10 keys # Filter metadata to comply with S3 Vector API limit of 10 keys
metadata = self._filter_metadata(metadata, item["id"]) metadata = self._filter_metadata(metadata, item["id"])