浏览代码

refac: s3 vector

Timothy Jaeryang Baek 2 月之前
父节点
当前提交
8b489cb31f
共有 1 个文件被更改,包括 7 次插入0 次删除
  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"])