@cogitator-ai/fastify 0.1.1

package/dist/types.js

@@ -0,0 +1,59 @@
+export const AgentRunRequestSchema = {
+    type: 'object',
+    properties: {
+        input: { type: 'string' },
+        context: { type: 'object', additionalProperties: true },
+        threadId: { type: 'string' },
+    },
+    required: ['input'],
+};
+export const AgentRunResponseSchema = {
+    type: 'object',
+    properties: {
+        output: { type: 'string' },
+        threadId: { type: 'string' },
+        usage: {
+            type: 'object',
+            properties: {
+                inputTokens: { type: 'number' },
+                outputTokens: { type: 'number' },
+                totalTokens: { type: 'number' },
+            },
+        },
+        toolCalls: { type: 'array' },
+    },
+};
+export const AddMessageRequestSchema = {
+    type: 'object',
+    properties: {
+        role: { type: 'string', enum: ['user', 'assistant', 'system'] },
+        content: { type: 'string' },
+        metadata: { type: 'object', additionalProperties: true },
+    },
+    required: ['role', 'content'],
+};
+export const WorkflowRunRequestSchema = {
+    type: 'object',
+    properties: {
+        input: { type: 'object', additionalProperties: true },
+        options: {
+            type: 'object',
+            properties: {
+                maxConcurrency: { type: 'number' },
+                maxIterations: { type: 'number' },
+                checkpoint: { type: 'boolean' },
+            },
+        },
+    },
+};
+export const SwarmRunRequestSchema = {
+    type: 'object',
+    properties: {
+        input: { type: 'string' },
+        context: { type: 'object', additionalProperties: true },
+        threadId: { type: 'string' },
+        timeout: { type: 'number' },
+    },
+    required: ['input'],
+};
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AA+RA,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACzB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACvD,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC7B;IACD,QAAQ,EAAE,CAAC,OAAO,CAAC;CACX,CAAC;AAEX,MAAM,CAAC,MAAM,sBAAsB,GAAG;IACpC,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC1B,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC5B,KAAK,EAAE;YACL,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC/B,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAChC,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAChC;SACF;QACD,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;KAC7B;CACO,CAAC;AAEX,MAAM,CAAC,MAAM,uBAAuB,GAAG;IACrC,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,EAAE;QAC/D,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC3B,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;KACzD;IACD,QAAQ,EAAE,CAAC,MAAM,EAAE,SAAS,CAAC;CACrB,CAAC;AAEX,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACtC,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACrD,OAAO,EAAE;YACP,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,cAAc,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAClC,aAAa,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACjC,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aAChC;SACF;KACF;CACO,CAAC;AAEX,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,IAAI,EAAE,QAAQ;IACd,UAAU,EAAE;QACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACzB,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,oBAAoB,EAAE,IAAI,EAAE;QACvD,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC5B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC5B;IACD,QAAQ,EAAE,CAAC,OAAO,CAAC;CACX,CAAC"}

package/dist/websocket/handler.d.ts

@@ -0,0 +1,7 @@
+import type { FastifyPluginAsync } from 'fastify';
+interface WebSocketRoutesOptions {
+    path?: string;
+}
+export declare const websocketRoutes: FastifyPluginAsync<WebSocketRoutesOptions>;
+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,kBAAkB,EAAkB,MAAM,SAAS,CAAC;AAiBlE,UAAU,sBAAsB;IAC9B,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,eAAO,MAAM,eAAe,EAAE,kBAAkB,CAAC,sBAAsB,CAgCtE,CAAC"}

package/dist/websocket/handler.js

@@ -0,0 +1,132 @@
+import { generateId } from '../streaming/helpers.js';
+export const websocketRoutes = async (fastify, opts) => {
+    const path = opts.path ?? '/ws';
+    fastify.get(path, { websocket: true }, (socket, _request) => {
+        const clientState = {
+            id: generateId('ws'),
+            subscriptions: new Map(),
+        };
+        socket.on('message', async (data) => {
+            try {
+                const message = JSON.parse(data.toString());
+                await handleMessage(socket, message, fastify, clientState);
+            }
+            catch (error) {
+                sendResponse(socket, {
+                    type: 'error',
+                    error: error instanceof Error ? error.message : 'Invalid message',
+                });
+            }
+        });
+        socket.on('close', () => {
+            clientState.abortController?.abort();
+        });
+        socket.on('error', () => {
+            clientState.abortController?.abort();
+        });
+    });
+};
+async function handleMessage(socket, message, fastify, state) {
+    switch (message.type) {
+        case 'ping':
+            sendResponse(socket, { type: 'pong' });
+            break;
+        case 'subscribe':
+            if (message.channel) {
+                state.subscriptions.set(message.channel, {
+                    channel: message.channel,
+                    callback: (data) => {
+                        sendResponse(socket, { type: 'event', channel: message.channel, payload: data });
+                    },
+                });
+                sendResponse(socket, { type: 'subscribed', channel: message.channel });
+            }
+            break;
+        case 'unsubscribe':
+            if (message.channel) {
+                state.subscriptions.delete(message.channel);
+                sendResponse(socket, { type: 'unsubscribed', channel: message.channel });
+            }
+            break;
+        case 'run':
+            await handleRun(socket, message, fastify, state);
+            break;
+        case 'stop':
+            state.abortController?.abort();
+            state.abortController = undefined;
+            break;
+    }
+}
+async function handleRun(socket, message, fastify, state) {
+    const payload = message.payload;
+    if (!payload?.type || !payload?.name || !payload?.input) {
+        sendResponse(socket, { type: 'error', id: message.id, error: 'Invalid run payload' });
+        return;
+    }
+    state.abortController = new AbortController();
+    try {
+        if (payload.type === 'agent') {
+            const agent = fastify.cogitator.agents[payload.name];
+            if (!agent) {
+                sendResponse(socket, {
+                    type: 'error',
+                    id: message.id,
+                    error: `Agent '${payload.name}' not found`,
+                });
+                return;
+            }
+            const result = await fastify.cogitator.runtime.run(agent, {
+                input: payload.input,
+                context: payload.context,
+                stream: true,
+                onToken: (token) => {
+                    sendResponse(socket, {
+                        type: 'event',
+                        id: message.id,
+                        payload: { type: 'token', delta: token },
+                    });
+                },
+                onToolCall: (toolCall) => {
+                    sendResponse(socket, {
+                        type: 'event',
+                        id: message.id,
+                        payload: { type: 'tool-call', ...toolCall },
+                    });
+                },
+                onToolResult: (toolResult) => {
+                    sendResponse(socket, {
+                        type: 'event',
+                        id: message.id,
+                        payload: { type: 'tool-result', ...toolResult },
+                    });
+                },
+            });
+            sendResponse(socket, {
+                type: 'event',
+                id: message.id,
+                payload: { type: 'complete', result },
+            });
+        }
+    }
+    catch (error) {
+        if (state.abortController?.signal.aborted) {
+            sendResponse(socket, { type: 'event', id: message.id, payload: { type: 'cancelled' } });
+        }
+        else {
+            sendResponse(socket, {
+                type: 'error',
+                id: message.id,
+                error: error instanceof Error ? error.message : 'Unknown error',
+            });
+        }
+    }
+}
+function sendResponse(socket, response) {
+    try {
+        if (socket.readyState === 1) {
+            socket.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":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAkBrD,MAAM,CAAC,MAAM,eAAe,GAA+C,KAAK,EAC9E,OAAO,EACP,IAAI,EACJ,EAAE;IACF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,KAAK,CAAC;IAEhC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,MAAiB,EAAE,QAAwB,EAAE,EAAE;QACrF,MAAM,WAAW,GAAgB;YAC/B,EAAE,EAAE,UAAU,CAAC,IAAI,CAAC;YACpB,aAAa,EAAE,IAAI,GAAG,EAAE;SACzB,CAAC;QAEF,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAY,EAAE,EAAE;YAC1C,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAqB,CAAC;gBAChE,MAAM,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,YAAY,CAAC,MAAM,EAAE;oBACnB,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB;iBAClE,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,WAAW,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACtB,WAAW,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,KAAK,UAAU,aAAa,CAC1B,MAAiB,EACjB,OAAyB,EACzB,OAA0C,EAC1C,KAAkB;IAElB,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;QACrB,KAAK,MAAM;YACT,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YACvC,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,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;oBACnF,CAAC;iBACF,CAAC,CAAC;gBACH,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACzE,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,MAAM,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3E,CAAC;YACD,MAAM;QAER,KAAK,KAAK;YACR,MAAM,SAAS,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACjD,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,MAAiB,EACjB,OAAyB,EACzB,OAA0C,EAC1C,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,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,KAAK,EAAE,qBAAqB,EAAE,CAAC,CAAC;QACtF,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,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,YAAY,CAAC,MAAM,EAAE;oBACnB,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,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE;gBACxD,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,MAAM,EAAE;wBACnB,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,MAAM,EAAE;wBACnB,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,MAAM,EAAE;wBACnB,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,MAAM,EAAE;gBACnB,IAAI,EAAE,OAAO;gBACb,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE;aACtC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1C,YAAY,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;QAC1F,CAAC;aAAM,CAAC;YACN,YAAY,CAAC,MAAM,EAAE;gBACnB,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,MAAiB,EAAE,QAA2B;IAClE,IAAI,CAAC;QACH,IAAI,MAAM,CAAC,UAAU,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AACZ,CAAC"}

package/package.json

@@ -0,0 +1,70 @@
+{
+  "name": "@cogitator-ai/fastify",
+  "version": "0.1.1",
+  "description": "Fastify 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": {
+    "fastify-plugin": "^5.0.0",
+    "@cogitator-ai/types": "0.19.3",
+    "@cogitator-ai/server-shared": "0.1.0",
+    "@cogitator-ai/core": "0.17.5"
+  },
+  "peerDependencies": {
+    "fastify": ">=5.0.0"
+  },
+  "optionalDependencies": {
+    "@fastify/swagger": "^9.0.0",
+    "@fastify/swagger-ui": "^5.0.0",
+    "@fastify/rate-limit": "^10.0.0",
+    "@fastify/websocket": "^11.0.0",
+    "@cogitator-ai/workflows": "0.5.4",
+    "@cogitator-ai/swarms": "0.4.7"
+  },
+  "devDependencies": {
+    "@fastify/swagger": "^9.4.2",
+    "@fastify/swagger-ui": "^5.2.2",
+    "@fastify/rate-limit": "^10.2.1",
+    "@fastify/websocket": "^11.0.2",
+    "@types/node": "^22.0.0",
+    "@types/ws": "^8.5.0",
+    "fastify": "^5.2.1",
+    "typescript": "^5.6.0",
+    "@cogitator-ai/workflows": "0.5.4",
+    "@cogitator-ai/swarms": "0.4.7"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/cogitator-ai/Cogitator-AI.git",
+    "directory": "packages/fastify"
+  },
+  "license": "MIT",
+  "keywords": [
+    "cogitator",
+    "ai",
+    "agents",
+    "fastify",
+    "server",
+    "api",
+    "streaming",
+    "sse",
+    "websocket",
+    "llm"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist"
+  }
+}