agenthub-multiagent-mcp 1.1.5 → 1.3.0

package/dist/websocket.d.ts

@@ -0,0 +1,66 @@
+/**
+ * WebSocket client for real-time push notifications from AgentHub server.
+ *
+ * Receives pushed tasks and messages and stores them in a queue
+ * for inclusion in tool responses.
+ */
+import type { Message, PendingTask } from "./client.js";
+export interface WSMessage {
+    type: "task" | "message" | "ping" | "pong" | "ack";
+    timestamp: string;
+    task?: PendingTask;
+    message?: Message;
+    data?: unknown;
+}
+export interface PushedItems {
+    tasks: PendingTask[];
+    messages: Message[];
+}
+export declare class WebSocketClient {
+    private ws;
+    private baseUrl;
+    private apiKey;
+    private agentId;
+    private reconnectAttempts;
+    private maxReconnectAttempts;
+    private reconnectDelay;
+    private reconnectTimer;
+    private isConnecting;
+    private isClosing;
+    private pushedTasks;
+    private pushedMessages;
+    constructor(baseUrl: string, apiKey: string);
+    /**
+     * Connect to WebSocket for a specific agent
+     */
+    connect(agentId: string): void;
+    /**
+     * Handle incoming WebSocket message
+     */
+    private handleMessage;
+    /**
+     * Send acknowledgment for received item
+     */
+    private sendAck;
+    /**
+     * Send a message over WebSocket
+     */
+    private send;
+    /**
+     * Schedule reconnection with exponential backoff
+     */
+    private scheduleReconnect;
+    /**
+     * Get and clear pushed items
+     */
+    getPushedItems(): PushedItems;
+    /**
+     * Check if connected
+     */
+    isConnected(): boolean;
+    /**
+     * Close the WebSocket connection
+     */
+    close(): void;
+}
+//# sourceMappingURL=websocket.d.ts.map

package/dist/websocket.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.d.ts","sourceRoot":"","sources":["../src/websocket.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAExD,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,GAAG,MAAM,GAAG,KAAK,CAAC;IACnD,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,IAAI,CAAC,EAAE,OAAO,CAAC;CAChB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,WAAW,EAAE,CAAC;IACrB,QAAQ,EAAE,OAAO,EAAE,CAAC;CACrB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAuB;IACtC,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,oBAAoB,CAAK;IACjC,OAAO,CAAC,cAAc,CAAQ;IAC9B,OAAO,CAAC,cAAc,CAA+B;IACrD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAS;IAG1B,OAAO,CAAC,WAAW,CAAqB;IACxC,OAAO,CAAC,cAAc,CAAiB;gBAE3B,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;IAS3C;;OAEG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAiD9B;;OAEG;IACH,OAAO,CAAC,aAAa;IA+BrB;;OAEG;IACH,OAAO,CAAC,OAAO;IAQf;;OAEG;IACH,OAAO,CAAC,IAAI;IAUZ;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;OAEG;IACH,cAAc,IAAI,WAAW;IAa7B;;OAEG;IACH,WAAW,IAAI,OAAO;IAItB;;OAEG;IACH,KAAK,IAAI,IAAI;CAuBd"}

package/dist/websocket.js

@@ -0,0 +1,196 @@
+/**
+ * WebSocket client for real-time push notifications from AgentHub server.
+ *
+ * Receives pushed tasks and messages and stores them in a queue
+ * for inclusion in tool responses.
+ */
+import WebSocket from "ws";
+export class WebSocketClient {
+    ws = null;
+    baseUrl;
+    apiKey;
+    agentId = null;
+    reconnectAttempts = 0;
+    maxReconnectAttempts = 5;
+    reconnectDelay = 1000; // Start with 1 second
+    reconnectTimer = null;
+    isConnecting = false;
+    isClosing = false;
+    // Queue for pushed items
+    pushedTasks = [];
+    pushedMessages = [];
+    constructor(baseUrl, apiKey) {
+        // Convert HTTP URL to WebSocket URL
+        this.baseUrl = baseUrl
+            .replace(/^https:/, "wss:")
+            .replace(/^http:/, "ws:")
+            .replace(/\/$/, "");
+        this.apiKey = apiKey;
+    }
+    /**
+     * Connect to WebSocket for a specific agent
+     */
+    connect(agentId) {
+        if (this.isConnecting || (this.ws && this.ws.readyState === WebSocket.OPEN)) {
+            return;
+        }
+        this.agentId = agentId;
+        this.isConnecting = true;
+        this.isClosing = false;
+        const url = `${this.baseUrl}/ws/agents/${agentId}?api_key=${encodeURIComponent(this.apiKey)}`;
+        try {
+            this.ws = new WebSocket(url);
+            this.ws.on("open", () => {
+                this.isConnecting = false;
+                this.reconnectAttempts = 0;
+                this.reconnectDelay = 1000;
+                console.error(`[AgentHub] WebSocket connected for agent: ${agentId}`);
+            });
+            this.ws.on("message", (data) => {
+                try {
+                    const msg = JSON.parse(data.toString());
+                    this.handleMessage(msg);
+                }
+                catch (error) {
+                    console.error("[AgentHub] Failed to parse WebSocket message:", error);
+                }
+            });
+            this.ws.on("close", () => {
+                this.isConnecting = false;
+                if (!this.isClosing) {
+                    console.error("[AgentHub] WebSocket disconnected, will reconnect...");
+                    this.scheduleReconnect();
+                }
+            });
+            this.ws.on("error", (error) => {
+                this.isConnecting = false;
+                console.error("[AgentHub] WebSocket error:", error.message);
+            });
+        }
+        catch (error) {
+            this.isConnecting = false;
+            console.error("[AgentHub] Failed to create WebSocket:", error);
+            this.scheduleReconnect();
+        }
+    }
+    /**
+     * Handle incoming WebSocket message
+     */
+    handleMessage(msg) {
+        switch (msg.type) {
+            case "task":
+                if (msg.task) {
+                    console.error(`[AgentHub] Received pushed task: ${msg.task.id}`);
+                    this.pushedTasks.push(msg.task);
+                    // Send ack
+                    this.sendAck(msg.task.id);
+                }
+                break;
+            case "message":
+                if (msg.message) {
+                    console.error(`[AgentHub] Received pushed message: ${msg.message.id}`);
+                    this.pushedMessages.push(msg.message);
+                    // Send ack
+                    this.sendAck(msg.message.id);
+                }
+                break;
+            case "ping":
+                // Respond with pong
+                this.send({ type: "pong", timestamp: new Date().toISOString() });
+                break;
+            default:
+                // Ignore other message types
+                break;
+        }
+    }
+    /**
+     * Send acknowledgment for received item
+     */
+    sendAck(itemId) {
+        this.send({
+            type: "ack",
+            timestamp: new Date().toISOString(),
+            data: { id: itemId },
+        });
+    }
+    /**
+     * Send a message over WebSocket
+     */
+    send(msg) {
+        if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+            try {
+                this.ws.send(JSON.stringify(msg));
+            }
+            catch (error) {
+                console.error("[AgentHub] Failed to send WebSocket message:", error);
+            }
+        }
+    }
+    /**
+     * Schedule reconnection with exponential backoff
+     */
+    scheduleReconnect() {
+        if (this.isClosing || !this.agentId) {
+            return;
+        }
+        if (this.reconnectAttempts >= this.maxReconnectAttempts) {
+            console.error("[AgentHub] Max reconnect attempts reached, giving up");
+            return;
+        }
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+        }
+        this.reconnectTimer = setTimeout(() => {
+            this.reconnectAttempts++;
+            console.error(`[AgentHub] Reconnect attempt ${this.reconnectAttempts}/${this.maxReconnectAttempts}`);
+            if (this.agentId) {
+                this.connect(this.agentId);
+            }
+        }, this.reconnectDelay);
+        // Exponential backoff with max of 30 seconds
+        this.reconnectDelay = Math.min(this.reconnectDelay * 2, 30000);
+    }
+    /**
+     * Get and clear pushed items
+     */
+    getPushedItems() {
+        const items = {
+            tasks: [...this.pushedTasks],
+            messages: [...this.pushedMessages],
+        };
+        // Clear the queues
+        this.pushedTasks = [];
+        this.pushedMessages = [];
+        return items;
+    }
+    /**
+     * Check if connected
+     */
+    isConnected() {
+        return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
+    }
+    /**
+     * Close the WebSocket connection
+     */
+    close() {
+        this.isClosing = true;
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+        if (this.ws) {
+            try {
+                this.ws.close();
+            }
+            catch (error) {
+                // Ignore close errors
+            }
+            this.ws = null;
+        }
+        this.agentId = null;
+        this.pushedTasks = [];
+        this.pushedMessages = [];
+        console.error("[AgentHub] WebSocket closed");
+    }
+}
+//# sourceMappingURL=websocket.js.map

package/dist/websocket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../src/websocket.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,SAAS,MAAM,IAAI,CAAC;AAgB3B,MAAM,OAAO,eAAe;IAClB,EAAE,GAAqB,IAAI,CAAC;IAC5B,OAAO,CAAS;IAChB,MAAM,CAAS;IACf,OAAO,GAAkB,IAAI,CAAC;IAC9B,iBAAiB,GAAG,CAAC,CAAC;IACtB,oBAAoB,GAAG,CAAC,CAAC;IACzB,cAAc,GAAG,IAAI,CAAC,CAAC,sBAAsB;IAC7C,cAAc,GAA0B,IAAI,CAAC;IAC7C,YAAY,GAAG,KAAK,CAAC;IACrB,SAAS,GAAG,KAAK,CAAC;IAE1B,yBAAyB;IACjB,WAAW,GAAkB,EAAE,CAAC;IAChC,cAAc,GAAc,EAAE,CAAC;IAEvC,YAAY,OAAe,EAAE,MAAc;QACzC,oCAAoC;QACpC,IAAI,CAAC,OAAO,GAAG,OAAO;aACnB,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC;aAC1B,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC;aACxB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,OAAe;QACrB,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5E,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QAEvB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,cAAc,OAAO,YAAY,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAE9F,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC;YAE7B,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACtB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBAC3B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;gBAC3B,OAAO,CAAC,KAAK,CAAC,6CAA6C,OAAO,EAAE,CAAC,CAAC;YACxE,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAuB,EAAE,EAAE;gBAChD,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAc,CAAC;oBACrD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;gBAC1B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,+CAA+C,EAAE,KAAK,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBACvB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACpB,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;oBACtE,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBAC5B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;gBAC1B,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAC1B,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;YAC/D,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,GAAc;QAClC,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,MAAM;gBACT,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,oCAAoC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;oBACjE,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAChC,WAAW;oBACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC5B,CAAC;gBACD,MAAM;YAER,KAAK,SAAS;gBACZ,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBAChB,OAAO,CAAC,KAAK,CAAC,uCAAuC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;oBACvE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACtC,WAAW;oBACX,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC/B,CAAC;gBACD,MAAM;YAER,KAAK,MAAM;gBACT,oBAAoB;gBACpB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;gBACjE,MAAM;YAER;gBACE,6BAA6B;gBAC7B,MAAM;QACV,CAAC;IACH,CAAC;IAED;;OAEG;IACK,OAAO,CAAC,MAAc;QAC5B,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,KAAK;YACX,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,GAAuB;QAClC,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;YACpC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,8CAA8C,EAAE,KAAK,CAAC,CAAC;YACvE,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB;QACvB,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACpC,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;YACtE,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE;YACpC,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACzB,OAAO,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,iBAAiB,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC,CAAC;YACrG,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAExB,6CAA6C;QAC7C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,EAAE,KAAK,CAAC,CAAC;IACjE,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,MAAM,KAAK,GAAG;YACZ,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;YAC5B,QAAQ,EAAE,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC;SACnC,CAAC;QAEF,mBAAmB;QACnB,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAEzB,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,EAAE,KAAK,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,CAAC;IACnE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,sBAAsB;YACxB,CAAC;YACD,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QAEzB,OAAO,CAAC,KAAK,CAAC,6BAA6B,CAAC,CAAC;IAC/C,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agenthub-multiagent-mcp",
-  "version": "1.1.5",
+  "version": "1.3.0",
   "description": "MCP server for AgentHub multi-agent communication",
   "type": "module",
   "main": "dist/index.js",
@@ -39,10 +39,13 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.0",
+    "open": "^11.0.0",
+    "ws": "^8.18.0",
     "zod": "^3.23.0"
   },
   "devDependencies": {
     "@types/node": "^20.0.0",
+    "@types/ws": "^8.5.13",
     "tsx": "^4.0.0",
     "typescript": "^5.0.0",
     "vitest": "^4.0.16"

package/src/client.ts

@@ -28,6 +28,76 @@ export interface PendingTask {
   created_at: string;
 }
 
+// Ticket system types
+export interface TicketTask {
+  id: string;
+  org_id: string;
+  project_id: string;
+  epic_id?: string;
+  story_id?: string;
+  key: string;
+  type: string;
+  title: string;
+  description?: string;
+  status: string;
+  priority: string;
+  points?: number;
+  review_mode: string;
+  assignee_id?: string;
+  assigned_agent?: string;
+  claimed_by?: string;
+  labels?: string[];
+  sprint_id?: string;
+  due_date?: string;
+  created_by: string;
+  created_at: string;
+  updated_at: string;
+}
+
+export interface TicketTaskContext {
+  task: TicketTask;
+  story?: {
+    id: string;
+    title: string;
+    description?: string;
+    status: string;
+  };
+  epic?: {
+    id: string;
+    title: string;
+    description?: string;
+    status: string;
+  };
+  project?: {
+    id: string;
+    key: string;
+    name: string;
+    description?: string;
+  };
+}
+
+export interface TaskCheckpoint {
+  id: string;
+  task_id: string;
+  task_key: string;
+  agent_id: string;
+  description: string;
+  status: string;
+  reviewed_by?: string;
+  review_notes?: string;
+  reviewed_at?: string;
+  created_at: string;
+}
+
+export interface CheckpointResponse {
+  id: string;
+  task_id: string;
+  task_key: string;
+  status: string;
+  slack_posted: boolean;
+  created_at: string;
+}
+
 export interface TaskCompletionResponse {
   task_id: string;
   elapsed_time: string;
@@ -150,11 +220,19 @@ export class ApiClient {
   async startWork(
     agentId: string,
     task: string,
-    project?: string
-  ): Promise<{ acknowledged: boolean; slack_posted: boolean }> {
+    project?: string,
+    taskId?: string
+  ): Promise<{
+    acknowledged: boolean;
+    slack_posted: boolean;
+    task_id?: string;
+    task_key?: string;
+    task_linked?: boolean;
+  }> {
     return this.request("POST", `/agents/${agentId}/start-work`, {
       task,
       project,
+      task_id: taskId,
     });
   }
 
@@ -294,4 +372,114 @@ export class ApiClient {
       reason,
     });
   }
+
+  // Ticket system methods (for agent integration)
+
+  async getAvailableTasks(params?: {
+    status?: string;
+    project_id?: string;
+    priority?: string;
+  }): Promise<{ tasks: TicketTask[]; total: number }> {
+    const query = new URLSearchParams();
+    if (params?.status) query.set("status", params.status);
+    if (params?.project_id) query.set("project_id", params.project_id);
+    if (params?.priority) query.set("priority", params.priority);
+    const queryStr = query.toString() ? `?${query}` : "";
+    return this.request("GET", `/agent-tickets/available${queryStr}`);
+  }
+
+  async getTicketTask(taskId: string): Promise<TicketTaskContext> {
+    return this.request("GET", `/agent-tickets/${taskId}`);
+  }
+
+  async claimTicketTask(
+    taskId: string,
+    agentId: string
+  ): Promise<{
+    success: boolean;
+    task: TicketTask;
+    task_id: string;
+    task_key: string;
+  }> {
+    return this.request("POST", `/agent-tickets/${taskId}/claim`, {
+      agent_id: agentId,
+    });
+  }
+
+  async createCheckpoint(
+    taskId: string,
+    agentId: string,
+    description: string
+  ): Promise<CheckpointResponse> {
+    return this.request("POST", `/agent-tickets/${taskId}/checkpoint`, {
+      agent_id: agentId,
+      description,
+    });
+  }
+
+  // Browser-based registration flow
+  async initRegistration(agentId?: string, agentName?: string): Promise<{
+    session_token: string;
+    dashboard_url: string;
+    expires_at: string;
+  }> {
+    return this.request("POST", "/agents/register-init", {
+      agent_id: agentId,
+      agent_name: agentName,
+    });
+  }
+
+  async pollRegistrationCallback(token: string): Promise<{
+    status: "pending" | "completed" | "expired";
+    agent_id?: string;
+    agent_name?: string;
+    agent_type?: string;
+    connect_token?: string;
+    expires_at?: string;
+  }> {
+    const url = `${this.baseUrl}/agents/register-callback/${token}`;
+    const response = await fetch(url, {
+      method: "GET",
+      headers: {
+        "Content-Type": "application/json",
+        "X-API-Key": this.apiKey,
+      },
+    });
+
+    if (response.status === 202) {
+      // Still pending
+      const data = await response.json() as { status: string; expires_at: string };
+      return { status: "pending", expires_at: data.expires_at };
+    }
+
+    if (response.status === 410) {
+      // Expired
+      return { status: "expired" };
+    }
+
+    if (!response.ok) {
+      const data = await response.json() as { message?: string };
+      throw new Error(data.message || `Request failed: ${response.status}`);
+    }
+
+    const data = await response.json() as {
+      status: string;
+      agent_id: string;
+      agent_name: string;
+      agent_type: string;
+      connect_token: string;
+    };
+
+    return {
+      status: "completed",
+      agent_id: data.agent_id,
+      agent_name: data.agent_name,
+      agent_type: data.agent_type,
+      connect_token: data.connect_token,
+    };
+  }
+
+  getBaseUrl(): string {
+    return this.baseUrl;
+  }
 }