Merge pull request restackio#40 from restackio/update-human-loop-example

leawn · web-flow · commit 6f84ffed77fa · 2024-11-08T17:55:47.000+01:00
Update human_loop example
diff --git a/examples/human_loop/pyproject.toml b/examples/human_loop/pyproject.toml
@@ -11,7 +11,7 @@ packages = [{include = "src"}]
 
 [tool.poetry.dependencies]
 python = "^3.9"
-restack-ai = "^0.0.25"
+restack-ai = "^0.0.26"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.2"  # Optional: Add if you want to include tests in your example
diff --git a/examples/human_loop/schedule_workflow.py b/examples/human_loop/schedule_workflow.py
@@ -1,13 +1,11 @@
 import asyncio
 import time
-from restack_ai import Restack
+from restack_ai import Restack, log
 
 async def main():
 
     client = Restack()
 
-    print(client)
-
     workflow_id = f"{int(time.time() * 1000)}-HumanLoopWorkflow"
     runId = await client.schedule_workflow(
         workflow_name="HumanLoopWorkflow",
@@ -22,7 +20,8 @@ async def main():
         workflow_id=workflow_id,
         run_id=runId,
     )
-    print(feedback)
+
+    log.info("Feedback sent", feedback=feedback)
 
     end = await client.send_workflow_event(
         event_name="event_end",
@@ -32,7 +31,8 @@ async def main():
         workflow_id=workflow_id,
         run_id=runId,
     )
-    print(end)
+
+    log.info("Workflow ended", end=end)
 
     exit(0)
 
diff --git a/examples/human_loop/src/workflows/workflow.py b/examples/human_loop/src/workflows/workflow.py
@@ -1,5 +1,6 @@
 from datetime import timedelta
-from restack_ai.workflow import workflow, log, import_functions
+from restack_ai.workflow import workflow, import_functions
+from restack_ai import log
 from dataclasses import dataclass
 
 with import_functions():
@@ -20,12 +21,13 @@ def __init__(self) -> None:
         self.feedbacks = []
     @workflow.event
     async def event_feedback(self, feedback: Feedback) -> Feedback:
-        log.info(f"Received feedback: {feedback.feedback}")
-        return await workflow.step(feedback_function, InputFeedback(feedback.feedback), start_to_close_timeout=timedelta(seconds=120))
+        result = await workflow.step(feedback_function, InputFeedback(feedback.feedback), start_to_close_timeout=timedelta(seconds=120))
+        log.info("Received feedback", result=result)
+        return result
     
     @workflow.event
     async def event_end(self, end: End) -> End:
-        log.info(f"Received end: {end.end}")
+        log.info("Received end", end=end)
         self.end_workflow = end.end
         return end
 
@@ -34,6 +36,8 @@ async def run(self):
         await workflow.condition(
             lambda: self.end_workflow
         )
-        return await workflow.step(goodbye, start_to_close_timeout=timedelta(seconds=120))
+        result = await workflow.step(goodbye, start_to_close_timeout=timedelta(seconds=120))
+        log.info("Workflow ended", result=result)
+        return result
 
 
