# Concept: handoff runs sub-agents in parallel, stores aggregated results in context,
# then returns a single "orchestrator" agent that reads the precomputed results.

import asyncio
from agents import Agent, Runner, handoff, function_tool, RunContextWrapper

# assume these exist
# tqf_p78_agent, tqf_p80_agent, tqf_p82_agent = Agent(...)

@function_tool
def get_aggregated_results(context: RunContextWrapper) -> str:
    return getattr(context.context, "aggregated_results", "No results available")

orchestrator_agent = Agent(
    name="TQF-Orchestrator",
    instructions="Call get_aggregated_results to obtain precomputed results and summarize them.",
    tools=[get_aggregated_results],
)

async def on_handoff_run_subagents(ctx: RunContextWrapper, input_json: str | None = None) -> Agent:
    task1 = Runner.run(tqf_p78_agent, input="forwarded user input / context")
    task2 = Runner.run(tqf_p80_agent, input="forwarded user input / context")
    task3 = Runner.run(tqf_p82_agent, input="forwarded user input / context")
    res1, res2, res3 = await asyncio.gather(task1, task2, task3)

    aggregated = f"P78:\n{res1.output}\n\nP80:\n{res2.output}\n\nP82:\n{res3.output}"
    ctx.context.aggregated_results = aggregated
    return orchestrator_agent

parallel_handoff = handoff(
    agent=orchestrator_agent,
    on_handoff=on_handoff_run_subagents,
    tool_description_override="Run P78, P80, P82 agents in parallel and aggregate results."
)