From 8dce54f3fc3f35ad8fc30433a7076c75c1135541 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Tue, 28 Oct 2025 00:34:53 -0700
Subject: [PATCH] refac/fix: kb image upload handling

---
 backend/open_webui/routers/files.py           |  4 ++++
 src/lib/apis/files/index.ts                   |  4 ++++
 .../workspace/Knowledge/KnowledgeBase.svelte  | 19 ++++++++++---------
 3 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/backend/open_webui/routers/files.py b/backend/open_webui/routers/files.py
index 84d8f841cf..2a5c3e5bb1 100644
--- a/backend/open_webui/routers/files.py
+++ b/backend/open_webui/routers/files.py
@@ -115,6 +115,10 @@ def process_uploaded_file(request, file, file_path, file_item, file_metadata, us
                 request.app.state.config.CONTENT_EXTRACTION_ENGINE == "external"
             ):
                 process_file(request, ProcessFileForm(file_id=file_item.id), user=user)
+            else:
+                raise Exception(
+                    f"File type {file.content_type} is not supported for processing"
+                )
         else:
             log.info(
                 f"File type {file.content_type} is not provided, but trying to process anyway"
diff --git a/src/lib/apis/files/index.ts b/src/lib/apis/files/index.ts
index 6a1763edb8..8351393e3c 100644
--- a/src/lib/apis/files/index.ts
+++ b/src/lib/apis/files/index.ts
@@ -63,6 +63,10 @@ export const uploadFile = async (token: string, file: File, metadata?: object |
 									console.error(data.error);
 									res.error = data.error;
 								}
+
+								if (res?.data) {
+									res.data = data;
+								}
 							}
 						}
 					}
diff --git a/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte b/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
index 7be697c96f..86ed49e420 100644
--- a/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
+++ b/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
@@ -185,12 +185,6 @@
 
 			if (uploadedFile) {
 				console.log(uploadedFile);
-
-				if (uploadedFile.error) {
-					console.warn('File upload warning:', uploadedFile.error);
-					toast.warning(uploadedFile.error);
-				}
-
 				knowledge.files = knowledge.files.map((item) => {
 					if (item.itemId === tempItemId) {
 						item.id = uploadedFile.id;
@@ -200,10 +194,17 @@
 					delete item.itemId;
 					return item;
 				});
-				if (syncMode) {
-					await syncFileHandler(uploadedFile.id);
+
+				if (uploadedFile.error) {
+					console.warn('File upload warning:', uploadedFile.error);
+					toast.warning(uploadedFile.error);
+					knowledge.files = knowledge.files.filter((file) => file.id !== uploadedFile.id);
 				} else {
-					await addFileHandler(uploadedFile.id);
+                    if (syncMode) {
+                        await syncFileHandler(uploadedFile.id);
+                    } else {
+                        await addFileHandler(uploadedFile.id);
+                    }
 				}
 			} else {
 				toast.error($i18n.t('Failed to upload file.'));