From a32a17965ca36f730ebf5ff53a2a8e17082b61a6 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Thu, 13 Aug 2026 21:06:51 -0600 Subject: [PATCH] refac --- backend/open_webui/utils/middleware.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 4f533b8eb7..9881d93d2a 100644 --- a/backend/open_webui/utils/middleware.py +++ b/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 = {k: v for k, v in message.items() if k != 'output'} + clean_message = dict(message) + 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) - clean.pop('context_summary', None) - clean.pop('usage', None) - clean.pop('id', None) + for key in ('id', 'files', 'output', 'contextSummary', 'context_summary', 'usage'): + clean.pop(key, None) stripped.append(clean) return stripped