Update child-workflows example

Author: leawn

child_workflows/pyproject.toml

@@ -7,9 +7,9 @@ requires-python = ">=3.10,<3.13"
 readme = "README.md"
 dependencies = [
     "pydantic>=2.10.6",
-    "restack-ai==0.0.62",
     "watchfiles>=1.0.4",
     "python-dotenv==1.0.1",
+    "restack-ai>=0.0.78",
 ]
 
 [project.scripts]

child_workflows/src/functions/function.py

@@ -1,4 +1,4 @@
-from restack_ai.function import function, log
+from restack_ai.function import function, log, NonRetryableError
 
 @function.defn(name="welcome")
 async def welcome(function_input: str) -> str:
@@ -7,4 +7,4 @@ async def welcome(function_input: str) -> str:
         return f"Hello, {function_input}!"
     except Exception as e:
         log.error("welcome function failed", error=e)
-        raise e
+        raise NonRetryableError(f"Welcome function failed: {e}") from e

child_workflows/src/workflows/child.py

@@ -1,6 +1,6 @@
 from datetime import timedelta
 from pydantic import BaseModel
-from restack_ai.workflow import workflow, import_functions, log
+from restack_ai.workflow import NonRetryableError, workflow, import_functions, log
 with import_functions():
     from src.functions.function import welcome
 
@@ -16,6 +16,15 @@ class ChildWorkflow:
     @workflow.run
     async def run(self, workflow_input: ChildInput) -> ChildOutput:
         log.info("ChildWorkflow started")
-        result = await workflow.step(function=welcome, function_input=workflow_input.name, start_to_close_timeout=timedelta(seconds=120))
-        log.info("ChildWorkflow completed", result=result)
-        return ChildOutput(result=result)
+        try:
+            result = await workflow.step(
+                function=welcome,
+                function_input=workflow_input.name,
+                start_to_close_timeout=timedelta(seconds=120)
+            )
+        except Exception as e:
+            error_message = f"Error during welcome: {e}"
+            raise NonRetryableError(error_message) from e
+        else:
+            log.info("ChildWorkflow completed", result=result)
+            return ChildOutput(result=result)

child_workflows/src/workflows/parent.py

@@ -1,4 +1,4 @@
-from restack_ai.workflow import workflow, log, workflow_info
+from restack_ai.workflow import workflow, log, workflow_info, NonRetryableError
 from pydantic import BaseModel
 from .child import ChildWorkflow, ChildInput
 
@@ -22,9 +22,18 @@ async def run(self, workflow_input: ParentInput) -> ParentOutput:
             # result = await workflow.child_start(ChildWorkflow, input=ChildInput(name="world"), workflow_id=f"{parent_workflow_id}-child-start")
 
             log.info("Start ChildWorkflow and wait for result")
-            result = await workflow.child_execute(workflow=ChildWorkflow, workflow_input=ChildInput(name="world"), workflow_id=f"{parent_workflow_id}-child-execute")
-            log.info("ChildWorkflow completed", result=result)
-            return ParentOutput(result="ParentWorkflow completed")
+            try:
+                result = await workflow.child_execute(
+                    workflow=ChildWorkflow,
+                    workflow_input=ChildInput(name="world"),
+                    workflow_id=f"{parent_workflow_id}-child-execute",
+                )
+            except Exception as e:
+                error_message = f"Error during child_execute: {e}"
+                raise NonRetryableError(error_message) from e
+            else:
+                log.info("ChildWorkflow completed", result=result)
+                return ParentOutput(result="ParentWorkflow completed")
 
         else:
             log.info("ParentWorkflow without starting or executing child workflow")