code-graph-context 2.5.2 → 2.5.3

package/dist/mcp/tools/swarm-claim-task.tool.js

@@ -66,9 +66,9 @@ const CLAIM_TASK_BY_ID_QUERY = `
  * Uses APOC locking to prevent race conditions between parallel workers
  */
 const CLAIM_NEXT_TASK_QUERY = `
-  // Find available tasks not blocked by dependencies
+  // Find available or blocked tasks (blocked tasks may have deps completed now)
   MATCH (t:SwarmTask {projectId: $projectId, swarmId: $swarmId})
-  WHERE t.status = 'available'
+  WHERE t.status IN ['available', 'blocked']
     AND ($types IS NULL OR size($types) = 0 OR t.type IN $types)
     AND ($minPriority IS NULL OR t.priorityScore >= $minPriority)
 
@@ -86,7 +86,7 @@ const CLAIM_NEXT_TASK_QUERY = `
   CALL apoc.lock.nodes([t])
 
   // Double-check status after acquiring lock (another worker may have claimed it)
-  WITH t WHERE t.status = 'available'
+  WITH t WHERE t.status IN ['available', 'blocked']
 
   // Atomic claim
   SET t.status = 'claimed',

package/dist/mcp/tools/swarm-orchestrate.tool.js

@@ -352,33 +352,129 @@ export const createSwarmOrchestrateTool = (server) => {
  */
 function generateWorkerInstructions(swarmId, projectId, maxAgents, taskCount) {
     const recommendedAgents = Math.min(maxAgents, Math.ceil(taskCount / 2), taskCount);
-    return `
-## Worker Agent Instructions
+    // Generate unique agent IDs for each worker
+    const agentIds = Array.from({ length: recommendedAgents }, (_, i) => `${swarmId}_worker_${i + 1}`);
+    const workerPrompt = `You are a swarm worker agent.
+- Agent ID: {AGENT_ID}
+- Swarm ID: ${swarmId}
+- Project: ${projectId}
 
-To execute this swarm, spawn ${recommendedAgents} worker agent(s) using the Task tool:
+## CRITICAL RULES
+1. NEVER fabricate node IDs - get them from graph tool responses
+2. ALWAYS use the blackboard task queue (swarm_claim_task, swarm_complete_task)
+3. Use graph tools (traverse_from_node, search_codebase) to understand context
+4. Exit when swarm_claim_task returns "no_tasks"
 
-\`\`\`
-Task({
-  subagent_type: "general-purpose",
-  prompt: "You are a swarm worker for swarm ${swarmId}. Project: ${projectId}.
+## WORKFLOW - Follow these steps exactly:
 
-Your workflow:
-1. swarm_sense({ projectId: '${projectId}', swarmId: '${swarmId}', types: ['modifying', 'claiming'] })
-2. swarm_claim_task({ projectId: '${projectId}', swarmId: '${swarmId}', agentId: '<your-id>' })
-3. If task claimed: execute it, then swarm_complete_task with action: 'complete'
-4. Loop until no tasks remain
+### Step 1: Claim a task from the blackboard
+swarm_claim_task({
+  projectId: "${projectId}",
+  swarmId: "${swarmId}",
+  agentId: "{AGENT_ID}"
+})
+// If returns "no_tasks" → exit, swarm is complete
+// Otherwise you now own the returned task
 
-Always leave pheromones when working. Exit when swarm_get_tasks shows no available/in_progress tasks.",
-  run_in_background: true
+### Step 2: Start working and check for conflicts
+swarm_claim_task({
+  projectId: "${projectId}",
+  swarmId: "${swarmId}",
+  agentId: "{AGENT_ID}",
+  taskId: "<TASK_ID_FROM_STEP_1>",
+  action: "start"
 })
+
+// Check if another agent is working on related code
+swarm_sense({
+  projectId: "${projectId}",
+  swarmId: "${swarmId}",
+  types: ["modifying", "warning"],
+  excludeAgentId: "{AGENT_ID}"
+})
+
+### Step 3: Understand the code context (USE GRAPH TOOLS!)
+// Use traverse_from_node to see relationships and callers
+traverse_from_node({
+  projectId: "${projectId}",
+  filePath: "<TARGET_FILE_FROM_TASK>",
+  maxDepth: 2,
+  includeCode: true
+})
+
+// Or search for related code
+search_codebase({
+  projectId: "${projectId}",
+  query: "<WHAT_YOU_NEED_TO_UNDERSTAND>"
+})
+
+### Step 4: Do the work
+- Use Read tool for full source code of files to modify
+- Use Edit tool to make changes
+- Mark nodes you're modifying:
+swarm_pheromone({
+  projectId: "${projectId}",
+  nodeId: "<NODE_ID_FROM_GRAPH>",
+  type: "modifying",
+  agentId: "{AGENT_ID}",
+  swarmId: "${swarmId}"
+})
+
+### Step 5: Complete the task via blackboard
+swarm_pheromone({
+  projectId: "${projectId}",
+  nodeId: "<NODE_ID>",
+  type: "completed",
+  agentId: "{AGENT_ID}",
+  swarmId: "${swarmId}",
+  data: { summary: "<WHAT_YOU_DID>" }
+})
+
+swarm_complete_task({
+  projectId: "${projectId}",
+  taskId: "<TASK_ID>",
+  agentId: "{AGENT_ID}",
+  action: "complete",
+  summary: "<DESCRIBE_WHAT_YOU_DID>",
+  filesChanged: ["<LIST_OF_FILES_YOU_MODIFIED>"]
+})
+
+### Step 6: Loop back to Step 1
+Claim the next available task. Continue until no tasks remain.
+
+## IF YOU GET STUCK
+swarm_complete_task({
+  projectId: "${projectId}",
+  taskId: "<TASK_ID>",
+  agentId: "{AGENT_ID}",
+  action: "fail",
+  reason: "<WHY_YOU_ARE_STUCK>",
+  retryable: true
+})
+Then claim another task.`;
+    const taskCalls = agentIds.map(agentId => {
+        const prompt = workerPrompt.replace(/\{AGENT_ID\}/g, agentId);
+        return `Task({
+  subagent_type: "general-purpose",
+  run_in_background: false,
+  prompt: \`${prompt}\`
+})`;
+    }).join('\n\n');
+    return `
+## Worker Agent Instructions
+
+Spawn ${recommendedAgents} worker agent(s) IN PARALLEL (all Task calls in ONE message):
+
+\`\`\`javascript
+${taskCalls}
 \`\`\`
 
-Launch ${recommendedAgents} agents in parallel for optimal execution.
+**CRITICAL:** Include ALL ${recommendedAgents} Task calls in a single message to run them in parallel.
 
 ## Monitoring Progress
 
 Check swarm progress:
-\`\`\`
+\`\`\`javascript
 swarm_get_tasks({
   projectId: "${projectId}",
   swarmId: "${swarmId}",
@@ -386,9 +482,9 @@ swarm_get_tasks({
 })
 \`\`\`
 
-## Cleanup (after completion)
+## Cleanup (after all workers complete)
 
-\`\`\`
+\`\`\`javascript
 swarm_cleanup({
   projectId: "${projectId}",
   swarmId: "${swarmId}"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "2.5.2",
+  "version": "2.5.3",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/drewdrewH/code-graph-context#readme",