agenthub-multiagent-mcp 1.1.2 → 1.1.4

package/src/tools/index.ts

@@ -1,666 +1,672 @@
-/**
- * MCP tool definitions and handlers
- */
-
-import { ApiClient, Message } from "../client.js";
-import type { Tool } from "@modelcontextprotocol/sdk/types.js";
-import * as state from "../state.js";
-
-export interface ToolContext {
-  getCurrentAgentId: () => string;
-  setCurrentAgentId: (id: string) => void;
-  stopHeartbeat: () => void;
-  getWorkingDir: () => string;
-}
-
-interface UrgentAction {
-  required: true;
-  instruction: string;
-  message: Message;
-}
-
-interface WrappedResponse {
-  result: unknown;
-  pending_messages: Message[];
-  pending_count: number;
-  has_more: boolean;
-  urgent_action: UrgentAction | null;
-}
-
-/**
- * Wraps a tool response with pending messages
- * This enables automatic message delivery without extra API calls
- */
-async function wrapWithMessages(
-  client: ApiClient,
-  agentId: string | null,
-  result: unknown
-): Promise<WrappedResponse | unknown> {
-  // If not registered, return raw result
-  if (!agentId) return result;
-
-  try {
-    // Fetch unread messages
-    const inbox = await client.getInbox(agentId, true, 10);
-    const messages = inbox.messages || [];
-
-    // Sort: urgent first, then by created_at
-    messages.sort((a, b) => {
-      const aUrgent = isUrgentMessage(a);
-      const bUrgent = isUrgentMessage(b);
-      if (aUrgent && !bUrgent) return -1;
-      if (!aUrgent && bUrgent) return 1;
-      return new Date(b.created_at).getTime() - new Date(a.created_at).getTime();
-    });
-
-    // Find urgent message for action
-    const urgentMsg = messages.find(isUrgentMessage);
-
-    let urgentAction: UrgentAction | null = null;
-    if (urgentMsg) {
-      urgentAction = {
-        required: true,
-        instruction: buildUrgentInstruction(urgentMsg),
-        message: urgentMsg,
-      };
-    }
-
-    return {
-      result,
-      pending_messages: messages,
-      pending_count: inbox.total,
-      has_more: inbox.total > messages.length,
-      urgent_action: urgentAction,
-    };
-  } catch (error) {
-    // Don't break tool call if message fetch fails
-    console.warn("Failed to fetch messages for wrapping:", error);
-    return result;
-  }
-}
-
-/**
- * Check if a message is urgent (requires immediate action)
- */
-function isUrgentMessage(msg: Message): boolean {
-  return (
-    msg.priority === "urgent" ||
-    (msg.priority === "high" && msg.type === "question")
-  );
-}
-
-/**
- * Build instruction text for urgent messages
- */
-function buildUrgentInstruction(msg: Message): string {
-  return `URGENT: Agent [${msg.from_agent}] needs immediate assistance.
-
-Query: ${msg.subject}
-
-Details: ${msg.body}
-
-Spawn a subagent with Task tool to research and respond to this query. Use send_message to reply when done.`;
-}
-
-export function registerTools(): Tool[] {
-  return [
-    // Registration tools
-    {
-      name: "agent_register",
-      description: "Register this agent with AgentHub. Must be called before using other tools.",
-      inputSchema: {
-        type: "object",
-        properties: {
-          id: {
-            type: "string",
-            description: "Unique agent identifier (e.g., 'backend-1', 'tester')",
-          },
-          name: {
-            type: "string",
-            description: "Human-friendly name for this agent",
-          },
-          model: {
-            type: "string",
-            description: "Model name (e.g., 'claude-opus-4.5', 'gpt-4-turbo', 'o3', 'gemini-2.0-pro'). Provider is auto-detected.",
-          },
-        },
-        required: ["id", "model"],
-      },
-    },
-    {
-      name: "agent_start_work",
-      description: "Declare what you're currently working on. Sends notification to Slack.",
-      inputSchema: {
-        type: "object",
-        properties: {
-          task: {
-            type: "string",
-            description: "Description of the task you're starting",
-          },
-          project: {
-            type: "string",
-            description: "Project name (optional)",
-          },
-        },
-        required: ["task"],
-      },
-    },
-    {
-      name: "agent_set_status",
-      description: "Update your status (online or busy)",
-      inputSchema: {
-        type: "object",
-        properties: {
-          status: {
-            type: "string",
-            enum: ["online", "busy"],
-            description: "New status",
-          },
-        },
-        required: ["status"],
-      },
-    },
-    {
-      name: "agent_disconnect",
-      description: "Gracefully disconnect from AgentHub",
-      inputSchema: {
-        type: "object",
-        properties: {},
-      },
-    },
-
-    // Messaging tools
-    {
-      name: "send_message",
-      description: "Send a direct message to another agent",
-      inputSchema: {
-        type: "object",
-        properties: {
-          to: {
-            type: "string",
-            description: "Target agent ID",
-          },
-          type: {
-            type: "string",
-            enum: ["task", "status", "question", "response", "context"],
-            description: "Type of message",
-          },
-          subject: {
-            type: "string",
-            description: "Message subject/summary",
-          },
-          body: {
-            type: "string",
-            description: "Full message content",
-          },
-          priority: {
-            type: "string",
-            enum: ["normal", "high", "urgent"],
-            description: "Message priority (default: normal)",
-          },
-        },
-        required: ["to", "type", "subject", "body"],
-      },
-    },
-    {
-      name: "send_to_channel",
-      description: "Send a message to all members of a channel",
-      inputSchema: {
-        type: "object",
-        properties: {
-          channel: {
-            type: "string",
-            description: "Target channel name",
-          },
-          type: {
-            type: "string",
-            enum: ["task", "status", "question", "response", "context"],
-            description: "Type of message",
-          },
-          subject: {
-            type: "string",
-            description: "Message subject/summary",
-          },
-          body: {
-            type: "string",
-            description: "Full message content",
-          },
-        },
-        required: ["channel", "type", "subject", "body"],
-      },
-    },
-    {
-      name: "broadcast",
-      description: "Send a message to all registered agents",
-      inputSchema: {
-        type: "object",
-        properties: {
-          type: {
-            type: "string",
-            enum: ["task", "status", "question", "response", "context"],
-            description: "Type of message",
-          },
-          subject: {
-            type: "string",
-            description: "Message subject/summary",
-          },
-          body: {
-            type: "string",
-            description: "Full message content",
-          },
-        },
-        required: ["type", "subject", "body"],
-      },
-    },
-    {
-      name: "check_inbox",
-      description: "Check for incoming messages",
-      inputSchema: {
-        type: "object",
-        properties: {
-          unread_only: {
-            type: "boolean",
-            description: "Only return unread messages (default: false)",
-          },
-        },
-      },
-    },
-    {
-      name: "mark_read",
-      description: "Mark a message as read",
-      inputSchema: {
-        type: "object",
-        properties: {
-          message_id: {
-            type: "string",
-            description: "ID of the message to mark as read",
-          },
-        },
-        required: ["message_id"],
-      },
-    },
-    {
-      name: "reply",
-      description: "Reply to a message",
-      inputSchema: {
-        type: "object",
-        properties: {
-          message_id: {
-            type: "string",
-            description: "ID of the message to reply to",
-          },
-          body: {
-            type: "string",
-            description: "Reply content",
-          },
-        },
-        required: ["message_id", "body"],
-      },
-    },
-
-    // Discovery tools
-    {
-      name: "list_agents",
-      description: "List all registered agents",
-      inputSchema: {
-        type: "object",
-        properties: {
-          status: {
-            type: "string",
-            enum: ["online", "busy", "offline"],
-            description: "Filter by status (optional)",
-          },
-        },
-      },
-    },
-    {
-      name: "get_agent",
-      description: "Get details about a specific agent",
-      inputSchema: {
-        type: "object",
-        properties: {
-          id: {
-            type: "string",
-            description: "Agent ID to look up",
-          },
-        },
-        required: ["id"],
-      },
-    },
-    {
-      name: "list_channels",
-      description: "List all available channels",
-      inputSchema: {
-        type: "object",
-        properties: {},
-      },
-    },
-    {
-      name: "join_channel",
-      description: "Subscribe to a channel to receive its messages",
-      inputSchema: {
-        type: "object",
-        properties: {
-          channel: {
-            type: "string",
-            description: "Channel name to join",
-          },
-        },
-        required: ["channel"],
-      },
-    },
-    {
-      name: "leave_channel",
-      description: "Unsubscribe from a channel",
-      inputSchema: {
-        type: "object",
-        properties: {
-          channel: {
-            type: "string",
-            description: "Channel name to leave",
-          },
-        },
-        required: ["channel"],
-      },
-    },
-
-    // Task completion tools
-    {
-      name: "agent_complete_task",
-      description:
-        "Mark current task as complete with a summary. Posts to Slack and notifies team.",
-      inputSchema: {
-        type: "object",
-        properties: {
-          summary: {
-            type: "string",
-            description: "Summary of what was accomplished",
-          },
-          outcome: {
-            type: "string",
-            enum: ["completed", "blocked", "partial", "needs_review", "handed_off"],
-            description: "Task outcome status",
-          },
-          files_changed: {
-            type: "array",
-            items: { type: "string" },
-            description: "List of files modified (optional)",
-          },
-          time_spent: {
-            type: "string",
-            description: "Override auto-calculated time (optional, e.g., '45 minutes')",
-          },
-          next_steps: {
-            type: "string",
-            description:
-              "What comes next, can @mention agents for handoff (optional)",
-          },
-        },
-        required: ["summary", "outcome"],
-      },
-    },
-    {
-      name: "get_pending_tasks",
-      description: "Get tasks assigned to you by other agents",
-      inputSchema: {
-        type: "object",
-        properties: {},
-      },
-    },
-    {
-      name: "accept_task",
-      description: "Accept a pending task and start working on it",
-      inputSchema: {
-        type: "object",
-        properties: {
-          task_id: {
-            type: "string",
-            description: "ID of the task to accept",
-          },
-        },
-        required: ["task_id"],
-      },
-    },
-    {
-      name: "decline_task",
-      description: "Decline a pending task with reason",
-      inputSchema: {
-        type: "object",
-        properties: {
-          task_id: {
-            type: "string",
-            description: "ID of the task to decline",
-          },
-          reason: {
-            type: "string",
-            description: "Reason for declining",
-          },
-        },
-        required: ["task_id", "reason"],
-      },
-    },
-  ];
-}
-
-export async function handleToolCall(
-  name: string,
-  args: Record<string, unknown>,
-  client: ApiClient,
-  context: ToolContext
-): Promise<unknown> {
-  const agentId = context.getCurrentAgentId();
-
-  switch (name) {
-    // Registration
-    case "agent_register": {
-      const workingDir = context.getWorkingDir();
-      const owner = state.getCurrentOwner();
-      const requestedId = args.id as string;
-
-      // Check for existing state file (reconnection)
-      const existingState = state.loadState(workingDir);
-
-      if (existingState) {
-        // Verify owner matches
-        if (existingState.owner !== owner) {
-          // Different user - delete state and register fresh
-          state.deleteState(workingDir);
-        } else {
-          // Try to reconnect
-          try {
-            const reconnectResult = await client.reconnectAgent(
-              existingState.agent_id,
-              existingState.token,
-              owner
-            );
-
-            context.setCurrentAgentId(existingState.agent_id);
-
-            // Update state with any new info
-            state.saveState(workingDir, {
-              ...existingState,
-              last_task: undefined, // Cleared on reconnect
-            });
-
-            return {
-              ...reconnectResult,
-              mode: "reconnected",
-              message: reconnectResult.was_offline
-                ? `Reconnected after timeout. You have ${reconnectResult.pending_tasks_count} pending tasks and ${reconnectResult.unread_messages_count} unread messages.`
-                : `Reconnected. You have ${reconnectResult.pending_tasks_count} pending tasks and ${reconnectResult.unread_messages_count} unread messages.`,
-            };
-          } catch (error) {
-            // Reconnection failed (invalid token, agent not found, etc.)
-            console.warn("Reconnection failed, registering fresh:", error);
-            state.deleteState(workingDir);
-          }
-        }
-      }
-
-      // Fresh registration
-      const result = await client.registerAgent(
-        requestedId,
-        args.name as string | undefined,
-        owner,
-        workingDir,
-        args.model as string
-      );
-
-      context.setCurrentAgentId(requestedId);
-
-      // Save state for future reconnection
-      state.saveState(workingDir, {
-        agent_id: requestedId,
-        owner,
-        token: result.token,
-        registered_at: result.registered_at,
-      });
-
-      return {
-        ...result,
-        mode: "registered",
-        message: `Registered as ${requestedId}. You have ${result.pending_tasks_count} pending tasks and ${result.unread_messages_count} unread messages.`,
-      };
-    }
-
-    case "agent_start_work": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.startWork(
-        agentId,
-        args.task as string,
-        args.project as string | undefined
-      );
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "agent_set_status": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.heartbeat(agentId, args.status as "online" | "busy");
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "agent_disconnect": {
-      // No wrapping - agent is going offline
-      if (!agentId) throw new Error("Not registered.");
-      context.stopHeartbeat();
-      return client.disconnect(agentId);
-    }
-
-    // Messaging
-    case "send_message": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.sendMessage({
-        from_agent: agentId,
-        to_agent: args.to as string,
-        type: args.type as string,
-        subject: args.subject as string,
-        body: args.body as string,
-        priority: args.priority as string | undefined,
-      });
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "send_to_channel": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.sendMessage({
-        from_agent: agentId,
-        to_channel: args.channel as string,
-        type: args.type as string,
-        subject: args.subject as string,
-        body: args.body as string,
-      });
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "broadcast": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.sendMessage({
-        from_agent: agentId,
-        broadcast: true,
-        type: args.type as string,
-        subject: args.subject as string,
-        body: args.body as string,
-      });
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "check_inbox": {
-      // No wrapping - already fetching messages
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      return client.getInbox(agentId, args.unread_only as boolean | undefined);
-    }
-
-    case "mark_read": {
-      const result = await client.markRead(args.message_id as string);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "reply": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.reply(args.message_id as string, agentId, args.body as string);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    // Discovery
-    case "list_agents": {
-      const result = await client.listAgents(args.status as string | undefined);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "get_agent": {
-      const result = await client.getAgent(args.id as string);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "list_channels": {
-      const result = await client.listChannels();
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "join_channel": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.joinChannel(args.channel as string, agentId);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "leave_channel": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.leaveChannel(args.channel as string, agentId);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    // Task completion
-    case "agent_complete_task": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.completeTask(agentId, {
-        summary: args.summary as string,
-        outcome: args.outcome as "completed" | "blocked" | "partial" | "needs_review" | "handed_off",
-        files_changed: args.files_changed as string[] | undefined,
-        time_spent: args.time_spent as string | undefined,
-        next_steps: args.next_steps as string | undefined,
-      });
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "get_pending_tasks": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.getPendingTasks(agentId);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "accept_task": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.acceptTask(agentId, args.task_id as string);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    case "decline_task": {
-      if (!agentId) throw new Error("Not registered. Call agent_register first.");
-      const result = await client.declineTask(agentId, args.task_id as string, args.reason as string);
-      return wrapWithMessages(client, agentId, result);
-    }
-
-    default:
-      throw new Error(`Unknown tool: ${name}`);
-  }
-}
+/**
+ * MCP tool definitions and handlers
+ */
+
+import { ApiClient, Message } from "../client.js";
+import type { Tool } from "@modelcontextprotocol/sdk/types.js";
+import * as state from "../state.js";
+
+export interface ToolContext {
+  getCurrentAgentId: () => string;
+  setCurrentAgentId: (id: string) => void;
+  stopHeartbeat: () => void;
+  getWorkingDir: () => string;
+}
+
+interface UrgentAction {
+  required: true;
+  instruction: string;
+  message: Message;
+}
+
+interface WrappedResponse {
+  result: unknown;
+  pending_messages: Message[];
+  pending_count: number;
+  has_more: boolean;
+  urgent_action: UrgentAction | null;
+}
+
+/**
+ * Wraps a tool response with pending messages
+ * This enables automatic message delivery without extra API calls
+ */
+async function wrapWithMessages(
+  client: ApiClient,
+  agentId: string | null,
+  result: unknown
+): Promise<WrappedResponse | unknown> {
+  // If not registered, return raw result
+  if (!agentId) return result;
+
+  try {
+    // Fetch unread messages
+    const inbox = await client.getInbox(agentId, true, 10);
+    const messages = inbox.messages || [];
+
+    // Sort: urgent first, then by created_at
+    messages.sort((a, b) => {
+      const aUrgent = isUrgentMessage(a);
+      const bUrgent = isUrgentMessage(b);
+      if (aUrgent && !bUrgent) return -1;
+      if (!aUrgent && bUrgent) return 1;
+      return new Date(b.created_at).getTime() - new Date(a.created_at).getTime();
+    });
+
+    // Find urgent message for action
+    const urgentMsg = messages.find(isUrgentMessage);
+
+    let urgentAction: UrgentAction | null = null;
+    if (urgentMsg) {
+      urgentAction = {
+        required: true,
+        instruction: buildUrgentInstruction(urgentMsg),
+        message: urgentMsg,
+      };
+    }
+
+    return {
+      result,
+      pending_messages: messages,
+      pending_count: inbox.total,
+      has_more: inbox.total > messages.length,
+      urgent_action: urgentAction,
+    };
+  } catch (error) {
+    // Don't break tool call if message fetch fails
+    console.warn("Failed to fetch messages for wrapping:", error);
+    return result;
+  }
+}
+
+/**
+ * Check if a message is urgent (requires immediate action)
+ */
+function isUrgentMessage(msg: Message): boolean {
+  return (
+    msg.priority === "urgent" ||
+    (msg.priority === "high" && msg.type === "question")
+  );
+}
+
+/**
+ * Build instruction text for urgent messages
+ */
+function buildUrgentInstruction(msg: Message): string {
+  return `URGENT: Agent [${msg.from_agent}] needs immediate assistance.
+
+Query: ${msg.subject}
+
+Details: ${msg.body}
+
+Spawn a subagent with Task tool to research and respond to this query. Use send_message to reply when done.`;
+}
+
+export function registerTools(): Tool[] {
+  return [
+    // Registration tools
+    {
+      name: "agent_register",
+      description: "Register this agent with AgentHub. Must be called before using other tools.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          id: {
+            type: "string",
+            description: "Unique agent identifier (e.g., 'backend-1', 'tester')",
+          },
+          name: {
+            type: "string",
+            description: "Human-friendly name for this agent",
+          },
+          model: {
+            type: "string",
+            description: "Model name (e.g., 'claude-opus-4.5', 'gpt-4-turbo', 'o3', 'gemini-2.0-pro'). Provider is auto-detected.",
+          },
+        },
+        required: ["id", "model"],
+      },
+    },
+    {
+      name: "agent_start_work",
+      description: "Declare what you're currently working on. Sends notification to Slack.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          task: {
+            type: "string",
+            description: "Description of the task you're starting",
+          },
+          project: {
+            type: "string",
+            description: "Project name (optional)",
+          },
+        },
+        required: ["task"],
+      },
+    },
+    {
+      name: "agent_set_status",
+      description: "Update your status (online or busy)",
+      inputSchema: {
+        type: "object",
+        properties: {
+          status: {
+            type: "string",
+            enum: ["online", "busy"],
+            description: "New status",
+          },
+        },
+        required: ["status"],
+      },
+    },
+    {
+      name: "agent_disconnect",
+      description: "Gracefully disconnect from AgentHub",
+      inputSchema: {
+        type: "object",
+        properties: {},
+      },
+    },
+
+    // Messaging tools
+    {
+      name: "send_message",
+      description: "Send a direct message to another agent",
+      inputSchema: {
+        type: "object",
+        properties: {
+          to: {
+            type: "string",
+            description: "Target agent ID",
+          },
+          type: {
+            type: "string",
+            enum: ["task", "status", "question", "response", "context"],
+            description: "Type of message",
+          },
+          subject: {
+            type: "string",
+            description: "Message subject/summary",
+          },
+          body: {
+            type: "string",
+            description: "Full message content",
+          },
+          priority: {
+            type: "string",
+            enum: ["normal", "high", "urgent"],
+            description: "Message priority (default: normal)",
+          },
+        },
+        required: ["to", "type", "subject", "body"],
+      },
+    },
+    {
+      name: "send_to_channel",
+      description: "Send a message to all members of a channel",
+      inputSchema: {
+        type: "object",
+        properties: {
+          channel: {
+            type: "string",
+            description: "Target channel name",
+          },
+          type: {
+            type: "string",
+            enum: ["task", "status", "question", "response", "context"],
+            description: "Type of message",
+          },
+          subject: {
+            type: "string",
+            description: "Message subject/summary",
+          },
+          body: {
+            type: "string",
+            description: "Full message content",
+          },
+        },
+        required: ["channel", "type", "subject", "body"],
+      },
+    },
+    {
+      name: "broadcast",
+      description: "Send a message to all registered agents",
+      inputSchema: {
+        type: "object",
+        properties: {
+          type: {
+            type: "string",
+            enum: ["task", "status", "question", "response", "context"],
+            description: "Type of message",
+          },
+          subject: {
+            type: "string",
+            description: "Message subject/summary",
+          },
+          body: {
+            type: "string",
+            description: "Full message content",
+          },
+        },
+        required: ["type", "subject", "body"],
+      },
+    },
+    {
+      name: "check_inbox",
+      description: "Check for incoming messages",
+      inputSchema: {
+        type: "object",
+        properties: {
+          unread_only: {
+            type: "boolean",
+            description: "Only return unread messages (default: false)",
+          },
+        },
+      },
+    },
+    {
+      name: "mark_read",
+      description: "Mark a message as read",
+      inputSchema: {
+        type: "object",
+        properties: {
+          message_id: {
+            type: "string",
+            description: "ID of the message to mark as read",
+          },
+        },
+        required: ["message_id"],
+      },
+    },
+    {
+      name: "reply",
+      description: "Reply to a message",
+      inputSchema: {
+        type: "object",
+        properties: {
+          message_id: {
+            type: "string",
+            description: "ID of the message to reply to",
+          },
+          body: {
+            type: "string",
+            description: "Reply content",
+          },
+        },
+        required: ["message_id", "body"],
+      },
+    },
+
+    // Discovery tools
+    {
+      name: "list_agents",
+      description: "List all registered agents",
+      inputSchema: {
+        type: "object",
+        properties: {
+          status: {
+            type: "string",
+            enum: ["online", "busy", "offline"],
+            description: "Filter by status (optional)",
+          },
+        },
+      },
+    },
+    {
+      name: "get_agent",
+      description: "Get details about a specific agent",
+      inputSchema: {
+        type: "object",
+        properties: {
+          id: {
+            type: "string",
+            description: "Agent ID to look up",
+          },
+        },
+        required: ["id"],
+      },
+    },
+    {
+      name: "list_channels",
+      description: "List all available channels",
+      inputSchema: {
+        type: "object",
+        properties: {},
+      },
+    },
+    {
+      name: "join_channel",
+      description: "Subscribe to a channel to receive its messages",
+      inputSchema: {
+        type: "object",
+        properties: {
+          channel: {
+            type: "string",
+            description: "Channel name to join",
+          },
+        },
+        required: ["channel"],
+      },
+    },
+    {
+      name: "leave_channel",
+      description: "Unsubscribe from a channel",
+      inputSchema: {
+        type: "object",
+        properties: {
+          channel: {
+            type: "string",
+            description: "Channel name to leave",
+          },
+        },
+        required: ["channel"],
+      },
+    },
+
+    // Task completion tools
+    {
+      name: "agent_complete_task",
+      description:
+        "Mark current task as complete with a summary. Posts to Slack and notifies team.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          summary: {
+            type: "string",
+            description: "Summary of what was accomplished",
+          },
+          outcome: {
+            type: "string",
+            enum: ["completed", "blocked", "partial", "needs_review", "handed_off"],
+            description: "Task outcome status",
+          },
+          files_changed: {
+            type: "array",
+            items: { type: "string" },
+            description: "List of files modified (optional)",
+          },
+          time_spent: {
+            type: "string",
+            description: "Override auto-calculated time (optional, e.g., '45 minutes')",
+          },
+          next_steps: {
+            type: "string",
+            description:
+              "What comes next, can @mention agents for handoff (optional)",
+          },
+        },
+        required: ["summary", "outcome"],
+      },
+    },
+    {
+      name: "get_pending_tasks",
+      description: "Get tasks assigned to you by other agents",
+      inputSchema: {
+        type: "object",
+        properties: {},
+      },
+    },
+    {
+      name: "accept_task",
+      description: "Accept a pending task and start working on it",
+      inputSchema: {
+        type: "object",
+        properties: {
+          task_id: {
+            type: "string",
+            description: "ID of the task to accept",
+          },
+        },
+        required: ["task_id"],
+      },
+    },
+    {
+      name: "decline_task",
+      description: "Decline a pending task with reason",
+      inputSchema: {
+        type: "object",
+        properties: {
+          task_id: {
+            type: "string",
+            description: "ID of the task to decline",
+          },
+          reason: {
+            type: "string",
+            description: "Reason for declining",
+          },
+        },
+        required: ["task_id", "reason"],
+      },
+    },
+  ];
+}
+
+export async function handleToolCall(
+  name: string,
+  args: Record<string, unknown>,
+  client: ApiClient,
+  context: ToolContext
+): Promise<unknown> {
+  const agentId = context.getCurrentAgentId();
+
+  switch (name) {
+    // Registration
+    case "agent_register": {
+      const workingDir = context.getWorkingDir();
+      const owner = state.getCurrentOwner();
+      const requestedId = args.id as string;
+
+      // Check for existing state file (reconnection)
+      const existingState = state.loadState(workingDir);
+
+      if (existingState) {
+        // Verify owner matches
+        if (existingState.owner !== owner) {
+          // Different user - delete state and register fresh
+          state.deleteState(workingDir);
+        } else {
+          // Try to reconnect
+          try {
+            const reconnectResult = await client.reconnectAgent(
+              existingState.agent_id,
+              existingState.token,
+              owner,
+              args.model as string // Pass the requested model to update it
+            );
+
+            context.setCurrentAgentId(existingState.agent_id);
+
+            // Update state with any new info
+            state.saveState(workingDir, {
+              ...existingState,
+              last_task: undefined, // Cleared on reconnect
+            });
+
+            return {
+              ...reconnectResult,
+              mode: "reconnected",
+              message: reconnectResult.was_offline
+                ? `Reconnected after timeout. You have ${reconnectResult.pending_tasks_count} pending tasks and ${reconnectResult.unread_messages_count} unread messages.`
+                : `Reconnected. You have ${reconnectResult.pending_tasks_count} pending tasks and ${reconnectResult.unread_messages_count} unread messages.`,
+            };
+          } catch (error) {
+            // Reconnection failed (invalid token, agent not found, etc.)
+            console.warn("Reconnection failed, registering fresh:", error);
+            state.deleteState(workingDir);
+          }
+        }
+      }
+
+      // Fresh registration
+      // Auto-detect model from environment or use provided value
+      const model = args.model as string ||
+                    process.env.CLAUDE_MODEL ||
+                    "claude-opus-4-5-20251101";
+
+      const result = await client.registerAgent(
+        requestedId,
+        args.name as string | undefined,
+        owner,
+        workingDir,
+        model
+      );
+
+      context.setCurrentAgentId(requestedId);
+
+      // Save state for future reconnection
+      state.saveState(workingDir, {
+        agent_id: requestedId,
+        owner,
+        token: result.token,
+        registered_at: result.registered_at,
+      });
+
+      return {
+        ...result,
+        mode: "registered",
+        message: `Registered as ${requestedId}. You have ${result.pending_tasks_count} pending tasks and ${result.unread_messages_count} unread messages.`,
+      };
+    }
+
+    case "agent_start_work": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.startWork(
+        agentId,
+        args.task as string,
+        args.project as string | undefined
+      );
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "agent_set_status": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.heartbeat(agentId, args.status as "online" | "busy");
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "agent_disconnect": {
+      // No wrapping - agent is going offline
+      if (!agentId) throw new Error("Not registered.");
+      context.stopHeartbeat();
+      return client.disconnect(agentId);
+    }
+
+    // Messaging
+    case "send_message": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.sendMessage({
+        from_agent: agentId,
+        to_agent: args.to as string,
+        type: args.type as string,
+        subject: args.subject as string,
+        body: args.body as string,
+        priority: args.priority as string | undefined,
+      });
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "send_to_channel": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.sendMessage({
+        from_agent: agentId,
+        to_channel: args.channel as string,
+        type: args.type as string,
+        subject: args.subject as string,
+        body: args.body as string,
+      });
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "broadcast": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.sendMessage({
+        from_agent: agentId,
+        broadcast: true,
+        type: args.type as string,
+        subject: args.subject as string,
+        body: args.body as string,
+      });
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "check_inbox": {
+      // No wrapping - already fetching messages
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      return client.getInbox(agentId, args.unread_only as boolean | undefined);
+    }
+
+    case "mark_read": {
+      const result = await client.markRead(args.message_id as string);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "reply": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.reply(args.message_id as string, agentId, args.body as string);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    // Discovery
+    case "list_agents": {
+      const result = await client.listAgents(args.status as string | undefined);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "get_agent": {
+      const result = await client.getAgent(args.id as string);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "list_channels": {
+      const result = await client.listChannels();
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "join_channel": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.joinChannel(args.channel as string, agentId);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "leave_channel": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.leaveChannel(args.channel as string, agentId);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    // Task completion
+    case "agent_complete_task": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.completeTask(agentId, {
+        summary: args.summary as string,
+        outcome: args.outcome as "completed" | "blocked" | "partial" | "needs_review" | "handed_off",
+        files_changed: args.files_changed as string[] | undefined,
+        time_spent: args.time_spent as string | undefined,
+        next_steps: args.next_steps as string | undefined,
+      });
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "get_pending_tasks": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.getPendingTasks(agentId);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "accept_task": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.acceptTask(agentId, args.task_id as string);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    case "decline_task": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+      const result = await client.declineTask(agentId, args.task_id as string, args.reason as string);
+      return wrapWithMessages(client, agentId, result);
+    }
+
+    default:
+      throw new Error(`Unknown tool: ${name}`);
+  }
+}