From 8faaf2cd1e0182fefd793c05b8bc2b413230a02a Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Wed, 5 Aug 2026 10:37:38 -0500 Subject: [PATCH] refac --- backend/open_webui/main.py | 1 + backend/open_webui/utils/automations.py | 1 + 2 files changed, 2 insertions(+) diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py index 781a6f4a71..f5c182eb3f 100644 --- a/backend/open_webui/main.py +++ b/backend/open_webui/main.py @@ -1198,6 +1198,7 @@ async def chat_completion( 'user_message_id': user_message.get('id') if user_message else None, 'assistant_message_id': form_data.pop('assistant_message_id', None), 'session_id': form_data.pop('session_id', None), + 'automation_id': form_data.pop('automation_id', None), 'folder_id': form_data.pop('folder_id', None), 'filter_ids': form_data.pop('filter_ids', []), 'tool_ids': form_data.get('tool_ids', None), diff --git a/backend/open_webui/utils/automations.py b/backend/open_webui/utils/automations.py index 38045d4096..712ea45096 100644 --- a/backend/open_webui/utils/automations.py +++ b/backend/open_webui/utils/automations.py @@ -547,6 +547,7 @@ async def execute_automation(app, automation: AutomationModel) -> None: 'content': prompt, }, 'session_id': f'automation:{automation.id}', + 'automation_id': automation.id, 'background_tasks': {}, } if tool_ids: