From f85cb27ef835aa76aff7de6176bf2159ba392061 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 13 Jun 2026 02:13:51 +0100 Subject: [PATCH] refac --- backend/open_webui/utils/headers.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/backend/open_webui/utils/headers.py b/backend/open_webui/utils/headers.py index 72a5d4c4d1..cb77636ca4 100644 --- a/backend/open_webui/utils/headers.py +++ b/backend/open_webui/utils/headers.py @@ -64,9 +64,18 @@ def get_custom_headers(custom_headers: dict, user=None, metadata: dict = None) - return {} metadata = metadata or {} + + # Extract user_message info for tree mapping + user_message = metadata.get('user_message') or {} + user_message_id = metadata.get('user_message_id', '') or (user_message.get('id', '') if user_message else '') + user_message_parent_id = user_message.get('parentId', '') if user_message else '' + template_vars = { '{{CHAT_ID}}': metadata.get('chat_id', '') or '', '{{MESSAGE_ID}}': metadata.get('message_id', '') or '', + '{{USER_MESSAGE_ID}}': user_message_id or '', + '{{USER_MESSAGE_PARENT_ID}}': user_message_parent_id or '', + '{{TASK}}': metadata.get('task', '') or '', '{{USER_ID}}': (user.id if user else '') or '', '{{USER_NAME}}': (user.name if user else '') or '', '{{USER_EMAIL}}': (user.email if user else '') or '',