@denispro2006/duke-ai-kernel 1.0.0

package/dist/workers/aiWorker.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runWithRetry = runWithRetry;
+exports.streamWithAbort = streamWithAbort;
+const logger_1 = require("../lib/logger");
+const helpers_1 = require("../utils/helpers");
+const config_1 = require("../config");
+async function runWithRetry(provider, request) {
+    let lastError;
+    for (let attempt = 0; attempt <= config_1.config.worker.retryLimit; attempt++) {
+        try {
+            const result = await Promise.race([
+                provider.complete(request),
+                new Promise((_, reject) => setTimeout(() => reject(new Error('TIMEOUT')), config_1.config.worker.timeoutMs)),
+            ]);
+            return result;
+        }
+        catch (err) {
+            lastError = err;
+            const status = err.status;
+            const isRetryable = status ? (0, helpers_1.isRetryableStatus)(status) : err.message === 'TIMEOUT';
+            if (attempt < config_1.config.worker.retryLimit && isRetryable) {
+                const delay = (0, helpers_1.exponentialBackoff)(attempt, config_1.config.worker.retryDelayMs);
+                logger_1.logger.warn({ attempt, delay, error: err.message }, 'Worker retrying');
+                await (0, helpers_1.sleep)(delay);
+            }
+            else {
+                break;
+            }
+        }
+    }
+    throw lastError ?? new Error('Worker: all retries exhausted');
+}
+async function* streamWithAbort(provider, request, abortSignal) {
+    const stream = provider.stream(request);
+    for await (const chunk of stream) {
+        if (abortSignal.aborted)
+            return;
+        yield chunk;
+    }
+}
+//# sourceMappingURL=aiWorker.js.map

package/dist/workers/aiWorker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aiWorker.js","sourceRoot":"","sources":["../../src/workers/aiWorker.ts"],"names":[],"mappings":";;AAKA,oCA+BC;AAED,0CAUC;AAhDD,0CAAsC;AACtC,8CAA+E;AAC/E,sCAAkC;AAG3B,KAAK,UAAU,YAAY,CAChC,QAAqB,EACrB,OAA0B;IAE1B,IAAI,SAA4B,CAAA;IAEhC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,eAAM,CAAC,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QACrE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBAChC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC1B,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAC/B,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,eAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CACxE;aACF,CAAC,CAAA;YACF,OAAO,MAAM,CAAA;QACf,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAY,CAAA;YACxB,MAAM,MAAM,GAAI,GAA2B,CAAC,MAAM,CAAA;YAClD,MAAM,WAAW,GAAG,MAAM,CAAC,CAAC,CAAC,IAAA,2BAAiB,EAAC,MAAM,CAAC,CAAC,CAAC,CAAE,GAAa,CAAC,OAAO,KAAK,SAAS,CAAA;YAE7F,IAAI,OAAO,GAAG,eAAM,CAAC,MAAM,CAAC,UAAU,IAAI,WAAW,EAAE,CAAC;gBACtD,MAAM,KAAK,GAAG,IAAA,4BAAkB,EAAC,OAAO,EAAE,eAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAA;gBACrE,eAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAG,GAAa,CAAC,OAAO,EAAE,EAAE,iBAAiB,CAAC,CAAA;gBACjF,MAAM,IAAA,eAAK,EAAC,KAAK,CAAC,CAAA;YACpB,CAAC;iBAAM,CAAC;gBACN,MAAK;YACP,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;AAC/D,CAAC;AAEM,KAAK,SAAS,CAAC,CAAC,eAAe,CACpC,QAAqB,EACrB,OAA0B,EAC1B,WAAiC;IAEjC,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;IACvC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,IAAI,WAAW,CAAC,OAAO;YAAE,OAAM;QAC/B,MAAM,KAAK,CAAA;IACb,CAAC;AACH,CAAC"}

package/dist/ws/gateway.d.ts

@@ -0,0 +1,3 @@
+import { WebSocketServer } from 'ws';
+export declare function createWsGateway(wss: WebSocketServer): void;
+//# sourceMappingURL=gateway.d.ts.map

package/dist/ws/gateway.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway.d.ts","sourceRoot":"","sources":["../../src/ws/gateway.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAa,MAAM,IAAI,CAAA;AA4B/C,wBAAgB,eAAe,CAAC,GAAG,EAAE,eAAe,GAAG,IAAI,CAgG1D"}

package/dist/ws/gateway.js

@@ -0,0 +1,110 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createWsGateway = createWsGateway;
+const ws_1 = require("ws");
+const kernel_1 = require("../kernel");
+const context_manager_1 = require("../context/context-manager");
+const config_1 = require("../config");
+const logger_1 = require("../lib/logger");
+const helpers_1 = require("../utils/helpers");
+function send(ws, data) {
+    if (ws.readyState === ws_1.WebSocket.OPEN) {
+        ws.send(JSON.stringify(data));
+    }
+}
+function authenticate(req) {
+    if (config_1.config.auth.bypass)
+        return true;
+    const auth = req.headers['authorization'];
+    if (!auth || !auth.startsWith('Bearer '))
+        return false;
+    const token = auth.slice(7);
+    return token === config_1.config.auth.secret;
+}
+function createWsGateway(wss) {
+    wss.on('connection', (ws, req) => {
+        if (!authenticate(req)) {
+            logger_1.logger.warn({ ip: req.socket.remoteAddress }, 'WebSocket auth failed');
+            ws.close(1008, 'Unauthorized');
+            return;
+        }
+        const sessionId = (0, helpers_1.generateId)('sess');
+        logger_1.logger.info({ sessionId }, 'WebSocket client connected');
+        send(ws, { type: 'connected', sessionId, timestamp: new Date() });
+        ws.on('message', async (raw) => {
+            let msg;
+            try {
+                msg = JSON.parse(raw.toString());
+            }
+            catch {
+                send(ws, { type: 'error', code: 'INVALID_MESSAGE', message: 'Invalid JSON' });
+                return;
+            }
+            switch (msg.type) {
+                case 'ping': {
+                    send(ws, { type: 'pong', timestamp: new Date() });
+                    break;
+                }
+                case 'clear': {
+                    await context_manager_1.contextManager.clear(msg.conversationId);
+                    send(ws, { type: 'done', conversationId: msg.conversationId, finishReason: 'stop' });
+                    break;
+                }
+                case 'abort': {
+                    kernel_1.kernel.abort(msg.conversationId);
+                    break;
+                }
+                case 'message': {
+                    if (!msg.content) {
+                        send(ws, { type: 'error', code: 'INVALID_MESSAGE', message: 'content is required' });
+                        return;
+                    }
+                    const conversationId = msg.conversationId ?? (0, helpers_1.generateId)('conv');
+                    let index = 0;
+                    kernel_1.kernel.onStream(conversationId, (chunk) => {
+                        send(ws, { type: 'chunk', conversationId, delta: chunk, index: index++ });
+                    });
+                    try {
+                        const response = await kernel_1.kernel.process({
+                            id: (0, helpers_1.generateId)('req'),
+                            conversationId,
+                            messages: [{ role: 'user', content: msg.content }],
+                            model: msg.model,
+                            options: msg.options,
+                        });
+                        send(ws, {
+                            type: 'done',
+                            conversationId,
+                            finishReason: response.finishReason,
+                            usage: response.usage,
+                            model: response.model,
+                        });
+                    }
+                    catch (err) {
+                        logger_1.logger.error({ err, conversationId, sessionId }, 'WebSocket generation error');
+                        send(ws, {
+                            type: 'error',
+                            conversationId,
+                            code: 'PROVIDER_ERROR',
+                            message: err.message,
+                        });
+                    }
+                    finally {
+                        kernel_1.kernel.offStream(conversationId);
+                    }
+                    break;
+                }
+                default: {
+                    send(ws, { type: 'error', code: 'INVALID_MESSAGE', message: 'Unknown message type' });
+                }
+            }
+        });
+        ws.on('close', () => {
+            logger_1.logger.info({ sessionId }, 'WebSocket client disconnected');
+        });
+        ws.on('error', (err) => {
+            logger_1.logger.error({ err, sessionId }, 'WebSocket error');
+        });
+    });
+}
+//# sourceMappingURL=gateway.js.map

package/dist/ws/gateway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gateway.js","sourceRoot":"","sources":["../../src/ws/gateway.ts"],"names":[],"mappings":";;AA4BA,0CAgGC;AA5HD,2BAA+C;AAE/C,sCAAkC;AAClC,gEAA2D;AAC3D,sCAAkC;AAClC,0CAAsC;AACtC,8CAA6C;AAQ7C,SAAS,IAAI,CAAC,EAAa,EAAE,IAAa;IACxC,IAAI,EAAE,CAAC,UAAU,KAAK,cAAS,CAAC,IAAI,EAAE,CAAC;QACrC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAA;IAC/B,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CAAC,GAAoB;IACxC,IAAI,eAAM,CAAC,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAA;IACnC,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC,CAAA;IACzC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,KAAK,CAAA;IACtD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;IAC3B,OAAO,KAAK,KAAK,eAAM,CAAC,IAAI,CAAC,MAAM,CAAA;AACrC,CAAC;AAED,SAAgB,eAAe,CAAC,GAAoB;IAClD,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,GAAoB,EAAE,EAAE;QAC3D,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;YACvB,eAAM,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,uBAAuB,CAAC,CAAA;YACtE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAA;YAC9B,OAAM;QACR,CAAC;QAED,MAAM,SAAS,GAAG,IAAA,oBAAU,EAAC,MAAM,CAAC,CAAA;QACpC,eAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,4BAA4B,CAAC,CAAA;QAExD,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAA;QAEjE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,GAAW,EAAE,EAAE;YACrC,IAAI,GAAc,CAAA;YAClB,IAAI,CAAC;gBACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAc,CAAA;YAC/C,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAA;gBAC7E,OAAM;YACR,CAAC;YAED,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,MAAM,CAAC,CAAC,CAAC;oBACZ,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC,CAAA;oBACjD,MAAK;gBACP,CAAC;gBAED,KAAK,OAAO,CAAC,CAAC,CAAC;oBACb,MAAM,gCAAc,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;oBAC9C,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,CAAC,cAAc,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAA;oBACpF,MAAK;gBACP,CAAC;gBAED,KAAK,OAAO,CAAC,CAAC,CAAC;oBACb,eAAM,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAA;oBAChC,MAAK;gBACP,CAAC;gBAED,KAAK,SAAS,CAAC,CAAC,CAAC;oBACf,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;wBACjB,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC,CAAA;wBACpF,OAAM;oBACR,CAAC;oBAED,MAAM,cAAc,GAAG,GAAG,CAAC,cAAc,IAAI,IAAA,oBAAU,EAAC,MAAM,CAAC,CAAA;oBAC/D,IAAI,KAAK,GAAG,CAAC,CAAA;oBAEb,eAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,KAAa,EAAE,EAAE;wBAChD,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAA;oBAC3E,CAAC,CAAC,CAAA;oBAEF,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,eAAM,CAAC,OAAO,CAAC;4BACpC,EAAE,EAAE,IAAA,oBAAU,EAAC,KAAK,CAAC;4BACrB,cAAc;4BACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;4BAClD,KAAK,EAAE,GAAG,CAAC,KAAK;4BAChB,OAAO,EAAE,GAAG,CAAC,OAA8C;yBAC5D,CAAC,CAAA;wBAEF,IAAI,CAAC,EAAE,EAAE;4BACP,IAAI,EAAE,MAAM;4BACZ,cAAc;4BACd,YAAY,EAAE,QAAQ,CAAC,YAAY;4BACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;4BACrB,KAAK,EAAE,QAAQ,CAAC,KAAK;yBACtB,CAAC,CAAA;oBACJ,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,cAAc,EAAE,SAAS,EAAE,EAAE,4BAA4B,CAAC,CAAA;wBAC9E,IAAI,CAAC,EAAE,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,cAAc;4BACd,IAAI,EAAE,gBAAgB;4BACtB,OAAO,EAAG,GAAa,CAAC,OAAO;yBAChC,CAAC,CAAA;oBACJ,CAAC;4BAAS,CAAC;wBACT,eAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;oBAClC,CAAC;oBACD,MAAK;gBACP,CAAC;gBAED,OAAO,CAAC,CAAC,CAAC;oBACR,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC,CAAA;gBACvF,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,eAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,EAAE,+BAA+B,CAAC,CAAA;QAC7D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;YAC5B,eAAM,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,iBAAiB,CAAC,CAAA;QACrD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC"}

package/package.json

@@ -0,0 +1,68 @@
+{
+  "name": "@denispro2006/duke-ai-kernel",
+  "version": "1.0.0",
+  "description": "AI kernel for the duke-messenger platform — handles LLM routing, context management, and real-time messaging via WebSocket.",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "config",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/denispro2006/duke-ai-kernel"
+  },
+  "scripts": {
+    "dev": "tsx watch src/index.ts",
+    "build": "tsc",
+    "start": "node dist/index.js",
+    "lint": "eslint .",
+    "lint:fix": "eslint . --fix",
+    "format": "prettier --write .",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage",
+    "db:migrate": "node dist/db/migrate.js"
+  },
+  "dependencies": {
+    "express": "^5.0.0",
+    "ws": "^8.18.0",
+    "bull": "^4.16.0",
+    "ioredis": "^5.4.1",
+    "openai": "^4.77.0",
+    "@anthropic-ai/sdk": "^0.36.0",
+    "pino": "^9.6.0",
+    "pino-http": "^10.3.0",
+    "dotenv": "^16.4.7",
+    "zod": "^3.24.1",
+    "uuid": "^11.0.5",
+    "pg": "^8.13.1"
+  },
+  "devDependencies": {
+    "@types/express": "^5.0.0",
+    "@types/ws": "^8.5.13",
+    "@types/bull": "^4.10.0",
+    "@types/node": "^22.10.0",
+    "@types/uuid": "^10.0.0",
+    "@types/pg": "^8.11.10",
+    "@types/supertest": "^6.0.2",
+    "typescript": "^5.7.2",
+    "tsx": "^4.19.2",
+    "vitest": "^2.1.8",
+    "@vitest/coverage-v8": "^2.1.8",
+    "supertest": "^7.0.0",
+    "eslint": "^9.16.0",
+    "@typescript-eslint/eslint-plugin": "^8.18.0",
+    "@typescript-eslint/parser": "^8.18.0",
+    "prettier": "^3.4.2"
+  },
+  "engines": {
+    "node": ">=20.0.0"
+  },
+  "license": "MIT"
+}