@cogitator-ai/express 0.2.14 → 0.3.1

package/dist/websocket/ws-handler.js

@@ -1,88 +1,206 @@
 import { generateId } from '../streaming/helpers.js';
-const clients = new Map();
-export async function setupWebSocket(server, ctx, config = {}) {
+const WS_OPEN = 1;
+const MAX_SUBSCRIPTIONS = 64;
+const MAX_CHANNEL_LENGTH = 256;
+const MESSAGE_TYPES = new Set([
+    'subscribe',
+    'unsubscribe',
+    'run',
+    'stop',
+    'ping',
+]);
+class ChannelHub {
+    channels = new Map();
+    subscribe(ws, channel) {
+        let members = this.channels.get(channel);
+        if (!members) {
+            members = new Set();
+            this.channels.set(channel, members);
+        }
+        members.add(ws);
+    }
+    unsubscribe(ws, channel) {
+        const members = this.channels.get(channel);
+        if (!members)
+            return;
+        members.delete(ws);
+        if (members.size === 0)
+            this.channels.delete(channel);
+    }
+    publish(channel, response, exclude) {
+        const members = this.channels.get(channel);
+        if (!members)
+            return;
+        for (const member of members) {
+            if (member !== exclude)
+                sendResponse(member, { ...response, channel });
+        }
+    }
+}
+function isPlainObject(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function parseMessage(raw) {
+    let data;
     try {
-        const { WebSocketServer } = await import('ws');
-        const wss = new WebSocketServer({
-            server,
-            path: config.path || `${ctx.config.basePath}/ws`,
-            maxPayload: config.maxPayloadSize || 1024 * 1024,
-        });
-        const pingInterval = config.pingInterval || 30000;
-        wss.on('connection', (ws) => {
-            const clientState = {
-                id: generateId('ws'),
-                subscriptions: new Map(),
-            };
-            clients.set(ws, clientState);
-            let alive = true;
-            const heartbeat = setInterval(() => {
-                if (!alive) {
-                    clearInterval(heartbeat);
-                    ws.terminate();
-                    return;
-                }
-                alive = false;
-                ws.ping();
-            }, pingInterval);
-            ws.on('pong', () => {
-                alive = true;
-            });
-            ws.on('message', async (data) => {
-                try {
-                    const message = JSON.parse(data.toString());
-                    await handleMessage(ws, message, ctx, clientState);
-                }
-                catch (error) {
-                    sendResponse(ws, {
-                        type: 'error',
-                        error: error instanceof Error ? error.message : 'Invalid message',
-                    });
-                }
-            });
-            ws.on('close', () => {
-                clearInterval(heartbeat);
-                clientState.abortController?.abort();
-                clients.delete(ws);
-            });
-            ws.on('error', () => {
-                clearInterval(heartbeat);
-                clientState.abortController?.abort();
-                clients.delete(ws);
-            });
-        });
-        console.log(`[CogitatorServer] WebSocket enabled at ${ctx.config.basePath}/ws`);
-        return wss;
+        data = JSON.parse(raw);
     }
     catch {
+        return null;
+    }
+    if (!isPlainObject(data) || typeof data.type !== 'string' || !MESSAGE_TYPES.has(data.type)) {
+        return null;
+    }
+    if (data.id !== undefined && typeof data.id !== 'string')
+        return null;
+    if (data.channel !== undefined && typeof data.channel !== 'string')
+        return null;
+    return {
+        type: data.type,
+        id: data.id,
+        channel: data.channel,
+        payload: data.payload,
+    };
+}
+function parseRunPayload(payload) {
+    if (!isPlainObject(payload))
+        return null;
+    const { type, name, input, context, threadId } = payload;
+    if (type !== 'agent' && type !== 'workflow' && type !== 'swarm')
+        return null;
+    if (typeof name !== 'string' || !name)
+        return null;
+    if (typeof input !== 'string' || !input.trim())
+        return null;
+    if (context !== undefined && !isPlainObject(context))
+        return null;
+    if (threadId !== undefined && typeof threadId !== 'string')
+        return null;
+    return { type, name, input, context, threadId };
+}
+function isModuleNotFound(error) {
+    return (typeof error === 'object' &&
+        error !== null &&
+        'code' in error &&
+        (error.code === 'ERR_MODULE_NOT_FOUND' || error.code === 'MODULE_NOT_FOUND'));
+}
+/**
+ * Attach the Cogitator WebSocket protocol to an HTTP server.
+ *
+ * When `ctx.config.auth` is set it is called with the upgrade request; connections whose
+ * auth function throws are rejected with 401. Returns `null` when the optional `ws`
+ * package is not installed.
+ */
+export async function setupWebSocket(server, ctx, config = {}) {
+    let wsModule;
+    try {
+        wsModule = await import('ws');
+    }
+    catch (error) {
+        if (!isModuleNotFound(error))
+            throw error;
         console.warn('[CogitatorServer] WebSocket setup failed (ws package not installed)');
         return null;
     }
+    const authFn = ctx.config.auth;
+    const authResults = new WeakMap();
+    const hub = new ChannelHub();
+    const path = config.path || `${ctx.config.basePath}/ws`;
+    const wss = new wsModule.WebSocketServer({
+        server,
+        path,
+        maxPayload: config.maxPayloadSize || 1024 * 1024,
+        verifyClient: authFn
+            ? (info, callback) => {
+                Promise.resolve()
+                    .then(() => authFn(info.req))
+                    .then((auth) => {
+                    authResults.set(info.req, auth);
+                    callback(true);
+                })
+                    .catch(() => callback(false, 401, 'Unauthorized'));
+            }
+            : undefined,
+    });
+    const pingInterval = config.pingInterval || 30000;
+    wss.on('connection', (ws, req) => {
+        const clientState = {
+            id: generateId('ws'),
+            auth: authResults.get(req),
+            subscriptions: new Set(),
+        };
+        let alive = true;
+        const heartbeat = setInterval(() => {
+            if (!alive) {
+                clearInterval(heartbeat);
+                ws.terminate();
+                return;
+            }
+            alive = false;
+            ws.ping();
+        }, pingInterval);
+        ws.on('pong', () => {
+            alive = true;
+        });
+        ws.on('message', (data) => {
+            const message = parseMessage(data.toString());
+            if (!message) {
+                sendResponse(ws, { type: 'error', error: 'Invalid message' });
+                return;
+            }
+            void handleMessage(ws, message, ctx, clientState, hub);
+        });
+        const cleanup = () => {
+            clearInterval(heartbeat);
+            clientState.abortController?.abort();
+            clientState.abortController = undefined;
+            for (const channel of clientState.subscriptions)
+                hub.unsubscribe(ws, channel);
+            clientState.subscriptions.clear();
+        };
+        ws.on('close', cleanup);
+        ws.on('error', cleanup);
+    });
+    wss.on('close', () => {
+        for (const client of wss.clients)
+            client.terminate();
+    });
+    console.log(`[CogitatorServer] WebSocket enabled at ${path}`);
+    return wss;
 }
-async function handleMessage(ws, message, ctx, state) {
+async function handleMessage(ws, message, ctx, state, hub) {
     switch (message.type) {
         case 'ping':
-            sendResponse(ws, { type: 'pong' });
+            sendResponse(ws, { type: 'pong', id: message.id });
             break;
-        case 'subscribe':
-            if (message.channel) {
-                state.subscriptions.set(message.channel, {
-                    channel: message.channel,
-                    callback: (data) => {
-                        sendResponse(ws, { type: 'event', channel: message.channel, payload: data });
-                    },
+        case 'subscribe': {
+            const channel = message.channel;
+            if (!channel || channel.length > MAX_CHANNEL_LENGTH) {
+                sendResponse(ws, { type: 'error', id: message.id, error: 'Invalid channel' });
+                break;
+            }
+            if (!state.subscriptions.has(channel) && state.subscriptions.size >= MAX_SUBSCRIPTIONS) {
+                sendResponse(ws, {
+                    type: 'error',
+                    id: message.id,
+                    error: `Subscription limit reached (max ${MAX_SUBSCRIPTIONS})`,
                 });
-                sendResponse(ws, { type: 'subscribed', channel: message.channel });
+                break;
             }
+            state.subscriptions.add(channel);
+            hub.subscribe(ws, channel);
+            sendResponse(ws, { type: 'subscribed', id: message.id, channel });
             break;
+        }
         case 'unsubscribe':
             if (message.channel) {
                 state.subscriptions.delete(message.channel);
-                sendResponse(ws, { type: 'unsubscribed', channel: message.channel });
+                hub.unsubscribe(ws, message.channel);
+                sendResponse(ws, { type: 'unsubscribed', id: message.id, channel: message.channel });
             }
             break;
         case 'run':
-            await handleRun(ws, message, ctx, state);
+            await handleRun(ws, message, ctx, state, hub);
             break;
         case 'stop':
             state.abortController?.abort();
@@ -90,57 +208,61 @@ async function handleMessage(ws, message, ctx, state) {
             break;
     }
 }
-async function handleRun(ws, message, ctx, state) {
-    const payload = message.payload;
-    if (!payload?.type || !payload?.name || !payload?.input) {
+async function handleRun(ws, message, ctx, state, hub) {
+    const payload = parseRunPayload(message.payload);
+    if (!payload) {
         sendResponse(ws, { type: 'error', id: message.id, error: 'Invalid run payload' });
         return;
     }
+    if (payload.type !== 'agent') {
+        sendResponse(ws, {
+            type: 'error',
+            id: message.id,
+            error: `Run type '${payload.type}' is not supported over WebSocket`,
+        });
+        return;
+    }
+    const agent = Object.hasOwn(ctx.agents, payload.name) ? ctx.agents[payload.name] : undefined;
+    if (!agent) {
+        sendResponse(ws, {
+            type: 'error',
+            id: message.id,
+            error: `Agent '${payload.name}' not found`,
+        });
+        return;
+    }
     state.abortController?.abort();
-    state.abortController = new AbortController();
+    const controller = new AbortController();
+    state.abortController = controller;
+    const channel = `agent:${payload.name}`;
+    const emit = (eventPayload) => {
+        const response = { type: 'event', id: message.id, payload: eventPayload };
+        sendResponse(ws, response);
+        hub.publish(channel, response, ws);
+    };
     try {
-        if (payload.type === 'agent') {
-            const agent = ctx.agents[payload.name];
-            if (!agent) {
-                sendResponse(ws, {
-                    type: 'error',
-                    id: message.id,
-                    error: `Agent '${payload.name}' not found`,
-                });
-                return;
-            }
-            const result = await ctx.cogitator.run(agent, {
-                input: payload.input,
-                context: payload.context,
-                stream: true,
-                onToken: (token) => {
-                    sendResponse(ws, {
-                        type: 'event',
-                        id: message.id,
-                        payload: { type: 'token', delta: token },
-                    });
-                },
-                onToolCall: (toolCall) => {
-                    sendResponse(ws, {
-                        type: 'event',
-                        id: message.id,
-                        payload: { type: 'tool-call', ...toolCall },
-                    });
-                },
-                onToolResult: (toolResult) => {
-                    sendResponse(ws, {
-                        type: 'event',
-                        id: message.id,
-                        payload: { type: 'tool-result', ...toolResult },
-                    });
-                },
-            });
-            sendResponse(ws, { type: 'event', id: message.id, payload: { type: 'complete', result } });
-        }
+        const result = await ctx.cogitator.run(agent, {
+            input: payload.input,
+            context: payload.context,
+            threadId: payload.threadId,
+            userId: state.auth?.userId,
+            signal: controller.signal,
+            stream: true,
+            onToken: (token) => {
+                emit({ type: 'token', delta: token });
+            },
+            onToolCall: (toolCall) => {
+                emit({ type: 'tool-call', ...toolCall });
+            },
+            onToolResult: (toolResult) => {
+                emit({ type: 'tool-result', ...toolResult });
+            },
+        });
+        emit({ type: 'complete', result });
     }
     catch (error) {
-        if (state.abortController?.signal.aborted) {
-            sendResponse(ws, { type: 'event', id: message.id, payload: { type: 'cancelled' } });
+        if (controller.signal.aborted) {
+            emit({ type: 'cancelled' });
         }
         else {
             sendResponse(ws, {
@@ -150,13 +272,16 @@ async function handleRun(ws, message, ctx, state) {
             });
         }
     }
+    finally {
+        if (state.abortController === controller)
+            state.abortController = undefined;
+    }
 }
-const WS_OPEN = 1;
 function sendResponse(ws, response) {
+    if (ws.readyState !== WS_OPEN)
+        return;
     try {
-        if (ws.readyState === WS_OPEN) {
-            ws.send(JSON.stringify(response));
-        }
+        ws.send(JSON.stringify(response));
     }
     catch { }
 }

package/dist/websocket/ws-handler.js.map

@@ -1 +1 @@
-{"version":3,"file":"ws-handler.js","sourceRoot":"","sources":["../../src/websocket/ws-handler.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAiBrD,MAAM,OAAO,GAAG,IAAI,GAAG,EAA8B,CAAC;AAEtD,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAkB,EAClB,GAAiB,EACjB,SAA0B,EAAE;IAE5B,IAAI,CAAC;QACH,MAAM,EAAE,eAAe,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QAE/C,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC;YAC9B,MAAM;YACN,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK;YAChD,UAAU,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI,GAAG,IAAI;SACjD,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC;QAElD,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAiB,EAAE,EAAE;YACzC,MAAM,WAAW,GAAgB;gBAC/B,EAAE,EAAE,UAAU,CAAC,IAAI,CAAC;gBACpB,aAAa,EAAE,IAAI,GAAG,EAAE;aACzB,CAAC;YACF,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAE7B,IAAI,KAAK,GAAG,IAAI,CAAC;YACjB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;gBACjC,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,aAAa,CAAC,SAAS,CAAC,CAAC;oBACzB,EAAE,CAAC,SAAS,EAAE,CAAC;oBACf,OAAO;gBACT,CAAC;gBACD,KAAK,GAAG,KAAK,CAAC;gBACd,EAAE,CAAC,IAAI,EAAE,CAAC;YACZ,CAAC,EAAE,YAAY,CAAC,CAAC;YAEjB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACjB,KAAK,GAAG,IAAI,CAAC;YACf,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE;gBACtC,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAqB,CAAC;oBAChE,MAAM,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,CAAC,CAAC;gBACrD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,YAAY,CAAC,EAAE,EAAE;wBACf,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB;qBAClE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAClB,aAAa,CAAC,SAAS,CAAC,CAAC;gBACzB,WAAW,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;gBACrC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAClB,aAAa,CAAC,SAAS,CAAC,CAAC;gBACzB,WAAW,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;gBACrC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACrB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,0CAA0C,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC,CAAC;QAChF,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;QACpF,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,EAAiB,EACjB,OAAyB,EACzB,GAAiB,EACjB,KAAkB;IAElB,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YACnC,MAAM;QAER,KAAK,WAAW;YACd,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE;oBACvC,OAAO,EAAE,OAAO,CAAC,OAAO;oBACxB,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE;wBACjB,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC/E,CAAC;iBACF,CAAC,CAAC;gBACH,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACrE,CAAC;YACD,MAAM;QAER,KAAK,aAAa;YAChB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC5C,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACvE,CAAC;YACD,MAAM;QAER,KAAK,KAAK;YACR,MAAM,SAAS,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;YACzC,MAAM;QAER,KAAK,MAAM;YACT,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;YAC/B,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;YAClC,MAAM;IACV,CAAC;AACH,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,EAAiB,EACjB,OAAyB,EACzB,GAAiB,EACjB,KAAkB;IAElB,MAAM,OAAO,GAAG,OAAO,CAAC,OAKvB,CAAC;IAEF,IAAI,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QACxD,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC,CAAC;QAClF,OAAO;IACT,CAAC;IAED,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;IAC/B,KAAK,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAE9C,IAAI,CAAC;QACH,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,YAAY,CAAC,EAAE,EAAE;oBACf,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,OAAO,CAAC,EAAE;oBACd,KAAK,EAAE,UAAU,OAAO,CAAC,IAAI,aAAa;iBAC3C,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;gBAC5C,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,MAAM,EAAE,IAAI;gBACZ,OAAO,EAAE,CAAC,KAAa,EAAE,EAAE;oBACzB,YAAY,CAAC,EAAE,EAAE;wBACf,IAAI,EAAE,OAAO;wBACb,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE;qBACzC,CAAC,CAAC;gBACL,CAAC;gBACD,UAAU,EAAE,CAAC,QAAkB,EAAE,EAAE;oBACjC,YAAY,CAAC,EAAE,EAAE;wBACf,IAAI,EAAE,OAAO;wBACb,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,EAAE;qBAC5C,CAAC,CAAC;gBACL,CAAC;gBACD,YAAY,EAAE,CAAC,UAA+C,EAAE,EAAE;oBAChE,YAAY,CAAC,EAAE,EAAE;wBACf,IAAI,EAAE,OAAO;wBACb,EAAE,EAAE,OAAO,CAAC,EAAE;wBACd,OAAO,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,UAAU,EAAE;qBAChD,CAAC,CAAC;gBACL,CAAC;aACF,CAAC,CAAC;YAEH,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC7F,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1C,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QACtF,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,EAAE,EAAE;gBACf,IAAI,EAAE,OAAO;gBACb,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,OAAO,GAAG,CAAC,CAAC;AAElB,SAAS,YAAY,CAAC,EAAiB,EAAE,QAA2B;IAClE,IAAI,CAAC;QACH,IAAI,EAAE,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;YAC9B,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACZ,CAAC"}
+{"version":3,"file":"ws-handler.js","sourceRoot":"","sources":["../../src/websocket/ws-handler.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAKrD,MAAM,OAAO,GAAG,CAAC,CAAC;AAClB,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAC7B,MAAM,kBAAkB,GAAG,GAAG,CAAC;AAC/B,MAAM,aAAa,GAAwB,IAAI,GAAG,CAAC;IACjD,WAAW;IACX,aAAa;IACb,KAAK;IACL,MAAM;IACN,MAAM;CACP,CAAC,CAAC;AAiBH,MAAM,UAAU;IACG,QAAQ,GAAG,IAAI,GAAG,EAA8B,CAAC;IAElE,SAAS,CAAC,EAAiB,EAAE,OAAe;QAC1C,IAAI,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;YACpB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAClB,CAAC;IAED,WAAW,CAAC,EAAiB,EAAE,OAAe;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACnB,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC;YAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,QAA2B,EAAE,OAAuB;QAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,KAAK,OAAO;gBAAE,YAAY,CAAC,MAAM,EAAE,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;CACF;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,YAAY,CAAC,GAAW;IAC/B,IAAI,IAAa,CAAC;IAClB,IAAI,CAAC;QACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3F,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,IAAI,CAAC,EAAE,KAAK,SAAS,IAAI,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACtE,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAChF,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAgC;QAC3C,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,OAAO,EAAE,IAAI,CAAC,OAAO;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,OAAgB;IACvC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IACzC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;IACzD,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,OAAO;QAAE,OAAO,IAAI,CAAC;IAC7E,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE;QAAE,OAAO,IAAI,CAAC;IAC5D,IAAI,OAAO,KAAK,SAAS,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;QAAE,OAAO,IAAI,CAAC;IAClE,IAAI,QAAQ,KAAK,SAAS,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IACxE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,MAAM,IAAI,KAAK;QACf,CAAC,KAAK,CAAC,IAAI,KAAK,sBAAsB,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAC7E,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAAkB,EAClB,GAAiB,EACjB,SAA0B,EAAE;IAE5B,IAAI,QAA6B,CAAC;IAClC,IAAI,CAAC;QACH,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;QACpF,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;IAC/B,MAAM,WAAW,GAAG,IAAI,OAAO,EAA4C,CAAC;IAC5E,MAAM,GAAG,GAAG,IAAI,UAAU,EAAE,CAAC;IAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK,CAAC;IAExD,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,eAAe,CAAC;QACvC,MAAM;QACN,IAAI;QACJ,UAAU,EAAE,MAAM,CAAC,cAAc,IAAI,IAAI,GAAG,IAAI;QAChD,YAAY,EAAE,MAAM;YAClB,CAAC,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACjB,OAAO,CAAC,OAAO,EAAE;qBACd,IAAI,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,GAAc,CAAC,CAAC;qBACvC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;oBACb,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;oBAChC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBACjB,CAAC,CAAC;qBACD,KAAK,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC;YACvD,CAAC;YACH,CAAC,CAAC,SAAS;KACd,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC;IAElD,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAiB,EAAE,GAAoB,EAAE,EAAE;QAC/D,MAAM,WAAW,GAAgB;YAC/B,EAAE,EAAE,UAAU,CAAC,IAAI,CAAC;YACpB,IAAI,EAAE,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC;YAC1B,aAAa,EAAE,IAAI,GAAG,EAAE;SACzB,CAAC;QAEF,IAAI,KAAK,GAAG,IAAI,CAAC;QACjB,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;YACjC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,aAAa,CAAC,SAAS,CAAC,CAAC;gBACzB,EAAE,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO;YACT,CAAC;YACD,KAAK,GAAG,KAAK,CAAC;YACd,EAAE,CAAC,IAAI,EAAE,CAAC;QACZ,CAAC,EAAE,YAAY,CAAC,CAAC;QAEjB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACjB,KAAK,GAAG,IAAI,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAY,EAAE,EAAE;YAChC,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC,CAAC;gBAC9D,OAAO;YACT,CAAC;YACD,KAAK,aAAa,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,aAAa,CAAC,SAAS,CAAC,CAAC;YACzB,WAAW,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;YACrC,WAAW,CAAC,eAAe,GAAG,SAAS,CAAC;YACxC,KAAK,MAAM,OAAO,IAAI,WAAW,CAAC,aAAa;gBAAE,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC9E,WAAW,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACxB,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QACnB,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO;YAAE,MAAM,CAAC,SAAS,EAAE,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,0CAA0C,IAAI,EAAE,CAAC,CAAC;IAC9D,OAAO,GAAG,CAAC;AACb,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,EAAiB,EACjB,OAAyB,EACzB,GAAiB,EACjB,KAAkB,EAClB,GAAe;IAEf,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;YACnD,MAAM;QAER,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;YAChC,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,kBAAkB,EAAE,CAAC;gBACpD,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC,CAAC;gBAC9E,MAAM;YACR,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,aAAa,CAAC,IAAI,IAAI,iBAAiB,EAAE,CAAC;gBACvF,YAAY,CAAC,EAAE,EAAE;oBACf,IAAI,EAAE,OAAO;oBACb,EAAE,EAAE,OAAO,CAAC,EAAE;oBACd,KAAK,EAAE,mCAAmC,iBAAiB,GAAG;iBAC/D,CAAC,CAAC;gBACH,MAAM;YACR,CAAC;YACD,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACjC,GAAG,CAAC,SAAS,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YAC3B,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAClE,MAAM;QACR,CAAC;QAED,KAAK,aAAa;YAChB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC5C,GAAG,CAAC,WAAW,CAAC,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;gBACrC,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACvF,CAAC;YACD,MAAM;QAER,KAAK,KAAK;YACR,MAAM,SAAS,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;YAC9C,MAAM;QAER,KAAK,MAAM;YACT,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;YAC/B,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;YAClC,MAAM;IACV,CAAC;AACH,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,EAAiB,EACjB,OAAyB,EACzB,GAAiB,EACjB,KAAkB,EAClB,GAAe;IAEf,MAAM,OAAO,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAEjD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC,CAAC;QAClF,OAAO;IACT,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC7B,YAAY,CAAC,EAAE,EAAE;YACf,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE,aAAa,OAAO,CAAC,IAAI,mCAAmC;SACpE,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7F,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,YAAY,CAAC,EAAE,EAAE;YACf,IAAI,EAAE,OAAO;YACb,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE,UAAU,OAAO,CAAC,IAAI,aAAa;SAC3C,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,KAAK,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;IAC/B,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,KAAK,CAAC,eAAe,GAAG,UAAU,CAAC;IAEnC,MAAM,OAAO,GAAG,SAAS,OAAO,CAAC,IAAI,EAAE,CAAC;IACxC,MAAM,IAAI,GAAG,CAAC,YAAqC,EAAE,EAAE;QACrD,MAAM,QAAQ,GAAsB,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;QAC7F,YAAY,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC3B,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE;YAC5C,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM;YAC1B,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,MAAM,EAAE,IAAI;YACZ,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;YACxC,CAAC;YACD,UAAU,EAAE,CAAC,QAAQ,EAAE,EAAE;gBACvB,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,EAAE,CAAC,CAAC;YAC3C,CAAC;YACD,YAAY,EAAE,CAAC,UAAU,EAAE,EAAE;gBAC3B,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,UAAU,EAAE,CAAC,CAAC;YAC/C,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC9B,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,EAAE,EAAE;gBACf,IAAI,EAAE,OAAO;gBACb,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;aAChE,CAAC,CAAC;QACL,CAAC;IACH,CAAC;YAAS,CAAC;QACT,IAAI,KAAK,CAAC,eAAe,KAAK,UAAU;YAAE,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;IAC9E,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,EAAiB,EAAE,QAA2B;IAClE,IAAI,EAAE,CAAC,UAAU,KAAK,OAAO;QAAE,OAAO;IACtC,IAAI,CAAC;QACH,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;IACpC,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACZ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cogitator-ai/express",
-  "version": "0.2.14",
+  "version": "0.3.1",
   "description": "Express.js server adapter for Cogitator AI runtime",
   "type": "module",
   "exports": {
@@ -15,17 +15,17 @@
     "dist"
   ],
   "dependencies": {
-    "@cogitator-ai/types": "0.22.2",
-    "@cogitator-ai/server-shared": "0.1.0",
-    "@cogitator-ai/core": "0.19.3"
+    "@cogitator-ai/core": "0.20.0",
+    "@cogitator-ai/types": "0.23.0",
+    "@cogitator-ai/server-shared": "0.1.0"
   },
   "peerDependencies": {
     "express": ">=4.18.0"
   },
   "optionalDependencies": {
     "ws": "^8.0.0",
-    "@cogitator-ai/workflows": "0.5.16",
-    "@cogitator-ai/swarms": "0.4.19"
+    "@cogitator-ai/swarms": "0.5.1",
+    "@cogitator-ai/workflows": "0.6.1"
   },
   "devDependencies": {
     "@types/express": "^4.17.21",
@@ -36,8 +36,8 @@
     "typescript": "^5.6.0",
     "vitest": "^4.0.18",
     "ws": "^8.18.0",
-    "@cogitator-ai/swarms": "0.4.19",
-    "@cogitator-ai/workflows": "0.5.16"
+    "@cogitator-ai/swarms": "0.5.1",
+    "@cogitator-ai/workflows": "0.6.1"
   },
   "repository": {
     "type": "git",
@@ -58,7 +58,7 @@
     "llm"
   ],
   "scripts": {
-    "build": "tsc",
+    "build": "tsc && tsx ../../scripts/fix-esm-imports.ts .",
     "dev": "tsc --watch",
     "test": "vitest run",
     "test:watch": "vitest",