omnara 0.1.0

package/dist/messaging/proto-codec.js

@@ -0,0 +1,387 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.encodeClientMessage = encodeClientMessage;
+exports.decodeClientMessage = decodeClientMessage;
+exports.encodeServerMessage = encodeServerMessage;
+exports.decodeServerMessage = decodeServerMessage;
+const relay_1 = require("../generated/relay");
+function statusStringToProto(status) {
+    switch (status?.toLowerCase()) {
+        case 'running':
+            return relay_1.WorkerStatus.WORKER_STATUS_RUNNING;
+        case 'completed':
+            return relay_1.WorkerStatus.WORKER_STATUS_COMPLETED;
+        case 'error':
+            return relay_1.WorkerStatus.WORKER_STATUS_ERROR;
+        case 'canceled':
+        case 'cancelled':
+            return relay_1.WorkerStatus.WORKER_STATUS_CANCELED;
+        default:
+            return relay_1.WorkerStatus.WORKER_STATUS_UNSPECIFIED;
+    }
+}
+function statusProtoToString(status) {
+    switch (status) {
+        case relay_1.WorkerStatus.WORKER_STATUS_RUNNING:
+            return 'running';
+        case relay_1.WorkerStatus.WORKER_STATUS_COMPLETED:
+            return 'completed';
+        case relay_1.WorkerStatus.WORKER_STATUS_ERROR:
+            return 'error';
+        case relay_1.WorkerStatus.WORKER_STATUS_CANCELED:
+            return 'canceled';
+        default:
+            return 'unspecified';
+    }
+}
+function originStringToProto(origin) {
+    switch (origin) {
+        case 'user':
+            return relay_1.AssistantMessageOrigin.ORIGIN_USER;
+        case 'worker':
+            return relay_1.AssistantMessageOrigin.ORIGIN_WORKER;
+        default:
+            return relay_1.AssistantMessageOrigin.ORIGIN_UNSPECIFIED;
+    }
+}
+function originProtoToString(origin) {
+    switch (origin) {
+        case relay_1.AssistantMessageOrigin.ORIGIN_USER:
+            return 'user';
+        case relay_1.AssistantMessageOrigin.ORIGIN_WORKER:
+            return 'worker';
+        default:
+            return undefined;
+    }
+}
+function safeStringify(value) {
+    if (typeof value === 'string') {
+        return value;
+    }
+    try {
+        return JSON.stringify(value ?? null);
+    }
+    catch {
+        return JSON.stringify(String(value));
+    }
+}
+function safeParse(value) {
+    try {
+        return JSON.parse(value);
+    }
+    catch {
+        return value;
+    }
+}
+function encodeClientMessage(message) {
+    const proto = {};
+    switch (message.type) {
+        case 'chat':
+            proto.payload = {
+                $case: 'chat',
+                chat: {
+                    sessionId: message.sessionId ?? '',
+                    text: message.text,
+                },
+            };
+            break;
+        case 'session_init':
+            proto.payload = {
+                $case: 'sessionInit',
+                sessionInit: {
+                    sessionId: message.sessionId,
+                    parentSessionId: message.parentSessionId ?? undefined,
+                    metadata: Object.fromEntries(Object.entries(message.metadata ?? {}).map(([key, value]) => [key, safeStringify(value)])),
+                },
+            };
+            break;
+        case 'join':
+            proto.payload = {
+                $case: 'join',
+                join: { sessionId: message.sessionId },
+            };
+            break;
+        case 'leave':
+            proto.payload = {
+                $case: 'leave',
+                leave: { sessionId: message.sessionId },
+            };
+            break;
+        case 'cancel_worker':
+            proto.payload = {
+                $case: 'cancelWorker',
+                cancelWorker: { workerId: message.workerId },
+            };
+            break;
+        case 'ping':
+            proto.payload = { $case: 'ping', ping: {} };
+            break;
+        default:
+            throw new Error(`Unsupported client message type: ${message.type}`);
+    }
+    return JSON.stringify(relay_1.ClientMessage.toJSON(proto));
+}
+function decodeClientMessage(raw) {
+    const parsed = JSON.parse(raw);
+    const proto = relay_1.ClientMessage.fromJSON(parsed);
+    const payload = proto.payload;
+    if (!payload) {
+        throw new Error('Received client message without payload');
+    }
+    const active = payload;
+    switch (active.$case) {
+        case 'chat':
+            return {
+                type: 'chat',
+                sessionId: active.chat.sessionId || undefined,
+                text: active.chat.text,
+            };
+        case 'sessionInit':
+            const sessionInitMetadata = active.sessionInit.metadata ?? {};
+            return {
+                type: 'session_init',
+                sessionId: active.sessionInit.sessionId,
+                parentSessionId: active.sessionInit.parentSessionId || undefined,
+                metadata: Object.fromEntries(Object.entries(sessionInitMetadata).map(([key, value]) => [key, safeParse(value)])),
+            };
+        case 'join':
+            return { type: 'join', sessionId: active.join.sessionId };
+        case 'leave':
+            return { type: 'leave', sessionId: active.leave.sessionId };
+        case 'cancelWorker':
+            return { type: 'cancel_worker', workerId: active.cancelWorker.workerId };
+        case 'ping':
+            return { type: 'ping' };
+        default:
+            throw new Error(`Unsupported client payload: ${active.$case}`);
+    }
+}
+function workerPayloadToProto(worker) {
+    return {
+        sessionId: worker.sessionId ? worker.sessionId : undefined,
+        worker: {
+            id: worker.id,
+            status: statusStringToProto(worker.status),
+            prompt: worker.prompt,
+            lastUpdateAt: worker.lastUpdateAt ?? 0,
+            latestMessage: worker.latestMessage ?? undefined,
+            result: worker.result ?? undefined,
+            error: worker.error ?? undefined,
+            originSessionId: worker.originSessionId ?? undefined,
+            sessionId: worker.sessionId,
+        },
+    };
+}
+function workerPayloadFromProto(env) {
+    const worker = env.worker;
+    return {
+        id: worker?.id ?? '',
+        status: statusProtoToString(worker?.status),
+        prompt: worker?.prompt ?? '',
+        lastUpdateAt: Number(worker?.lastUpdateAt ?? 0n),
+        latestMessage: worker?.latestMessage ?? undefined,
+        result: worker?.result ?? undefined,
+        error: worker?.error ?? undefined,
+        originSessionId: worker?.originSessionId ?? undefined,
+        sessionId: worker?.sessionId ?? env.sessionId ?? '',
+    };
+}
+function encodeServerMessage(message) {
+    const proto = {};
+    switch (message.type) {
+        case 'session_created':
+            proto.payload = { $case: 'sessionCreated', sessionCreated: { sessionId: message.sessionId } };
+            break;
+        case 'session_joined':
+            proto.payload = { $case: 'sessionJoined', sessionJoined: { sessionId: message.sessionId } };
+            break;
+        case 'session_init':
+            proto.payload = {
+                $case: 'sessionInit',
+                sessionInit: {
+                    sessionId: message.sessionId,
+                    parentSessionId: message.parentSessionId ?? undefined,
+                    metadata: Object.fromEntries(Object.entries(message.metadata ?? {}).map(([key, value]) => [key, safeStringify(value)])),
+                },
+            };
+            break;
+        case 'session_initialized':
+            proto.payload = {
+                $case: 'sessionInitialized',
+                sessionInitialized: {
+                    sessionId: message.sessionId,
+                    parentSessionId: message.parentSessionId ?? undefined,
+                },
+            };
+            break;
+        case 'user_text':
+            proto.payload = {
+                $case: 'userText',
+                userText: {
+                    sessionId: message.sessionId,
+                    text: message.text,
+                    origin: originStringToProto(message.origin),
+                },
+            };
+            break;
+        case 'assistant_text':
+            proto.payload = {
+                $case: 'assistantText',
+                assistantText: { sessionId: message.sessionId, text: message.text },
+            };
+            break;
+        case 'assistant_complete':
+            proto.payload = {
+                $case: 'assistantComplete',
+                assistantComplete: { sessionId: message.sessionId },
+            };
+            break;
+        case 'tool_call':
+            proto.payload = {
+                $case: 'toolCall',
+                toolCall: {
+                    sessionId: message.sessionId,
+                    toolName: message.toolName,
+                    toolUseId: message.toolUseId,
+                    inputJson: safeStringify(message.input),
+                },
+            };
+            break;
+        case 'tool_result':
+            proto.payload = {
+                $case: 'toolResult',
+                toolResult: {
+                    sessionId: message.sessionId,
+                    toolUseId: message.toolUseId,
+                    contentJson: safeStringify(message.content),
+                    isError: message.isError ?? false,
+                },
+            };
+            break;
+        case 'result':
+            proto.payload = {
+                $case: 'result',
+                result: {
+                    sessionId: message.sessionId,
+                    success: message.success,
+                    message: message.message ?? '',
+                    durationMs: message.durationMs ?? 0,
+                    costUsd: message.costUsd ?? 0,
+                },
+            };
+            break;
+        case 'system':
+            proto.payload = {
+                $case: 'system',
+                system: { sessionId: message.sessionId, subtype: message.subtype },
+            };
+            break;
+        case 'error':
+            proto.payload = {
+                $case: 'error',
+                error: { sessionId: message.sessionId ?? undefined, message: message.message },
+            };
+            break;
+        case 'worker_update':
+            proto.payload = { $case: 'workerUpdate', workerUpdate: workerPayloadToProto(message.worker) };
+            break;
+        case 'pong':
+            proto.payload = { $case: 'pong', pong: { sessionId: message.sessionId ?? undefined } };
+            break;
+        default:
+            throw new Error(`Unsupported server message type: ${message.type}`);
+    }
+    return JSON.stringify(relay_1.ServerMessage.toJSON(proto));
+}
+function decodeServerMessage(raw) {
+    const parsed = JSON.parse(raw);
+    const proto = relay_1.ServerMessage.fromJSON(parsed);
+    const payload = proto.payload;
+    if (!payload) {
+        throw new Error('Received server message without payload');
+    }
+    const active = payload;
+    switch (active.$case) {
+        case 'sessionCreated':
+            return { type: 'session_created', sessionId: active.sessionCreated.sessionId };
+        case 'sessionJoined':
+            return { type: 'session_joined', sessionId: active.sessionJoined.sessionId };
+        case 'sessionInit':
+            const serverInitMetadata = active.sessionInit.metadata ?? {};
+            return {
+                type: 'session_init',
+                sessionId: active.sessionInit.sessionId,
+                parentSessionId: active.sessionInit.parentSessionId ?? undefined,
+                metadata: Object.fromEntries(Object.entries(serverInitMetadata).map(([key, value]) => [key, safeParse(value)])),
+            };
+        case 'sessionInitialized':
+            return {
+                type: 'session_initialized',
+                sessionId: active.sessionInitialized.sessionId,
+                parentSessionId: active.sessionInitialized.parentSessionId ?? undefined,
+            };
+        case 'userText':
+            return {
+                type: 'user_text',
+                sessionId: active.userText.sessionId,
+                text: active.userText.text,
+                origin: originProtoToString(active.userText.origin),
+            };
+        case 'assistantText':
+            return {
+                type: 'assistant_text',
+                sessionId: active.assistantText.sessionId,
+                text: active.assistantText.text,
+            };
+        case 'assistantComplete':
+            return { type: 'assistant_complete', sessionId: active.assistantComplete.sessionId };
+        case 'toolCall':
+            return {
+                type: 'tool_call',
+                sessionId: active.toolCall.sessionId,
+                toolName: active.toolCall.toolName,
+                toolUseId: active.toolCall.toolUseId,
+                input: safeParse(active.toolCall.inputJson),
+            };
+        case 'toolResult':
+            return {
+                type: 'tool_result',
+                sessionId: active.toolResult.sessionId,
+                toolUseId: active.toolResult.toolUseId,
+                content: safeParse(active.toolResult.contentJson),
+                isError: active.toolResult.isError ?? undefined,
+            };
+        case 'result':
+            return {
+                type: 'result',
+                sessionId: active.result.sessionId,
+                success: active.result.success,
+                message: active.result.message || undefined,
+                durationMs: active.result.durationMs || undefined,
+                costUsd: active.result.costUsd || undefined,
+            };
+        case 'system':
+            return {
+                type: 'system',
+                sessionId: active.system.sessionId,
+                subtype: active.system.subtype,
+            };
+        case 'error':
+            return {
+                type: 'error',
+                sessionId: active.error.sessionId ?? undefined,
+                message: active.error.message,
+            };
+        case 'workerUpdate':
+            return {
+                type: 'worker_update',
+                sessionId: active.workerUpdate.sessionId ?? undefined,
+                worker: workerPayloadFromProto(active.workerUpdate),
+            };
+        case 'pong':
+            return { type: 'pong', sessionId: active.pong.sessionId ?? undefined };
+        default:
+            throw new Error(`Unsupported server payload: ${active.$case}`);
+    }
+}
+//# sourceMappingURL=proto-codec.js.map

package/dist/messaging/proto-codec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proto-codec.js","sourceRoot":"","sources":["../../src/messaging/proto-codec.ts"],"names":[],"mappings":";;AAqFA,kDAiDC;AAED,kDAoCC;AAkCD,kDA4GC;AAED,kDA4FC;AAxZD,8CAO4B;AAM5B,SAAS,mBAAmB,CAAC,MAA0B;IACrD,QAAQ,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC;QAC9B,KAAK,SAAS;YACZ,OAAO,oBAAY,CAAC,qBAAqB,CAAC;QAC5C,KAAK,WAAW;YACd,OAAO,oBAAY,CAAC,uBAAuB,CAAC;QAC9C,KAAK,OAAO;YACV,OAAO,oBAAY,CAAC,mBAAmB,CAAC;QAC1C,KAAK,UAAU,CAAC;QAChB,KAAK,WAAW;YACd,OAAO,oBAAY,CAAC,sBAAsB,CAAC;QAC7C;YACE,OAAO,oBAAY,CAAC,yBAAyB,CAAC;IAClD,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,MAAgC;IAC3D,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,oBAAY,CAAC,qBAAqB;YACrC,OAAO,SAAS,CAAC;QACnB,KAAK,oBAAY,CAAC,uBAAuB;YACvC,OAAO,WAAW,CAAC;QACrB,KAAK,oBAAY,CAAC,mBAAmB;YACnC,OAAO,OAAO,CAAC;QACjB,KAAK,oBAAY,CAAC,sBAAsB;YACtC,OAAO,UAAU,CAAC;QACpB;YACE,OAAO,aAAa,CAAC;IACzB,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA0B;IACrD,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,MAAM;YACT,OAAO,8BAAsB,CAAC,WAAW,CAAC;QAC5C,KAAK,QAAQ;YACX,OAAO,8BAAsB,CAAC,aAAa,CAAC;QAC9C;YACE,OAAO,8BAAsB,CAAC,kBAAkB,CAAC;IACrD,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,MAA8B;IACzD,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,8BAAsB,CAAC,WAAW;YACrC,OAAO,MAAM,CAAC;QAChB,KAAK,8BAAsB,CAAC,aAAa;YACvC,OAAO,QAAQ,CAAC;QAClB;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC;IACvC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,KAAa;IAC9B,IAAI,CAAC;QACH,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAgB,mBAAmB,CAAC,OAAsB;IACxD,MAAM,KAAK,GAAuB,EAAE,CAAC;IACrC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE;oBACJ,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,EAAE;oBAClC,IAAI,EAAE,OAAO,CAAC,IAAI;iBACnB;aACF,CAAC;YACF,MAAM;QACR,KAAK,cAAc;YACjB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,aAAa;gBACpB,WAAW,EAAE;oBACX,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,SAAS;oBACrD,QAAQ,EAAE,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAC1F;iBACF;aACF,CAAC;YACF,MAAM;QACR,KAAK,MAAM;YACT,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,MAAM;gBACb,IAAI,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;aACvC,CAAC;YACF,MAAM;QACR,KAAK,OAAO;YACV,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,KAAK,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;aACxC,CAAC;YACF,MAAM;QACR,KAAK,eAAe;YAClB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,cAAc;gBACrB,YAAY,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE;aAC7C,CAAC;YACF,MAAM;QACR,KAAK,MAAM;YACT,KAAK,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAgB,EAAE,CAAC;YAC1D,MAAM;QACR;YACE,MAAM,IAAI,KAAK,CAAC,oCAAqC,OAA4B,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9F,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,qBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,SAAgB,mBAAmB,CAAC,GAAW;IAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,qBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;IACD,MAAM,MAAM,GAAG,OAAc,CAAC;IAC9B,QAAQ,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,SAAS;gBAC7C,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,IAAI;aACvB,CAAC;QACL,KAAK,aAAa;YACf,MAAM,mBAAmB,GAA2B,MAAM,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;YACtF,OAAO;gBACL,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,SAAS;gBACvC,eAAe,EAAE,MAAM,CAAC,WAAW,CAAC,eAAe,IAAI,SAAS;gBAChE,QAAQ,EAAE,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CACnF;aACF,CAAC;QACJ,KAAK,MAAM;YACT,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QAC5D,KAAK,OAAO;YACV,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;QAC9D,KAAK,cAAc;YACjB,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC3E,KAAK,MAAM;YACT,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC1B;YACE,MAAM,IAAI,KAAK,CAAC,+BAA+B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IACnE,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,MAA2B;IACvD,OAAO;QACL,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QAC1D,MAAM,EAAE;YACN,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,MAAM,EAAE,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC;YAC1C,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC;YACtC,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,SAAS;YAChD,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,SAAS;YAClC,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS;YAChC,eAAe,EAAE,MAAM,CAAC,eAAe,IAAI,SAAS;YACpD,SAAS,EAAE,MAAM,CAAC,SAAS;SAC5B;KACF,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAAC,GAAyB;IACvD,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;IAC1B,OAAO;QACL,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE;QACpB,MAAM,EAAE,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC;QAC3C,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE;QAC5B,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,IAAI,EAAE,CAAC;QAChD,aAAa,EAAE,MAAM,EAAE,aAAa,IAAI,SAAS;QACjD,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,SAAS;QACnC,KAAK,EAAE,MAAM,EAAE,KAAK,IAAI,SAAS;QACjC,eAAe,EAAE,MAAM,EAAE,eAAe,IAAI,SAAS;QACrD,SAAS,EAAE,MAAM,EAAE,SAAS,IAAI,GAAG,CAAC,SAAS,IAAI,EAAE;KACpD,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CAAC,OAAsB;IACxD,MAAM,KAAK,GAAuB,EAAE,CAAC;IACrC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,iBAAiB;YACpB,KAAK,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,gBAAgB,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YAC9F,MAAM;QACR,KAAK,gBAAgB;YACnB,KAAK,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YAC5F,MAAM;QACR,KAAK,cAAc;YACjB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,aAAa;gBACpB,WAAW,EAAE;oBACX,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,SAAS;oBACrD,QAAQ,EAAE,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAC1F;iBACF;aACF,CAAC;YACF,MAAM;QACR,KAAK,qBAAqB;YACxB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,oBAAoB;gBAC3B,kBAAkB,EAAE;oBAClB,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,SAAS;iBACtD;aACF,CAAC;YACF,MAAM;QACR,KAAK,WAAW;YACd,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE;oBACR,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;oBAClB,MAAM,EAAE,mBAAmB,CAAC,OAAO,CAAC,MAAM,CAAC;iBAC5C;aACF,CAAC;YACF,MAAM;QACR,KAAK,gBAAgB;YACnB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,eAAe;gBACtB,aAAa,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE;aACpE,CAAC;YACF,MAAM;QACR,KAAK,oBAAoB;YACvB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,mBAAmB;gBAC1B,iBAAiB,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE;aACpD,CAAC;YACF,MAAM;QACR,KAAK,WAAW;YACd,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE;oBACR,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,KAAK,CAAC;iBACxC;aACF,CAAC;YACF,MAAM;QACR,KAAK,aAAa;YAChB,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,YAAY;gBACnB,UAAU,EAAE;oBACV,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,WAAW,EAAE,aAAa,CAAC,OAAO,CAAC,OAAO,CAAC;oBAC3C,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,KAAK;iBAClC;aACF,CAAC;YACF,MAAM;QACR,KAAK,QAAQ;YACX,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,QAAQ;gBACf,MAAM,EAAE;oBACN,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;oBAC9B,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,CAAC;oBACnC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,CAAC;iBAC9B;aACF,CAAC;YACF,MAAM;QACR,KAAK,QAAQ;YACX,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,QAAQ;gBACf,MAAM,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;aACnE,CAAC;YACF,MAAM;QACR,KAAK,OAAO;YACV,KAAK,CAAC,OAAO,GAAG;gBACd,KAAK,EAAE,OAAO;gBACd,KAAK,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;aAC/E,CAAC;YACF,MAAM;QACR,KAAK,eAAe;YAClB,KAAK,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,cAAc,EAAE,YAAY,EAAE,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9F,MAAM;QACR,KAAK,MAAM;YACT,KAAK,CAAC,OAAO,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,SAAS,EAAE,EAAE,CAAC;YACvF,MAAM;QACR;YACE,MAAM,IAAI,KAAK,CAAC,oCAAqC,OAA4B,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9F,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,qBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,SAAgB,mBAAmB,CAAC,GAAW;IAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/B,MAAM,KAAK,GAAG,qBAAkB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAC9B,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;IAC7D,CAAC;IACD,MAAM,MAAM,GAAG,OAAc,CAAC;IAC9B,QAAQ,MAAM,CAAC,KAAK,EAAE,CAAC;QACrB,KAAK,gBAAgB;YACnB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC;QACjF,KAAK,eAAe;YAClB,OAAO,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;QAC/E,KAAK,aAAa;YAChB,MAAM,kBAAkB,GAA2B,MAAM,CAAC,WAAW,CAAC,QAAQ,IAAI,EAAE,CAAC;YACrF,OAAO;gBACL,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,SAAS;gBACvC,eAAe,EAAE,MAAM,CAAC,WAAW,CAAC,eAAe,IAAI,SAAS;gBAChE,QAAQ,EAAE,MAAM,CAAC,WAAW,CAC1B,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAClF;aACF,CAAC;QACJ,KAAK,oBAAoB;YACvB,OAAO;gBACL,IAAI,EAAE,qBAAqB;gBAC3B,SAAS,EAAE,MAAM,CAAC,kBAAkB,CAAC,SAAS;gBAC9C,eAAe,EAAE,MAAM,CAAC,kBAAkB,CAAC,eAAe,IAAI,SAAS;aACxE,CAAC;QACJ,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;gBACpC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAC1B,MAAM,EAAE,mBAAmB,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;aACpD,CAAC;QACJ,KAAK,eAAe;YAClB,OAAO;gBACL,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,MAAM,CAAC,aAAa,CAAC,SAAS;gBACzC,IAAI,EAAE,MAAM,CAAC,aAAa,CAAC,IAAI;aAChC,CAAC;QACJ,KAAK,mBAAmB;YACtB,OAAO,EAAE,IAAI,EAAE,oBAAoB,EAAE,SAAS,EAAE,MAAM,CAAC,iBAAiB,CAAC,SAAS,EAAE,CAAC;QACvF,KAAK,UAAU;YACb,OAAO;gBACL,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;gBACpC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ;gBAClC,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,SAAS;gBACpC,KAAK,EAAE,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC;aAC5C,CAAC;QACJ,KAAK,YAAY;YACf,OAAO;gBACL,IAAI,EAAE,aAAa;gBACnB,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,SAAS;gBACtC,SAAS,EAAE,MAAM,CAAC,UAAU,CAAC,SAAS;gBACtC,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC;gBACjD,OAAO,EAAE,MAAM,CAAC,UAAU,CAAC,OAAO,IAAI,SAAS;aAChD,CAAC;QACJ,KAAK,QAAQ;YACX,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS;gBAClC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO;gBAC9B,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,SAAS;gBAC3C,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,IAAI,SAAS;gBACjD,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,SAAS;aAC5C,CAAC;QACJ,KAAK,QAAQ;YACX,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,SAAS,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS;gBAClC,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO;aAC/B,CAAC;QACJ,KAAK,OAAO;YACV,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,IAAI,SAAS;gBAC9C,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;aAC9B,CAAC;QACJ,KAAK,cAAc;YACjB,OAAO;gBACL,IAAI,EAAE,eAAe;gBACrB,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,SAAS,IAAI,SAAS;gBACrD,MAAM,EAAE,sBAAsB,CAAC,MAAM,CAAC,YAAY,CAAC;aACpD,CAAC;QACJ,KAAK,MAAM;YACT,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,SAAS,IAAI,SAAS,EAAE,CAAC;QACzE;YACE,MAAM,IAAI,KAAK,CAAC,+BAA+B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IACnE,CAAC;AACH,CAAC"}

package/dist/orchestrator/agent-orchestrator.d.ts

@@ -0,0 +1,57 @@
+import { type AgentClient } from './ai-client';
+import { Session } from './session';
+import { WorkerManager } from './worker-manager';
+import type { InboundMessage, OutboundMessage } from './types';
+interface AgentOrchestratorDependencies {
+    createMainClient?: () => AgentClient;
+    createWorkerClient?: () => AgentClient;
+    workerManager?: WorkerManager;
+}
+export interface AgentOrchestratorClientAdapter {
+    send: (message: OutboundMessage) => void;
+    canSend?: () => boolean;
+    close?: () => void;
+}
+export interface AgentOrchestratorConnection {
+    id: string;
+    dispatch(message: InboundMessage): void;
+    disconnect(): void;
+}
+export declare class AgentOrchestrator {
+    private readonly aiClient;
+    private readonly workerManager;
+    private readonly sessions;
+    private readonly clients;
+    private readonly toolUseMap;
+    private started;
+    constructor(dependencies?: AgentOrchestratorDependencies);
+    ensureSessionExists(sessionId: string): Session;
+    getSession(sessionId: string): Session | undefined;
+    getWorkerManager(): WorkerManager;
+    addClient(adapter: AgentOrchestratorClientAdapter): AgentOrchestratorConnection;
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    private handleDisconnect;
+    private handleClientMessage;
+    private handleSessionInit;
+    private handleChatMessage;
+    private handleJoin;
+    private handleLeave;
+    private handleCancelWorker;
+    private createSession;
+    private registerToolUse;
+    private consumeToolUse;
+    private handleSessionEmpty;
+    private registerWorkerListeners;
+    private broadcastWorkerUpdate;
+    private toWorkerPayload;
+    private handleWorkerSessionStart;
+    private handleWorkerMessage;
+    private handleWorkerCompletion;
+    private buildWorkerSummary;
+    private broadcastToAll;
+    private findSessionIdBySdkSession;
+    private sendToConnection;
+}
+export declare function startAgentOrchestrator(dependencies?: AgentOrchestratorDependencies): Promise<AgentOrchestrator>;
+export {};