junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-30 13:10:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Two-parameter subtask handler completes without TypeError 🟩

Browse source
This commit is contained in:
Daniel Roth 2026-06-10 08:40:32 +00:00
parent 51cf545776
commit bb1c8c88ce
1 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
utilities/aws_lambda/subtask_handler.py
View file

@ -4,6 +4,7 @@ Translates an AWS Lambda invocation (SQS-shaped or direct) into
TaskOrchestrator.run_subtask(...) calls. TaskOrchestrator.run_subtask(...) calls.
""" """
import inspect
import json import json
import logging import logging
import os import os
@ -37,6 +38,8 @@ def subtask_handler(
factory = orchestrator_cm or default_orchestrator factory = orchestrator_cm or default_orchestrator
def decorator(func: Callable[..., Any]) -> Callable[..., Any]: def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
_wants_orchestrator = len(inspect.signature(func).parameters) >= 3
@wraps(func) @wraps(func)
def wrapper(event: dict[str, Any], context: Any) -> None: def wrapper(event: dict[str, Any], context: Any) -> None:
cloud_logs_url = _cloudwatch_url() cloud_logs_url = _cloudwatch_url()
@ -45,12 +48,22 @@ def subtask_handler(
body = _parse_body(record) body = _parse_body(record)
trigger = SubtaskTriggerBody.model_validate(body) trigger = SubtaskTriggerBody.model_validate(body)
logger.info("Running subtask %s", trigger.sub_task_id) logger.info("Running subtask %s", trigger.sub_task_id)
def _work_with(
_body: dict[str, Any] = body,
_o: TaskOrchestrator = orchestrator,
) -> Any:
return func(_body, context, _o)
def _work_without(_body: dict[str, Any] = body) -> Any:
return func(_body, context)
work: Callable[[], Any] = (
_work_with if _wants_orchestrator else _work_without
)
try: try:
orchestrator.run_subtask( orchestrator.run_subtask(
trigger.sub_task_id, trigger.sub_task_id,
work=lambda body=body, o=orchestrator: func( work=work,
body, context, o
),
cloud_logs_url=cloud_logs_url, cloud_logs_url=cloud_logs_url,
) )
except Exception: except Exception: