Evan Mattson dd3d085539 Python: Include reasoning messages in MESSAGES_SNAPSHOT events (#4844) ...

* Include reasoning messages in MESSAGES_SNAPSHOT (#4843)

FlowState now tracks reasoning messages emitted during a run.
_emit_text_reasoning() persists reasoning (including encrypted_value)
into flow.reasoning_messages, and _build_messages_snapshot() appends
them to the final MESSAGES_SNAPSHOT event.

Changes:
- Add reasoning_messages field to FlowState
- Update _emit_text_reasoning() to accept optional flow parameter
- Include reasoning_messages in _build_messages_snapshot()
- Add 'reasoning' to ALLOWED_AGUI_ROLES so normalize_agui_role()
  preserves the role through snapshot round-trips
- Skip reasoning messages in agui_messages_to_agent_framework() since
  they are UI-only state and should not be forwarded to LLM providers
- Add regression tests for snapshot emission, encrypted value
  preservation, and multi-turn round-trip with reasoning

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Python: Include reasoning messages in MESSAGES_SNAPSHOT events

Fixes #4843

* Fix PR review feedback for reasoning persistence (#4843)

- Accumulate reasoning text per message_id (append deltas) instead of
  storing only the current chunk, matching flow.accumulated_text pattern
- Use camelCase encryptedValue in snapshot JSON to match AG-UI protocol
  conventions (toolCallId, encryptedValue)
- Normalize snake_case encrypted_value to encryptedValue in
  agui_messages_to_snapshot_format for input compatibility
- Update normalize_agui_role docstring to include reasoning role
- Add tests for incremental reasoning accumulation and key normalization

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

* Address review feedback for #4843: Python: agent-framework-ag-ui: include reasoning messages in MESSAGES_SNAPSHOT

---------

Co-authored-by: Copilot <copilot@github.com>
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

dd3d085539 · 2026-03-26 05:56:10 +00:00

History

..

golden

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

conftest.py

Python: [BREAKING] Refactor middleware layering and split Anthropic raw client (#4746)

2026-03-20 00:43:37 +00:00

event_stream.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

sse_helpers.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

test_ag_ui_client.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

test_agent_wrapper_comprehensive.py

[BREAKING] Python: clean up kwargs across agents, chat clients, tools, and sessions (#4581)

2026-03-13 08:58:32 +00:00

test_approval_result_event.py

Python: Emit TOOL_CALL_RESULT events when resuming after tool approval (#4758)

2026-03-20 00:41:46 +00:00

test_endpoint.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

test_event_converters.py

Python: Unify tool results as Content items with rich content support (#4331)

2026-03-12 22:30:09 +00:00

test_helpers.py

Python: [BREAKING] Simplify API: ChatAgent -> Agent, ChatMessage -> Message (#3747)

2026-02-10 23:04:32 +00:00

test_http_round_trip.py

Python: Emit AG-UI events for MCP tool calls, results, and text reasoning (#4760)

2026-03-20 00:41:37 +00:00

test_http_service.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_message_adapters.py

Python: Include reasoning messages in MESSAGES_SNAPSHOT events (#4844)

2026-03-26 05:56:10 +00:00

test_message_hygiene.py

Python: (ag-ui): fix approval payloads being re-processed on subsequent conversation turns (#4232)

2026-02-26 04:18:46 +00:00

test_multi_turn.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

test_predictive_state.py

Python: [BREAKING] Moved to a single get_response and run API (#3379)

2026-02-05 20:09:58 +00:00

test_public_exports.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_run_common.py

Python: Improve ag-ui tests and coverage (#4442)

2026-03-06 07:06:56 +00:00

test_run.py

Python: Include reasoning messages in MESSAGES_SNAPSHOT events (#4844)

2026-03-26 05:56:10 +00:00

test_service_thread_id.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_structured_output.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_subgraphs_example_agent.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_tooling.py

fix duplicate names between supplied tools and mcp servers (#4649)

2026-03-13 08:22:56 +00:00

test_types.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_utils.py

Python: Include reasoning messages in MESSAGES_SNAPSHOT events (#4844)

2026-03-26 05:56:10 +00:00

test_workflow_agent.py

Python: (ag-ui): Add Workflow Support, Harden Streaming Semantics, and add Dynamic Handoff Demo (#3911)

2026-02-23 11:59:56 +00:00

test_workflow_run.py

Python: Fix function_approval_response extraction in AG-UI workflow path (#4550)

2026-03-11 22:54:16 +00:00