mcp-subagents-opencode 1.0.0

package/build/services/sdk-session-adapter.d.ts

@@ -0,0 +1,203 @@
+/**
+ * SDK Session Adapter - Bridges Copilot SDK sessions to the MCP server's TaskState model.
+ *
+ * This adapter:
+ * - Maps SDK session events to TaskState updates using SDK's native types
+ * - Handles mid-session rate limit detection via session.error events
+ * - Triggers account rotation and session resume on rate limits
+ * - Manages streaming output accumulation
+ * - Provides unified error handling with proper typing
+ * - Collects completion metrics from session.shutdown
+ * - Tracks quota info from assistant.usage
+ * - Monitors tool execution and subagent activity
+ */
+import type { CopilotSession } from '@github/copilot-sdk';
+import { type QuotaInfo, type ToolMetrics, type SubagentInfo } from '../types.js';
+type RotationRequestCallback = (taskId: string, sessionId: string, reason: string, statusCode?: number) => Promise<{
+    rotated: boolean;
+    newSession?: CopilotSession;
+}>;
+interface SessionBinding {
+    taskId: string;
+    session: CopilotSession;
+    sessionId: string;
+    unsubscribe: () => void;
+    outputBuffer: string[];
+    reasoningBuffer: string[];
+    lastMessageId?: string;
+    startTime: Date;
+    isCompleted: boolean;
+    isPaused: boolean;
+    error?: string;
+    rotationAttempts: number;
+    maxRotationAttempts: number;
+    rotationInProgress: boolean;
+    rateLimitInfo?: {
+        statusCode: number;
+        resetDate?: string;
+        remainingPercentage?: number;
+    };
+    pendingPrompt?: string;
+    turnCount: number;
+    totalTokens: {
+        input: number;
+        output: number;
+    };
+    toolMetrics: Map<string, ToolMetrics>;
+    toolStartTimes: Map<string, number>;
+    toolCallIdToName: Map<string, string>;
+    activeSubagents: Map<string, SubagentInfo>;
+    completedSubagents: SubagentInfo[];
+    quotas: Map<string, QuotaInfo>;
+    isUnbound: boolean;
+}
+declare class SDKSessionAdapter {
+    private bindings;
+    private rotationCallback?;
+    /**
+     * Set the callback for rotation requests.
+     * This is called when mid-session rate limits are detected.
+     */
+    onRotationRequest(callback: RotationRequestCallback): void;
+    /**
+     * Perform a health check on the current account by creating a test session.
+     * Uses claude-haiku-4.5 for fast/cheap verification.
+     * Returns true if the account can successfully respond.
+     */
+    performHealthCheck(cwd: string): Promise<boolean>;
+    /**
+     * Bind a SDK session to a task, setting up event handlers.
+     */
+    bind(taskId: string, session: CopilotSession, pendingPrompt?: string): void;
+    /**
+     * Handle SDK session events and map them to task updates.
+     * Uses type narrowing for type-safe event handling.
+     */
+    private handleEvent;
+    /**
+     * Handle session.start event
+     */
+    private handleSessionStart;
+    /**
+     * Handle assistant.turn_start event - track turn count
+     */
+    private handleTurnStart;
+    /**
+     * Handle session.idle event - indicates completion
+     */
+    private handleSessionIdle;
+    /**
+     * Handle session.error event - key for mid-session rate limit detection
+     * Now stores structured failure context
+     */
+    private handleSessionError;
+    /**
+     * Attempt to rotate to a new account and resume the session
+     */
+    private attemptRotationAndResume;
+    /**
+     * Rebind a task with a new session after rotation
+     */
+    private rebindWithNewSession;
+    /**
+     * Mark task as rate limited (for when rotation is exhausted)
+     */
+    private markAsRateLimited;
+    /**
+     * Mark task as failed with structured failure context
+     */
+    private markAsFailed;
+    /**
+     * Handle assistant.message_delta event (streaming)
+     */
+    private handleMessageDelta;
+    /**
+     * Handle assistant.message event (final message)
+     * Also checks for string-based rate limit detection as fallback
+     */
+    private handleAssistantMessage;
+    /**
+     * Handle assistant.usage event for quota tracking
+     * Stores structured quota info and updates session metrics
+     */
+    private handleUsage;
+    /**
+     * Handle tool.execution_start event - track tool metrics
+     */
+    private handleToolStart;
+    /**
+     * Handle tool.execution_complete event - finalize tool metrics
+     */
+    private handleToolComplete;
+    /**
+     * Handle subagent.started event
+     */
+    private handleSubagentStarted;
+    /**
+     * Handle subagent.completed event
+     */
+    private handleSubagentCompleted;
+    /**
+     * Handle subagent.failed event
+     */
+    private handleSubagentFailed;
+    /**
+     * Handle session.shutdown event - extract completion metrics
+     */
+    private handleSessionShutdown;
+    /**
+     * Handle abort event
+     */
+    private handleAbort;
+    /**
+     * Update session metrics in task state
+     * Uses SDK's UsageMetricsTracker for token/request metrics, combined with our custom tracking
+     */
+    private updateSessionMetrics;
+    /**
+     * Unbind a session from a task.
+     * Also destroys the session and removes it from the client manager's tracking
+     * to prevent PTY file descriptor leaks.
+     */
+    unbind(taskId: string): void;
+    /**
+     * Get the binding for a task.
+     */
+    getBinding(taskId: string): SessionBinding | undefined;
+    /**
+     * Get the session for a task.
+     */
+    getSession(taskId: string): CopilotSession | undefined;
+    /**
+     * Check if a task has an active session.
+     */
+    hasSession(taskId: string): boolean;
+    /**
+     * Get accumulated output for a task.
+     */
+    getOutput(taskId: string): string[];
+    /**
+     * Mark a task as timed out.
+     */
+    markTimedOut(taskId: string, timeoutMs: number): void;
+    /**
+     * Cleanup all bindings.
+     * Destroys all sessions to release PTY file descriptors.
+     */
+    cleanup(): void;
+    /**
+     * Get statistics.
+     */
+    getStats(): {
+        activeBindings: number;
+        totalRotations: number;
+        totalTurns: number;
+        totalTokens: {
+            input: number;
+            output: number;
+        };
+    };
+}
+export declare const sdkSessionAdapter: SDKSessionAdapter;
+export {};
+//# sourceMappingURL=sdk-session-adapter.d.ts.map

package/build/services/sdk-session-adapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sdk-session-adapter.d.ts","sourceRoot":"","sources":["../../src/services/sdk-session-adapter.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAgB,MAAM,qBAAqB,CAAC;AAGxE,OAAO,EAQL,KAAK,SAAS,EACd,KAAK,WAAW,EAChB,KAAK,YAAY,EAElB,MAAM,aAAa,CAAC;AAyBrB,KAAK,uBAAuB,GAAG,CAC7B,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EACd,UAAU,CAAC,EAAE,MAAM,KAChB,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,UAAU,CAAC,EAAE,cAAc,CAAA;CAAE,CAAC,CAAC;AA0BhE,UAAU,cAAc;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,cAAc,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,IAAI,CAAC;IACxB,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,IAAI,CAAC;IAChB,WAAW,EAAE,OAAO,CAAC;IACrB,QAAQ,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,gBAAgB,EAAE,MAAM,CAAC;IACzB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,kBAAkB,EAAE,OAAO,CAAC;IAC5B,aAAa,CAAC,EAAE;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,mBAAmB,CAAC,EAAE,MAAM,CAAC;KAC9B,CAAC;IACF,aAAa,CAAC,EAAE,MAAM,CAAC;IAGvB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/C,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACtC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,gBAAgB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACtC,eAAe,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC3C,kBAAkB,EAAE,YAAY,EAAE,CAAC;IACnC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/B,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,cAAM,iBAAiB;IACrB,OAAO,CAAC,QAAQ,CAA0C;IAC1D,OAAO,CAAC,gBAAgB,CAAC,CAA0B;IAEnD;;;OAGG;IACH,iBAAiB,CAAC,QAAQ,EAAE,uBAAuB,GAAG,IAAI;IAI1D;;;;OAIG;IACG,kBAAkB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAoCvD;;OAEG;IACH,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI;IA2D3E;;;OAGG;YACW,WAAW;IAyIzB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;OAEG;IACH,OAAO,CAAC,eAAe;IAQvB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA8BzB;;;OAGG;YACW,kBAAkB;IAwGhC;;OAEG;YACW,wBAAwB;IAsHtC;;OAEG;YACW,oBAAoB;IA4ElC;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAsCzB;;OAEG;IACH,OAAO,CAAC,YAAY;IAuBpB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAW1B;;;OAGG;YACW,sBAAsB;IA4EpC;;;OAGG;IACH,OAAO,CAAC,WAAW;IA0EnB;;OAEG;IACH,OAAO,CAAC,eAAe;IA2BvB;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAwC1B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAiB7B;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAe/B;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAgB5B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAqF7B;;OAEG;IACH,OAAO,CAAC,WAAW;IA+BnB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;IA0B5B;;;;OAIG;IACH,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAgB5B;;OAEG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAItD;;OAEG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAItD;;OAEG;IACH,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;IAInC;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE;IAInC;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAmCrD;;;OAGG;IACH,OAAO,IAAI,IAAI;IAWf;;OAEG;IACH,QAAQ,IAAI;QACV,cAAc,EAAE,MAAM,CAAC;QACvB,cAAc,EAAE,MAAM,CAAC;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC;KAChD;CAmBF;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}