[Bug]: taskId-only continuation still fails required contextId inference in SDK 1.3.0 (follow-up to #1206) #1411

import asyncio
from uuid import uuid4
from a2a.server.agent_execution import AgentExecutor
from a2a.server.context import ServerCallContext
from a2a.server.request_handlers import DefaultRequestHandler
from a2a.server.tasks import InMemoryTaskStore, TaskUpdater
from a2a.types import (AgentCard, AgentCapabilities, AgentInterface, AgentSkill,
    Message, Part, SendMessageRequest, Task, TaskState, TaskStatus, GetTaskRequest)

class Agent(AgentExecutor):
    def __init__(self):
        self.contexts = []

    async def execute(self, context, queue):
        self.contexts.append(context.context_id)
        if context.current_task is None:
            await queue.enqueue_event(Task(id=context.task_id, context_id=context.context_id,
                status=TaskStatus(state=TaskState.TASK_STATE_SUBMITTED), history=[context.message]))
        updater = TaskUpdater(queue, context.task_id, context.context_id)
        if context.current_task is None:
            await updater.requires_input()
        else:
            await updater.start_work()
            await updater.complete()

    async def cancel(self, context, queue):
        await TaskUpdater(queue, context.task_id, context.context_id).cancel()

async def main():
    agent = Agent()
    card = AgentCard(name='Repro', description='Task-only context inference', version='1',
        capabilities=AgentCapabilities(streaming=True),
        default_input_modes=['text/plain'], default_output_modes=['text/plain'],
        supported_interfaces=[AgentInterface(url='http://localhost/a2a', protocol_binding='JSONRPC', protocol_version='1.0')],
        skills=[AgentSkill(id='repro', name='Repro', description='Pause and resume', tags=['repro'])])
    handler = DefaultRequestHandler(agent_executor=agent, task_store=InMemoryTaskStore(), agent_card=card)
    caller = ServerCallContext()
    def request(text, task_id=''):
        return SendMessageRequest(message=Message(message_id=str(uuid4()), role=1,
            parts=[Part(text=text)], task_id=task_id))
    try:
        first = await handler.on_message_send(request('Analyze'), caller)
        events = [event async for event in handler.on_message_send_stream(request('EU', first.id), caller)]
        canonical = await handler.on_get_task(GetTaskRequest(id=first.id), caller)
        print('stored task context:', first.context_id)
        print('continuation executor context:', agent.contexts[-1])
        print('continuation event contexts:', [event.context_id for event in events])
        print('canonical task context:', canonical.context_id)
        print('inherited:', agent.contexts[-1] == first.context_id)
        assert agent.contexts[-1] != first.context_id
        assert canonical.context_id == first.context_id
    finally:
        await handler.aclose()

asyncio.run(main())