From fbb8c111edb863bdae8bef70a4e9fba1abda054d Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Mon, 11 Aug 2025 00:39:12 +0400
Subject: [PATCH] refac

---
 backend/open_webui/utils/middleware.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py
index 0810e85a40..d90488722f 100644
--- a/backend/open_webui/utils/middleware.py
+++ b/backend/open_webui/utils/middleware.py
@@ -1344,14 +1344,14 @@ async def process_chat_response(
                         **response_data,
                     }
 
+                if isinstance(response, dict):
+                    response = response_data
                 if isinstance(response, JSONResponse):
                     response = JSONResponse(
                         content=response_data,
                         headers=response.headers,
                         status_code=response.status_code,
                     )
-                else:
-                    response = response_data
 
             return response
         else: