refac

backend/open_webui/utils/middleware.py

@@ -2055,8 +2055,9 @@ def process_messages_with_output(
                 processed.extend(output_messages)
                 continue
 
-        # Strip 'output' field before adding (LLM shouldn't see it)
+        clean_message = dict(message)
-        clean_message = {k: v for k, v in message.items() if k != 'output'}
+        for key in ('id', 'files', 'output', 'contextSummary', 'context_summary', 'usage'):
+            clean_message.pop(key, None)
         processed.append(clean_message)
 
     return processed
@@ -2066,10 +2067,8 @@ def strip_compaction_fields(messages: list[dict]) -> list[dict]:
     stripped = []
     for message in messages:
         clean = dict(message)
-        clean.pop('contextSummary', None)
+        for key in ('id', 'files', 'output', 'contextSummary', 'context_summary', 'usage'):
-        clean.pop('context_summary', None)
+            clean.pop(key, None)
-        clean.pop('usage', None)
-        clean.pop('id', None)
         stripped.append(clean)
     return stripped