A2AController

class A2AController(APIRouter):
    def __init__(
        self,
        name: Optional[str] = None,
        description: Optional[str] = None,
        url: Optional[str] = None,
        version: str = "0.0.1",
        skills: Optional[List[AgentSkill]] = None,
        capabilities: Optional[AgentCapabilities] = None,
        **kwargs,
    )

def add_handler(
    self,
    method: Literal["message/send", "message/stream", "tasks/get", "tasks/cancel"],
    handler: Callable,
)

def get_handler(
    self,
    method: Literal["message/send", "message/stream", "tasks/get", "tasks/cancel"],
) -> Callable

async def run(self, a2a_request: JSONRPCRequest, request: Request)

from agentor.a2a import A2AController
from a2a.types import AgentSkill, JSONRPCResponse
from fastapi import FastAPI

# Create the A2A controller
controller = A2AController(
    name="My Agent",
    description="A helpful AI assistant",
    url="http://localhost:8000",
    skills=[
        AgentSkill(
            name="answer_questions",
            description="Answer user questions"
        )
    ],
)

# Register a message handler
async def handle_message(request):
    # Process the message
    return JSONRPCResponse(
        id=request.id,
        result={"message": "Response text"}
    )

controller.add_handler("message/send", handle_message)

# Mount to FastAPI app
app = FastAPI()
app.include_router(controller)