Merge pull request #131 from restackio/fix-production-demo

Author: martinibach

production_demo/src/workflows/child.py

@@ -3,7 +3,7 @@
 from restack_ai.workflow import workflow, import_functions, log
 
 with import_functions():
-    from src.functions.function import example_function
+    from src.functions.function import example_function, ExampleFunctionInput
     from src.functions.generate import llm_generate, GenerateInput
     from src.functions.evaluate import llm_evaluate, EvaluateInput
 
@@ -15,7 +15,7 @@ class ChildWorkflow:
     @workflow.run
     async def run(self, input: ChildWorkflowInput):
         log.info("ChildWorkflow started")
-        await workflow.step(example_function, input=input, start_to_close_timeout=timedelta(minutes=2))
+        await workflow.step(example_function, input=ExampleFunctionInput(name='John Doe'), start_to_close_timeout=timedelta(minutes=2))
 
         await workflow.sleep(1)