omnara 0.1.0

package/dist/orchestrator/agent-orchestrator.js

@@ -0,0 +1,401 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentOrchestrator = void 0;
+exports.startAgentOrchestrator = startAgentOrchestrator;
+const crypto_1 = require("crypto");
+const ai_client_1 = require("./ai-client");
+const session_1 = require("./session");
+const worker_manager_1 = require("./worker-manager");
+const worker_tools_1 = require("./tools/worker-tools");
+const WORKER_TOOL_NAME = 'mcp__worker__launch_worker';
+class AgentOrchestrator {
+    constructor(dependencies) {
+        this.sessions = new Map();
+        this.clients = new Map();
+        this.toolUseMap = new Map();
+        this.started = false;
+        const resolveSessionIdBySdk = (sdkSessionId) => this.findSessionIdBySdkSession(sdkSessionId);
+        const workerClientFactory = dependencies?.createWorkerClient ??
+            (() => new ai_client_1.AIClient({
+                defaults: {
+                    allowedTools: [...ai_client_1.BASE_ALLOWED_TOOLS],
+                },
+            }));
+        this.workerManager =
+            dependencies?.workerManager ??
+                new worker_manager_1.WorkerManager({
+                    createClient: workerClientFactory,
+                    resolveSessionIdBySdk,
+                });
+        const workerMcpServer = (0, worker_tools_1.createWorkerMcpServer)(this.workerManager, {
+            resolveSessionIdBySdk,
+            resolveSessionIdByToolUse: (toolUseId) => this.consumeToolUse(toolUseId),
+        });
+        const mainClientFactory = dependencies?.createMainClient ??
+            (() => new ai_client_1.AIClient({
+                defaults: {
+                    allowedTools: [...ai_client_1.BASE_ALLOWED_TOOLS, WORKER_TOOL_NAME],
+                    mcpServers: {
+                        worker: workerMcpServer,
+                    },
+                },
+            }));
+        this.aiClient = mainClientFactory();
+        this.registerWorkerListeners();
+    }
+    ensureSessionExists(sessionId) {
+        const existing = this.sessions.get(sessionId);
+        if (existing) {
+            return existing;
+        }
+        const session = new session_1.Session(sessionId, this.aiClient, (id) => this.handleSessionEmpty(id), (toolUseId) => this.registerToolUse(toolUseId, sessionId));
+        this.sessions.set(sessionId, session);
+        return session;
+    }
+    getSession(sessionId) {
+        return this.sessions.get(sessionId);
+    }
+    getWorkerManager() {
+        return this.workerManager;
+    }
+    addClient(adapter) {
+        const clientId = (0, crypto_1.randomUUID)();
+        const connection = {
+            id: clientId,
+            sessionIds: new Set(),
+            send: adapter.send,
+            canSend: adapter.canSend,
+            close: adapter.close,
+        };
+        this.clients.set(clientId, connection);
+        let disconnected = false;
+        const disconnect = () => {
+            if (disconnected) {
+                return;
+            }
+            disconnected = true;
+            this.handleDisconnect(connection);
+            try {
+                connection.close?.();
+            }
+            catch (error) {
+                console.error('Failed to close client connection:', error);
+            }
+        };
+        return {
+            id: clientId,
+            dispatch: (message) => {
+                this.handleClientMessage(connection, message);
+            },
+            disconnect,
+        };
+    }
+    async start() {
+        if (this.started) {
+            return;
+        }
+        this.started = true;
+    }
+    async stop() {
+        for (const connection of Array.from(this.clients.values())) {
+            this.handleDisconnect(connection);
+            try {
+                connection.close?.();
+            }
+            catch (error) {
+                console.error('Failed to close client connection:', error);
+            }
+        }
+        this.clients.clear();
+        for (const session of this.sessions.values()) {
+            session.close();
+        }
+        this.sessions.clear();
+        this.started = false;
+    }
+    handleDisconnect(connection) {
+        for (const sessionId of connection.sessionIds) {
+            const session = this.sessions.get(sessionId);
+            if (session) {
+                session.unsubscribe(connection.id);
+            }
+        }
+        this.clients.delete(connection.id);
+    }
+    handleClientMessage(connection, message) {
+        switch (message.type) {
+            case 'chat':
+                this.handleChatMessage(connection, message);
+                break;
+            case 'session_init':
+                this.handleSessionInit(connection, message);
+                break;
+            case 'join':
+                this.handleJoin(connection, message.sessionId);
+                break;
+            case 'leave':
+                this.handleLeave(connection, message.sessionId);
+                break;
+            case 'ping':
+                this.sendToConnection(connection, { type: 'pong' });
+                break;
+            case 'cancel_worker':
+                this.handleCancelWorker(connection, message.workerId);
+                break;
+            default:
+                this.sendToConnection(connection, {
+                    type: 'error',
+                    message: `Unsupported message type: ${message.type}`,
+                });
+        }
+    }
+    handleSessionInit(connection, message) {
+        this.sendToConnection(connection, {
+            type: 'session_initialized',
+            sessionId: message.sessionId,
+            parentSessionId: message.parentSessionId ?? null,
+        });
+    }
+    handleChatMessage(connection, message) {
+        let session;
+        let sessionId = message.sessionId;
+        if (sessionId) {
+            session = this.sessions.get(sessionId);
+            if (!session) {
+                this.sendToConnection(connection, {
+                    type: 'error',
+                    sessionId,
+                    message: 'Session not found.',
+                });
+                return;
+            }
+        }
+        else {
+            session = this.createSession();
+            sessionId = session.id;
+            this.sendToConnection(connection, {
+                type: 'session_created',
+                sessionId,
+            });
+        }
+        if (!connection.sessionIds.has(sessionId)) {
+            session.subscribe({
+                id: connection.id,
+                send: (payload) => {
+                    this.sendToConnection(connection, payload);
+                },
+                canSend: connection.canSend,
+            });
+            connection.sessionIds.add(sessionId);
+        }
+        session.logInbound({ type: 'chat', text: message.text });
+        session.announceInboundMessage(message.text, 'user');
+        session.enqueueUserMessage(message.text);
+    }
+    handleJoin(connection, sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            this.sendToConnection(connection, {
+                type: 'error',
+                sessionId,
+                message: 'Session not found.',
+            });
+            return;
+        }
+        if (!connection.sessionIds.has(sessionId)) {
+            session.subscribe({
+                id: connection.id,
+                send: (payload) => {
+                    this.sendToConnection(connection, payload);
+                },
+                canSend: connection.canSend,
+            });
+            connection.sessionIds.add(sessionId);
+        }
+        else {
+            this.sendToConnection(connection, {
+                type: 'session_joined',
+                sessionId,
+            });
+        }
+    }
+    handleLeave(connection, sessionId) {
+        if (!connection.sessionIds.has(sessionId)) {
+            this.sendToConnection(connection, {
+                type: 'error',
+                sessionId,
+                message: 'You are not subscribed to this session.',
+            });
+            return;
+        }
+        connection.sessionIds.delete(sessionId);
+        const session = this.sessions.get(sessionId);
+        if (session) {
+            session.unsubscribe(connection.id);
+        }
+    }
+    handleCancelWorker(connection, workerId) {
+        const canceled = this.workerManager.cancelWorker(workerId);
+        if (!canceled) {
+            this.sendToConnection(connection, {
+                type: 'error',
+                message: `Worker ${workerId} not found or not running.`,
+            });
+        }
+    }
+    createSession() {
+        const sessionId = (0, crypto_1.randomUUID)();
+        return this.ensureSessionExists(sessionId);
+    }
+    registerToolUse(toolUseId, sessionId) {
+        this.toolUseMap.set(toolUseId, sessionId);
+    }
+    consumeToolUse(toolUseId) {
+        const sessionId = this.toolUseMap.get(toolUseId);
+        if (sessionId) {
+            this.toolUseMap.delete(toolUseId);
+        }
+        return sessionId;
+    }
+    handleSessionEmpty(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return;
+        }
+        if (!session.hasSubscribers()) {
+            session.close();
+            this.sessions.delete(sessionId);
+        }
+    }
+    registerWorkerListeners() {
+        this.workerManager.on('session_start', (record) => {
+            this.handleWorkerSessionStart(record);
+        });
+        this.workerManager.on('message', (event) => {
+            this.handleWorkerMessage(event);
+        });
+        this.workerManager.on('update', (record) => {
+            this.broadcastWorkerUpdate(record);
+        });
+        this.workerManager.on('completed', (record) => {
+            this.broadcastWorkerUpdate(record);
+            this.handleWorkerCompletion(record);
+        });
+    }
+    broadcastWorkerUpdate(record) {
+        const payload = this.toWorkerPayload(record);
+        this.broadcastToAll({
+            type: 'worker_update',
+            sessionId: record.sessionId,
+            worker: payload,
+        });
+    }
+    toWorkerPayload(record) {
+        const latest = record.progress.length > 0 ? record.progress[record.progress.length - 1]?.message : undefined;
+        return {
+            id: record.id,
+            status: record.status,
+            prompt: record.prompt,
+            lastUpdateAt: record.lastUpdateAt,
+            latestMessage: latest,
+            result: record.result,
+            error: record.error,
+            originSessionId: record.originSessionId,
+            sessionId: record.sessionId,
+        };
+    }
+    handleWorkerSessionStart(record) {
+        this.broadcastToAll({
+            type: 'session_init',
+            sessionId: record.sessionId,
+            parentSessionId: record.originSessionId ?? null,
+            metadata: record.metadata ?? undefined,
+        });
+    }
+    handleWorkerMessage(event) {
+        switch (event.messageType) {
+            case 'assistant_text':
+                if (!event.text) {
+                    return;
+                }
+                this.broadcastToAll({
+                    type: 'assistant_text',
+                    sessionId: event.record.sessionId,
+                    text: event.text,
+                });
+                break;
+            case 'assistant_complete':
+                this.broadcastToAll({
+                    type: 'assistant_complete',
+                    sessionId: event.record.sessionId,
+                });
+                break;
+            default:
+                break;
+        }
+    }
+    handleWorkerCompletion(record) {
+        const sessionId = record.originSessionId;
+        if (!sessionId) {
+            return;
+        }
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return;
+        }
+        const summary = this.buildWorkerSummary(record);
+        if (!summary) {
+            return;
+        }
+        session.logInbound({ type: 'worker_result', workerId: record.id, summary });
+        session.announceInboundMessage(summary, 'worker');
+        session.enqueueUserMessage(summary);
+    }
+    buildWorkerSummary(record) {
+        switch (record.status) {
+            case 'completed':
+                return record.result ?? `Worker ${record.id} completed.`;
+            case 'error':
+                return `Worker ${record.id} failed: ${record.error ?? 'Unknown error.'}`;
+            case 'canceled':
+                return `Worker ${record.id} was canceled.`;
+            default:
+                return null;
+        }
+    }
+    broadcastToAll(message) {
+        for (const connection of this.clients.values()) {
+            this.sendToConnection(connection, message);
+        }
+    }
+    findSessionIdBySdkSession(sdkSessionId) {
+        if (!sdkSessionId) {
+            return undefined;
+        }
+        for (const [serverId, session] of this.sessions.entries()) {
+            if (session.sdkSessionIdentifier === sdkSessionId) {
+                return serverId;
+            }
+        }
+        return undefined;
+    }
+    sendToConnection(connection, message) {
+        if (connection.canSend && !connection.canSend()) {
+            return;
+        }
+        try {
+            connection.send(message);
+        }
+        catch (error) {
+            console.error('Failed to send message to client:', error);
+            this.handleDisconnect(connection);
+        }
+    }
+}
+exports.AgentOrchestrator = AgentOrchestrator;
+async function startAgentOrchestrator(dependencies) {
+    const server = new AgentOrchestrator(dependencies);
+    await server.start();
+    return server;
+}
+//# sourceMappingURL=agent-orchestrator.js.map

package/dist/orchestrator/agent-orchestrator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-orchestrator.js","sourceRoot":"","sources":["../../src/orchestrator/agent-orchestrator.ts"],"names":[],"mappings":";;;AAseA,wDAMC;AA5eD,mCAAoC;AACpC,2CAA6E;AAC7E,uCAAoC;AACpC,qDAA6F;AAC7F,uDAA6D;AAS7D,MAAM,gBAAgB,GAAG,4BAA4B,CAAC;AAsBtD,MAAa,iBAAiB;IAS5B,YAAY,YAA4C;QANvC,aAAQ,GAAG,IAAI,GAAG,EAAmB,CAAC;QACtC,YAAO,GAAG,IAAI,GAAG,EAA4B,CAAC;QAC9C,eAAU,GAAG,IAAI,GAAG,EAAkB,CAAC;QAEhD,YAAO,GAAG,KAAK,CAAC;QAGtB,MAAM,qBAAqB,GAAG,CAAC,YAAqB,EAAE,EAAE,CACtD,IAAI,CAAC,yBAAyB,CAAC,YAAY,CAAC,CAAC;QAE/C,MAAM,mBAAmB,GACvB,YAAY,EAAE,kBAAkB;YAChC,CAAC,GAAG,EAAE,CACJ,IAAI,oBAAQ,CAAC;gBACX,QAAQ,EAAE;oBACR,YAAY,EAAE,CAAC,GAAG,8BAAkB,CAAC;iBACtC;aACF,CAAC,CAAC,CAAC;QAER,IAAI,CAAC,aAAa;YAChB,YAAY,EAAE,aAAa;gBAC3B,IAAI,8BAAa,CAAC;oBAChB,YAAY,EAAE,mBAAmB;oBACjC,qBAAqB;iBACtB,CAAC,CAAC;QAEL,MAAM,eAAe,GAAG,IAAA,oCAAqB,EAAC,IAAI,CAAC,aAAa,EAAE;YAChE,qBAAqB;YACrB,yBAAyB,EAAE,CAAC,SAAiB,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC;SACjF,CAAC,CAAC;QAEH,MAAM,iBAAiB,GACrB,YAAY,EAAE,gBAAgB;YAC9B,CAAC,GAAG,EAAE,CACJ,IAAI,oBAAQ,CAAC;gBACX,QAAQ,EAAE;oBACR,YAAY,EAAE,CAAC,GAAG,8BAAkB,EAAE,gBAAgB,CAAC;oBACvD,UAAU,EAAE;wBACV,MAAM,EAAE,eAAe;qBACxB;iBACF;aACF,CAAC,CAAC,CAAC;QAER,IAAI,CAAC,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QAEpC,IAAI,CAAC,uBAAuB,EAAE,CAAC;IACjC,CAAC;IAED,mBAAmB,CAAC,SAAiB;QACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QACC,MAAM,OAAO,GAAG,IAAI,iBAAO,CACzB,SAAS,EACT,IAAI,CAAC,QAAQ,EACb,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,EACnC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,SAAS,EAAE,SAAS,CAAC,CAC1D,CAAC;QACJ,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACtC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACtC,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,SAAS,CAAC,OAAuC;QAC/C,MAAM,QAAQ,GAAG,IAAA,mBAAU,GAAE,CAAC;QAC9B,MAAM,UAAU,GAAqB;YACnC,EAAE,EAAE,QAAQ;YACZ,UAAU,EAAE,IAAI,GAAG,EAAE;YACrB,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAEvC,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,MAAM,UAAU,GAAG,GAAS,EAAE;YAC5B,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO;YACT,CAAC;YACD,YAAY,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAClC,IAAI,CAAC;gBACH,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC;YACvB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC;QAEF,OAAO;YACL,EAAE,EAAE,QAAQ;YACZ,QAAQ,EAAE,CAAC,OAAuB,EAAE,EAAE;gBACpC,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAChD,CAAC;YACD,UAAU;SACX,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACtB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAC3D,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAClC,IAAI,CAAC;gBACH,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC;YACvB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC7C,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACtB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;IACvB,CAAC;IAEO,gBAAgB,CAAC,UAA4B;QACnD,KAAK,MAAM,SAAS,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;YAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC7C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IAEO,mBAAmB,CAAC,UAA4B,EAAE,OAAuB;QAC/E,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;YACrB,KAAK,MAAM;gBACT,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAC5C,MAAM;YACR,KAAK,cAAc;gBACjB,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAC5C,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC/C,MAAM;YACR,KAAK,OAAO;gBACV,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBAChD,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBACpD,MAAM;YACR,KAAK,eAAe;gBAClB,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACtD,MAAM;YACR;gBACE,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;oBAChC,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,6BAA8B,OAA4B,CAAC,IAAI,EAAE;iBAC3E,CAAC,CAAC;QACP,CAAC;IACH,CAAC;IAEO,iBAAiB,CACvB,UAA4B,EAC5B,OAA0D;QAE1D,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;YAChC,IAAI,EAAE,qBAAqB;YAC3B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,IAAI;SACjD,CAAC,CAAC;IACL,CAAC;IAEO,iBAAiB,CACvB,UAA4B,EAC5B,OAAkD;QAElD,IAAI,OAA4B,CAAC;QACjC,IAAI,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QAElC,IAAI,SAAS,EAAE,CAAC;YACd,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;oBAChC,IAAI,EAAE,OAAO;oBACb,SAAS;oBACT,OAAO,EAAE,oBAAoB;iBAC9B,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;YAC/B,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;gBAChC,IAAI,EAAE,iBAAiB;gBACvB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,SAAS,CAAC;gBAChB,EAAE,EAAE,UAAU,CAAC,EAAE;gBACjB,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE;oBAChB,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC,CAAC;YACH,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC;QAED,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACzD,OAAO,CAAC,sBAAsB,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACrD,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC;IAEO,UAAU,CAAC,UAA4B,EAAE,SAAiB;QAChE,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;gBAChC,IAAI,EAAE,OAAO;gBACb,SAAS;gBACT,OAAO,EAAE,oBAAoB;aAC9B,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,OAAO,CAAC,SAAS,CAAC;gBAChB,EAAE,EAAE,UAAU,CAAC,EAAE;gBACjB,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE;oBAChB,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAC7C,CAAC;gBACD,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC,CAAC;YACH,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;gBAChC,IAAI,EAAE,gBAAgB;gBACtB,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,UAA4B,EAAE,SAAiB;QACjE,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;gBAChC,IAAI,EAAE,OAAO;gBACb,SAAS;gBACT,OAAO,EAAE,yCAAyC;aACnD,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,UAA4B,EAAE,QAAgB;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC3D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE;gBAChC,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,UAAU,QAAQ,4BAA4B;aACxD,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,aAAa;QACnB,MAAM,SAAS,GAAG,IAAA,mBAAU,GAAE,CAAC;QAC/B,OAAO,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IAEO,eAAe,CAAC,SAAiB,EAAE,SAAiB;QAC1D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;IAEO,cAAc,CAAC,SAAiB;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,kBAAkB,CAAC,SAAiB;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE,CAAC;YAC9B,OAAO,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAEO,uBAAuB;QAC7B,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,EAAE;YAChD,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,KAAyB,EAAE,EAAE;YAC7D,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE;YACzC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,MAAM,EAAE,EAAE;YAC5C,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YACnC,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAAC,MAAoB;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC7C,IAAI,CAAC,cAAc,CAAC;YAClB,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;IACL,CAAC;IAEO,eAAe,CAAC,MAAoB;QAC1C,MAAM,MAAM,GACV,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAChG,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,aAAa,EAAE,MAAM;YACrB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,eAAe,EAAE,MAAM,CAAC,eAAe;YACvC,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAAC,MAAoB;QACnD,IAAI,CAAC,cAAc,CAAC;YAClB,IAAI,EAAE,cAAc;YACpB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,IAAI;YAC/C,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,SAAS;SACvC,CAAC,CAAC;IACL,CAAC;IAEO,mBAAmB,CAAC,KAAyB;QACnD,QAAQ,KAAK,CAAC,WAAW,EAAE,CAAC;YAC1B,KAAK,gBAAgB;gBACnB,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;oBAChB,OAAO;gBACT,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC;oBAClB,IAAI,EAAE,gBAAgB;oBACtB,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS;oBACjC,IAAI,EAAE,KAAK,CAAC,IAAI;iBACjB,CAAC,CAAC;gBACH,MAAM;YACR,KAAK,oBAAoB;gBACvB,IAAI,CAAC,cAAc,CAAC;oBAClB,IAAI,EAAE,oBAAoB;oBAC1B,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,SAAS;iBAClC,CAAC,CAAC;gBACH,MAAM;YACR;gBACE,MAAM;QACV,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,MAAoB;QACjD,MAAM,SAAS,GAAG,MAAM,CAAC,eAAe,CAAC;QACzC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,OAAO,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5E,OAAO,CAAC,sBAAsB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAClD,OAAO,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAEO,kBAAkB,CAAC,MAAoB;QAC7C,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;YACtB,KAAK,WAAW;gBACd,OAAO,MAAM,CAAC,MAAM,IAAI,UAAU,MAAM,CAAC,EAAE,aAAa,CAAC;YAC3D,KAAK,OAAO;gBACV,OAAO,UAAU,MAAM,CAAC,EAAE,YAAY,MAAM,CAAC,KAAK,IAAI,gBAAgB,EAAE,CAAC;YAC3E,KAAK,UAAU;gBACb,OAAO,UAAU,MAAM,CAAC,EAAE,gBAAgB,CAAC;YAC7C;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAEO,cAAc,CAAC,OAAwB;QAC7C,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC/C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,yBAAyB,CAAC,YAAqB;QACrD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,KAAK,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;YAC1D,IAAI,OAAO,CAAC,oBAAoB,KAAK,YAAY,EAAE,CAAC;gBAClD,OAAO,QAAQ,CAAC;YAClB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,gBAAgB,CAAC,UAA4B,EAAE,OAAwB;QAC7E,IAAI,UAAU,CAAC,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;YAC1D,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;CACF;AAjcD,8CAicC;AAEM,KAAK,UAAU,sBAAsB,CAC1C,YAA4C;IAE5C,MAAM,MAAM,GAAG,IAAI,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACnD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;IACrB,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/orchestrator/ai-client.d.ts

@@ -0,0 +1,16 @@
+import { type Options, type Query, type SDKMessage } from '@anthropic-ai/claude-agent-sdk';
+export interface AIClientConfig {
+    defaults?: Partial<Options>;
+}
+export interface AgentClient {
+    streamUserMessage(prompt: string, overrides?: Partial<Options>): AsyncIterable<SDKMessage>;
+    createQuery(prompt: string, overrides?: Partial<Options>): Query;
+}
+export declare const BASE_ALLOWED_TOOLS: readonly ["Task", "Bash", "Glob", "Grep", "LS", "ExitPlanMode", "Read", "Edit", "MultiEdit", "Write", "NotebookEdit", "WebFetch", "TodoWrite", "WebSearch", "BashOutput", "KillBash", "Skill", "SlashCommand"];
+export declare class AIClient implements AgentClient {
+    private readonly baseOptions;
+    constructor(config?: AIClientConfig);
+    streamUserMessage(prompt: string, overrides?: Partial<Options>): AsyncIterable<SDKMessage>;
+    createQuery(prompt: string, overrides?: Partial<Options>): Query;
+    private ensureWorkingDirectory;
+}

package/dist/orchestrator/ai-client.js

@@ -0,0 +1,74 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIClient = exports.BASE_ALLOWED_TOOLS = void 0;
+const fs_1 = __importDefault(require("fs"));
+const claude_agent_sdk_1 = require("@anthropic-ai/claude-agent-sdk");
+exports.BASE_ALLOWED_TOOLS = [
+    'Task',
+    'Bash',
+    'Glob',
+    'Grep',
+    'LS',
+    'ExitPlanMode',
+    'Read',
+    'Edit',
+    'MultiEdit',
+    'Write',
+    'NotebookEdit',
+    'WebFetch',
+    'TodoWrite',
+    'WebSearch',
+    'BashOutput',
+    'KillBash',
+    'Skill',
+    'SlashCommand',
+];
+class AIClient {
+    constructor(config) {
+        const defaultOptions = {
+            maxTurns: 100,
+            model: 'opus',
+            cwd: process.cwd(),
+            includePartialMessages: false,
+            allowedTools: [...exports.BASE_ALLOWED_TOOLS],
+        };
+        this.baseOptions = {
+            ...defaultOptions,
+            ...config?.defaults,
+        };
+        this.ensureWorkingDirectory(this.baseOptions.cwd);
+    }
+    async *streamUserMessage(prompt, overrides) {
+        const q = this.createQuery(prompt, overrides);
+        for await (const message of q) {
+            yield message;
+        }
+    }
+    createQuery(prompt, overrides) {
+        const merged = {
+            ...this.baseOptions,
+            ...overrides,
+        };
+        this.ensureWorkingDirectory(merged.cwd);
+        return (0, claude_agent_sdk_1.query)({
+            prompt,
+            options: merged,
+        });
+    }
+    ensureWorkingDirectory(cwd) {
+        if (!cwd) {
+            return;
+        }
+        try {
+            fs_1.default.mkdirSync(cwd, { recursive: true });
+        }
+        catch (error) {
+            console.warn(`Failed to ensure working directory ${cwd}:`, error);
+        }
+    }
+}
+exports.AIClient = AIClient;
+//# sourceMappingURL=ai-client.js.map

package/dist/orchestrator/ai-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-client.js","sourceRoot":"","sources":["../../src/orchestrator/ai-client.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,qEAAkG;AAWrF,QAAA,kBAAkB,GAAG;IAChC,MAAM;IACN,MAAM;IACN,MAAM;IACN,MAAM;IACN,IAAI;IACJ,cAAc;IACd,MAAM;IACN,MAAM;IACN,WAAW;IACX,OAAO;IACP,cAAc;IACd,UAAU;IACV,WAAW;IACX,WAAW;IACX,YAAY;IACZ,UAAU;IACV,OAAO;IACP,cAAc;CACN,CAAC;AAEX,MAAa,QAAQ;IAGnB,YAAY,MAAuB;QACjC,MAAM,cAAc,GAAY;YAC9B,QAAQ,EAAE,GAAG;YACb,KAAK,EAAE,MAAM;YACb,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE;YAClB,sBAAsB,EAAE,KAAK;YAC7B,YAAY,EAAE,CAAC,GAAG,0BAAkB,CAAC;SACtC,CAAC;QAEF,IAAI,CAAC,WAAW,GAAG;YACjB,GAAG,cAAc;YACjB,GAAG,MAAM,EAAE,QAAQ;SACpB,CAAC;QAEF,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,CAAC,iBAAiB,CACtB,MAAc,EACd,SAA4B;QAE5B,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC9C,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,OAAO,CAAC;QAChB,CAAC;IACH,CAAC;IAED,WAAW,CAAC,MAAc,EAAE,SAA4B;QACtD,MAAM,MAAM,GAAY;YACtB,GAAG,IAAI,CAAC,WAAW;YACnB,GAAG,SAAS;SACb,CAAC;QAEF,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAExC,OAAO,IAAA,wBAAK,EAAC;YACX,MAAM;YACN,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAEO,sBAAsB,CAAC,GAAY;QACzC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,sCAAsC,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;CACF;AAtDD,4BAsDC"}

package/dist/orchestrator/session.d.ts

@@ -0,0 +1,37 @@
+import type { AgentClient } from './ai-client';
+import type { SessionSubscriber } from './types';
+type StreamEvent = {
+    type: string;
+    [key: string]: unknown;
+};
+type ToolUseCallback = (toolUseId: string, sessionId: string) => void;
+export declare class Session {
+    readonly id: string;
+    private readonly aiClient;
+    private readonly subscribers;
+    private readonly onEmpty;
+    private readonly onToolUse?;
+    private readonly logger;
+    private sdkSessionId;
+    private inflight;
+    private lastActivity;
+    constructor(id: string, aiClient: AgentClient, onEmpty: (sessionId: string) => void, onToolUse?: ToolUseCallback);
+    announceInboundMessage(text: string, origin?: 'user' | 'worker'): void;
+    get lastActiveAt(): number;
+    get sdkSessionIdentifier(): string | null;
+    subscribe(subscriber: SessionSubscriber): void;
+    unsubscribe(subscriberId: string): void;
+    hasSubscribers(): boolean;
+    close(): void;
+    enqueueUserMessage(text: string): void;
+    handleStreamEvent(event: StreamEvent): void;
+    private processUserMessage;
+    private routeSdkMessage;
+    private handleAssistantMessage;
+    private handleResultMessage;
+    private handleSystemMessage;
+    logInbound(payload: unknown): void;
+    private broadcast;
+    private safeSend;
+}
+export {};