Update templates/workflow_template.py.j2

2025-05-23 01:50:25 +00:00 · 2025-05-23 01:50:25 +00:00 · c146639802
commit c146639802
parent 12aa38b027
1 changed files with 2 additions and 1 deletions
--- a/templates/workflow_template.py.j2
+++ b/templates/workflow_template.py.j2
@ -6,6 +6,7 @@ import json
 import datetime
 import re
 import jmespath
+from temporalio import workflow

 # Configure logging
 logging.basicConfig(level=logging.INFO,
@ -16,7 +17,7 @@ logger = logging.getLogger(__name__)
 class {{ workflow_class_name }}:
    @temporalio.workflow.run
    async def run(self, root_inputs: Dict[str, Any]) -> Dict[str, Any]:
-        await temporalio.workflow.sleep(0)
+        await workflow.sleep(0)
        workflow_info = temporalio.workflow.info()
        workflow_output: Dict[str, Any] = {
            "workflow_id": workflow_info.workflow_id,