from orxhestra import LlmAgent, InMemorySessionService
from orxhestra.a2a import A2AServer, AgentSkill

agent = LlmAgent(name="MyAgent", model=model, tools=[...])

server = A2AServer(
    agent=agent,
    session_service=InMemorySessionService(),
    app_name="my-agent-service",
    skills=[
        AgentSkill(
            id="qa", name="Q&A",
            description="Answers general questions.",
            tags=["general"],
        ),
    ],
)

app = server.as_fastapi_app()
# uvicorn my_module:app --host 0.0.0.0 --port 8000

from orxhestra.agents.a2a_agent import A2AAgent

remote = A2AAgent(
    name="RemoteAgent",
    description="A remote research agent.",
    url="http://localhost:9000",
)

async for event in remote.astream("What is quantum computing?"):
    print(event.text)

agents:
  remote_researcher:
    type: a2a
    description: "Remote research agent"
    url: "http://localhost:9000"

  orchestrator:
    type: llm
    tools:
      - agent: remote_researcher

main_agent: orchestrator