Refactor child and parent workflows with output models and service options

Author: martinibach

features/child_workflows/.env.Example

@@ -4,6 +4,6 @@ BEARER_TOKEN=<your-bearer-token>
 # Restack Cloud (Optional)
 
 # RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
 # RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
+# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>

features/child_workflows/src/services.py

@@ -11,10 +11,7 @@
 async def main():
     await client.start_service(
         workflows= [ParentWorkflow, ChildWorkflow],
-        functions= [welcome],
-        options=ServiceOptions(
-            endpoints= True
-        ),
+        functions= [welcome]
     )
 
 def run_services():

features/child_workflows/src/workflows/child.py

@@ -8,11 +8,14 @@
 class ChildInput(BaseModel):
     name: str = "world"
 
+class ChildOutput(BaseModel):
+    result: str
+
 @workflow.defn()
 class ChildWorkflow:
     @workflow.run
-    async def run(self, input: ChildInput):
+    async def run(self, input: ChildInput) -> ChildOutput:
         log.info("ChildWorkflow started")
         result = await workflow.step(welcome, input=input.name, start_to_close_timeout=timedelta(seconds=120))
         log.info("ChildWorkflow completed", result=result)
-        return result
+        return ChildOutput(result=result)

features/child_workflows/src/workflows/parent.py

@@ -5,10 +5,13 @@
 class ParentInput(BaseModel):
     child: bool = True
 
+class ParentOutput(BaseModel):
+    result: str
+
 @workflow.defn()
 class ParentWorkflow:
     @workflow.run
-    async def run(self, input: ParentInput):
+    async def run(self, input: ParentInput) -> ParentOutput:
         if input.child:
             # use the parent run id to create child workflow ids
             parent_workflow_id = workflow_info().workflow_id
@@ -19,8 +22,8 @@ async def run(self, input: ParentInput):
             log.info("Start ChildWorkflow and wait for result")
             result = await workflow.child_execute(ChildWorkflow, input=ChildInput(name="world"), workflow_id=f"{parent_workflow_id}-child-execute")
             log.info("ChildWorkflow completed", result=result)
-            return "ParentWorkflow completed"
+            return ParentOutput(result="ParentWorkflow completed")
 
         else:
             log.info("ParentWorkflow without starting or executing child workflow")
-            return "ParentWorkflow completed"
+            return ParentOutput(result="ParentWorkflow completed")