ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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
Show stats 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"])