Python: fix(ag-ui): dedupe client-replayed transcripts on resume

[LI (ktz03)]

Motivation & Context

On AG-UI resume, clients often replay the full transcript in messages while also sending resume. Without dedupe, the same turn is persisted twice and replayed twice after hydrate (#8140). Replayed approval retries can also fall through to a fresh agent run after retained results are available.

Fixes #8140.

Description & Review Guide

• What are the major changes? Dedupe client-replayed transcripts on resume; when approval resume entries are fully handled via retained results, re-emit those results even if reconstruction refilled raw_messages.
• What is the impact of these changes? Prevents duplicate turns and avoids re-invoking the agent on approval retry.
• What do you want reviewers to focus on? resume reconstruction / retained-approval short-circuit in _agent_run.py.

Related Issue

Fixes #8140

No other open PR targets this issue.

Contribution Checklist

• The code builds clean without any errors or warnings
• All unit tests pass, and I have added new tests where possible
• The PR follows the Contribution Guidelines
• This PR is linked to an issue and there is no other open PR for this issue (see Related Issue above).
• This is not a breaking change. If it is a breaking change, add the breaking change label (or add "[BREAKING]" to the title prefix, before or after any language prefix) — a workflow keeps the label and title prefix in sync automatically.

[Copilot]

🟡 Changes recommended

Empty-message resumes can lose stored history in both agent and workflow paths.

Once you've addressed the issues Copilot identified, you can request another Copilot review.

Pull request overview

Updates AG-UI resume handling to deduplicate client-replayed transcripts using snapshot reconstruction.

Changes:

• Applies reconstruction to agent and workflow resume history.
• Adds transcript deduplication regression tests.

File summaries

File	Review
python/packages/ag-ui/tests/ag_ui/test_resume_transcript_dedupe.py	Adds tests, but misses empty-message integration coverage and the required copyright header.
python/packages/ag-ui/agent_framework_ag_ui/_workflow.py	Reconstruction drops stored history for empty checkpoint-resume input.
python/packages/ag-ui/agent_framework_ag_ui/_agent_run.py	Reconstruction drops stored history for empty approval-resume input.

Review details

Suppressed comments (1)

python/packages/ag-ui/tests/ag_ui/test_resume_transcript_dedupe.py:1

• This new Python file is missing the required first-line copyright header mandated by python/CODING_STANDARD.md:453 and used throughout this test directory.

"""Regression for AG-UI resume with client-replayed transcript (#8140)."""

• Files reviewed: 3/3 changed files
• Comments generated: 3
• Review effort level: Balanced

---

💡 Add a code-review agent skill for context-aware, tailored reviews. Learn more in the docs.

[LI (ktz03)]

@microsoft-github-policy-service agree

[LI (ktz03)]

Addressed Copilot review:

1. Empty messages on resume — agent (_agent_run) and workflow (_workflow) now use _reconstruct_messages_from_thread_snapshot only when the client sent a non-empty transcript; empty approval/checkpoint resumes still go through resume_seeded_messages so stored history is kept.
2. Tests — added the required copyright header, documented the empty-input reconstructor contract, and clarified the partial-turn case.

Existing test_endpoint / test_workflow_agent empty-resume assertions should stay green with this branching.

[LI (ktz03)]

Addressed on head 3bb98c18:

• Empty messages on approval/checkpoint resume again goes through resume_seeded_messages (agent + workflow), so stored history is preserved.
• Non-empty / client-replayed transcripts still use _reconstruct_messages_from_thread_snapshot for dedupe.
• Added a unit test documenting the reconstructor’s empty-input contract ([] unchanged → call sites must seed).

Existing test_endpoint / test_workflow_agent empty-resume assertions should stay green with this branching. Please re-run Copilot review on the latest head when convenient.

[LI (ktz03)]

Thanks Evan Mattson (@moonbox3) — addressed both points on this head:

1. ThreadSnapshotSession.reconcile_resume_messages now owns the empty-vs-nonempty resume decision (empty → resume_seeded_messages, non-empty → overlap reconstructor). Agent approval resume and workflow resume/checkpoint seeding both call this single method.
2. Generic non-service resumes (resume_payload without approval-interrupt ids) also go through reconcile and set seeded_resume_from_snapshot = True, so save-time resume_seeded_messages no longer double-prepends after a client-replayed transcript.

Added a unit test covering the session-owned empty vs replayed shapes.

[Evan Mattson (moonbox3)]

Please resolve individual comments with a note if addressed or not. Then resolve when complete.

[LI (ktz03)]

Evan Mattson (@moonbox3) Done — replied on each review thread with addressed/partial notes and resolved all five threads. Please take another look when convenient.

[Evan Mattson (moonbox3)]

Please also fix the failing CI/CD checks.

[LI (ktz03)]

Evan Mattson (@moonbox3) Fixed on head \�682021\ — empty generic resumes no longer seed via
econcile_resume_messages([])\ (that was causing the confirm_changes \call_count\ regression / CI failure). Only non-empty client-replayed generic resumes are reconciled and marked seeded; empty confirm_changes stays on the save-time prepend path as you described. Please re-check when convenient.

[Eduard van Valkenburg (eavanvalkenburg)]

LI (@ktz03) make sure to use the defined PR template

[LI (ktz03)]

Eduard van Valkenburg (@eavanvalkenburg) Updated the description to follow the repo PR template — please take another look when convenient.

[LI (ktz03)]

Eduard van Valkenburg (@eavanvalkenburg) Addressed both review points on latest head:

1. No separate reconcile_* method — resume_seeded_messages stays blind prepend; non-empty client replays use _reconstruct_messages_from_thread_snapshot at the call site.
2. Removed the small dedicated test file; coverage folded into test_snapshot_session.

This should also clear the confirm_changes / prepend CI regressions from the earlier combine attempt. Please re-check when convenient.

[LI (ktz03)]

Typing CI failed only on the folded replay fixture in test_snapshot_session (mypy inferred Mapping from nested toolCalls). Annotated stored/incoming as list[dict[str, Any]] on latest head — unit tests were already green across versions.

[LI (ktz03)]

Eduard van Valkenburg (@eavanvalkenburg) Applied your suggestion on aeff82b: collapsed the duplicate _reconstruct_messages_from_thread_snapshot branches. Generic/non-service path always reconstructs; seeded_resume_from_snapshot is set only when resume_payload carries a non-empty client-replayed transcript (empty confirm_changes still relies on save-time prepend).

[LI (ktz03)]

Eduard van Valkenburg (@eavanvalkenburg) Evan Mattson (@moonbox3) Friendly ping: eavan's duplicate-reconstruct suggestion is applied on aeff82b (CI was green on that head). Happy to take any remaining nits — otherwise a re-review when convenient would help move #8149 forward.

[lsmlhi_25 (FOWEPJF255)]

Following up: CI is still green after the review threads. Dedupe of client-replayed transcripts on AG-UI resume remains an important HITL correctness fix.

Anything left from the earlier maintainer notes that still needs a push, or is this waiting on a final review pass?

[LI (ktz03)]

Eduard van Valkenburg (@eavanvalkenburg) Updated PR body to the repository template and addressed the latest review feedback on this head — please re-review when convenient.