agenthub-multiagent-mcp 1.1.2 → 1.1.3

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.3",
+  },
+  {
+    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);
+});