Timothy Jaeryang Baek 2 月之前
父节点
当前提交
e1435501aa
共有 3 个文件被更改,包括 14 次插入11 次删除
  1. 1 0
      backend/open_webui/utils/chat.py
  2. 8 7
      backend/open_webui/utils/middleware.py
  3. 5 4
      backend/open_webui/utils/misc.py

+ 1 - 0
backend/open_webui/utils/chat.py

@@ -161,6 +161,7 @@ async def generate_chat_completion(
     user: Any,
     bypass_filter: bool = False,
 ):
+    log.debug(f"generate_chat_completion: {form_data}")
     if BYPASS_MODEL_ACCESS_CONTROL:
         bypass_filter = True
 

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

@@ -1180,12 +1180,14 @@ async def process_chat_response(
                         temp_blocks.append(block)
 
                 if temp_blocks:
-                    messages.append(
-                        {
-                            "role": "assistant",
-                            "content": serialize_content_blocks(temp_blocks),
-                        }
-                    )
+                    content = serialize_content_blocks(temp_blocks)
+                    if content:
+                        messages.append(
+                            {
+                                "role": "assistant",
+                                "content": content,
+                            }
+                        )
 
                 return messages
 
@@ -1580,7 +1582,6 @@ async def process_chat_response(
 
                     results = []
                     for tool_call in response_tool_calls:
-                        print("\n\n" + str(tool_call) + "\n\n")
                         tool_call_id = tool_call.get("id", "")
                         tool_name = tool_call.get("function", {}).get("name", "")
 

+ 5 - 4
backend/open_webui/utils/misc.py

@@ -225,10 +225,11 @@ def openai_chat_completion_message_template(
     template = openai_chat_message_template(model)
     template["object"] = "chat.completion"
     if message is not None:
-        template["choices"][0]["message"] = {"content": message, "role": "assistant"}
-
-    if tool_calls:
-        template["choices"][0]["tool_calls"] = tool_calls
+        template["choices"][0]["message"] = {
+            "content": message,
+            "role": "assistant",
+            **({"tool_calls": tool_calls} if tool_calls else {}),
+        }
 
     template["choices"][0]["finish_reason"] = "stop"