agenthub-multiagent-mcp 1.4.1 → 1.5.1

package/src/tools/index.ts

@@ -17,6 +17,7 @@ export interface ToolContext {
   setCurrentAgentId: (id: string) => void;
   stopHeartbeat: () => void;
   getWorkingDir: () => string;
+  setWorkingDir: (dir: string) => void;
   getPushedItems: () => PushedItems;
 }
 
@@ -178,6 +179,14 @@ export function registerTools(): Tool[] {
       inputSchema: {
         type: "object",
         properties: {
+          working_dir: {
+            type: "string",
+            description: "Project directory path where .agenthub state will be saved (required)",
+          },
+          project: {
+            type: "string",
+            description: "Project key to bind this agent to (e.g., 'KRISHI'). Required.",
+          },
           id: {
             type: "string",
             description: "Unique agent identifier (e.g., 'backend-1', 'tester')",
@@ -191,7 +200,7 @@ export function registerTools(): Tool[] {
             description: "Model name (e.g., 'claude-opus-4.5', 'gpt-4-turbo', 'o3', 'gemini-2.0-pro'). Provider is auto-detected.",
           },
         },
-        required: ["id", "model"],
+        required: ["working_dir", "project", "model"],
       },
     },
     {
@@ -594,14 +603,10 @@ export function registerTools(): Tool[] {
     // Ticket creation tools
     {
       name: "create_epic",
-      description: "Create an epic (large body of work) in a project",
+      description: "Create an epic (large body of work) in the bound project. Automatically uploads the OpenSpec file as an attachment. (requires agent_register first)",
       inputSchema: {
         type: "object",
         properties: {
-          project_id: {
-            type: "string",
-            description: "Project ID to create epic in",
-          },
           title: {
             type: "string",
             description: "Epic title",
@@ -612,25 +617,21 @@ export function registerTools(): Tool[] {
           },
           source_file: {
             type: "string",
-            description: "Optional: path to OpenSpec file this came from",
+            description: "Path to OpenSpec file (e.g., 'openspec/changes/feature-name/proposal.md'). Will be uploaded as attachment.",
           },
         },
-        required: ["project_id", "title"],
+        required: ["title", "source_file"],
       },
     },
     {
       name: "create_story",
-      description: "Create a story under an epic",
+      description: "Create a story in the bound project (requires agent_register first)",
       inputSchema: {
         type: "object",
         properties: {
-          project_id: {
-            type: "string",
-            description: "Project ID",
-          },
           epic_id: {
             type: "string",
-            description: "Epic ID to add story to",
+            description: "Epic ID to add story to (optional)",
           },
           title: {
             type: "string",
@@ -651,22 +652,18 @@ export function registerTools(): Tool[] {
             description: "Additional labels",
           },
         },
-        required: ["project_id", "title"],
+        required: ["title"],
       },
     },
     {
       name: "create_task",
-      description: "Create a task under a story",
+      description: "Create a task in the bound project (requires agent_register first)",
       inputSchema: {
         type: "object",
         properties: {
-          project_id: {
-            type: "string",
-            description: "Project ID",
-          },
           story_id: {
             type: "string",
-            description: "Story ID to add task to",
+            description: "Story ID to add task to (optional)",
           },
           title: {
             type: "string",
@@ -682,7 +679,7 @@ export function registerTools(): Tool[] {
             description: "Required skill type",
           },
         },
-        required: ["project_id", "title"],
+        required: ["title"],
       },
     },
     {
@@ -748,7 +745,17 @@ export async function handleToolCall(
   switch (name) {
     // Registration
     case "agent_register": {
-      const workingDir = context.getWorkingDir();
+      // Get required params
+      const workingDir = args.working_dir as string;
+      if (!workingDir) {
+        throw new Error("working_dir is required");
+      }
+
+      const projectKey = args.project as string;
+      if (!projectKey) {
+        throw new Error("project is required");
+      }
+
       const owner = state.getCurrentOwner();
       const requestedId = args.id as string;
 
@@ -779,6 +786,9 @@ export async function handleToolCall(
             // Set connect token on client for org operations
             client.setConnectToken(existingState.token);
 
+            // Store working_dir in context for other tools to use
+            context.setWorkingDir(workingDir);
+
             // Update state with returned name from server
             state.saveState(workingDir, {
               ...existingState,
@@ -805,7 +815,8 @@ export async function handleToolCall(
       // Step 1: Initialize registration session
       const initResult = await client.initRegistration(
         requestedId,
-        args.name as string | undefined
+        args.name as string | undefined,
+        projectKey
       );
 
       // Step 2: Open browser to dashboard
@@ -851,6 +862,9 @@ export async function handleToolCall(
       // Set connect token on client for org operations
       client.setConnectToken(registrationResult.connect_token!);
 
+      // Store working_dir in context for other tools to use
+      context.setWorkingDir(workingDir);
+
       // Save state for future reconnection
       state.saveState(workingDir, {
         agent_id: agentId,
@@ -858,6 +872,9 @@ export async function handleToolCall(
         owner,
         token: registrationResult.connect_token!,
         agent_type: registrationResult.agent_type,
+        project_id: registrationResult.project_id!,
+        project_key: registrationResult.project_key!,
+        org_id: registrationResult.org_id!,
         registered_at: new Date().toISOString(),
       });
 
@@ -1055,21 +1072,65 @@ export async function handleToolCall(
 
     // Ticket creation tools
     case "create_epic": {
+      if (!agentId) throw new Error("Not registered. Call agent_register first.");
+
+      // Get project from state
+      const workingDir = context.getWorkingDir();
+      const agentState = state.loadState(workingDir);
+
+      if (!agentState?.project_id) {
+        throw new Error("Not registered with a project. Call agent_register first.");
+      }
+
+      const sourceFile = args.source_file as string;
+      if (!sourceFile) {
+        throw new Error("source_file is required. Provide the path to the OpenSpec file.");
+      }
+
+      // Create the epic
       const result = await client.createEpic({
-        project_id: args.project_id as string,
+        project_id: agentState.project_id,
         title: args.title as string,
         description: args.description as string | undefined,
-        source_file: args.source_file as string | undefined,
+        source_file: sourceFile,
       });
+
+      // Auto-upload the OpenSpec file as attachment
+      let attachmentResult = null;
+      try {
+        attachmentResult = await client.uploadAttachment({
+          ticket_id: result.id,
+          ticket_type: "epic",
+          ticket_key: result.key,
+          file_path: sourceFile,
+          description: "OpenSpec source document",
+          agent_id: agentId,
+        });
+      } catch (uploadError) {
+        // Log but don't fail epic creation if upload fails
+        console.error("Failed to upload OpenSpec attachment:", uploadError);
+      }
+
       return {
         ...result,
-        message: `Created epic ${result.key}: ${args.title}`,
+        attachment: attachmentResult,
+        message: attachmentResult
+          ? `Created epic ${result.key}: ${args.title} (OpenSpec uploaded)`
+          : `Created epic ${result.key}: ${args.title} (OpenSpec upload failed)`,
       };
     }
 
     case "create_story": {
+      // Get project from state
+      const workingDir = context.getWorkingDir();
+      const agentState = state.loadState(workingDir);
+
+      if (!agentState?.project_id) {
+        throw new Error("Not registered with a project. Call agent_register first.");
+      }
+
       const result = await client.createStory({
-        project_id: args.project_id as string,
+        project_id: agentState.project_id,  // From state, not args
         epic_id: args.epic_id as string | undefined,
         title: args.title as string,
         description: args.description as string | undefined,
@@ -1083,8 +1144,16 @@ export async function handleToolCall(
     }
 
     case "create_task": {
+      // Get project from state
+      const workingDir = context.getWorkingDir();
+      const agentState = state.loadState(workingDir);
+
+      if (!agentState?.project_id) {
+        throw new Error("Not registered with a project. Call agent_register first.");
+      }
+
       const result = await client.createTask({
-        project_id: args.project_id as string,
+        project_id: agentState.project_id,  // From state, not args
         story_id: args.story_id as string | undefined,
         title: args.title as string,
         description: args.description as string | undefined,

package/src/tools/tools.test.ts

@@ -40,21 +40,25 @@ const mockClient = {
 // Mock context
 function createMockContext(): ToolContext {
   let currentAgentId = "";
+  let currentWorkingDir = "/tmp/test-workdir";
   return {
     getCurrentAgentId: () => currentAgentId,
     setCurrentAgentId: (id: string) => {
       currentAgentId = id;
     },
     stopHeartbeat: vi.fn(),
-    getWorkingDir: () => "/tmp/test-workdir",
+    getWorkingDir: () => currentWorkingDir,
+    setWorkingDir: (dir: string) => {
+      currentWorkingDir = dir;
+    },
     getPushedItems: () => ({ tasks: [], messages: [] }),
   };
 }
 
 describe("registerTools", () => {
-  it("should return all 19 tools", () => {
+  it("should return all 28 tools", () => {
     const tools = registerTools();
-    expect(tools).toHaveLength(19);
+    expect(tools).toHaveLength(28);
   });
 
   it("should have required tool names", () => {
@@ -100,40 +104,33 @@ describe("handleToolCall", () => {
   });
 
   describe("agent_register", () => {
-    it("should register agent and set current agent ID", async () => {
-      const mockResult = {
-        agent_id: "test-agent",
-        name: "Test Agent",
-        token: "test-token-uuid",
-        model: "claude-opus-4.5",
-        model_provider: "anthropic",
-        registered_at: "2024-01-01T00:00:00Z",
-        slack_notified: false,
-        pending_tasks_count: 0,
-        unread_messages_count: 0,
-      };
-      vi.mocked(mockClient.registerAgent).mockResolvedValue(mockResult);
+    // Note: The agent_register flow now uses browser-based registration
+    // which requires mocking initRegistration, pollRegistrationCallback, and open()
+    // These are integration tests that are skipped in unit tests
+    it.skip("should register agent and set current agent ID", async () => {
+      // Browser-based flow - tested via e2e tests
+    });
 
-      const result = await handleToolCall(
-        "agent_register",
-        { id: "test-agent", name: "Test Agent", model: "claude-opus-4.5" },
-        mockClient,
-        context
-      );
+    it("should require working_dir parameter", async () => {
+      await expect(
+        handleToolCall(
+          "agent_register",
+          { project: "TEST", model: "claude-opus-4.5" },
+          mockClient,
+          context
+        )
+      ).rejects.toThrow("working_dir is required");
+    });
 
-      expect(mockClient.registerAgent).toHaveBeenCalledWith(
-        "test-agent",
-        "Test Agent",
-        "test-user",
-        "/tmp/test-workdir",
-        "claude-opus-4.5"
-      );
-      expect(context.getCurrentAgentId()).toBe("test-agent");
-      // Result is extended with mode and message
-      expect(result).toMatchObject({
-        ...mockResult,
-        mode: "registered",
-      });
+    it("should require project parameter", async () => {
+      await expect(
+        handleToolCall(
+          "agent_register",
+          { working_dir: "/tmp/test", model: "claude-opus-4.5" },
+          mockClient,
+          context
+        )
+      ).rejects.toThrow("project is required");
     });
   });
 
@@ -164,7 +161,8 @@ describe("handleToolCall", () => {
       expect(mockClient.startWork).toHaveBeenCalledWith(
         "test-agent",
         "Building feature X",
-        "TestProject"
+        "TestProject",
+        undefined
       );
       // Result is wrapped with messages
       expect(result).toMatchObject({ result: mockResult });