Timothy Jaeryang Baek 1 week ago
parent
commit
26d069080a
1 changed files with 11 additions and 0 deletions
  1. 11 0
      backend/open_webui/main.py

+ 11 - 0
backend/open_webui/main.py

@@ -1373,6 +1373,17 @@ async def chat_completion(
             request, response, form_data, user, metadata, model, events, tasks
             request, response, form_data, user, metadata, model, events, tasks
         )
         )
     except Exception as e:
     except Exception as e:
+        log.debug(f"Error in chat completion: {e}")
+        if metadata.get("chat_id") and metadata.get("message_id"):
+            # Update the chat message with the error
+            Chats.upsert_message_to_chat_by_id_and_message_id(
+                metadata["chat_id"],
+                metadata["message_id"],
+                {
+                    "error": {"content": str(e)},
+                },
+            )
+
         raise HTTPException(
         raise HTTPException(
             status_code=status.HTTP_400_BAD_REQUEST,
             status_code=status.HTTP_400_BAD_REQUEST,
             detail=str(e),
             detail=str(e),