@cogitator-ai/koa 0.1.0

package/dist/types.d.ts

@@ -0,0 +1,164 @@
+import type { Context, Next } from 'koa';
+import type { Cogitator, Agent } from '@cogitator-ai/core';
+import type { Message, ToolCall, ToolResult, RunResult, Workflow, WorkflowResult, WorkflowState, SwarmConfig, SwarmResult, SwarmRunOptions, StreamingWorkflowEvent, SwarmEvent, SwarmMessage } from '@cogitator-ai/types';
+export type { Message, ToolCall, ToolResult, RunResult, Workflow, WorkflowResult, WorkflowState, SwarmConfig, SwarmResult, SwarmRunOptions, StreamingWorkflowEvent, SwarmEvent, SwarmMessage, };
+export interface AuthContext {
+    userId?: string;
+    roles?: string[];
+    permissions?: string[];
+    metadata?: Record<string, unknown>;
+}
+export type AuthFunction = (ctx: Context) => Promise<AuthContext | undefined> | AuthContext | undefined;
+export interface WebSocketConfig {
+    path?: string;
+    pingInterval?: number;
+    pingTimeout?: number;
+    maxPayloadSize?: number;
+}
+export interface CogitatorAppOptions {
+    cogitator: Cogitator;
+    agents?: Record<string, Agent>;
+    workflows?: Record<string, Workflow<WorkflowState>>;
+    swarms?: Record<string, SwarmConfig>;
+    auth?: AuthFunction;
+    enableWebSocket?: boolean;
+    websocket?: WebSocketConfig;
+}
+export interface CogitatorState {
+    cogitator: RouteContext;
+    auth?: AuthContext;
+    requestId: string;
+    startTime: number;
+}
+export interface RouteContext {
+    cogitator: Cogitator;
+    agents: Record<string, Agent>;
+    workflows: Record<string, Workflow<WorkflowState>>;
+    swarms: Record<string, SwarmConfig>;
+}
+export type KoaMiddleware = (ctx: Context, next: Next) => Promise<void>;
+export interface AgentListResponse {
+    agents: Array<{
+        name: string;
+        description?: string;
+        tools: string[];
+    }>;
+}
+export interface AgentRunRequest {
+    input: string;
+    context?: Record<string, unknown>;
+    threadId?: string;
+}
+export interface AgentRunResponse {
+    output: string;
+    threadId?: string;
+    usage: {
+        inputTokens: number;
+        outputTokens: number;
+        totalTokens: number;
+    };
+    toolCalls: ToolCall[];
+}
+export interface ThreadResponse {
+    id: string;
+    messages: Message[];
+    metadata?: Record<string, unknown>;
+    createdAt: number;
+    updatedAt: number;
+}
+export interface AddMessageRequest {
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+    metadata?: Record<string, unknown>;
+}
+export interface ToolListResponse {
+    tools: Array<{
+        name: string;
+        description?: string;
+        parameters: Record<string, unknown>;
+    }>;
+}
+export interface WorkflowListResponse {
+    workflows: Array<{
+        name: string;
+        entryPoint: string;
+        nodes: string[];
+    }>;
+}
+export interface WorkflowRunRequest {
+    input?: Record<string, unknown>;
+    options?: {
+        maxConcurrency?: number;
+        maxIterations?: number;
+        checkpoint?: boolean;
+    };
+}
+export interface WorkflowRunResponse {
+    workflowId: string;
+    workflowName: string;
+    state: WorkflowState;
+    duration: number;
+    nodeResults: Record<string, {
+        output: unknown;
+        duration: number;
+    }>;
+}
+export interface SwarmListResponse {
+    swarms: Array<{
+        name: string;
+        strategy: string;
+        agents: string[];
+    }>;
+}
+export interface SwarmRunRequest {
+    input: string;
+    context?: Record<string, unknown>;
+    threadId?: string;
+    timeout?: number;
+}
+export interface SwarmRunResponse {
+    swarmId: string;
+    swarmName: string;
+    strategy: string;
+    output: unknown;
+    agentResults: Record<string, unknown>;
+    usage: {
+        totalTokens: number;
+        totalCost: number;
+        elapsedTime: number;
+    };
+}
+export interface BlackboardResponse {
+    sections: Record<string, unknown>;
+}
+export interface HealthResponse {
+    status: 'ok' | 'degraded' | 'error';
+    version?: string;
+    uptime: number;
+    timestamp: number;
+    checks?: Record<string, {
+        status: 'ok' | 'error';
+        message?: string;
+    }>;
+}
+export interface ErrorResponse {
+    error: {
+        message: string;
+        code?: string;
+        details?: unknown;
+    };
+}
+export interface WebSocketMessage {
+    type: 'subscribe' | 'unsubscribe' | 'run' | 'stop' | 'ping';
+    id?: string;
+    channel?: string;
+    payload?: unknown;
+}
+export interface WebSocketResponse {
+    type: 'subscribed' | 'unsubscribed' | 'event' | 'error' | 'pong';
+    id?: string;
+    channel?: string;
+    payload?: unknown;
+    error?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,KAAK,CAAC;AACzC,OAAO,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EACV,OAAO,EACP,QAAQ,EACR,UAAU,EACV,SAAS,EACT,QAAQ,EACR,cAAc,EACd,aAAa,EACb,WAAW,EACX,WAAW,EACX,eAAe,EACf,sBAAsB,EACtB,UAAU,EACV,YAAY,EACb,MAAM,qBAAqB,CAAC;AAE7B,YAAY,EACV,OAAO,EACP,QAAQ,EACR,UAAU,EACV,SAAS,EACT,QAAQ,EACR,cAAc,EACd,aAAa,EACb,WAAW,EACX,WAAW,EACX,eAAe,EACf,sBAAsB,EACtB,UAAU,EACV,YAAY,GACb,CAAC;AAEF,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,OAAO,KACT,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,WAAW,GAAG,SAAS,CAAC;AAEhE,MAAM,WAAW,eAAe;IAC9B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;IACpD,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IACrC,IAAI,CAAC,EAAE,YAAY,CAAC;IACpB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,SAAS,CAAC,EAAE,eAAe,CAAC;CAC7B;AAED,MAAM,WAAW,cAAc;IAC7B,SAAS,EAAE,YAAY,CAAC;IACxB,IAAI,CAAC,EAAE,WAAW,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IAC9B,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;IACnD,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;CACrC;AAED,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AAExE,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,MAAM,EAAE,CAAC;KACjB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,SAAS,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,KAAK,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,KAAK,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,EAAE,CAAC;KACjB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE;QACR,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,UAAU,CAAC,EAAE,OAAO,CAAC;KACtB,CAAC;CACH;AAED,MAAM,WAAW,mBAAmB;IAClC,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,aAAa,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,OAAO,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACpE;AAED,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE,KAAK,CAAC;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC,CAAC;CACJ;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,OAAO,CAAC;IAChB,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACtC,KAAK,EAAE;QACL,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,CAAC;QAClB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;CACH;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,IAAI,GAAG,UAAU,GAAG,OAAO,CAAC;IACpC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CACb,MAAM,EACN;QACE,MAAM,EAAE,IAAI,GAAG,OAAO,CAAC;QACvB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CACF,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE;QACL,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,OAAO,CAAC;KACnB,CAAC;CACH;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,WAAW,GAAG,aAAa,GAAG,KAAK,GAAG,MAAM,GAAG,MAAM,CAAC;IAC5D,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,YAAY,GAAG,cAAc,GAAG,OAAO,GAAG,OAAO,GAAG,MAAM,CAAC;IACjE,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB"}

package/dist/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":""}

package/dist/websocket/handler.d.ts

@@ -0,0 +1,6 @@
+import type { Server as HttpServer } from 'http';
+import type { RouteContext, WebSocketConfig } from '../types.js';
+type WebSocketServerType = import('ws').WebSocketServer;
+export declare function setupWebSocket(server: HttpServer, ctx: RouteContext, config?: WebSocketConfig): Promise<WebSocketServerType | null>;
+export {};
+//# sourceMappingURL=handler.d.ts.map

package/dist/websocket/handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../src/websocket/handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,MAAM,CAAC;AACjD,OAAO,KAAK,EAGV,YAAY,EACZ,eAAe,EAChB,MAAM,aAAa,CAAC;AAKrB,KAAK,mBAAmB,GAAG,OAAO,IAAI,EAAE,eAAe,CAAC;AAexD,wBAAsB,cAAc,CAClC,MAAM,EAAE,UAAU,EAClB,GAAG,EAAE,YAAY,EACjB,MAAM,GAAE,eAAoB,GAC3B,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC,CAiErC"}

package/dist/websocket/handler.js

@@ -0,0 +1,161 @@
+import { generateId } from '@cogitator-ai/server-shared';
+const clients = new Map();
+export async function setupWebSocket(server, ctx, config = {}) {
+    try {
+        const { WebSocketServer } = await import('ws');
+        const wss = new WebSocketServer({
+            server,
+            path: config.path || '/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('[CogitatorKoa] WebSocket enabled');
+        return wss;
+    }
+    catch {
+        console.warn('[CogitatorKoa] WebSocket setup failed (ws package not installed)');
+        return null;
+    }
+}
+async function handleMessage(ws, message, ctx, state) {
+    switch (message.type) {
+        case 'ping':
+            sendResponse(ws, { type: 'pong' });
+            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 });
+                    },
+                });
+                sendResponse(ws, { type: 'subscribed', channel: message.channel });
+            }
+            break;
+        case 'unsubscribe':
+            if (message.channel) {
+                state.subscriptions.delete(message.channel);
+                sendResponse(ws, { type: 'unsubscribed', channel: message.channel });
+            }
+            break;
+        case 'run':
+            await handleRun(ws, message, ctx, state);
+            break;
+        case 'stop':
+            state.abortController?.abort();
+            state.abortController = undefined;
+            break;
+    }
+}
+async function handleRun(ws, message, ctx, state) {
+    const payload = message.payload;
+    if (!payload?.type || !payload?.name || !payload?.input) {
+        sendResponse(ws, { type: 'error', id: message.id, error: 'Invalid run payload' });
+        return;
+    }
+    state.abortController = new AbortController();
+    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 } });
+        }
+    }
+    catch (error) {
+        if (state.abortController?.signal.aborted) {
+            sendResponse(ws, { type: 'event', id: message.id, payload: { type: 'cancelled' } });
+        }
+        else {
+            sendResponse(ws, {
+                type: 'error',
+                id: message.id,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            });
+        }
+    }
+}
+function sendResponse(ws, response) {
+    try {
+        if (ws.readyState === 1) {
+            ws.send(JSON.stringify(response));
+        }
+    }
+    catch { }
+}
+//# sourceMappingURL=handler.js.map

package/dist/websocket/handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler.js","sourceRoot":"","sources":["../../src/websocket/handler.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,UAAU,EAAE,MAAM,6BAA6B,CAAC;AAiBzD,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,KAAK;YAC1B,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,kCAAkC,CAAC,CAAC;QAChD,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,IAAI,CAAC,kEAAkE,CAAC,CAAC;QACjF,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,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,SAAS,YAAY,CAAC,EAAiB,EAAE,QAA2B;IAClE,IAAI,CAAC;QACH,IAAI,EAAE,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;YACxB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACZ,CAAC"}

package/dist/websocket/index.d.ts

@@ -0,0 +1,2 @@
+export { setupWebSocket } from './handler.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/websocket/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/websocket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC"}

package/dist/websocket/index.js

@@ -0,0 +1,2 @@
+export { setupWebSocket } from './handler.js';
+//# sourceMappingURL=index.js.map

package/dist/websocket/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/websocket/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC"}

package/package.json

@@ -0,0 +1,67 @@
+{
+  "name": "@cogitator-ai/koa",
+  "version": "0.1.0",
+  "description": "Koa server adapter for Cogitator AI runtime",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@cogitator-ai/core": "0.17.5",
+    "@cogitator-ai/types": "0.19.3",
+    "@cogitator-ai/server-shared": "0.1.0"
+  },
+  "peerDependencies": {
+    "koa": ">=2.15.0",
+    "@koa/router": ">=12.0.0"
+  },
+  "optionalDependencies": {
+    "ws": "^8.0.0",
+    "@cogitator-ai/swarms": "0.4.7",
+    "@cogitator-ai/workflows": "0.5.4"
+  },
+  "devDependencies": {
+    "@types/koa": "^2.15.0",
+    "@types/koa__router": "^12.0.0",
+    "@types/ws": "^8.5.0",
+    "koa": "^2.15.0",
+    "@koa/router": "^13.1.0",
+    "typescript": "^5.6.0",
+    "ws": "^8.18.0",
+    "@cogitator-ai/swarms": "0.4.7",
+    "@cogitator-ai/workflows": "0.5.4"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/cogitator-ai/Cogitator-AI.git",
+    "directory": "packages/koa"
+  },
+  "license": "MIT",
+  "keywords": [
+    "cogitator",
+    "ai",
+    "agents",
+    "koa",
+    "server",
+    "api",
+    "streaming",
+    "sse",
+    "websocket",
+    "llm",
+    "middleware"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist"
+  }
+}