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

[LI (ktz03)]

Motivation & Context

AG-UI resume can receive a client-replayed transcript that already overlaps the thread snapshot. Naively concatenating snapshot history with the resume payload duplicated message ids and broke HITL / confirm_changes resume semantics.

Fixes the non-use_service_session resume path so overlapping transcripts are reconciled instead of appended twice. Scenario: AG-UI client resumes after interrupt with a replayed conversation history.

Fixes #8140

Description & Review Guide

• What are the major changes?

  • Seed history via _reconstruct_messages_from_thread_snapshot / ThreadSnapshotSession.reconcile_resume_messages instead of naive concatenation on non-service AG-UI resume.
  • Apply the same reconstructor for workflow resume / checkpoint builder seeding when a stored snapshot exists.
  • Empty generic resumes stay on the save-time prepend path (do not reconcile []); only non-empty client-replayed resumes are reconciled and marked seeded.
  • Add a focused regression test for replayed-transcript vs interrupt-only resume shapes.

• What is the impact of these changes?

  • Prevents duplicate message ids on AG-UI resume when the client replays the transcript.
  • Keeps interrupt-only / empty confirm_changes resume behavior unchanged.

• What do you want reviewers to focus on?

  • Empty vs non-empty resume branching in reconcile_resume_messages / generic resume path.
  • That workflow checkpoint seeding uses the same reconstructor when a snapshot exists.

Related Issue

Fixes #8140

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) Folded reconcile_resume_messages into resume_seeded_messages on 40a0733 as suggested — one method owns both empty-prepend and non-empty reconstruct. 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.