awb-agent-manager 1.0.0

package/dist/lib/base-session-manager.d.ts

@@ -0,0 +1,127 @@
+import { type ChildProcessByStdio } from 'node:child_process';
+import type { Readable, Writable } from 'node:stream';
+import { type CliAdapter, type HarnessSpec, type ParseResult, type ResolvedEffortPreset, type TurnImage } from './cli-adapters/base.js';
+import type { AwbConfig } from './rest.js';
+import type { SubagentMonitor, SubagentTapHandle } from './subagent-monitor.js';
+export interface BaseSessionOptions {
+    keyField: string;
+    logTag: string;
+    cfgPrefix: string;
+    kindLabel: 'chat_session' | 'ticket_session';
+}
+export interface SessionDelegationConfig {
+    enabled?: boolean;
+    maxConcurrent?: number;
+    ttlMinutes?: number;
+    idleMinutes?: number;
+    maxTurnsPerSession?: number;
+    claudeBin?: string;
+    persistentChatSessions?: boolean;
+    persistentTicketSessions?: boolean;
+    worktreeIsolation?: boolean;
+}
+export interface SessionAwareConfig extends AwbConfig {
+    delegation: SessionDelegationConfig;
+}
+export interface MonitorMeta {
+    ticket_id?: string;
+    ticket_title?: string;
+    role?: string;
+    trigger_source?: string;
+}
+export interface SpawnOpts {
+    onProgress?: (stage: string) => void;
+    monitorMeta?: MonitorMeta;
+    agentContext?: {
+        agent_id: string;
+        api_key: string;
+        cwd: string;
+        mcp_config_path: string;
+        cli: string;
+        cli_home_dir: string;
+        model?: string | null;
+        extra_env?: Record<string, string>;
+        credential_provider?: string | null;
+    };
+    firstTurnImages?: TurnImage[];
+    harness?: HarnessSpec | null;
+    effortPreset?: ResolvedEffortPreset | null;
+    envVars?: Record<string, string>;
+}
+interface TurnState {
+    onProgress: (stage: string) => void;
+    stage: string | null;
+    fired: {
+        thinking: boolean;
+        composing: boolean;
+    };
+    heartbeatTimer: NodeJS.Timeout | null;
+}
+export interface SessionRecord {
+    [key: string]: any;
+    pid: number;
+    cli_type: string;
+    adapter: CliAdapter;
+    child: ChildProcessByStdio<Writable, Readable, Readable>;
+    configPath: string | null;
+    configPathIsTemp: boolean;
+    pidPath: string | null;
+    turnCount: number;
+    startedAt: number;
+    lastTouchedAt: number;
+    idleTimer: NodeJS.Timeout | null;
+    unrespondedTurnCount: number;
+    unrespondedSince: number | null;
+    unhealthyKilled: boolean;
+    tap: SubagentTapHandle | null;
+    _currentTurn?: TurnState | null;
+    onResult?: (raw: any) => void;
+    splitRequested?: boolean;
+    splitReason?: string;
+    _effectiveApiKey?: string;
+    _lastLivenessPostAtMs?: number;
+}
+export interface InflightReservation {
+    agentId?: string;
+    ticketId?: string;
+    roomId?: string;
+}
+export declare class BaseSessionManager {
+    #private;
+    protected readonly _config: SessionAwareConfig;
+    protected readonly _sessions: Map<string, SessionRecord>;
+    protected readonly _inflight: Map<string, InflightReservation>;
+    protected readonly _outputRings: Map<number, string[]>;
+    constructor(config: SessionAwareConfig, options: BaseSessionOptions);
+    protected get _adapter(): CliAdapter;
+    protected _adapterFor(cli: string | null | undefined): CliAdapter;
+    setMonitor(monitor: SubagentMonitor | null): void;
+    protected _getSession(sessionKey: string): SessionRecord | undefined;
+    protected _isPidAlive(pid: number): boolean;
+    protected _getLiveSession(sessionKey: string): SessionRecord | undefined;
+    protected _ensureCapacity(): boolean;
+    protected _spawnSession(sessionKey: string, rolePrompt: string, firstTurnText: string, { onProgress, monitorMeta, agentContext, firstTurnImages, harness: rawHarness, effortPreset, envVars }?: SpawnOpts): Promise<SessionRecord | null>;
+    protected _sendFollowUp(sess: SessionRecord, turnText: string, { checkMaxTurns, onProgress, images, }?: {
+        checkMaxTurns?: boolean;
+        onProgress?: (stage: string) => void;
+        images?: TurnImage[];
+    }): void;
+    protected _writeTurn(sess: SessionRecord, text: string, images?: TurnImage[]): void;
+    protected _collectOutputTail(pid: number, maxChars: number): string;
+    protected _clearOutputBuffer(pid: number): void;
+    protected _resetIdleTimer(sess: SessionRecord): void;
+    protected _onStdoutParsed(_sess: SessionRecord, _parsed: ParseResult, _rawLine: string): void;
+    protected _onStderrLine(_sess: SessionRecord, _line: string): void;
+    protected _onChildExit(_sess: SessionRecord, _code: number | null, _signal: NodeJS.Signals | null): Promise<void>;
+    protected _rememberDedup(key: string): boolean;
+    protected _forgetDedup(key: string): void;
+    stopForAgent(agentId: string): Promise<{
+        count: number;
+        inflight: Array<{
+            ticketId: string;
+            role: string;
+        }>;
+    }>;
+    stop(): Promise<void>;
+}
+export {};