@idevconn/ai-chat-be 0.1.2

package/dist/providers/openai.provider.js

@@ -0,0 +1,87 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIProvider = void 0;
+const openai_1 = __importDefault(require("openai"));
+const logger_1 = require("../utils/logger");
+class OpenAIProvider {
+    s;
+    id = 'openai';
+    supportsEmbedding = true;
+    client;
+    constructor(s) {
+        this.s = s;
+        this.client = new openai_1.default({
+            apiKey: s.apiKey,
+            ...(s.baseUrl ? { baseURL: s.baseUrl } : {}),
+        });
+    }
+    async *chatStream(messages, options) {
+        let stream;
+        try {
+            stream = await this.client.chat.completions.create({
+                model: this.s.chatModel,
+                messages: messages.map((m) => ({
+                    role: m.role,
+                    content: m.content,
+                })),
+                stream: true,
+                temperature: options?.temperature ?? 0.7,
+                max_tokens: options?.maxTokens ?? 2048,
+            });
+        }
+        catch (error) {
+            (0, logger_1.logStreamError)('OpenAI', error);
+            yield {
+                type: 'error',
+                error: 'Failed to connect to OpenAI API',
+            };
+            return;
+        }
+        let totalTokens = 0;
+        try {
+            for await (const chunk of stream) {
+                const delta = chunk.choices[0]?.delta;
+                if (delta?.content) {
+                    totalTokens += Math.ceil(delta.content.length / 4);
+                    yield { type: 'token', content: delta.content };
+                }
+                if (chunk.usage) {
+                    yield {
+                        type: 'done',
+                        usage: {
+                            promptTokens: chunk.usage.prompt_tokens,
+                            completionTokens: chunk.usage.completion_tokens,
+                            totalTokens: chunk.usage.total_tokens,
+                        },
+                    };
+                    return;
+                }
+            }
+        }
+        catch (error) {
+            (0, logger_1.logStreamError)('OpenAI', error);
+            yield { type: 'error', error: 'Stream interrupted' };
+            return;
+        }
+        yield {
+            type: 'done',
+            usage: {
+                promptTokens: 0,
+                completionTokens: totalTokens,
+                totalTokens,
+            },
+        };
+    }
+    async embed(texts) {
+        const response = await this.client.embeddings.create({
+            model: this.s.embeddingModel,
+            input: texts,
+        });
+        return response.data.map((item) => item.embedding);
+    }
+}
+exports.OpenAIProvider = OpenAIProvider;
+//# sourceMappingURL=openai.provider.js.map

package/dist/providers/openai.provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.provider.js","sourceRoot":"","sources":["../../src/providers/openai.provider.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAE5B,4CAAiD;AAGjD,MAAa,cAAc;IAMI;IALpB,EAAE,GAAG,QAAiB,CAAC;IACvB,iBAAiB,GAAG,IAAI,CAAC;IAEjB,MAAM,CAAS;IAEhC,YAA6B,CAAmB;QAAnB,MAAC,GAAD,CAAC,CAAkB;QAC9C,IAAI,CAAC,MAAM,GAAG,IAAI,gBAAM,CAAC;YACvB,MAAM,EAAE,CAAC,CAAC,MAAM;YAChB,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC7C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CACf,QAAuB,EACvB,OAAuB;QAEvB,IAAI,MAAkE,CAAC;QAEvE,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;gBACjD,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,SAAS;gBACvB,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC7B,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;iBACnB,CAAC,CAAC;gBACH,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,OAAO,EAAE,WAAW,IAAI,GAAG;gBACxC,UAAU,EAAE,OAAO,EAAE,SAAS,IAAI,IAAI;aACvC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAA,uBAAc,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAChC,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,iCAAiC;aACzC,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;gBACtC,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;oBACnB,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBACnD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClD,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,MAAM;wBACJ,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE;4BACL,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa;4BACvC,gBAAgB,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB;4BAC/C,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;yBACtC;qBACF,CAAC;oBACF,OAAO;gBACT,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAA,uBAAc,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YAChC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;YACrD,OAAO;QACT,CAAC;QAED,MAAM;YACJ,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE;gBACL,YAAY,EAAE,CAAC;gBACf,gBAAgB,EAAE,WAAW;gBAC7B,WAAW;aACZ;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAe;QACzB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YACnD,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,cAAe;YAC7B,KAAK,EAAE,KAAK;SACb,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;CACF;AArFD,wCAqFC"}

package/dist/providers/provider-registry.d.ts

@@ -0,0 +1,17 @@
+import { ILlmProvider, ProviderId } from '../types';
+import { ProviderSettings, RetryPolicy } from './provider-settings';
+export interface ProviderDefaults {
+    chatModel: string;
+    embeddingModel?: string;
+    baseUrl?: string;
+    retry?: RetryPolicy;
+}
+export interface ProviderDescriptor {
+    id: ProviderId;
+    label: string;
+    defaults: ProviderDefaults;
+    factory: (s: ProviderSettings) => ILlmProvider;
+}
+export declare const BUILT_IN_PROVIDERS: readonly ProviderDescriptor[];
+export declare function findDescriptor(id: ProviderId): ProviderDescriptor | undefined;
+//# sourceMappingURL=provider-registry.d.ts.map

package/dist/providers/provider-registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-registry.d.ts","sourceRoot":"","sources":["../../src/providers/provider-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAKpE,MAAM,WAAW,gBAAgB;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AASD,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,UAAU,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,EAAE,CAAC,CAAC,EAAE,gBAAgB,KAAK,YAAY,CAAC;CAChD;AASD,eAAO,MAAM,kBAAkB,EAAE,SAAS,kBAAkB,EA8BlD,CAAC;AAEX,wBAAgB,cAAc,CAAC,EAAE,EAAE,UAAU,GAAG,kBAAkB,GAAG,SAAS,CAE7E"}

package/dist/providers/provider-registry.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BUILT_IN_PROVIDERS = void 0;
+exports.findDescriptor = findDescriptor;
+const gemini_provider_1 = require("./gemini.provider");
+const openai_provider_1 = require("./openai.provider");
+const grok_provider_1 = require("./grok.provider");
+exports.BUILT_IN_PROVIDERS = [
+    {
+        id: 'gemini',
+        label: 'Gemini',
+        defaults: {
+            chatModel: 'gemini-2.5-flash',
+            embeddingModel: 'gemini-embedding-001',
+            baseUrl: 'https://generativelanguage.googleapis.com/v1beta',
+            retry: { maxAttempts: 2, baseDelayMs: 500 },
+        },
+        factory: (s) => new gemini_provider_1.GeminiProvider(s),
+    },
+    {
+        id: 'openai',
+        label: 'OpenAI',
+        defaults: {
+            chatModel: 'gpt-4o',
+            embeddingModel: 'text-embedding-3-small',
+        },
+        factory: (s) => new openai_provider_1.OpenAIProvider(s),
+    },
+    {
+        id: 'grok',
+        label: 'Grok',
+        defaults: {
+            chatModel: 'grok-3-mini',
+            baseUrl: 'https://api.x.ai/v1',
+        },
+        factory: (s) => new grok_provider_1.GrokProvider(s),
+    },
+];
+function findDescriptor(id) {
+    return exports.BUILT_IN_PROVIDERS.find((d) => d.id === id);
+}
+//# sourceMappingURL=provider-registry.js.map

package/dist/providers/provider-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-registry.js","sourceRoot":"","sources":["../../src/providers/provider-registry.ts"],"names":[],"mappings":";;;AAkEA,wCAEC;AAlED,uDAAmD;AACnD,uDAAmD;AACnD,mDAA+C;AA8BlC,QAAA,kBAAkB,GAAkC;IAC/D;QACE,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE;YACR,SAAS,EAAE,kBAAkB;YAC7B,cAAc,EAAE,sBAAsB;YACtC,OAAO,EAAE,kDAAkD;YAC3D,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;SAC5C;QACD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,gCAAc,CAAC,CAAC,CAAC;KACtC;IACD;QACE,EAAE,EAAE,QAAQ;QACZ,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE;YACR,SAAS,EAAE,QAAQ;YACnB,cAAc,EAAE,wBAAwB;SACzC;QACD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,gCAAc,CAAC,CAAC,CAAC;KACtC;IACD;QACE,EAAE,EAAE,MAAM;QACV,KAAK,EAAE,MAAM;QACb,QAAQ,EAAE;YACR,SAAS,EAAE,aAAa;YACxB,OAAO,EAAE,qBAAqB;SAC/B;QACD,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,4BAAY,CAAC,CAAC,CAAC;KACpC;CACO,CAAC;AAEX,SAAgB,cAAc,CAAC,EAAc;IAC3C,OAAO,0BAAkB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;AACrD,CAAC"}

package/dist/providers/provider-settings.d.ts

@@ -0,0 +1,12 @@
+export interface RetryPolicy {
+    maxAttempts: number;
+    baseDelayMs: number;
+}
+export interface ProviderSettings {
+    apiKey: string;
+    chatModel: string;
+    embeddingModel?: string;
+    baseUrl?: string;
+    retry?: RetryPolicy;
+}
+//# sourceMappingURL=provider-settings.d.ts.map

package/dist/providers/provider-settings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-settings.d.ts","sourceRoot":"","sources":["../../src/providers/provider-settings.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,WAAW;IAC1B,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;CACrB;AAOD,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB"}

package/dist/providers/provider-settings.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=provider-settings.js.map

package/dist/providers/provider-settings.js.map

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

package/dist/providers/provider.factory.d.ts

@@ -0,0 +1,18 @@
+import { OnModuleInit } from '@nestjs/common';
+import { AiChatModuleConfig, ILlmProvider, ProviderId } from '../types';
+export declare class ProviderFactory implements OnModuleInit {
+    private readonly config;
+    private providers;
+    private activeProvider;
+    constructor(config: AiChatModuleConfig);
+    onModuleInit(): void;
+    private registerProviders;
+    private buildSettings;
+    private resolveApiKey;
+    private selectActive;
+    getProvider(): ILlmProvider | null;
+    getProviderById(id: ProviderId): ILlmProvider | null;
+    isAvailable(): boolean;
+    clampMaxTokens(requested: number | undefined): number | undefined;
+}
+//# sourceMappingURL=provider.factory.d.ts.map

package/dist/providers/provider.factory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.factory.d.ts","sourceRoot":"","sources":["../../src/providers/provider.factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAClE,OAAO,EAAkB,kBAAkB,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AAUxF,qBACa,eAAgB,YAAW,YAAY;IAId,OAAO,CAAC,QAAQ,CAAC,MAAM;IAH3D,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,cAAc,CAA6B;gBAEE,MAAM,EAAE,kBAAkB;IAE/E,YAAY,IAAI,IAAI;IAIpB,OAAO,CAAC,iBAAiB;IAwBzB,OAAO,CAAC,aAAa;IAgBrB,OAAO,CAAC,aAAa;IAYrB,OAAO,CAAC,YAAY;IAYpB,WAAW,IAAI,YAAY,GAAG,IAAI;IAIlC,eAAe,CAAC,EAAE,EAAE,UAAU,GAAG,YAAY,GAAG,IAAI;IAIpD,WAAW,IAAI,OAAO;IAStB,cAAc,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,GAAG,SAAS;CASlE"}

package/dist/providers/provider.factory.js

@@ -0,0 +1,107 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProviderFactory = void 0;
+const common_1 = require("@nestjs/common");
+const types_1 = require("../types");
+const provider_registry_1 = require("./provider-registry");
+const logger_1 = require("../utils/logger");
+let ProviderFactory = class ProviderFactory {
+    config;
+    providers = new Map();
+    activeProvider = null;
+    constructor(config) {
+        this.config = config;
+    }
+    onModuleInit() {
+        this.registerProviders();
+    }
+    registerProviders() {
+        for (const desc of provider_registry_1.BUILT_IN_PROVIDERS) {
+            const settings = this.buildSettings(desc);
+            if (!settings) {
+                (0, logger_1.logProviderDisabled)(desc.label, `no API key for ${desc.id}`);
+                continue;
+            }
+            this.providers.set(desc.id, desc.factory(settings));
+            (0, logger_1.logProviderEnabled)(desc.label, settings.chatModel);
+        }
+        this.activeProvider = this.selectActive();
+        if (!this.activeProvider) {
+            (0, logger_1.logBootError)('No LLM providers configured. Chat endpoint will return 503.');
+        }
+    }
+    buildSettings(desc) {
+        const apiKey = this.resolveApiKey(desc.id);
+        if (!apiKey)
+            return null;
+        return {
+            apiKey,
+            chatModel: this.config[`${desc.id}Model`] ||
+                this.config.model ||
+                desc.defaults.chatModel,
+            embeddingModel: this.config.embeddingModel || desc.defaults.embeddingModel,
+            baseUrl: this.config.baseUrl || desc.defaults.baseUrl,
+            retry: this.config.retry || desc.defaults.retry,
+        };
+    }
+    resolveApiKey(id) {
+        const perProvider = this.config[`${id}ApiKey`];
+        if (perProvider)
+            return perProvider;
+        const provider = this.config.provider || 'gemini';
+        if (provider === id || provider === 'auto') {
+            return this.config.apiKey;
+        }
+        return undefined;
+    }
+    selectActive() {
+        const provider = this.config.provider || 'gemini';
+        if (provider !== 'auto') {
+            return this.providers.get(provider) || null;
+        }
+        for (const desc of provider_registry_1.BUILT_IN_PROVIDERS) {
+            const p = this.providers.get(desc.id);
+            if (p)
+                return p;
+        }
+        return null;
+    }
+    getProvider() {
+        return this.activeProvider;
+    }
+    getProviderById(id) {
+        return this.providers.get(id) || null;
+    }
+    isAvailable() {
+        return this.activeProvider !== null;
+    }
+    clampMaxTokens(requested) {
+        const ceiling = this.config.limits?.maxOutputTokens ?? 8192;
+        if (requested === undefined)
+            return undefined;
+        if (requested > ceiling) {
+            logger_1.aiChatLogger.warn(`maxTokens=${requested} clamped to hard limit ${ceiling}`);
+            return ceiling;
+        }
+        return requested;
+    }
+};
+exports.ProviderFactory = ProviderFactory;
+exports.ProviderFactory = ProviderFactory = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)(types_1.AI_CHAT_CONFIG)),
+    __metadata("design:paramtypes", [Object])
+], ProviderFactory);
+//# sourceMappingURL=provider.factory.js.map

package/dist/providers/provider.factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.factory.js","sourceRoot":"","sources":["../../src/providers/provider.factory.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAkE;AAClE,oCAAwF;AACxF,2DAA6E;AAE7E,4CAKyB;AAGlB,IAAM,eAAe,GAArB,MAAM,eAAe;IAI2B;IAH7C,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;IAChD,cAAc,GAAwB,IAAI,CAAC;IAEnD,YAAqD,MAA0B;QAA1B,WAAM,GAAN,MAAM,CAAoB;IAAG,CAAC;IAEnF,YAAY;QACV,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,iBAAiB;QACvB,KAAK,MAAM,IAAI,IAAI,sCAAkB,EAAE,CAAC;YACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,IAAA,4BAAmB,EAAC,IAAI,CAAC,KAAK,EAAE,kBAAkB,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7D,SAAS;YACX,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YACpD,IAAA,2BAAkB,EAAC,IAAI,CAAC,KAAK,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAA,qBAAY,EAAC,6DAA6D,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;IAQO,aAAa,CAAC,IAAwB;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAEzB,OAAO;YACL,MAAM;YACN,SAAS,EACN,IAAI,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,EAAE,OAAmC,CAAwB;gBAClF,IAAI,CAAC,MAAM,CAAC,KAAK;gBACjB,IAAI,CAAC,QAAQ,CAAC,SAAS;YACzB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc;YAC1E,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO;YACrD,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK;SAChD,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,EAAc;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,QAAoC,CAE5D,CAAC;QACd,IAAI,WAAW;YAAE,OAAO,WAAW,CAAC;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAClD,IAAI,QAAQ,KAAK,EAAE,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAC5B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,YAAY;QAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,QAAQ,CAAC;QAClD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAsB,CAAC,IAAI,IAAI,CAAC;QAC5D,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,sCAAkB,EAAE,CAAC;YACtC,MAAM,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtC,IAAI,CAAC;gBAAE,OAAO,CAAC,CAAC;QAClB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,eAAe,CAAC,EAAc;QAC5B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,cAAc,KAAK,IAAI,CAAC;IACtC,CAAC;IAOD,cAAc,CAAC,SAA6B;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC;QAC5D,IAAI,SAAS,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAC9C,IAAI,SAAS,GAAG,OAAO,EAAE,CAAC;YACxB,qBAAY,CAAC,IAAI,CAAC,aAAa,SAAS,0BAA0B,OAAO,EAAE,CAAC,CAAC;YAC7E,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF,CAAA;AApGY,0CAAe;0BAAf,eAAe;IAD3B,IAAA,mBAAU,GAAE;IAKE,WAAA,IAAA,eAAM,EAAC,sBAAc,CAAC,CAAA;;GAJxB,eAAe,CAoG3B"}

package/dist/rag/chunker.d.ts

@@ -0,0 +1,7 @@
+export interface TextChunk {
+    id: string;
+    content: string;
+    metadata: Record<string, string>;
+}
+export declare function splitMarkdownIntoChunks(text: string, source: string, maxChars?: number): TextChunk[];
+//# sourceMappingURL=chunker.d.ts.map

package/dist/rag/chunker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunker.d.ts","sourceRoot":"","sources":["../../src/rag/chunker.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,SAAM,GAAG,SAAS,EAAE,CA6BjG"}

package/dist/rag/chunker.js

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.splitMarkdownIntoChunks = splitMarkdownIntoChunks;
+function splitMarkdownIntoChunks(text, source, maxChars = 800) {
+    const paragraphs = text.split(/\n{2,}/);
+    const chunks = [];
+    let current = '';
+    let index = 0;
+    for (const paragraph of paragraphs) {
+        if (current.length + paragraph.length > maxChars && current) {
+            chunks.push({
+                id: `${source}:${index}`,
+                content: current.trim(),
+                metadata: { source, chunk: String(index) },
+            });
+            current = paragraph;
+            index++;
+        }
+        else {
+            current += (current ? '\n\n' : '') + paragraph;
+        }
+    }
+    if (current.trim()) {
+        chunks.push({
+            id: `${source}:${index}`,
+            content: current.trim(),
+            metadata: { source, chunk: String(index) },
+        });
+    }
+    return chunks;
+}
+//# sourceMappingURL=chunker.js.map

package/dist/rag/chunker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunker.js","sourceRoot":"","sources":["../../src/rag/chunker.ts"],"names":[],"mappings":";;AAMA,0DA6BC;AA7BD,SAAgB,uBAAuB,CAAC,IAAY,EAAE,MAAc,EAAE,QAAQ,GAAG,GAAG;IAClF,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxC,MAAM,MAAM,GAAgB,EAAE,CAAC;IAC/B,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,IAAI,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;YAC5D,MAAM,CAAC,IAAI,CAAC;gBACV,EAAE,EAAE,GAAG,MAAM,IAAI,KAAK,EAAE;gBACxB,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE;gBACvB,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;aAC3C,CAAC,CAAC;YACH,OAAO,GAAG,SAAS,CAAC;YACpB,KAAK,EAAE,CAAC;QACV,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC;QACjD,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC;YACV,EAAE,EAAE,GAAG,MAAM,IAAI,KAAK,EAAE;YACxB,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE;YACvB,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;SAC3C,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/rag/rag.service.d.ts

@@ -0,0 +1,20 @@
+import { OnApplicationBootstrap } from '@nestjs/common';
+import { AiChatModuleConfig } from '../types';
+import { ProviderFactory } from '../providers/provider.factory';
+import { TextChunk } from './chunker';
+export declare class RagService implements OnApplicationBootstrap {
+    private readonly config;
+    private readonly providerFactory;
+    private chunks;
+    private loaded;
+    private topK;
+    constructor(config: AiChatModuleConfig, providerFactory: ProviderFactory);
+    onApplicationBootstrap(): Promise<void>;
+    isLoaded(): boolean;
+    load(): Promise<void>;
+    query(text: string): Promise<TextChunk[]>;
+    queryAsContext(text: string): Promise<string | null>;
+    getCorpusAverageEmbedding(): number[] | null;
+    private getEmbedFn;
+}
+//# sourceMappingURL=rag.service.d.ts.map

package/dist/rag/rag.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.service.d.ts","sourceRoot":"","sources":["../../src/rag/rag.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAG5E,OAAO,EAAkB,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAGhE,OAAO,EAA2B,SAAS,EAAE,MAAM,WAAW,CAAC;AAM/D,qBACa,UAAW,YAAW,sBAAsB;IAM7B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAC/C,OAAO,CAAC,QAAQ,CAAC,eAAe;IANlC,OAAO,CAAC,MAAM,CAAsB;IACpC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,IAAI,CAAS;gBAGsB,MAAM,EAAE,kBAAkB,EAClD,eAAe,EAAE,eAAe;IAK7C,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAY7C,QAAQ,IAAI,OAAO;IAIb,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA8CrB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAmBzC,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAM1D,yBAAyB,IAAI,MAAM,EAAE,GAAG,IAAI;IAe5C,OAAO,CAAC,UAAU;CAOnB"}

package/dist/rag/rag.service.js

@@ -0,0 +1,171 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RagService = void 0;
+const common_1 = require("@nestjs/common");
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const types_1 = require("../types");
+const provider_factory_1 = require("../providers/provider.factory");
+const cosine_similarity_1 = require("../utils/cosine-similarity");
+const logger_1 = require("../utils/logger");
+const chunker_1 = require("./chunker");
+let RagService = class RagService {
+    config;
+    providerFactory;
+    chunks = [];
+    loaded = false;
+    topK;
+    constructor(config, providerFactory) {
+        this.config = config;
+        this.providerFactory = providerFactory;
+        this.topK = config.rag?.topK ?? 4;
+    }
+    async onApplicationBootstrap() {
+        if (!this.config.rag)
+            return;
+        try {
+            await this.load();
+        }
+        catch (error) {
+            (0, logger_1.logBootError)(`Failed to load RAG documents: ${error instanceof Error ? error.message : error}`);
+        }
+    }
+    isLoaded() {
+        return this.loaded;
+    }
+    async load() {
+        if (this.loaded)
+            return;
+        const docsPath = this.config.rag?.docsPath || './chatdocs';
+        if (!this.config.rag)
+            return;
+        const resolvedPath = path.resolve(docsPath);
+        if (!fs.existsSync(resolvedPath)) {
+            (0, logger_1.logBootWarn)(`RAG docs path not found: ${resolvedPath}`);
+            return;
+        }
+        const files = fs.readdirSync(resolvedPath).filter((f) => f.endsWith('.md'));
+        if (files.length === 0) {
+            (0, logger_1.logBootWarn)(`No .md files found in ${resolvedPath}`);
+            return;
+        }
+        const allChunks = [];
+        const chunkSize = this.config.rag?.chunkSize ?? 800;
+        for (const file of files) {
+            const content = fs.readFileSync(path.join(resolvedPath, file), 'utf-8');
+            const chunks = (0, chunker_1.splitMarkdownIntoChunks)(content, file, chunkSize);
+            allChunks.push(...chunks);
+        }
+        const embedFn = this.getEmbedFn();
+        if (!embedFn) {
+            (0, logger_1.logBootWarn)('No embedding provider available. RAG disabled.');
+            return;
+        }
+        const batchSize = this.config.rag?.embedBatchSize ?? 50;
+        for (let i = 0; i < allChunks.length; i += batchSize) {
+            const batch = allChunks.slice(i, i + batchSize);
+            const embeddings = await embedFn(batch.map((c) => c.content));
+            for (let j = 0; j < batch.length; j++) {
+                this.chunks.push({ ...batch[j], embedding: embeddings[j] });
+            }
+        }
+        this.loaded = true;
+        (0, logger_1.logRagLoaded)(this.chunks.length, files.length, resolvedPath);
+    }
+    async query(text) {
+        if (!this.loaded || this.chunks.length === 0)
+            return [];
+        const embedFn = this.getEmbedFn();
+        if (!embedFn)
+            return [];
+        const [queryEmbedding] = await embedFn([text]);
+        const scored = this.chunks
+            .map((chunk) => ({
+            chunk,
+            score: (0, cosine_similarity_1.cosineSimilarity)(queryEmbedding, chunk.embedding),
+        }))
+            .sort((a, b) => b.score - a.score)
+            .slice(0, this.topK);
+        return scored.map((s) => s.chunk);
+    }
+    async queryAsContext(text) {
+        const docs = await this.query(text);
+        if (docs.length === 0)
+            return null;
+        return docs.map((d) => d.content).join('\n\n---\n\n');
+    }
+    getCorpusAverageEmbedding() {
+        if (this.chunks.length === 0)
+            return null;
+        const dim = this.chunks[0].embedding.length;
+        const avg = new Array(dim).fill(0);
+        for (const chunk of this.chunks) {
+            for (let i = 0; i < dim; i++) {
+                avg[i] += chunk.embedding[i];
+            }
+        }
+        for (let i = 0; i < dim; i++) {
+            avg[i] /= this.chunks.length;
+        }
+        return avg;
+    }
+    getEmbedFn() {
+        const provider = this.providerFactory.getProvider();
+        if (provider?.embed) {
+            return (texts) => provider.embed(texts);
+        }
+        return null;
+    }
+};
+exports.RagService = RagService;
+exports.RagService = RagService = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)(types_1.AI_CHAT_CONFIG)),
+    __metadata("design:paramtypes", [Object, provider_factory_1.ProviderFactory])
+], RagService);
+//# sourceMappingURL=rag.service.js.map

package/dist/rag/rag.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.service.js","sourceRoot":"","sources":["../../src/rag/rag.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2CAA4E;AAC5E,uCAAyB;AACzB,2CAA6B;AAC7B,oCAA8D;AAC9D,oEAAgE;AAChE,kEAA8D;AAC9D,4CAA0E;AAC1E,uCAA+D;AAOxD,IAAM,UAAU,GAAhB,MAAM,UAAU;IAMsB;IACxB;IANX,MAAM,GAAmB,EAAE,CAAC;IAC5B,MAAM,GAAG,KAAK,CAAC;IACf,IAAI,CAAS;IAErB,YAC2C,MAA0B,EAClD,eAAgC;QADR,WAAM,GAAN,MAAM,CAAoB;QAClD,oBAAe,GAAf,eAAe,CAAiB;QAEjD,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,sBAAsB;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG;YAAE,OAAO;QAE7B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAA,qBAAY,EACV,iCAAiC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,CAClF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,QAAQ,IAAI,YAAY,CAAC;QAC3D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG;YAAE,OAAO;QAE7B,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YACjC,IAAA,oBAAW,EAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5E,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,IAAA,oBAAW,EAAC,yBAAyB,YAAY,EAAE,CAAC,CAAC;YACrD,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAgB,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,SAAS,IAAI,GAAG,CAAC;QAEpD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,IAAA,iCAAuB,EAAC,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;YACjE,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAA,oBAAW,EAAC,gDAAgD,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,cAAc,IAAI,EAAE,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACrD,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAChD,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAC9D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAA,qBAAY,EAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAExD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAExB,MAAM,CAAC,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAE/C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM;aACvB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACf,KAAK;YACL,KAAK,EAAE,IAAA,oCAAgB,EAAC,cAAc,EAAE,KAAK,CAAC,SAAS,CAAC;SACzD,CAAC,CAAC;aACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAEvB,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,IAAY;QAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACnC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACxD,CAAC;IAED,yBAAyB;QACvB,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC;QAC5C,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC7B,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7B,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAC/B,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,UAAU;QAChB,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,WAAW,EAAE,CAAC;QACpD,IAAI,QAAQ,EAAE,KAAK,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAM,CAAC,KAAK,CAAC,CAAC;QAC3C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;CACF,CAAA;AAzHY,gCAAU;qBAAV,UAAU;IADtB,IAAA,mBAAU,GAAE;IAOR,WAAA,IAAA,eAAM,EAAC,sBAAc,CAAC,CAAA;6CACW,kCAAe;GAPxC,UAAU,CAyHtB"}

package/dist/session/session.service.d.ts

@@ -0,0 +1,32 @@
+import { OnModuleDestroy } from '@nestjs/common';
+import { AiChatModuleConfig, ChatMessage } from '../types';
+export declare class SessionLimitError extends Error {
+    constructor(message?: string);
+}
+export declare class InvalidSessionError extends Error {
+    constructor(message?: string);
+}
+export declare class SessionService implements OnModuleDestroy {
+    private readonly config;
+    private readonly sessions;
+    private readonly maxHistory;
+    private readonly ttlMs;
+    private readonly maxTurns;
+    private readonly secret;
+    private readonly allowLegacy;
+    private evictionTimer;
+    constructor(config: AiChatModuleConfig);
+    createSession(): string;
+    resolveToken(token: string): {
+        rawId: string;
+        legacy: boolean;
+    } | null;
+    getHistory(token: string): ChatMessage[];
+    addMessage(token: string, message: ChatMessage): void;
+    deleteSession(token: string): void;
+    hasSession(token: string): boolean;
+    hashToken(token: string): string;
+    private startEviction;
+    onModuleDestroy(): void;
+}
+//# sourceMappingURL=session.service.d.ts.map

package/dist/session/session.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.service.d.ts","sourceRoot":"","sources":["../../src/session/session.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAErE,OAAO,EAAkB,kBAAkB,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAU3E,qBAAa,iBAAkB,SAAQ,KAAK;gBAC9B,OAAO,SAA+B;CAInD;AAED,qBAAa,mBAAoB,SAAQ,KAAK;gBAChC,OAAO,SAAsC;CAI1D;AAED,qBACa,cAAe,YAAW,eAAe;IAShB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAR3D,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAmC;IAC5D,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAS;IAClC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAU;IACtC,OAAO,CAAC,aAAa,CAA+C;gBAEf,MAAM,EAAE,kBAAkB;IAsB/E,aAAa,IAAI,MAAM;IAevB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,CAAA;KAAE,GAAG,IAAI;IAMtE,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,WAAW,EAAE;IASxC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,IAAI;IAwCrD,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAMlC,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAUlC,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAMhC,OAAO,CAAC,aAAa;IAWrB,eAAe,IAAI,IAAI;CAKxB"}