fix(agents-api): Fix wait for input step (#522)

<!-- ELLIPSIS_HIDDEN -->



> [!IMPORTANT]
> Removes unnecessary transition call in `WaitForInputStep` case in
`TaskExecutionWorkflow.run()` and updates tests accordingly.
> 
>   - **Behavior**:
> - Removes `await transition(context, type="wait", output=output)` from
`WaitForInputStep` case in `run()` in `__init__.py`.
>   - **Tests**:
> - Updates `test_execution_workflow.py` to ensure workflow behaves
correctly without the removed transition call.
> 
> <sup>This description was created by </sup>[<img alt="Ellipsis"
src="https://img.shields.io/badge/Ellipsis-blue?color=175173">](https://www.ellipsis.dev?ref=julep-ai%2Fjulep&utm_source=github&utm_medium=referral)<sup>
for 6600cee. It will automatically
update as commits are pushed.</sup>

<!-- ELLIPSIS_HIDDEN -->

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

@@ -381,8 +381,6 @@ async def run(
             case WaitForInputStep(), StepOutcome(output=output):
                 workflow.logger.info("Wait for input step: Waiting for external input")
 
-                await transition(context, type="wait", output=output)
-
                 result = await workflow.execute_activity(
                     task_steps.raise_complete_async,
                     args=[context, output],

agents-api/tests/test_execution_workflow.py

@@ -1085,4 +1085,3 @@ async def _(
             mock_run_task_execution_workflow.assert_called_once()
 
             await handle.result()
-