refac: s3 vector

This commit is contained in:
Timothy Jaeryang Baek 2025-08-08 12:24:47 +04:00
parent 601f1af0a5
commit 8b489cb31f

View file

@ -1,3 +1,4 @@
from backend.open_webui.retrieval.vector.utils import stringify_metadata
from open_webui.retrieval.vector.main import (
VectorDBBase,
VectorItem,
@ -183,6 +184,9 @@ class S3VectorClient(VectorDBBase):
# Add the text field to metadata so it's available for retrieval
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
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
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
metadata = self._filter_metadata(metadata, item["id"])