agenthub-multiagent-mcp 1.1.2 → 1.1.4

package/src/heartbeat.ts

@@ -1,5 +1,5 @@
 /**
- * Background heartbeat manager
+ * Background heartbeat manager with enhanced logging for pending tasks/messages
  */
 
 import { ApiClient } from "./client.js";
@@ -9,6 +9,9 @@ export class HeartbeatManager {
   private intervalId?: NodeJS.Timeout;
   private agentId?: string;
   private status: "online" | "busy" = "online";
+  private lastPendingTasks = 0;
+  private lastUnreadMessages = 0;
+  private heartbeatCount = 0;
 
   constructor(client: ApiClient) {
     this.client = client;
@@ -17,13 +20,16 @@ export class HeartbeatManager {
   start(agentId: string): void {
     this.stop(); // Clear any existing interval
     this.agentId = agentId;
+    this.heartbeatCount = 0;
+    this.lastPendingTasks = 0;
+    this.lastUnreadMessages = 0;
 
     // Send heartbeat every 30 seconds
     this.intervalId = setInterval(async () => {
       if (!this.agentId) return;
 
       try {
-        await this.client.heartbeat(this.agentId, this.status);
+        await this.sendHeartbeat();
       } catch (error) {
         console.error("Heartbeat failed:", error);
       }
@@ -58,9 +64,45 @@ export class HeartbeatManager {
     if (!this.agentId) return;
 
     try {
-      await this.client.heartbeat(this.agentId, this.status);
+      const response = await this.client.heartbeat(this.agentId, this.status);
+      this.heartbeatCount++;
+
+      const { pending_tasks_count, unread_messages_count } = response;
+
+      // Log if there are new pending tasks
+      if (pending_tasks_count > 0 && pending_tasks_count !== this.lastPendingTasks) {
+        console.error(`\n🔔 [AgentHub] You have ${pending_tasks_count} pending task(s) waiting! Use get_pending_tasks to view them.\n`);
+      }
+
+      // Log if there are new unread messages
+      if (unread_messages_count > 0 && unread_messages_count !== this.lastUnreadMessages) {
+        console.error(`\n📬 [AgentHub] You have ${unread_messages_count} unread message(s)! Use check_inbox to read them.\n`);
+      }
+
+      // Periodic reminder every 5 heartbeats (2.5 minutes) if there are pending items
+      if (this.heartbeatCount % 5 === 0) {
+        if (pending_tasks_count > 0 || unread_messages_count > 0) {
+          const parts = [];
+          if (pending_tasks_count > 0) parts.push(`${pending_tasks_count} pending task(s)`);
+          if (unread_messages_count > 0) parts.push(`${unread_messages_count} unread message(s)`);
+          console.error(`\n⏰ [AgentHub] Reminder: You have ${parts.join(" and ")}.\n`);
+        }
+      }
+
+      this.lastPendingTasks = pending_tasks_count;
+      this.lastUnreadMessages = unread_messages_count;
     } catch (error) {
       console.error("Heartbeat failed:", error);
     }
   }
+
+  /**
+   * Get the current pending counts (for use by tools)
+   */
+  getPendingCounts(): { pendingTasks: number; unreadMessages: number } {
+    return {
+      pendingTasks: this.lastPendingTasks,
+      unreadMessages: this.lastUnreadMessages,
+    };
+  }
 }

package/src/index.ts

@@ -1,123 +1,123 @@
-#!/usr/bin/env node
-
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from "@modelcontextprotocol/sdk/types.js";
-
-import { ApiClient } from "./client.js";
-import { registerTools, handleToolCall } from "./tools/index.js";
-import { HeartbeatManager } from "./heartbeat.js";
-
-// Environment configuration
-const AGENTHUB_URL = process.env.AGENTHUB_URL || "http://localhost:8787";
-const AGENTHUB_API_KEY = process.env.AGENTHUB_API_KEY || "";
-const AGENTHUB_AGENT_ID = process.env.AGENTHUB_AGENT_ID || "";
-
-// Validate configuration
-if (!AGENTHUB_API_KEY) {
-  console.error("Error: AGENTHUB_API_KEY environment variable is required");
-  process.exit(1);
-}
-
-// Initialize API client
-const client = new ApiClient(AGENTHUB_URL, AGENTHUB_API_KEY);
-
-// Initialize heartbeat manager
-const heartbeat = new HeartbeatManager(client);
-
-// Track current agent ID
-let currentAgentId = AGENTHUB_AGENT_ID;
-
-// Create MCP server
-const server = new Server(
-  {
-    name: "agenthub",
-    version: "1.1.2",
-  },
-  {
-    capabilities: {
-      tools: {},
-    },
-  }
-);
-
-// List available tools
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-  return {
-    tools: registerTools(),
-  };
-});
-
-// Handle tool calls
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  const { name, arguments: args } = request.params;
-
-  try {
-    const result = await handleToolCall(name, args || {}, client, {
-      getCurrentAgentId: () => currentAgentId,
-      setCurrentAgentId: (id: string) => {
-        currentAgentId = id;
-        heartbeat.start(id);
-      },
-      stopHeartbeat: () => heartbeat.stop(),
-      getWorkingDir: () => process.cwd(),
-    });
-
-    return {
-      content: [
-        {
-          type: "text",
-          text: JSON.stringify(result, null, 2),
-        },
-      ],
-    };
-  } catch (error) {
-    const message = error instanceof Error ? error.message : String(error);
-    return {
-      content: [
-        {
-          type: "text",
-          text: `Error: ${message}`,
-        },
-      ],
-      isError: true,
-    };
-  }
-});
-
-// Graceful shutdown
-process.on("SIGINT", () => {
-  heartbeat.stop();
-  process.exit(0);
-});
-
-process.on("SIGTERM", () => {
-  heartbeat.stop();
-  process.exit(0);
-});
-
-// Start server
-async function main() {
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
-  console.error("AgentHub MCP server running");
-
-  // Auto-register if agent ID is provided
-  if (AGENTHUB_AGENT_ID) {
-    try {
-      await client.registerAgent(AGENTHUB_AGENT_ID);
-      heartbeat.start(AGENTHUB_AGENT_ID);
-      console.error(`Auto-registered as agent: ${AGENTHUB_AGENT_ID}`);
-    } catch (error) {
-      console.error(`Failed to auto-register: ${error}`);
-    }
-  }
-}
-
-main().catch((error) => {
-  console.error("Fatal error:", error);
-  process.exit(1);
-});
+#!/usr/bin/env node
+
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} from "@modelcontextprotocol/sdk/types.js";
+
+import { ApiClient } from "./client.js";
+import { registerTools, handleToolCall } from "./tools/index.js";
+import { HeartbeatManager } from "./heartbeat.js";
+
+// Environment configuration
+const AGENTHUB_URL = process.env.AGENTHUB_URL || "http://localhost:8787";
+const AGENTHUB_API_KEY = process.env.AGENTHUB_API_KEY || "";
+const AGENTHUB_AGENT_ID = process.env.AGENTHUB_AGENT_ID || "";
+
+// Validate configuration
+if (!AGENTHUB_API_KEY) {
+  console.error("Error: AGENTHUB_API_KEY environment variable is required");
+  process.exit(1);
+}
+
+// Initialize API client
+const client = new ApiClient(AGENTHUB_URL, AGENTHUB_API_KEY);
+
+// Initialize heartbeat manager
+const heartbeat = new HeartbeatManager(client);
+
+// Track current agent ID
+let currentAgentId = AGENTHUB_AGENT_ID;
+
+// Create MCP server
+const server = new Server(
+  {
+    name: "agenthub",
+    version: "1.1.4",
+  },
+  {
+    capabilities: {
+      tools: {},
+    },
+  }
+);
+
+// List available tools
+server.setRequestHandler(ListToolsRequestSchema, async () => {
+  return {
+    tools: registerTools(),
+  };
+});
+
+// Handle tool calls
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+
+  try {
+    const result = await handleToolCall(name, args || {}, client, {
+      getCurrentAgentId: () => currentAgentId,
+      setCurrentAgentId: (id: string) => {
+        currentAgentId = id;
+        heartbeat.start(id);
+      },
+      stopHeartbeat: () => heartbeat.stop(),
+      getWorkingDir: () => process.cwd(),
+    });
+
+    return {
+      content: [
+        {
+          type: "text",
+          text: JSON.stringify(result, null, 2),
+        },
+      ],
+    };
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    return {
+      content: [
+        {
+          type: "text",
+          text: `Error: ${message}`,
+        },
+      ],
+      isError: true,
+    };
+  }
+});
+
+// Graceful shutdown
+process.on("SIGINT", () => {
+  heartbeat.stop();
+  process.exit(0);
+});
+
+process.on("SIGTERM", () => {
+  heartbeat.stop();
+  process.exit(0);
+});
+
+// Start server
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("AgentHub MCP server running");
+
+  // Auto-register if agent ID is provided
+  if (AGENTHUB_AGENT_ID) {
+    try {
+      await client.registerAgent(AGENTHUB_AGENT_ID);
+      heartbeat.start(AGENTHUB_AGENT_ID);
+      console.error(`Auto-registered as agent: ${AGENTHUB_AGENT_ID}`);
+    } catch (error) {
+      console.error(`Failed to auto-register: ${error}`);
+    }
+  }
+}
+
+main().catch((error) => {
+  console.error("Fatal error:", error);
+  process.exit(1);
+});