Timothy J. Baek 7 meses atrás
pai
commit
a0fb4a9b84

+ 2 - 6
backend/open_webui/apps/retrieval/main.py

@@ -729,13 +729,9 @@ def process_file(
         text_content = " ".join([doc.page_content for doc in docs])
         log.debug(f"text_content: {text_content}")
 
-        Files.update_files_metadata_by_id(
+        Files.update_files_data_by_id(
             form_data.file_id,
-            {
-                "content": {
-                    "text": text_content,
-                }
-            },
+            {"content": text_content},
         )
 
         try:

+ 11 - 0
backend/open_webui/apps/webui/models/files.py

@@ -112,6 +112,17 @@ class FilesTable:
                 for file in db.query(File).filter_by(user_id=user_id).all()
             ]
 
+    def update_files_data_by_id(self, id: str, data: dict) -> Optional[FileModel]:
+        with get_db() as db:
+            try:
+                file = db.query(File).filter_by(id=id).first()
+                file.data = {**file.data, **data}
+                db.commit()
+
+                return FileModel.model_validate(file)
+            except Exception:
+                return None
+
     def update_files_metadata_by_id(self, id: str, meta: dict) -> Optional[FileModel]:
         with get_db() as db:
             try: