- Changed the function's input parameters.
- Callback now gets state and output directly.
- Removed use of the old agent instance object.
- Get the current step number from new input.
- Updated how agent actions are displayed in the output.
- Show observation from the browser state.