Timothy Jaeryang Baek 4 月之前
父节点
当前提交
cf0aca1487
共有 1 个文件被更改,包括 9 次插入1 次删除
  1. 9 1
      backend/open_webui/utils/middleware.py

+ 9 - 1
backend/open_webui/utils/middleware.py

@@ -528,7 +528,14 @@ async def process_chat_response(request, response, user, events, metadata, tasks
         return response
         return response
 
 
     event_emitter = None
     event_emitter = None
-    if "session_id" in metadata:
+    if (
+        "session_id" in metadata
+        and metadata["session_id"]
+        and "chat_id" in metadata
+        and metadata["chat_id"]
+        and "message_id" in metadata
+        and metadata["message_id"]
+    ):
         event_emitter = get_event_emitter(metadata)
         event_emitter = get_event_emitter(metadata)
 
 
     if event_emitter:
     if event_emitter:
@@ -701,6 +708,7 @@ async def process_chat_response(request, response, user, events, metadata, tasks
         return {"status": True, "task_id": task_id}
         return {"status": True, "task_id": task_id}
 
 
     else:
     else:
+
         # Fallback to the original response
         # Fallback to the original response
         async def stream_wrapper(original_generator, events):
         async def stream_wrapper(original_generator, events):
             def wrap_item(item):
             def wrap_item(item):