agent-pool-mcp 1.0.0 → 1.2.0

package/src/server.js

@@ -17,6 +17,8 @@ import { getSystemLoad } from './runner/process-manager.js';
 import { createTask, completeTask, failTask, formatTaskResult, getActiveTasks, cancelTask } from './tools/results.js';
 import { listSkills, createSkill, deleteSkill, installSkill, provisionSkill } from './tools/skills.js';
 import { consultPeer } from './tools/consult.js';
+import { addSchedule, listSchedules, removeSchedule, getScheduledResults, getDaemonStatus } from './scheduler/scheduler.js';
+import { createPipeline, listPipelines, runPipeline, getRun, listRuns, cancelRun, signalStepComplete, bounceBack } from './scheduler/pipeline.js';
 
 import { TOOL_DEFINITIONS } from './tool-definitions.js';
 
@@ -105,7 +107,7 @@ export function createServer() {
   }
 
   const server = new Server(
-    { name: 'agent-pool', version: '1.0.0' },
+    { name: 'agent-pool', version: '1.2.0' },
     { capabilities: { tools: {} } },
   );
 
@@ -145,6 +147,28 @@ export function createServer() {
           response = handleDeleteSkill(args); break;
         case 'install_skill':
           response = handleInstallSkill(args); break;
+        case 'schedule_task':
+          response = handleScheduleTask(args); break;
+        case 'list_schedules':
+          response = handleListSchedules(args); break;
+        case 'cancel_schedule':
+          response = handleCancelSchedule(args); break;
+        case 'get_scheduled_results':
+          response = handleGetScheduledResults(args); break;
+        case 'create_pipeline':
+          response = handleCreatePipeline(args); break;
+        case 'run_pipeline':
+          response = handleRunPipeline(args); break;
+        case 'list_pipelines':
+          response = handleListPipelines(args); break;
+        case 'get_pipeline_status':
+          response = handleGetPipelineStatus(args); break;
+        case 'cancel_pipeline':
+          response = handleCancelPipeline(args); break;
+        case 'signal_step_complete':
+          response = handleSignalStepComplete(args); break;
+        case 'bounce_back':
+          response = handleBounceBack(args); break;
         default:
           response = { content: [{ type: 'text', text: `Unknown tool: ${name}` }], isError: true };
       }
@@ -348,3 +372,247 @@ function handleInstallSkill(args) {
     }],
   };
 }
+
+// ─── Scheduler Handlers ─────────────────────────────────────────────
+
+/** @param {object} args */
+function handleScheduleTask(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  try {
+    const result = addSchedule(cwd, {
+      prompt: args.prompt,
+      cron: args.cron,
+      skill: args.skill,
+      approvalMode: args.approval_mode,
+      catchup: args.catchup,
+      taskCwd: args.cwd,
+    });
+
+    return {
+      content: [{
+        type: 'text',
+        text: `⏰ Task scheduled.\n\n- **Schedule ID**: \`${result.scheduleId}\`\n- **Cron**: \`${args.cron}\`\n- **Next run**: ${result.nextRun || 'unknown'}\n- **Prompt**: ${args.prompt.substring(0, 100)}...\n\nDaemon is running in the background. Results will be saved to \`.agent/scheduled-results/\`.\nUse \`list_schedules\` to see all schedules, \`get_scheduled_results\` to read outputs.`,
+      }],
+    };
+  } catch (error) {
+    return { content: [{ type: 'text', text: `❌ Failed to schedule: ${error.message}` }], isError: true };
+  }
+}
+
+/** @param {object} args */
+function handleListSchedules(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const schedules = listSchedules(cwd);
+  const daemon = getDaemonStatus(cwd);
+
+  if (schedules.length === 0) {
+    return { content: [{ type: 'text', text: `No scheduled tasks.\n\nDaemon: ${daemon.running ? `running (pid ${daemon.pid})` : 'not running'}` }] };
+  }
+
+  const lines = schedules.map((s) =>
+    `- **${s.id}** | \`${s.cron}\` | next: ${s.nextRun || '?'} | last: ${s.lastRun || 'never'}\n  ${s.prompt.substring(0, 80)}`,
+  );
+
+  return {
+    content: [{
+      type: 'text',
+      text: `## Scheduled Tasks (${schedules.length})\n\nDaemon: ${daemon.running ? `✅ running (pid ${daemon.pid})` : '❌ not running'}\n\n${lines.join('\n')}`,
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleCancelSchedule(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const removed = removeSchedule(cwd, args.schedule_id);
+  return {
+    content: [{
+      type: 'text',
+      text: removed
+        ? `✅ Schedule \`${args.schedule_id}\` cancelled. Daemon will auto-exit when no schedules remain.`
+        : `❌ Schedule \`${args.schedule_id}\` not found.`,
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleGetScheduledResults(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const results = getScheduledResults(cwd, args.schedule_id);
+
+  if (results.length === 0) {
+    return { content: [{ type: 'text', text: 'No scheduled results yet.' }] };
+  }
+
+  const lines = results.map((r) =>
+    `### ${r.scheduleId} — ${r.executedAt}\n- Exit: ${r.exitCode}\n- Events: ${r.totalEvents}\n\n\`\`\`\n${(r.response || '').substring(0, 500)}\n\`\`\``,
+  );
+
+  return {
+    content: [{
+      type: 'text',
+      text: `## Scheduled Results (${results.length})\n\n${lines.join('\n\n')}`,
+    }],
+  };
+}
+
+// ─── Pipeline Handlers ─────────────────────────────────────────────
+
+/** @param {object} args */
+function handleCreatePipeline(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const result = createPipeline(cwd, args);
+  return {
+    content: [{
+      type: 'text',
+      text: `✅ Pipeline created.\n\n- **Pipeline ID**: \`${result.pipelineId}\`\n- **Steps**: ${args.steps.length}`
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleRunPipeline(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const result = runPipeline(cwd, args.pipeline_id);
+  if (!result) {
+    return {
+      content: [{ type: 'text', text: `❌ Pipeline \`${args.pipeline_id}\` not found.` }],
+      isError: true,
+    };
+  }
+  return {
+    content: [{
+      type: 'text',
+      text: `🚀 Pipeline started.\n\n- **Run ID**: \`${result.runId}\``
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleListPipelines(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const pipelines = listPipelines(cwd);
+  const runs = listRuns(cwd);
+  
+  const parts = [];
+
+  if (pipelines.length > 0) {
+    const lines = pipelines.map(p => `- **${p.name}** (\`${p.id}\`) — ${p.steps.length} steps`);
+    parts.push(`## Pipelines (${pipelines.length})\n\n${lines.join('\n')}`);
+  }
+
+  const activeRuns = runs.filter(r => r.status === 'running');
+  const recentRuns = runs.filter(r => r.status !== 'running').slice(0, 5);
+
+  if (activeRuns.length > 0) {
+    const emojiMap = { success: '✅', failed: '❌', running: '🔄', pending: '⏸️', bounce_pending: '↩️', waiting_bounce: '⏳', skipped: '⏭️', cancelled: '🛑' };
+    const lines = activeRuns.map(r => {
+      const stepsSummary = Object.entries(r.steps).map(([n, s]) => `${emojiMap[s.status] || '❓'}${n}`).join(' → ');
+      return `- \`${r.id}\` (${r.pipelineName}) ${stepsSummary}`;
+    });
+    parts.push(`## Active Runs (${activeRuns.length})\n\n${lines.join('\n')}`);
+  }
+
+  if (recentRuns.length > 0) {
+    const lines = recentRuns.map(r => `- \`${r.id}\` ${r.status === 'success' ? '✅' : '❌'} ${r.pipelineName} (${r.completedAt || ''})`);
+    parts.push(`## Recent Runs\n\n${lines.join('\n')}`);
+  }
+
+  if (parts.length === 0) {
+    return { content: [{ type: 'text', text: 'No pipelines found.' }] };
+  }
+
+  return {
+    content: [{ type: 'text', text: parts.join('\n\n') }],
+  };
+}
+
+/** @param {object} args */
+function handleGetPipelineStatus(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const run = getRun(cwd, args.run_id);
+  
+  if (!run) {
+    return { content: [{ type: 'text', text: `❌ Pipeline run \`${args.run_id}\` not found.` }], isError: true };
+  }
+
+  const emojiMap = {
+    success: '✅',
+    failed: '❌',
+    running: '🔄',
+    pending: '⏸️',
+    bounce_pending: '↩️',
+    waiting_bounce: '⏳',
+    skipped: '⏭️',
+    cancelled: '🛑',
+  };
+
+  const lines = Object.entries(run.steps).map(([name, s]) => {
+    const emoji = emojiMap[s.status] || s.status;
+    return `- ${emoji} **${name}**: ${s.status}`;
+  });
+
+  return {
+    content: [{
+      type: 'text',
+      text: `## Pipeline Status: \`${args.run_id}\`\n**Status**: ${run.status}\n\n${lines.join('\n')}`
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleCancelPipeline(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const success = cancelRun(cwd, args.run_id);
+  return {
+    content: [{
+      type: 'text',
+      text: success ? `✅ Pipeline \`${args.run_id}\` cancelled.` : `❌ Failed to cancel pipeline \`${args.run_id}\` (not found or not running).`
+    }],
+  };
+}
+
+/** @param {object} args */
+function handleSignalStepComplete(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const result = signalStepComplete(cwd, args.step_name, args.output, args.run_id);
+  if (result.success) {
+    return {
+      content: [{
+        type: 'text',
+        text: `✅ Step \`${args.step_name}\` marked as complete.`
+      }],
+    };
+  } else {
+    return {
+      content: [{
+        type: 'text',
+        text: `❌ Failed to signal step completion. Step might not be running or run not found.`
+      }],
+      isError: true,
+    };
+  }
+}
+
+/** @param {object} args */
+function handleBounceBack(args) {
+  const cwd = args.cwd ?? defaultCwd;
+  const info = bounceBack(cwd, args.step_name, args.reason, args.run_id);
+  if (info.success) {
+    return {
+      content: [{
+        type: 'text',
+        text: `↩️ Task bounced back to \`${args.step_name}\`.\nReason: ${args.reason}\nBounces: ${info.bounceCount}/${info.maxBounces}`
+      }],
+    };
+  } else {
+    return {
+      content: [{
+        type: 'text',
+        text: `❌ Failed to bounce back. Pipeline might have reached max bounces or step not found.`
+      }],
+      isError: true,
+    };
+  }
+}
+

package/src/tool-definitions.js

@@ -173,4 +173,172 @@ export const TOOL_DEFINITIONS = [
       required: ['skill_name'],
     },
   },
+  // ─── Scheduler Tools ────────────────────────────────────
+  {
+    name: 'schedule_task',
+    description: [
+      'Schedule a Gemini CLI agent to run on a cron schedule or as a delayed one-shot.',
+      'Spawns a persistent daemon that survives IDE/CLI restarts.',
+      'Results are saved to .agent/scheduled-results/ and can be retrieved with get_scheduled_results.',
+      '',
+      'Cron format: standard 5-field (minute hour day month weekday).',
+      'Examples: "*/30 * * * *" (every 30 min), "0 9 * * MON-FRI" (9am weekdays), "0 */2 * * *" (every 2 hours).',
+    ].join('\n'),
+    inputSchema: {
+      type: 'object',
+      properties: {
+        prompt: { type: 'string', description: 'Task prompt for the Gemini agent.' },
+        cron: { type: 'string', description: 'Cron expression (5-field). E.g. "0 9 * * *" for daily at 9am.' },
+        cwd: { type: 'string', description: 'Working directory for the scheduled task. Defaults to current directory.' },
+        skill: { type: 'string', description: 'Skill to activate for each run.' },
+        approval_mode: {
+          type: 'string',
+          enum: ['yolo', 'auto_edit', 'plan'],
+          description: 'Approval mode for scheduled runs. Default: yolo.',
+        },
+        catchup: { type: 'boolean', description: 'If true, run missed schedules on daemon restart. Default: false (skip missed).' },
+      },
+      required: ['prompt', 'cron'],
+    },
+  },
+  {
+    name: 'list_schedules',
+    description: 'List all scheduled tasks with their cron expressions, next run times, and daemon status.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+    },
+  },
+  {
+    name: 'cancel_schedule',
+    description: 'Cancel a scheduled task by ID. Removes it from the schedule. The daemon will auto-exit when no schedules remain.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        schedule_id: { type: 'string', description: 'Schedule ID to cancel.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['schedule_id'],
+    },
+  },
+  {
+    name: 'get_scheduled_results',
+    description: 'Get results from scheduled task executions. Returns the last 20 results, newest first.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        schedule_id: { type: 'string', description: 'Filter results by schedule ID. Omit to get all.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+    },
+  },
+  // ─── Pipeline Tools ────────────────────────────────────
+  {
+    name: 'create_pipeline',
+    description: 'Create a pipeline definition with sequential steps.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Pipeline name.' },
+        steps: {
+          type: 'array',
+          items: {
+            type: 'object',
+            properties: {
+              name: { type: 'string', description: 'Step name.' },
+              prompt: { type: 'string', description: 'Step prompt.' },
+              trigger: { type: 'string', description: 'Trigger condition.' },
+              skill: { type: 'string', description: 'Skill to use.' },
+              approval_mode: { type: 'string', description: 'Approval mode.' },
+              timeout: { type: 'number', description: 'Timeout in seconds.' },
+              max_bounces: { type: 'number', description: 'Maximum bounces allowed.' },
+              expected_output: { type: 'string', description: 'Expected output description.' },
+            },
+            required: ['name', 'prompt'],
+          },
+          description: 'Array of pipeline steps.',
+        },
+        on_error: { type: 'string', enum: ['stop', 'skip'], description: 'What to do on error. Default is stop.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['name', 'steps'],
+    },
+  },
+  {
+    name: 'run_pipeline',
+    description: 'Start executing a pipeline.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        pipeline_id: { type: 'string', description: 'Pipeline ID to run.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['pipeline_id'],
+    },
+  },
+  {
+    name: 'list_pipelines',
+    description: 'List all pipeline definitions.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+    },
+  },
+  {
+    name: 'get_pipeline_status',
+    description: 'Get status of a pipeline run.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        run_id: { type: 'string', description: 'Pipeline run ID.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['run_id'],
+    },
+  },
+  {
+    name: 'cancel_pipeline',
+    description: 'Cancel a running pipeline.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        run_id: { type: 'string', description: 'Pipeline run ID to cancel.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['run_id'],
+    },
+  },
+  {
+    name: 'signal_step_complete',
+    description: 'Signal that a pipeline step is complete (called BY agents running inside pipeline steps).',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        step_name: { type: 'string', description: 'Name of the completed step.' },
+        run_id: { type: 'string', description: 'Pipeline run ID for precise targeting. Provided in the task prompt.' },
+        output: { type: 'string', description: 'Optional output message or result.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['step_name'],
+    },
+  },
+  {
+    name: 'bounce_back',
+    description: 'Return task to a previous pipeline step with feedback about missing/insufficient data.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        step_name: { type: 'string', description: 'Name of the step to return to.' },
+        reason: { type: 'string', description: 'Feedback about missing/insufficient data.' },
+        run_id: { type: 'string', description: 'Pipeline run ID for precise targeting. Provided in the task prompt.' },
+        cwd: { type: 'string', description: 'Project directory. Defaults to current working directory.' },
+      },
+      required: ['step_name', 'reason'],
+    },
+  },
 ];
+