@rk0429/agentic-relay 0.12.1 → 0.12.3

package/dist/relay.mjs

@@ -46,6 +46,129 @@ var init_logger = __esm({
   }
 });
 
+// src/mcp-server/deferred-cleanup-task-store.ts
+import { isTerminal } from "@modelcontextprotocol/sdk/experimental/tasks/interfaces.js";
+import { randomBytes } from "crypto";
+var DeferredCleanupTaskStore;
+var init_deferred_cleanup_task_store = __esm({
+  "src/mcp-server/deferred-cleanup-task-store.ts"() {
+    "use strict";
+    DeferredCleanupTaskStore = class {
+      tasks = /* @__PURE__ */ new Map();
+      cleanupTimers = /* @__PURE__ */ new Map();
+      generateTaskId() {
+        return randomBytes(16).toString("hex");
+      }
+      async createTask(taskParams, requestId, request, _sessionId) {
+        const taskId = this.generateTaskId();
+        if (this.tasks.has(taskId)) {
+          throw new Error(`Task with ID ${taskId} already exists`);
+        }
+        const actualTtl = taskParams.ttl ?? null;
+        const createdAt = (/* @__PURE__ */ new Date()).toISOString();
+        const task = {
+          taskId,
+          status: "working",
+          ttl: actualTtl,
+          createdAt,
+          lastUpdatedAt: createdAt,
+          pollInterval: taskParams.pollInterval ?? 1e3
+        };
+        this.tasks.set(taskId, { task, request, requestId });
+        return { ...task };
+      }
+      async getTask(taskId, _sessionId) {
+        const stored = this.tasks.get(taskId);
+        return stored ? { ...stored.task } : null;
+      }
+      async storeTaskResult(taskId, status, result, _sessionId) {
+        const stored = this.tasks.get(taskId);
+        if (!stored) {
+          throw new Error(`Task with ID ${taskId} not found`);
+        }
+        if (isTerminal(stored.task.status)) {
+          throw new Error(
+            `Cannot store result for task ${taskId} in terminal status '${stored.task.status}'. Task results can only be stored once.`
+          );
+        }
+        stored.result = result;
+        stored.task.status = status;
+        stored.task.lastUpdatedAt = (/* @__PURE__ */ new Date()).toISOString();
+        this.scheduleCleanup(taskId, stored.task.ttl);
+      }
+      async getTaskResult(taskId, _sessionId) {
+        const stored = this.tasks.get(taskId);
+        if (!stored) {
+          throw new Error(`Task with ID ${taskId} not found`);
+        }
+        if (!stored.result) {
+          throw new Error(`Task ${taskId} has no result stored`);
+        }
+        return stored.result;
+      }
+      async updateTaskStatus(taskId, status, statusMessage, _sessionId) {
+        const stored = this.tasks.get(taskId);
+        if (!stored) {
+          throw new Error(`Task with ID ${taskId} not found`);
+        }
+        if (isTerminal(stored.task.status)) {
+          throw new Error(
+            `Cannot update task ${taskId} from terminal status '${stored.task.status}'.`
+          );
+        }
+        stored.task.status = status;
+        if (statusMessage) {
+          stored.task.statusMessage = statusMessage;
+        }
+        stored.task.lastUpdatedAt = (/* @__PURE__ */ new Date()).toISOString();
+        if (isTerminal(status)) {
+          this.scheduleCleanup(taskId, stored.task.ttl);
+        }
+      }
+      async listTasks(cursor, _sessionId) {
+        const PAGE_SIZE = 10;
+        const allTaskIds = Array.from(this.tasks.keys());
+        let startIndex = 0;
+        if (cursor) {
+          const cursorIndex = allTaskIds.indexOf(cursor);
+          if (cursorIndex >= 0) {
+            startIndex = cursorIndex + 1;
+          } else {
+            throw new Error(`Invalid cursor: ${cursor}`);
+          }
+        }
+        const pageTaskIds = allTaskIds.slice(startIndex, startIndex + PAGE_SIZE);
+        const tasks = pageTaskIds.map((id) => {
+          const stored = this.tasks.get(id);
+          return { ...stored.task };
+        });
+        const nextCursor = startIndex + PAGE_SIZE < allTaskIds.length ? pageTaskIds[pageTaskIds.length - 1] : void 0;
+        return { tasks, nextCursor };
+      }
+      /** Cleanup all timers (for testing or graceful shutdown) */
+      cleanup() {
+        for (const timer of this.cleanupTimers.values()) {
+          clearTimeout(timer);
+        }
+        this.cleanupTimers.clear();
+        this.tasks.clear();
+      }
+      scheduleCleanup(taskId, ttl) {
+        if (!ttl) return;
+        const existingTimer = this.cleanupTimers.get(taskId);
+        if (existingTimer) {
+          clearTimeout(existingTimer);
+        }
+        const timer = setTimeout(() => {
+          this.tasks.delete(taskId);
+          this.cleanupTimers.delete(taskId);
+        }, ttl);
+        this.cleanupTimers.set(taskId, timer);
+      }
+    };
+  }
+});
+
 // src/mcp-server/recursion-guard.ts
 import { createHash } from "crypto";
 var RecursionGuard;
@@ -56,7 +179,7 @@ var init_recursion_guard = __esm({
       constructor(config = {
         maxDepth: 5,
         maxCallsPerSession: 50,
-        timeoutSec: 1800
+        timeoutSec: 86400
       }) {
         this.config = config;
       }
@@ -1158,13 +1281,12 @@ __export(server_exports, {
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
-import { InMemoryTaskStore } from "@modelcontextprotocol/sdk/experimental/tasks/stores/in-memory.js";
 import { InMemoryTaskMessageQueue } from "@modelcontextprotocol/sdk/experimental/tasks/stores/in-memory.js";
 import { createServer } from "http";
 import { randomUUID } from "crypto";
 import { z as z5 } from "zod";
 function createMcpServerOptions() {
-  const taskStore = new InMemoryTaskStore();
+  const taskStore = new DeferredCleanupTaskStore();
   return {
     capabilities: { tasks: { requests: { tools: { call: {} } } } },
     taskStore,
@@ -1176,6 +1298,7 @@ var spawnAgentsParallelInputShape, MAX_CHILD_HTTP_SESSIONS, RelayMCPServer;
 var init_server = __esm({
   "src/mcp-server/server.ts"() {
     "use strict";
+    init_deferred_cleanup_task_store();
     init_recursion_guard();
     init_spawn_agent();
     init_spawn_agents_parallel();
@@ -1201,7 +1324,7 @@ var init_server = __esm({
         this.guard = new RecursionGuard(guardConfig);
         this.backendSelector = new BackendSelector();
         this.server = new McpServer(
-          { name: "agentic-relay", version: "0.12.1" },
+          { name: "agentic-relay", version: "0.12.3" },
           createMcpServerOptions()
         );
         this.registerTools(this.server);
@@ -1565,7 +1688,7 @@ var init_server = __esm({
             sessionIdGenerator: () => randomUUID()
           });
           const server = new McpServer(
-            { name: "agentic-relay", version: "0.12.1" },
+            { name: "agentic-relay", version: "0.12.3" },
             createMcpServerOptions()
           );
           this.registerTools(server);
@@ -4343,7 +4466,7 @@ function createMCPCommand(configManager2, registry2, sessionManager2, hooksEngin
               guardConfig = {
                 maxDepth: config.mcpServerMode.maxDepth ?? 5,
                 maxCallsPerSession: config.mcpServerMode.maxCallsPerSession ?? 20,
-                timeoutSec: config.mcpServerMode.timeoutSec ?? 300
+                timeoutSec: config.mcpServerMode.timeoutSec ?? 86400
               };
             }
           } catch {
@@ -4517,7 +4640,7 @@ function createVersionCommand(registry2) {
       description: "Show relay and backend versions"
     },
     async run() {
-      const relayVersion = "0.12.1";
+      const relayVersion = "0.12.3";
       console.log(`agentic-relay v${relayVersion}`);
       console.log("");
       console.log("Backends:");
@@ -4867,7 +4990,7 @@ void configManager.getConfig().then((config) => {
 var main = defineCommand10({
   meta: {
     name: "relay",
-    version: "0.12.1",
+    version: "0.12.3",
     description: "Unified CLI proxy for Claude Code, Codex CLI, and Gemini CLI"
   },
   subCommands: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rk0429/agentic-relay",
-  "version": "0.12.1",
+  "version": "0.12.3",
   "description": "Unified CLI proxy for Claude Code, Codex CLI, and Gemini CLI with MCP-based multi-layer sub-agent orchestration",
   "type": "module",
   "license": "Apache-2.0",