Sfoglia il codice sorgente

remove files from message for tasks

Alexander Grimm 5 mesi fa
parent
commit
cf5201aada
1 ha cambiato i file con 14 aggiunte e 0 eliminazioni
  1. 14 0
      backend/open_webui/utils/middleware.py

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

@@ -977,6 +977,20 @@ async def process_chat_response(
         if message:
             messages = get_message_list(message_map, message.get("id"))
 
+            # Remove reasoning details and files from the messages.
+            # as get_message_list creates a new list, it does not affect 
+            # the original messages outside of this handler
+            for message in messages:
+                message["content"] = re.sub(
+                    r"<details\s+type=\"reasoning\"[^>]*>.*?<\/details>",
+                    "",
+                    message["content"],
+                    flags=re.S,
+                ).strip()
+
+                if message.get('files'):
+                    message['files'] = []
+
             if tasks and messages:
                 if TASKS.TITLE_GENERATION in tasks:
                     if tasks[TASKS.TITLE_GENERATION]: