kompass-sdk 0.1.0

package/src/a2a/server.ts

@@ -0,0 +1,234 @@
+/**
+ * Kompass A2A Server
+ * Exposes Kompass discovery + coordination as an A2A-compatible agent.
+ * Any A2A client can send tasks to Kompass to discover or coordinate agents.
+ */
+
+import express from "express";
+import { createKompassAgentCard } from "./agent-card.js";
+import { KompassUnified } from "../unified.js";
+import { A2ABridge, type CoordinationStep } from "./bridge.js";
+import { ProtocolBridge } from "../bridge.js";
+import { AGENT_CARD_PATH } from "@a2a-js/sdk";
+
+export interface KompassA2AServerOptions {
+  port?: number;
+  hostname?: string;
+  kompass?: KompassUnified;
+}
+
+export async function startA2AServer(options: KompassA2AServerOptions) {
+  const port = options.port ?? 3000;
+  const hostname = options.hostname ?? "localhost";
+  const baseUrl = `http://${hostname}:${port}`;
+
+  // Initialize Kompass if not provided
+  const kompass = options.kompass ?? (await KompassUnified.create());
+
+  // Create agent card
+  const agentCard = createKompassAgentCard({ url: baseUrl });
+
+  // Create A2A bridge
+  const protocolBridge = new ProtocolBridge();
+  const a2aBridge = new A2ABridge(protocolBridge);
+
+  const app = express();
+  app.use(express.json());
+
+  // ── Agent Card (Discovery) ─────────────────────────────
+
+  app.get(AGENT_CARD_PATH, (_req, res) => {
+    res.json(agentCard);
+  });
+
+  // Also serve at /agent-card.json for convenience
+  app.get("/agent-card.json", (_req, res) => {
+    res.json(agentCard);
+  });
+
+  // ── A2A JSON-RPC Endpoint ──────────────────────────────
+
+  app.post("/", async (req, res) => {
+    const { method, params, id } = req.body;
+
+    try {
+      let result: unknown;
+
+      switch (method) {
+        case "message/send":
+        case "SendMessage": {
+          const message = params?.message;
+          const text = extractText(message);
+
+          if (!text) {
+            res.json(jsonRpcError(id, -32602, "No text in message"));
+            return;
+          }
+
+          // Parse intent from message
+          result = await handleTask(text, kompass, a2aBridge);
+          break;
+        }
+
+        case "tasks/get":
+        case "GetTask": {
+          // For now, return task not found (we're stateless)
+          res.json(jsonRpcError(id, -32001, "Task not found — Kompass is stateless"));
+          return;
+        }
+
+        case "tasks/cancel":
+        case "CancelTask": {
+          res.json(jsonRpcError(id, -32001, "Task not cancelable"));
+          return;
+        }
+
+        default:
+          res.json(jsonRpcError(id, -32601, `Method not found: ${method}`));
+          return;
+      }
+
+      res.json({
+        jsonrpc: "2.0",
+        id,
+        result,
+      });
+    } catch (err) {
+      res.json(jsonRpcError(id, -32603, err instanceof Error ? err.message : String(err)));
+    }
+  });
+
+  // ── Health & Info ──────────────────────────────────────
+
+  app.get("/health", async (_req, res) => {
+    const sources = await kompass.sources();
+    const online = sources.filter((s) => s.healthy).length;
+    res.json({
+      status: "healthy",
+      sources: `${online}/${sources.length} online`,
+      protocol: "a2a",
+      version: "1.0.0",
+    });
+  });
+
+  // ── Start ──────────────────────────────────────────────
+
+  app.listen(port, () => {
+    console.log(`\n🟢 Kompass A2A Server running at ${baseUrl}`);
+    console.log(`   Agent Card: ${baseUrl}${AGENT_CARD_PATH}`);
+    console.log(`   JSON-RPC: POST ${baseUrl}/`);
+    console.log(`   Health: GET ${baseUrl}/health\n`);
+  });
+
+  return { app, baseUrl, agentCard };
+}
+
+// ── Helpers ──────────────────────────────────────────────
+
+function extractText(message: any): string | null {
+  if (!message?.parts) return null;
+  for (const part of message.parts) {
+    if (part.type === "text" && part.text) return part.text;
+  }
+  return null;
+}
+
+async function handleTask(text: string, kompass: KompassUnified, bridge: A2ABridge) {
+  const lower = text.toLowerCase();
+
+  // Detect intent: discovery, hire, or coordination
+  if (lower.startsWith("find ") || lower.startsWith("discover ") || lower.startsWith("search ")) {
+    // Discovery task
+    const query = text.replace(/^(find|discover|search)\s+/i, "");
+    const result = await kompass.find(query, { limit: 10 });
+
+    return {
+      id: `kompass-${Date.now()}`,
+      status: { state: "completed" },
+      artifacts: [
+        {
+          parts: [
+            {
+              type: "text",
+              text: `Found ${result.agents.length} agents across ${result.sources.length} registries:\n\n` +
+                result.agents
+                  .map(
+                    (a, i) =>
+                      `${i + 1}. ${a.name} [${a.protocol.toUpperCase()}] (${a.source})\n   ${a.description.slice(0, 100)}`
+                  )
+                  .join("\n\n"),
+            },
+            {
+              type: "data",
+              data: {
+                agents: result.agents,
+                sources: result.sources,
+                queryTimeMs: result.queryTimeMs,
+              },
+            },
+          ],
+        },
+      ],
+    };
+  }
+
+  if (lower.startsWith("coordinate ") || lower.startsWith("orchestrate ")) {
+    // Coordination task
+    const intent = text.replace(/^(coordinate|orchestrate)\s+/i, "");
+    const steps: CoordinationStep[] = [
+      { id: "research", role: "research", query: `research ${intent}` },
+      { id: "analysis", role: "analysis", query: `analyze ${intent}`, dependsOn: "research" },
+    ];
+
+    const result = await bridge.coordinate(steps, async (query) => {
+      const found = await kompass.find(query, { limit: 3 });
+      return found.agents[0] ?? null;
+    });
+
+    return {
+      id: `kompass-coord-${Date.now()}`,
+      status: { state: result.completedCount > 0 ? "completed" : "failed" },
+      artifacts: [
+        {
+          parts: [
+            {
+              type: "text",
+              text: `Coordination complete: ${result.completedCount}/${result.totalCount} steps succeeded.`,
+            },
+            { type: "data", data: result },
+          ],
+        },
+      ],
+    };
+  }
+
+  // Default: treat as a discovery query
+  const result = await kompass.find(text, { limit: 10 });
+
+  return {
+    id: `kompass-${Date.now()}`,
+    status: { state: "completed" },
+    artifacts: [
+      {
+        parts: [
+          {
+            type: "text",
+            text: `Found ${result.agents.length} agents for "${text}":\n\n` +
+              result.agents
+                .map((a) => `• ${a.name} [${a.protocol}] — ${a.description.slice(0, 80)}`)
+                .join("\n"),
+          },
+          { type: "data", data: { agents: result.agents, sources: result.sources } },
+        ],
+      },
+    ],
+  };
+}
+
+function jsonRpcError(id: unknown, code: number, message: string) {
+  return {
+    jsonrpc: "2.0",
+    id,
+    error: { code, message },
+  };
+}