Python: [Bug]: AgentExecutor full_conversation drops the user turn and the approval-request turn after a tool-approval resume #10288

"""AgentExecutorResponse.full_conversation after an approval resume drops the original user turn
and the tool-call/approval-request turn."""
import asyncio
from typing import Any
from agent_framework import (AgentExecutor, AgentExecutorResponse, AgentResponse, AgentResponseUpdate,
                             AgentSession, BaseAgent, Content, Executor, Message, ResponseStream,
                             WorkflowBuilder, WorkflowContext, handler)

class ApprovalAgent(BaseAgent):
    def __init__(self) -> None:
        super().__init__(name="approver")
        self.calls = 0
    def run(self, messages=None, *, stream=False, session: AgentSession | None = None, **kw: Any):
        self.calls += 1
        if self.calls == 1:
            call = Content.from_function_call(call_id="c1", name="delete_db", arguments="{}")
            contents = [Content.from_function_approval_request(id="r1", function_call=call)]
        else:
            contents = [Content.from_text("Deleted the DB as approved.")]
        async def _run():
            return AgentResponse(messages=[Message(role="assistant", contents=contents)])
        async def _stream():
            yield AgentResponseUpdate(role="assistant", contents=contents)
        return ResponseStream(_stream(), finalizer=AgentResponse.from_updates) if stream else _run()

class Capture(Executor):
    seen: list = []
    @handler
    async def h(self, r: AgentExecutorResponse, ctx: WorkflowContext[None, str]) -> None:
        Capture.seen = [(m.role, [c.type for c in m.contents], m.text) for m in r.full_conversation]
        await ctx.yield_output("ok")

async def main():
    ae = AgentExecutor(ApprovalAgent(), id="agent")
    cap = Capture(id="cap")
    wf = WorkflowBuilder(start_executor=ae).add_edge(ae, cap).build()
    r = await asyncio.wait_for(wf.run("please delete the db"), 5)
    req = r.get_request_info_events()[0]
    await asyncio.wait_for(wf.run(responses={req.request_id: req.data.to_function_approval_response(approved=True)}), 5)
    print("downstream full_conversation:")
    for row in Capture.seen:
        print("  ", row)
    print("expected: starts with ('user', ['text'], 'please delete the db') and includes the approval request")

asyncio.run(main())

class RecordingAgent(BaseAgent):
    def __init__(self) -> None:
        super().__init__(name="reviewer")
        self.received: list = []
    def run(self, messages=None, *, stream=False, session=None, **kw: Any):
        self.received = [(m.role, [c.type for c in m.contents], m.text) for m in messages]
        async def _run():
            return AgentResponse(messages=[Message(role="assistant", contents=[Content.from_text("reviewed")])])
        return _run()

async def chained():
    a = AgentExecutor(ApprovalAgent(), id="agent")
    rec = RecordingAgent()
    b = AgentExecutor(rec, id="reviewer")  # default context_mode="full"
    wf = WorkflowBuilder(start_executor=a).add_edge(a, b).build()
    r = await asyncio.wait_for(wf.run("please delete the db"), 5)
    req = r.get_request_info_events()[0]
    await asyncio.wait_for(wf.run(responses={req.request_id: req.data.to_function_approval_response(approved=True)}), 5)
    print("\nchained AgentExecutor (context_mode='full') received:")
    for row in rec.received:
        print("  ", row)

asyncio.run(chained())