A2A Protocol

// A2A Task Lifecycle
type TaskStatus =
  | "submitted"   // задача отправлена агенту
  | "working"     // агент выполняет задачу
  | "input_required" // агенту нужны уточнения
  | "completed"   // задача выполнена, есть результат
  | "failed"      // ошибка выполнения
  | "canceled"    // отменено клиентом

interface A2ATask {
  id: string
  status: TaskStatus
  created: string
  updated?: string
  artifacts?: A2AArtifact[]
  error?: { code: string; message: string }
}

interface A2AArtifact {
  type: "text" | "image" | "json" | "file"
  content: string
  content_type?: string  // MIME type, e.g. "text/markdown"
}

interface A2AMessage {
  role: "client" | "agent"
  content: string
  parts?: Array<{
    type: string
    data: unknown
  }>
}

// A2A Agent Card — "визитка" агента, доступна по /.well-known/agent.json
{
  "name": "pizza-ordering-agent",
  "description": "Принимает и обрабатывает заказы на пиццу",
  "url": "https://pizza.example.com/a2a",
  "version": "1.0.0",
  "capabilities": {
    "skills": [
      {
        "id": "order_pizza",
        "name": "Заказ пиццы",
        "description": "Оформляет заказ: выбирает размер, топпинги, адрес доставки"
      },
      {
        "id": "check_status",
        "name": "Статус заказа",
        "description": "Проверяет статус текущего заказа"
      }
    ],
    "streaming": true,
    "push_notifications": false
  },
  "authentication": {
    "schemes": [
      { "type": "bearer" }
    ]
  }
}

// A2A Server — минимальная реализация
import express from "express"
const app = express()
app.use(express.json())

// Agent Card — эндпоинт для discovery
app.get("/.well-known/agent.json", (req, res) => {
  res.json(agentCard)
})

// A2A Task: принять и выполнить задачу
app.post("/a2a/task", async (req, res) => {
  const { task_id, query, context } = req.body

  // Создаём задачу
  const task = {
    id: task_id,
    status: "working",
    created: new Date().toISOString(),
  }
  res.json(task)

  // Выполняем в фоне (для long-running задач)
  const result = await executeTask(query, context)

  // Отправляем результат через polling/streaming
  await updateTask(task_id, {
    status: "completed",
    artifacts: [{ type: "text", content: result }],
  })
})

// Polling: клиент проверяет статус
app.get("/a2a/task/:id", (req, res) => {
  const task = getTask(req.params.id)
  res.json(task)
})

// A2A Client — находит агента и отправляет задачу
class A2AClient {
  async discoverAgent(agentUrl: string) {
    const res = await fetch(`${agentUrl}/.well-known/agent.json`)
    return res.json() // Agent Card
  }

  async submitTask(
    agentUrl: string,
    query: string,
    context?: Record<string, unknown>
  ) {
    const taskId = crypto.randomUUID()

    // Отправляем задачу
    const res = await fetch(`${agentUrl}/a2a/task`, {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify({
        task_id: taskId,
        query,
        context,
      }),
    })
    return { taskId, status: res.status }
  }

  async pollTask(agentUrl: string, taskId: string) {
    while (true) {
      const res = await fetch(`${agentUrl}/a2a/task/${taskId}`)
      const task = await res.json()

      if (task.status === "completed" || task.status === "failed") {
        return task
      }
      await new Promise(r => setTimeout(r, 500)) // polling interval
    }
  }
}

// Использование:
const client = new A2AClient()
const agentCard = await client.discoverAgent("https://pizza.example.com")
// → узнаём, что агент умеет: order_pizza, check_status

const { taskId } = await client.submitTask(
  "https://pizza.example.com",
  "Закажи большую пепперони на ул. Ленина, 10"
)

const result = await client.pollTask("https://pizza.example.com", taskId)
console.log(result.artifacts) // → результат выполнения