Run check

Author: leawn

child_workflows/schedule_workflow.py

@@ -1,7 +1,9 @@
 import asyncio
 import time
+
 from restack_ai import Restack
 
+
 async def main():
 
     client = Restack()
@@ -23,4 +25,4 @@ def run_schedule_workflow():
     asyncio.run(main())
 
 if __name__ == "__main__":
-    run_schedule_workflow()
+    run_schedule_workflow()

child_workflows/src/client.py

@@ -1,7 +1,9 @@
 import os
+
+from dotenv import load_dotenv
 from restack_ai import Restack
 from restack_ai.restack import CloudConnectionOptions
-from dotenv import load_dotenv  
+
 # Load environment variables from a .env file
 load_dotenv()
 
@@ -18,4 +20,4 @@
     api_address=api_address
 )
 print(connection_options)
-client = Restack(connection_options)
+client = Restack(connection_options)

child_workflows/src/functions/function.py

@@ -1,4 +1,5 @@
-from restack_ai.function import function, log, NonRetryableError
+from restack_ai.function import NonRetryableError, function, log
+
 
 @function.defn(name="welcome")
 async def welcome(function_input: str) -> str:

child_workflows/src/services.py

@@ -1,11 +1,14 @@
 import asyncio
 import os
 import webbrowser
+
 from watchfiles import run_process
-from src.functions.function import welcome
+
 from src.client import client
-from src.workflows.parent import ParentWorkflow
+from src.functions.function import welcome
 from src.workflows.child import ChildWorkflow
+from src.workflows.parent import ParentWorkflow
+
 
 async def main():
     await client.start_service(
@@ -27,4 +30,4 @@ def watch_services():
     run_process(watch_path, recursive=True, target=run_services)
 
 if __name__ == "__main__":
-       run_services()
+       run_services()

child_workflows/src/workflows/child.py

@@ -1,9 +1,11 @@
 from datetime import timedelta
+
 from pydantic import BaseModel
-from restack_ai.workflow import NonRetryableError, workflow, import_functions, log
+from restack_ai.workflow import NonRetryableError, import_functions, log, workflow
+
 with import_functions():
     from src.functions.function import welcome
-    
+
 
 class ChildInput(BaseModel):
     name: str = "world"

child_workflows/src/workflows/parent.py

@@ -1,6 +1,8 @@
-from restack_ai.workflow import workflow, log, workflow_info, NonRetryableError
 from pydantic import BaseModel
-from .child import ChildWorkflow, ChildInput
+from restack_ai.workflow import NonRetryableError, log, workflow, workflow_info
+
+from .child import ChildInput, ChildWorkflow
+
 
 class ParentInput(BaseModel):
     child: bool = True
@@ -12,15 +14,15 @@ class ParentOutput(BaseModel):
 class ParentWorkflow:
     @workflow.run
     async def run(self, workflow_input: ParentInput) -> ParentOutput:
-        
+
         log.info("ParentWorkflow started", workflow_input=workflow_input)
         if workflow_input.child:
             # use the parent run id to create child workflow ids
             parent_workflow_id = workflow_info().workflow_id
 
             log.info("Start ChildWorkflow and dont wait for result")
             # 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")
             try:
                 result = await workflow.child_execute(
@@ -34,7 +36,7 @@ async def run(self, workflow_input: ParentInput) -> ParentOutput:
             else:
                 log.info("ChildWorkflow completed", result=result)
                 return ParentOutput(result="ParentWorkflow completed")
-        
+
         else:
             log.info("ParentWorkflow without starting or executing child workflow")
             return ParentOutput(result="ParentWorkflow completed")