SIENTIAPDE-1231

SIENTIAPDE-1231: Add timeout configuration for Temporal tasks

- Introduced new timeout parameters in values.yaml for task execution.
- Updated TemporalManager to utilize execution and task timeout settings in seconds.
- Enhanced orchestrator_functions.py to include default timeout values for better orchestration control.
This commit is contained in:
vitor-aignosi
2025-10-10 15:55:03 -03:00
parent 5fdabd27c0
commit 310cae4c2f
3 changed files with 16 additions and 3 deletions

View File

@@ -183,6 +183,10 @@ class TemporalManager(BaseActivity):
workflow_type = schedule['workflow_type'] workflow_type = schedule['workflow_type']
try: try:
execution_timeout_seconds = schedule.get(
'execution_timeout_seconds', 300)
task_timeout_seconds = schedule.get(
'task_timeout_seconds', 300)
self.debug( self.debug(
f"Creating schedule {schedule_name}:", metadata=metadata) f"Creating schedule {schedule_name}:", metadata=metadata)
self.debug( self.debug(
@@ -197,11 +201,11 @@ class TemporalManager(BaseActivity):
id=schedule_name, id=schedule_name,
task_queue=f"{workflow_type}-queue", task_queue=f"{workflow_type}-queue",
execution_timeout=timedelta( execution_timeout=timedelta(
minutes=self.execution_timeout), seconds=execution_timeout_seconds),
run_timeout=timedelta( run_timeout=timedelta(
minutes=self.run_timeout), seconds=execution_timeout_seconds),
task_timeout=timedelta( task_timeout=timedelta(
minutes=self.task_timeout), seconds=task_timeout_seconds),
typed_search_attributes=search_attributes typed_search_attributes=search_attributes
), ),
spec=ScheduleSpec( spec=ScheduleSpec(

View File

@@ -27,6 +27,9 @@ def common_config(config: dict[str, Any]):
"model_id": config['model_id'], "model_id": config['model_id'],
"model_name": model['name'], "model_name": model['name'],
"model_config": model.get('model_config', {}), "model_config": model.get('model_config', {}),
"execution_timeout_seconds": config.get('execution_timeout_seconds', 300),
"task_timeout_seconds": config.get('task_timeout_seconds', 300),
} }

View File

@@ -235,6 +235,12 @@ env:
value: "scouter" value: "scouter"
- name: TEMPORAL_LABORIOUS_NAMESPACE - name: TEMPORAL_LABORIOUS_NAMESPACE
value: "laborious" value: "laborious"
- name: TEMPORAL_TASK_TIMEOUT_MINUTES
value: "10"
- name: TEMPORAL_RUN_TIMEOUT_MINUTES
value: "20"
- name: TEMPORAL_EXECUTION_TIMEOUT_MINUTES
value: "20"
ssh: ssh:
enabled: true enabled: true