opencode-swarm-plugin 0.2.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-swarm-plugin",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "Multi-agent swarm coordination for OpenCode with learning capabilities, beads integration, and Agent Mail",
   "type": "module",
   "main": "dist/index.js",
@@ -18,13 +18,12 @@
     "test:watch": "bun test --watch",
     "test:integration": "bun test src/*.integration.test.ts",
     "test:all": "bun test",
-    "docker:test": "docker compose up --build --abort-on-container-exit --exit-code-from test-runner",
-    "docker:down": "docker compose down -v",
     "typecheck": "tsc --noEmit",
     "clean": "rm -rf dist"
   },
   "dependencies": {
     "@opencode-ai/plugin": "^1.0.134",
+    "ioredis": "^5.4.1",
     "zod": "4.1.8"
   },
   "devDependencies": {

package/src/agent-mail.ts

@@ -17,6 +17,7 @@
 import { tool } from "@opencode-ai/plugin";
 import { z } from "zod";
 import { isToolAvailable, warnMissingTool } from "./tool-availability";
+import { getRateLimiter, type RateLimiter } from "./rate-limiter";
 
 // ============================================================================
 // Configuration
@@ -157,6 +158,24 @@ export class FileReservationConflictError extends Error {
   }
 }
 
+export class RateLimitExceededError extends Error {
+  constructor(
+    public readonly endpoint: string,
+    public readonly remaining: number,
+    public readonly resetAt: number,
+  ) {
+    const resetDate = new Date(resetAt);
+    const waitMs = Math.max(0, resetAt - Date.now());
+    const waitSec = Math.ceil(waitMs / 1000);
+    super(
+      `Rate limit exceeded for ${endpoint}. ` +
+        `${remaining} remaining. ` +
+        `Retry in ${waitSec}s (at ${resetDate.toISOString()})`,
+    );
+    this.name = "RateLimitExceededError";
+  }
+}
+
 // ============================================================================
 // MCP Client
 // ============================================================================
@@ -190,6 +209,73 @@ export function resetAgentMailCache(): void {
   agentMailAvailable = null;
 }
 
+/** Cached rate limiter instance */
+let rateLimiter: RateLimiter | null = null;
+
+/** Whether rate limiting is enabled (can be disabled via env var) */
+const RATE_LIMITING_ENABLED =
+  process.env.OPENCODE_RATE_LIMIT_DISABLED !== "true";
+
+/**
+ * Check rate limit for an endpoint and throw if exceeded
+ *
+ * @param agentName - The agent making the request
+ * @param endpoint - The endpoint being accessed (e.g., "send", "inbox")
+ * @throws RateLimitExceededError if rate limit is exceeded
+ */
+async function checkRateLimit(
+  agentName: string,
+  endpoint: string,
+): Promise<void> {
+  if (!RATE_LIMITING_ENABLED) {
+    return;
+  }
+
+  if (!rateLimiter) {
+    rateLimiter = await getRateLimiter();
+  }
+
+  const result = await rateLimiter.checkLimit(agentName, endpoint);
+  if (!result.allowed) {
+    throw new RateLimitExceededError(
+      endpoint,
+      result.remaining,
+      result.resetAt,
+    );
+  }
+}
+
+/**
+ * Record a request against the rate limit (call after successful request)
+ *
+ * @param agentName - The agent making the request
+ * @param endpoint - The endpoint being accessed
+ */
+async function recordRateLimitedRequest(
+  agentName: string,
+  endpoint: string,
+): Promise<void> {
+  if (!RATE_LIMITING_ENABLED) {
+    return;
+  }
+
+  if (!rateLimiter) {
+    rateLimiter = await getRateLimiter();
+  }
+
+  await rateLimiter.recordRequest(agentName, endpoint);
+}
+
+/**
+ * Reset rate limiter (for testing)
+ */
+export async function resetRateLimiterCache(): Promise<void> {
+  if (rateLimiter) {
+    await rateLimiter.close();
+    rateLimiter = null;
+  }
+}
+
 /**
  * Call an Agent Mail MCP tool
  *
@@ -379,6 +465,9 @@ export const agentmail_send = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit before sending
+    await checkRateLimit(state.agentName, "send");
+
     await mcpCall("send_message", {
       project_key: state.projectKey,
       sender_name: state.agentName,
@@ -390,6 +479,9 @@ export const agentmail_send = tool({
       ack_required: args.ack_required || false,
     });
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "send");
+
     return `Message sent to ${args.to.join(", ")}`;
   },
 });
@@ -417,6 +509,9 @@ export const agentmail_inbox = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "inbox");
+
     // CRITICAL: Enforce context-safe defaults
     const limit = Math.min(args.limit || MAX_INBOX_LIMIT, MAX_INBOX_LIMIT);
 
@@ -429,6 +524,9 @@ export const agentmail_inbox = tool({
       since_ts: args.since_ts,
     });
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "inbox");
+
     return JSON.stringify(messages, null, 2);
   },
 });
@@ -444,6 +542,9 @@ export const agentmail_read_message = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "read_message");
+
     // Mark as read
     await mcpCall("mark_message_read", {
       project_key: state.projectKey,
@@ -465,6 +566,9 @@ export const agentmail_read_message = tool({
       return `Message ${args.message_id} not found`;
     }
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "read_message");
+
     return JSON.stringify(message, null, 2);
   },
 });
@@ -484,6 +588,9 @@ export const agentmail_summarize_thread = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "summarize_thread");
+
     const summary = await mcpCall<ThreadSummary>("summarize_thread", {
       project_key: state.projectKey,
       thread_id: args.thread_id,
@@ -491,6 +598,9 @@ export const agentmail_summarize_thread = tool({
       llm_mode: true, // Use LLM for better summaries
     });
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "summarize_thread");
+
     return JSON.stringify(summary, null, 2);
   },
 });
@@ -520,6 +630,9 @@ export const agentmail_reserve = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "reserve");
+
     const result = await mcpCall<ReservationResult>("file_reservation_paths", {
       project_key: state.projectKey,
       agent_name: state.agentName,
@@ -563,6 +676,9 @@ export const agentmail_reserve = tool({
     state.reservations = [...state.reservations, ...reservationIds];
     setState(ctx.sessionID, state);
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "reserve");
+
     if (granted.length === 0) {
       return "No paths were reserved (empty granted list)";
     }
@@ -591,6 +707,9 @@ export const agentmail_release = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "release");
+
     const result = await mcpCall<{ released: number; released_at: string }>(
       "release_file_reservations",
       {
@@ -605,6 +724,9 @@ export const agentmail_release = tool({
     state.reservations = [];
     setState(ctx.sessionID, state);
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "release");
+
     return `Released ${result.released} reservation(s)`;
   },
 });
@@ -620,12 +742,18 @@ export const agentmail_ack = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "ack");
+
     await mcpCall("acknowledge_message", {
       project_key: state.projectKey,
       agent_name: state.agentName,
       message_id: args.message_id,
     });
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "ack");
+
     return `Acknowledged message ${args.message_id}`;
   },
 });
@@ -647,12 +775,18 @@ export const agentmail_search = tool({
   async execute(args, ctx) {
     const state = requireState(ctx.sessionID);
 
+    // Check rate limit
+    await checkRateLimit(state.agentName, "search");
+
     const results = await mcpCall<MessageHeader[]>("search_messages", {
       project_key: state.projectKey,
       query: args.query,
       limit: args.limit || 20,
     });
 
+    // Record successful request
+    await recordRateLimitedRequest(state.agentName, "search");
+
     return JSON.stringify(results, null, 2);
   },
 });

package/src/index.ts

@@ -263,7 +263,9 @@ export {
   SwarmError,
   DecompositionError,
   formatSubtaskPrompt,
+  formatSubtaskPromptV2,
   formatEvaluationPrompt,
+  SUBTASK_PROMPT_V2,
 } from "./swarm";
 
 /**