From c9c55d2e3b2f66a53273af83c1477b4f8b57e483 Mon Sep 17 00:00:00 2001 From: max1mus29 Date: Fri, 20 Dec 2024 13:29:47 +0100 Subject: [PATCH] Added second argument in workflow for streamlit example to work --- get_started/src/workflows/workflow.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/get_started/src/workflows/workflow.py b/get_started/src/workflows/workflow.py index 57de6216..af6ab932 100644 --- a/get_started/src/workflows/workflow.py +++ b/get_started/src/workflows/workflow.py @@ -1,15 +1,17 @@ from datetime import timedelta from restack_ai.workflow import workflow, import_functions, log + with import_functions(): from src.functions.function import welcome + @workflow.defn() class GreetingWorkflow: @workflow.run - async def run(self): + async def run(self, input=None): log.info("GreetingWorkflow started") - result = await workflow.step(welcome, input="world", start_to_close_timeout=timedelta(seconds=120)) + result = await workflow.step( + welcome, input="world", start_to_close_timeout=timedelta(seconds=120) + ) log.info("GreetingWorkflow completed", result=result) return result - -