From f524a6a8e7d10754d21b81a59fbf75f4cfa63213 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      | 15 ++++++++-------
 3 files changed, 16 insertions(+), 7 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 3c494e7609..0054eb2964 100644
--- a/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
+++ b/src/lib/components/workspace/Knowledge/KnowledgeBase.svelte
@@ -184,12 +184,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;
@@ -199,7 +193,14 @@
 					delete item.itemId;
 					return item;
 				});
-				await addFileHandler(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);
+				}
 			} else {
 				toast.error($i18n.t('Failed to upload file.'));
 			}