Merge pull request #484 from julep-ai/f/set-get-step

agents-api/agents_api/workflows/task_execution/__init__.py

@@ -8,6 +8,7 @@
 from temporalio import workflow
 from temporalio.exceptions import ApplicationError
 
+
 with workflow.unsafe.imports_passed_through():
     from ...activities import task_steps
     from ...autogen.openapi_model import (
@@ -101,6 +102,7 @@
 # TODO: find a way to transition to error if workflow or activity times out.
 
 
+
 async def continue_as_child(
     execution_input: ExecutionInput,
     start: TransitionTarget,
@@ -498,14 +500,6 @@ async def run(
                 workflow.logger.info("Set step: Updating user state")
                 self.update_user_state(evaluated_output)
 
-                print("-" * 100)
-                print("user_state", self.user_state)
-                print()
-                print("-" * 100)
-                print()
-                print("evaluated_output", evaluated_output)
-                print("-" * 100)
-
                 # Pass along the previous output unchanged
                 state = PartialTransition(output=context.current_input)
 
@@ -514,10 +508,6 @@ async def run(
                 value = self.get_user_state_by_key(key)
                 workflow.logger.debug(f"Retrieved value: {value}")
 
-                print("-" * 100)
-                print("user_state", self.user_state)
-                print("-" * 100)
-
                 state = PartialTransition(output=value)
 
             case EmbedStep(), _: