Agent War Room

import asyncio
from agent_framework.azure import AzureOpenAIChatClient
from azure.identity import AzureCliCredential

client = AzureOpenAIChatClient(credential=AzureCliCredential())

# --- Specialist roster: name -> instructions ---------------------------
SPECIALISTS = {
    "researcher": "Gather facts and sources for the subtask. Be concise.",
    "analyst":    "Analyze data, quantify impact, and flag risks.",
    "writer":     "Draft clear, on-brand content from the inputs.",
    "reviewer":   "Check accuracy, tone, and compliance. Approve or request one fix.",
}

def specialist(name: str):
    return client.create_agent(name=name, instructions=SPECIALISTS[name])

async def run_specialist(name: str, goal: str, task: str) -> str:
    agent = specialist(name)
    reply = await agent.run(f"GOAL: {goal}\nYOUR SUBTASK: {task}")
    return reply.text

async def war_room(goal: str, mode: str = "concurrent") -> str:
    planner = client.create_agent(
        name="planner",
        instructions="Decompose the goal into one subtask per specialist, "
                     "then merge their results into a single deliverable.",
    )
    # 1) Decompose: ask the planner for a subtask per specialist
    plan = await planner.run(
        f"GOAL: {goal}\nSpecialists: {', '.join(SPECIALISTS)}.\n"
        "Return one short subtask per specialist, 'name: task' per line."
    )
    tasks = dict(
        line.split(":", 1) for line in plan.text.splitlines() if ":" in line
    )
    tasks = {k.strip().lower(): v.strip() for k, v in tasks.items()
             if k.strip().lower() in SPECIALISTS}

    # 2) Delegate + run (concurrent fan-out / fan-in, or sequential)
    results = {}
    if mode == "concurrent":
        outs = await asyncio.gather(
            *(run_specialist(n, goal, t) for n, t in tasks.items())
        )
        results = dict(zip(tasks, outs))
    else:  # sequential hand-off chain
        carry = ""
        for n, t in tasks.items():
            results[n] = await run_specialist(n, goal, f"{t}\nContext so far: {carry}")
            carry += f"\n[{n}] {results[n]}"

    # 3) Synthesize
    summary = "\n".join(f"[{n}] {o}" for n, o in results.items())
    final = await planner.run(f"GOAL: {goal}\nSpecialist outputs:\n{summary}\n"
                              "Write the final deliverable.")
    return final.text

if __name__ == "__main__":
    print(asyncio.run(war_room("Plan the launch of our new AI feature.")))

agent-framework
azure-identity