Protocolo Agente-para-Agente (A2A)

O Protocolo Agente-para-Agente (A2A) permite que agentes se comuniquem e colaborem através de limites de rede. O adk-rust oferece suporte completo tanto para expor agentes via A2A quanto para consumir agentes A2A remotos.

Visão Geral

A2A é útil quando:

  • Integrando com serviços de Agent de terceiros
  • Construindo arquiteturas de microsserviços com agentes especializados
  • Habilitando a comunicação de agentes entre diferentes linguagens
  • Impondo contratos formais entre sistemas de Agent

Para organização interna simples, utilize sub-agents locais em vez de A2A para um melhor desempenho.

Cartões de Agente

Todo Agent A2A expõe um cartão de Agent que descreve suas capacidades. O cartão é gerado automaticamente e servido em /.well-known/agent.json.

use adk_server::a2a::build_agent_card;

let agent_card = build_agent_card(&agent, "http://localhost:8080");

println!("Agent: {}", agent_card.name);
println!("Skills: {}", agent_card.skills.len());
println!("Streaming: {}", agent_card.capabilities.streaming);

O cartão de Agent inclui:

  • Nome e descrição do Agent
  • URL base para comunicação
  • Capacidades (streaming, histórico de estado, etc.)
  • Skills derivados do Agent e seus sub-agents

Expondo um Agent via A2A

Para expor um Agent para que outros agentes possam usá-lo, crie um servidor HTTP com endpoints A2A:

use adk_server::{create_app_with_a2a, ServerConfig};
use adk_agent::LlmAgentBuilder;
use adk_model::gemini::GeminiModel;
use std::sync::Arc;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Create your agent
    let model = GeminiModel::new(&api_key, "gemini-2.5-flash")?;
    let agent = LlmAgentBuilder::new("weather_agent")
        .description("Answers weather questions")
        .model(Arc::new(model))
        .build()?;

    // Create server config
    let config = ServerConfig::new(
        Arc::new(SingleAgentLoader::new(Arc::new(agent))),
        Arc::new(InMemorySessionService::new()),
    );

    // Create app with A2A support
    let app = create_app_with_a2a(config, Some("http://localhost:8080"));

    // Serve
    let listener = tokio::net::TcpListener::bind("0.0.0.0:8080").await?;
    axum::serve(listener, app).await?;

    Ok(())
}

Isso expõe:

  • GET /.well-known/agent.json - Cartão de Agent
  • POST /a2a - Endpoint JSON-RPC para o protocolo A2A
  • POST /a2a/stream - Endpoint de streaming SSE

Consumindo um Agent Remoto

Use RemoteA2aAgent para se comunicar com um Agent A2A remoto:

use adk_server::a2a::RemoteA2aAgent;

let remote_agent = RemoteA2aAgent::builder("prime_checker")
    .description("Checks if numbers are prime")
    .agent_url("http://localhost:8001")
    .build()?;

// Use as a sub-agent
let root_agent = LlmAgentBuilder::new("root")
    .model(Arc::new(model))
    .sub_agent(Arc::new(remote_agent))
    .build()?;

O RemoteA2aAgent:

  • Busca automaticamente o cartão de Agent da URL remota
  • Converte eventos ADK de/para mensagens de protocolo A2A
  • Lida com respostas de streaming
  • Funciona perfeitamente como um sub-agent

Cliente A2A

Para comunicação direta com agents remotos, use o A2aClient:

use adk_server::a2a::{A2aClient, Message, Part, Role};

// Create client from URL (fetches agent card)
let client = A2aClient::from_url("http://localhost:8080").await?;

// Build a message
let message = Message::builder()
    .role(Role::User)
    .parts(vec![Part::text("What's the weather?".to_string())])
    .message_id(uuid::Uuid::new_v4().to_string())
    .build();

// Send message (blocking)
let response = client.send_message(message.clone()).await?;

// Or send with streaming
let mut stream = client.send_streaming_message(message).await?;
while let Some(event) = stream.next().await {
    match event? {
        UpdateEvent::TaskArtifactUpdate(artifact) => {
            println!("Artifact: {:?}", artifact);
        }
        UpdateEvent::TaskStatusUpdate(status) => {
            println!("Status: {:?}", status.status.state);
        }
    }
}

Protocolo JSON-RPC

adk-rust implementa o protocolo A2A usando JSON-RPC 2.0. Métodos suportados:

message/send

Envie uma mensagem para o Agent:

{
  "jsonrpc": "2.0",
  "method": "message/send",
  "params": {
    "message": {
      "role": "user",
      "messageId": "msg-123",
      "parts": [{"text": "Hello!"}]
    }
  },
  "id": 1
}

A resposta inclui um objeto task com status e artifacts.

message/stream

Mesmo que message/send, mas retorna Server-Sent Events (SSE) para respostas de streaming.

tasks/cancel

Cancele uma task em execução:

{
  "jsonrpc": "2.0",
  "method": "tasks/cancel",
  "params": {
    "taskId": "task-123"
  },
  "id": 1
}

Exemplo Multi-Agent

Combine Agents locais e remotos:

// Local agent
let roll_agent = LlmAgentBuilder::new("roll_agent")
    .description("Rolls dice")
    .model(Arc::new(model.clone()))
    .tool(Arc::new(roll_die_tool))
    .build()?;

// Remote agent
let prime_agent = RemoteA2aAgent::builder("prime_agent")
    .description("Checks if numbers are prime")
    .agent_url("http://localhost:8001")
    .build()?;

// Root agent orchestrates both
let root_agent = LlmAgentBuilder::new("root_agent")
    .instruction("Delegate dice rolling to roll_agent and prime checking to prime_agent")
    .model(Arc::new(model))
    .sub_agent(Arc::new(roll_agent))
    .sub_agent(Arc::new(prime_agent))
    .build()?;

Tratamento de Erros

Operações A2A retornam erros ADK padrão:

match client.send_message(message).await {
    Ok(response) => {
        if let Some(error) = response.error {
            eprintln!("RPC error: {} (code: {})", error.message, error.code);
        }
    }
    Err(e) => {
        eprintln!("Request failed: {}", e);
    }
}

Códigos de erro comuns:

  • -32600: Invalid request
  • -32601: Method not found
  • -32602: Invalid params
  • -32603: Internal error

Melhores Práticas

  1. Use agent cards: Sempre obtenha e valide os agent cards antes da comunicação
  2. Lide com streaming: Use streaming para operações de longa duração
  3. Recuperação de erros: Implemente lógica de retry para falhas de rede
  4. Timeouts: Defina timeouts apropriados para chamadas remotas
  5. Segurança: Use HTTPS em produção e implemente autenticação

Configuração de Segurança

Configure CORS, timeouts e security headers para implantações de produção:

use adk_server::{ServerConfig, SecurityConfig};
use std::time::Duration;

// Production configuration
let config = ServerConfig::new(agent_loader, session_service)
    .with_allowed_origins(vec![
        "https://myapp.com".to_string(),
        "https://admin.myapp.com".to_string(),
    ])
    .with_request_timeout(Duration::from_secs(30))
    .with_max_body_size(10 * 1024 * 1024);  // 10MB

// Or use presets
let dev_config = ServerConfig::new(agent_loader, session_service)
    .with_security(SecurityConfig::development());  // Permissive for dev

let prod_config = ServerConfig::new(agent_loader, session_service)
    .with_security(SecurityConfig::production(allowed_origins));

Recursos de segurança incluem:

  • CORS: Origens permitidas configuráveis (padrão: permissivo para desenvolvimento)
  • Request timeout: Timeout configurável (padrão: 30 segundos)
  • Body size limit: Tamanho máximo do corpo da requisição (padrão: 10MB)
  • Security headers: X-Content-Type-Options, X-Frame-Options, X-XSS-Protection
  • Error sanitization: Erros internos são logados, mas não expostos a clientes em produção

Relacionado


Anterior: ← Servidor | Próximo: Avaliação →