gitea_admin/sientia-dataops-orchestrator_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1646

Browse Source 
SIENTIAPDE-1646 Add support for runtime workflows in TemporalManager by introducing RUNTIME_WORKFLOWS constant and updating schedule creation logic to handle runtime names appropriately.
This commit is contained in:
vitor-aignosi
2026-05-25 10:12:50 -03:00
parent 08e28441dc
commit feca52a357
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
orchestrator/activities/temporal_manager.py
View File

@@ -25,6 +25,8 @@ with workflow.unsafe.imports_passed_through():
from orchestrator.utils.converters import parse_frequency from orchestrator.utils.converters import parse_frequency
RUNTIME_WORKFLOWS = ['predictions_batch']
class TemporalManager(SientiaMonitoring): class TemporalManager(SientiaMonitoring):
""" """
Temporal workflow and schedule management activity. Temporal workflow and schedule management activity.
@@ -213,6 +215,8 @@ class TemporalManager(SientiaMonitoring):
self.debug( self.debug(
f'{json.dumps(schedule, indent=4, sort_keys=True)}', metadata=metadata f'{json.dumps(schedule, indent=4, sort_keys=True)}', metadata=metadata
) )
runtime_name = schedule.get('runtime', 'legacy') if workflow_type in RUNTIME_WORKFLOWS else None
await client.create_schedule( await client.create_schedule(
schedule_name, schedule_name,
@@ -222,7 +226,7 @@ class TemporalManager(SientiaMonitoring):
schedule, schedule,
id=schedule_name, id=schedule_name,
task_queue=build_queue_name( task_queue=build_queue_name(
workflow_type, schedule.get('runtime', 'legacy') workflow_type, runtime_name
), ),
execution_timeout=timedelta(seconds=execution_timeout_seconds), execution_timeout=timedelta(seconds=execution_timeout_seconds),
run_timeout=timedelta(seconds=execution_timeout_seconds), run_timeout=timedelta(seconds=execution_timeout_seconds),