Преглед на файлове

refac/fix: error message

Timothy Jaeryang Baek преди 1 месец
родител
ревизия
e7c7c65227
променени са 2 файла, в които са добавени 9 реда и са изтрити 0 реда
  1. 7 0
      backend/open_webui/utils/middleware.py
  2. 2 0
      src/lib/components/chat/Chat.svelte

+ 7 - 0
backend/open_webui/utils/middleware.py

@@ -1305,6 +1305,13 @@ async def process_chat_response(
                             "error": {"content": error},
                         },
                     )
+                    if isinstance(error, str) or isinstance(error, dict):
+                        await event_emitter(
+                            {
+                                "type": "chat:message:error",
+                                "data": {"error": {"content": error}},
+                            },
+                        )
 
                 if "selected_model_id" in response_data:
                     Chats.upsert_message_to_chat_by_id_and_message_id(

+ 2 - 0
src/lib/components/chat/Chat.svelte

@@ -324,6 +324,8 @@
 					message.content = data.content;
 				} else if (type === 'chat:message:files' || type === 'files') {
 					message.files = data.files;
+				} else if (type === 'chat:message:error') {
+					message.error = data.error;
 				} else if (type === 'chat:message:follow_ups') {
 					message.followUps = data.follow_ups;