Browse Source

refac: s3 vector

Timothy Jaeryang Baek 2 tháng trước cách đây
mục cha
commit
8b489cb31f
1 tập tin đã thay đổi với 7 bổ sung0 xóa
  1. 7 0
      backend/open_webui/retrieval/vector/dbs/s3vector.py

+ 7 - 0
backend/open_webui/retrieval/vector/dbs/s3vector.py

@@ -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"])