refactor(agents-api): Refactor context.model_dump() to be more lightweight

[creatorrr]

Signed-off-by: Diwank Singh Tomer [email protected]

---

Important

Refactor context.model_dump() to context.prepare_for_step() across multiple task step files for improved efficiency and add minor logging and type hint updates.

• Refactoring:
  • Replace context.model_dump() with context.prepare_for_step() in evaluate_step.py, for_each_step.py, if_else_step.py, log_step.py, map_reduce_step.py, prompt_step.py, return_step.py, set_value_step.py, switch_step.py, tool_call_step.py, wait_for_input_step.py, and yield_step.py.
• Logging:
  • Change logging from logging.error to activity.logger.error in for_each_step.py and map_reduce_step.py.
• Type Hints:
  • Add beartype type hints to run_task_execution_workflow() in temporal.py.
  • Update type hints for start and previous_inputs in run_task_execution_workflow() to use | None syntax.
• Miscellaneous:
  • Add prepare_for_step() method in tasks.py to merge execution inputs and current input.

^{This description was created by for 9bc0f3d. It will automatically update as commits are pushed.}

[ellipsis-dev]

👍 Looks good to me! Reviewed everything up to 9bc0f3d in 41 seconds

More details

• Looked at 258 lines of code in 14 files
• Skipped 0 files when reviewing.
• Skipped posting 2 drafted comments based on config settings.

1. agents-api/agents_api/activities/task_steps/return_step.py:24

• Draft comment:
  The error message here mistakenly refers to 'log_step'. It should be 'return_step'.

        activity.logger.error(f"Error in return_step: {e}")

• Reason this comment was not posted:
  Comment was on unchanged code.

2. agents-api/agents_api/activities/task_steps/map_reduce_step.py:27

• Draft comment:
  Use activity.logger for consistency with other step files.

activity.logger.error(f"Error in map_reduce_step: {e}")

• Reason this comment was not posted:
  Comment was on unchanged code.

Workflow ID: wflow_wMeQBheANsNlN7gh

---

You can customize Ellipsis with 👍 / 👎 feedback, review rules, user-specific overrides, quiet mode, and more.