diff --git a/orchestrator/activities/temporal_manager.py b/orchestrator/activities/temporal_manager.py index ab0cfcf..c54251d 100644 --- a/orchestrator/activities/temporal_manager.py +++ b/orchestrator/activities/temporal_manager.py @@ -183,6 +183,10 @@ class TemporalManager(BaseActivity): workflow_type = schedule['workflow_type'] try: + execution_timeout_seconds = schedule.get( + 'execution_timeout_seconds', 300) + task_timeout_seconds = schedule.get( + 'task_timeout_seconds', 300) self.debug( f"Creating schedule {schedule_name}:", metadata=metadata) self.debug( @@ -197,11 +201,11 @@ class TemporalManager(BaseActivity): id=schedule_name, task_queue=f"{workflow_type}-queue", execution_timeout=timedelta( - minutes=self.execution_timeout), + seconds=execution_timeout_seconds), run_timeout=timedelta( - minutes=self.run_timeout), + seconds=execution_timeout_seconds), task_timeout=timedelta( - minutes=self.task_timeout), + seconds=task_timeout_seconds), typed_search_attributes=search_attributes ), spec=ScheduleSpec( diff --git a/orchestrator/utils/orchestrator_functions.py b/orchestrator/utils/orchestrator_functions.py index 64499b0..38ca1f4 100644 --- a/orchestrator/utils/orchestrator_functions.py +++ b/orchestrator/utils/orchestrator_functions.py @@ -27,6 +27,9 @@ def common_config(config: dict[str, Any]): "model_id": config['model_id'], "model_name": model['name'], "model_config": model.get('model_config', {}), + + "execution_timeout_seconds": config.get('execution_timeout_seconds', 300), + "task_timeout_seconds": config.get('task_timeout_seconds', 300), } diff --git a/values.yaml b/values.yaml index 2a8dc9c..96d900d 100644 --- a/values.yaml +++ b/values.yaml @@ -235,6 +235,12 @@ env: value: "scouter" - name: TEMPORAL_LABORIOUS_NAMESPACE value: "laborious" + - name: TEMPORAL_TASK_TIMEOUT_MINUTES + value: "10" + - name: TEMPORAL_RUN_TIMEOUT_MINUTES + value: "20" + - name: TEMPORAL_EXECUTION_TIMEOUT_MINUTES + value: "20" ssh: enabled: true