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: