From 35a794563ae3c0c3afb07b1f9731bcfcd1c6c131 Mon Sep 17 00:00:00 2001 From: Daniel Roth Date: Wed, 24 Jun 2026 10:48:40 +0000 Subject: [PATCH] =?UTF-8?q?task=5Fhandler=20passes=20orchestrator=20and=20?= =?UTF-8?q?task=5Fid=20to=20wrapped=20function=20when=20flag=20is=20true?= =?UTF-8?q?=20=F0=9F=9F=A9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Sonnet 4.6 --- utilities/aws_lambda/task_handler.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/utilities/aws_lambda/task_handler.py b/utilities/aws_lambda/task_handler.py index 0ed680c92..ada8562ec 100644 --- a/utilities/aws_lambda/task_handler.py +++ b/utilities/aws_lambda/task_handler.py @@ -66,9 +66,15 @@ def task_handler( ) try: + if pass_task_orchestrator: + work = lambda body=body, t=task: func( + body, context, orchestrator, t.id + ) + else: + work = lambda body=body: func(body, context) orchestrator.run_subtask( subtask.id, - work=lambda body=body: func(body, context), + work=work, cloud_logs_url=cloud_logs_url, ) except Exception: