diff --git a/agents-api/agents_api/workflows/task_execution/__init__.py b/agents-api/agents_api/workflows/task_execution/__init__.py index 10af6c44b..dbbf0b8c6 100644 --- a/agents-api/agents_api/workflows/task_execution/__init__.py +++ b/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(), _: diff --git a/agents-api/tests/test_set_get_workflow.py b/agents-api/tests/test_set_get_workflow.py deleted file mode 100644 index e69de29bb..000000000