@plumbus/core 0.1.0

package/dist/ai/__tests__/ai-service.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ai-service.test.d.ts.map

package/dist/ai/__tests__/ai-service.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-service.test.d.ts","sourceRoot":"","sources":["../../../src/ai/__tests__/ai-service.test.ts"],"names":[],"mappings":""}

package/dist/ai/__tests__/ai-service.test.js

@@ -0,0 +1,347 @@
+import { describe, expect, it, vi } from 'vitest';
+import { z } from 'zod';
+import { definePrompt } from '../../define/definePrompt.js';
+import { createAIService, singleProviderConfig } from '../ai-service.js';
+import { createCostTracker } from '../cost-tracker.js';
+import { createExplainabilityTracker } from '../explainability.js';
+import { PromptRegistry } from '../prompt-registry.js';
+import { createInMemoryVectorStore, createRAGPipeline } from '../rag/pipeline.js';
+import { createMockProvider } from './provider.test.js';
+describe('AI Service (ctx.ai)', () => {
+    function setupService(opts) {
+        const provider = createMockProvider({
+            complete: vi.fn(async () => ({
+                content: '{"result":"hello"}',
+                model: 'mock-model',
+                usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+                finishReason: 'stop',
+            })),
+            embed: vi.fn(async (req) => ({
+                embeddings: req.texts.map(() => [0.1, 0.2, 0.3]),
+                model: 'mock-embed',
+                usage: { totalTokens: 5 },
+            })),
+        });
+        const promptRegistry = opts?.promptRegistry ? new PromptRegistry() : undefined;
+        const costTracker = opts?.costTracker ? createCostTracker() : undefined;
+        const explainability = opts?.explainability ? createExplainabilityTracker() : undefined;
+        let ragPipeline;
+        if (opts?.rag) {
+            const vectorStore = createInMemoryVectorStore();
+            ragPipeline = createRAGPipeline({ provider, vectorStore });
+        }
+        const service = createAIService(singleProviderConfig(provider, {
+            promptRegistry,
+            costTracker,
+            ragPipeline,
+            defaultModel: 'mock-model',
+            budget: { tenantId: 't1', actor: 'user1' },
+        }));
+        return { service, provider, promptRegistry, costTracker, explainability };
+    }
+    describe('generate', () => {
+        it('generates a response with raw prompt', async () => {
+            const { service } = setupService();
+            const result = await service.generate({
+                prompt: 'Say hello',
+                input: {},
+            });
+            expect(result).toBe('{"result":"hello"}');
+        });
+        it('uses prompt registry when available', async () => {
+            const { service, promptRegistry, provider } = setupService({ promptRegistry: true });
+            promptRegistry?.register(definePrompt({
+                name: 'greet',
+                description: 'Say hello to {{name}}',
+                input: z.object({ name: z.string() }),
+                output: z.object({ result: z.string() }),
+            }));
+            const result = await service.generate({
+                prompt: 'greet',
+                input: { name: 'Alice' },
+            });
+            expect(result).toEqual({ result: 'hello' });
+            expect(provider.complete).toHaveBeenCalled();
+        });
+        it('tracks cost when cost tracker is present', async () => {
+            const { service, costTracker } = setupService({ costTracker: true });
+            await service.generate({ prompt: 'test', input: {} });
+            expect(costTracker?.getRecords()).toHaveLength(1);
+            expect(costTracker?.getRecords()[0]?.operation).toBe('generate');
+        });
+    });
+    describe('extract', () => {
+        it('extracts structured data from text', async () => {
+            const provider = createMockProvider({
+                complete: vi.fn(async () => ({
+                    content: '{"name":"Alice","age":30}',
+                    model: 'mock',
+                    usage: { inputTokens: 20, outputTokens: 10, totalTokens: 30 },
+                    finishReason: 'stop',
+                })),
+            });
+            const service = createAIService(singleProviderConfig(provider));
+            const schema = z.object({ name: z.string(), age: z.number() });
+            const result = await service.extract({ schema, text: 'Alice is 30 years old' });
+            expect(result).toEqual({ name: 'Alice', age: 30 });
+        });
+    });
+    describe('classify', () => {
+        it('classifies text into labels', async () => {
+            const provider = createMockProvider({
+                complete: vi.fn(async () => ({
+                    content: '["urgent","billing"]',
+                    model: 'mock',
+                    usage: { inputTokens: 15, outputTokens: 5, totalTokens: 20 },
+                    finishReason: 'stop',
+                })),
+            });
+            const service = createAIService(singleProviderConfig(provider));
+            const result = await service.classify({
+                labels: ['urgent', 'billing', 'support'],
+                text: 'My payment failed and I need help now!',
+            });
+            expect(result).toEqual(['urgent', 'billing']);
+        });
+        it('filters out invalid labels', async () => {
+            const provider = createMockProvider({
+                complete: vi.fn(async () => ({
+                    content: '["urgent","invalid_label"]',
+                    model: 'mock',
+                    usage: { inputTokens: 10, outputTokens: 5, totalTokens: 15 },
+                    finishReason: 'stop',
+                })),
+            });
+            const service = createAIService(singleProviderConfig(provider));
+            const result = await service.classify({
+                labels: ['urgent', 'billing'],
+                text: 'test',
+            });
+            expect(result).toEqual(['urgent']);
+        });
+    });
+    describe('retrieve', () => {
+        it('throws when RAG pipeline not configured', async () => {
+            const { service } = setupService();
+            await expect(service.retrieve({ query: 'test' })).rejects.toThrow('RAG pipeline not configured');
+        });
+        it('retrieves documents when RAG is configured', async () => {
+            const { service } = setupService({ rag: true });
+            // The in-memory store is empty so results will be empty, but it should not throw
+            const results = await service.retrieve({ query: 'test' });
+            expect(Array.isArray(results)).toBe(true);
+        });
+    });
+    describe('budget enforcement', () => {
+        it('throws when budget is exceeded', async () => {
+            const costTracker = createCostTracker({ dailyCostLimit: 0.001 });
+            costTracker.record({
+                model: 'gpt-4o',
+                provider: 'mock',
+                operation: 'generate',
+                usage: { inputTokens: 1000, outputTokens: 500, totalTokens: 1500 },
+                cost: 0.001,
+                latencyMs: 200,
+            });
+            const provider = createMockProvider();
+            const service = createAIService(singleProviderConfig(provider, {
+                costTracker,
+                budget: { tenantId: 't1' },
+            }));
+            await expect(service.generate({ prompt: 'test', input: {} })).rejects.toThrow('AI budget exceeded');
+        });
+    });
+    describe('multi-provider routing', () => {
+        function createNamedProvider(name, content = '{"result":"ok"}') {
+            return createMockProvider({
+                name,
+                complete: vi.fn(async () => ({
+                    content,
+                    model: `${name}-model`,
+                    usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+                    finishReason: 'stop',
+                })),
+            });
+        }
+        it('routes to the prompt-specified provider', async () => {
+            const openai = createNamedProvider('openai', '{"result":"from-openai"}');
+            const anthropic = createNamedProvider('anthropic', '{"result":"from-anthropic"}');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'write-bio',
+                description: 'Write a bio for {{name}}',
+                input: z.object({ name: z.string() }),
+                output: z.object({ result: z.string() }),
+                model: { name: 'claude-sonnet', provider: 'anthropic' },
+            }));
+            const service = createAIService({
+                providers: { openai, anthropic },
+                defaultProvider: 'openai',
+                promptRegistry,
+            });
+            await service.generate({ prompt: 'write-bio', input: { name: 'Alice' } });
+            expect(anthropic.complete).toHaveBeenCalled();
+            expect(openai.complete).not.toHaveBeenCalled();
+        });
+        it('falls back to default provider when prompt has no provider', async () => {
+            const openai = createNamedProvider('openai');
+            const anthropic = createNamedProvider('anthropic');
+            const service = createAIService({
+                providers: { openai, anthropic },
+                defaultProvider: 'openai',
+            });
+            await service.generate({ prompt: 'raw prompt', input: {} });
+            expect(openai.complete).toHaveBeenCalled();
+            expect(anthropic.complete).not.toHaveBeenCalled();
+        });
+        it('throws when prompt specifies an unknown provider', async () => {
+            const openai = createNamedProvider('openai');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'failing',
+                description: 'This will fail',
+                input: z.object({}),
+                output: z.object({ result: z.string() }),
+                model: { name: 'some-model', provider: 'nonexistent' },
+            }));
+            const service = createAIService({
+                providers: { openai },
+                defaultProvider: 'openai',
+                promptRegistry,
+            });
+            await expect(service.generate({ prompt: 'failing', input: {} })).rejects.toThrow('AI provider "nonexistent" not configured');
+        });
+        it('records provider name in cost tracker entries', async () => {
+            const anthropic = createNamedProvider('anthropic');
+            const costTracker = createCostTracker();
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'tracked',
+                description: 'Tracked prompt',
+                input: z.object({}),
+                output: z.object({ result: z.string() }),
+                model: { name: 'claude-sonnet', provider: 'anthropic' },
+            }));
+            const service = createAIService({
+                providers: { anthropic },
+                defaultProvider: 'anthropic',
+                promptRegistry,
+                costTracker,
+            });
+            await service.generate({ prompt: 'tracked', input: {} });
+            const records = costTracker.getRecords();
+            expect(records).toHaveLength(1);
+            expect(records[0]?.provider).toBe('anthropic');
+        });
+        it('extract and classify use default provider', async () => {
+            const openai = createNamedProvider('openai', '{"name":"Alice","age":30}');
+            const anthropic = createNamedProvider('anthropic');
+            const service = createAIService({
+                providers: { openai, anthropic },
+                defaultProvider: 'openai',
+            });
+            const schema = z.object({ name: z.string(), age: z.number() });
+            await service.extract({ schema, text: 'Alice is 30' });
+            expect(openai.complete).toHaveBeenCalled();
+            expect(anthropic.complete).not.toHaveBeenCalled();
+        });
+    });
+    describe('prompt override resolution', () => {
+        function createNamedProvider(name, content = '{"result":"ok"}') {
+            return createMockProvider({
+                name,
+                complete: vi.fn(async () => ({
+                    content,
+                    model: `${name}-model`,
+                    usage: { inputTokens: 10, outputTokens: 20, totalTokens: 30 },
+                    finishReason: 'stop',
+                })),
+            });
+        }
+        it('uses promptOverrides to route to a different provider', async () => {
+            const openai = createNamedProvider('openai');
+            const anthropic = createNamedProvider('anthropic', '{"result":"from-anthropic"}');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'write-bio',
+                description: 'Write a bio for {{name}}',
+                input: z.object({ name: z.string() }),
+                output: z.object({ result: z.string() }),
+                model: { name: 'gpt-4o', provider: 'openai' },
+            }));
+            const service = createAIService({
+                providers: { openai, anthropic },
+                defaultProvider: 'openai',
+                promptRegistry,
+                promptOverrides: {
+                    'write-bio': { provider: 'anthropic', model: 'claude-sonnet' },
+                },
+            });
+            await service.generate({ prompt: 'write-bio', input: { name: 'Alice' } });
+            expect(anthropic.complete).toHaveBeenCalled();
+            expect(openai.complete).not.toHaveBeenCalled();
+        });
+        it('uses promptOverrides model over prompt-defined model', async () => {
+            const openai = createNamedProvider('openai');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'write-bio',
+                description: 'Write a bio for {{name}}',
+                input: z.object({ name: z.string() }),
+                output: z.object({ result: z.string() }),
+                model: { name: 'gpt-4o', provider: 'openai' },
+            }));
+            const service = createAIService({
+                providers: { openai },
+                defaultProvider: 'openai',
+                promptRegistry,
+                promptOverrides: {
+                    'write-bio': { model: 'gpt-4o-mini' },
+                },
+            });
+            await service.generate({ prompt: 'write-bio', input: { name: 'Alice' } });
+            const call = openai.complete.mock.calls[0]?.[0];
+            expect(call?.model).toBe('gpt-4o-mini');
+        });
+        it('falls back to defaultModel when neither override nor prompt defines model', async () => {
+            const openai = createNamedProvider('openai');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'simple',
+                description: 'A simple prompt',
+                input: z.object({}),
+                output: z.object({ result: z.string() }),
+            }));
+            const service = createAIService({
+                providers: { openai },
+                defaultProvider: 'openai',
+                promptRegistry,
+                defaultModel: 'gpt-4o-mini',
+            });
+            await service.generate({ prompt: 'simple', input: {} });
+            const call = openai.complete.mock.calls[0]?.[0];
+            expect(call?.model).toBe('gpt-4o-mini');
+        });
+        it('uses prompt model when no override is specified', async () => {
+            const openai = createNamedProvider('openai');
+            const promptRegistry = new PromptRegistry();
+            promptRegistry.register(definePrompt({
+                name: 'write-bio',
+                description: 'Write a bio for {{name}}',
+                input: z.object({ name: z.string() }),
+                output: z.object({ result: z.string() }),
+                model: { name: 'gpt-4o', provider: 'openai' },
+            }));
+            const service = createAIService({
+                providers: { openai },
+                defaultProvider: 'openai',
+                promptRegistry,
+                defaultModel: 'fallback-model',
+            });
+            await service.generate({ prompt: 'write-bio', input: { name: 'Alice' } });
+            const call = openai.complete.mock.calls[0]?.[0];
+            expect(call?.model).toBe('gpt-4o');
+        });
+    });
+});
+//# sourceMappingURL=ai-service.test.js.map

package/dist/ai/__tests__/ai-service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-service.test.js","sourceRoot":"","sources":["../../../src/ai/__tests__/ai-service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD,OAAO,EAAE,yBAAyB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAClF,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AAExD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,SAAS,YAAY,CAAC,IAKrB;QACC,MAAM,QAAQ,GAAG,kBAAkB,CAAC;YAClC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC3B,OAAO,EAAE,oBAAoB;gBAC7B,KAAK,EAAE,YAAY;gBACnB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7D,YAAY,EAAE,MAAM;aACrB,CAAC,CAAC;YACH,KAAK,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC3B,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;gBAChD,KAAK,EAAE,YAAY;gBACnB,KAAK,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE;aAC1B,CAAC,CAAC;SACJ,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/E,MAAM,WAAW,GAAG,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QACxE,MAAM,cAAc,GAAG,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAExF,IAAI,WAA6D,CAAC;QAClE,IAAI,IAAI,EAAE,GAAG,EAAE,CAAC;YACd,MAAM,WAAW,GAAG,yBAAyB,EAAE,CAAC;YAChD,WAAW,GAAG,iBAAiB,CAAC,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,OAAO,GAAG,eAAe,CAC7B,oBAAoB,CAAC,QAAQ,EAAE;YAC7B,cAAc;YACd,WAAW;YACX,WAAW;YACX,YAAY,EAAE,YAAY;YAC1B,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE;SAC3C,CAAC,CACH,CAAC;QAEF,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC;IAC5E,CAAC;IAED,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,CAAC;YACnC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;gBACpC,MAAM,EAAE,WAAW;gBACnB,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;YACrF,cAAc,EAAE,QAAQ,CACtB,YAAY,CAAC;gBACX,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,uBAAuB;gBACpC,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACrC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;aACzC,CAAC,CACH,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;gBACpC,MAAM,EAAE,OAAO;gBACf,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,YAAY,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;YACrE,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAEtD,MAAM,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClD,MAAM,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO,EAAE,2BAA2B;oBACpC,KAAK,EAAE,MAAM;oBACb,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;oBAC7D,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;aACJ,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,eAAe,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC;YAChE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAEhF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO,EAAE,sBAAsB;oBAC/B,KAAK,EAAE,MAAM;oBACb,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;oBAC5D,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;aACJ,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,eAAe,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC;YAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;gBACpC,MAAM,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC;gBACxC,IAAI,EAAE,wCAAwC;aAC/C,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,QAAQ,GAAG,kBAAkB,CAAC;gBAClC,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO,EAAE,4BAA4B;oBACrC,KAAK,EAAE,MAAM;oBACb,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE;oBAC5D,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;aACJ,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,eAAe,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC;YAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC;gBACpC,MAAM,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC;gBAC7B,IAAI,EAAE,MAAM;aACb,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,CAAC;YACnC,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC/D,6BAA6B,CAC9B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC;YAEhD,iFAAiF;YACjF,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,WAAW,GAAG,iBAAiB,CAAC,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;YACjE,WAAW,CAAC,MAAM,CAAC;gBACjB,KAAK,EAAE,QAAQ;gBACf,QAAQ,EAAE,MAAM;gBAChB,SAAS,EAAE,UAAU;gBACrB,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,GAAG,EAAE,WAAW,EAAE,IAAI,EAAE;gBAClE,IAAI,EAAE,KAAK;gBACX,SAAS,EAAE,GAAG;aACf,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,kBAAkB,EAAE,CAAC;YACtC,MAAM,OAAO,GAAG,eAAe,CAC7B,oBAAoB,CAAC,QAAQ,EAAE;gBAC7B,WAAW;gBACX,MAAM,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;aAC3B,CAAC,CACH,CAAC;YAEF,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3E,oBAAoB,CACrB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,SAAS,mBAAmB,CAAC,IAAY,EAAE,OAAO,GAAG,iBAAiB;YACpE,OAAO,kBAAkB,CAAC;gBACxB,IAAI;gBACJ,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO;oBACP,KAAK,EAAE,GAAG,IAAI,QAAQ;oBACtB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;oBAC7D,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,EAAE,0BAA0B,CAAC,CAAC;YACzE,MAAM,SAAS,GAAG,mBAAmB,CAAC,WAAW,EAAE,6BAA6B,CAAC,CAAC;YAElF,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,0BAA0B;gBACvC,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACrC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,WAAW,EAAE;aACxD,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;gBAChC,eAAe,EAAE,QAAQ;gBACzB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;gBAChC,eAAe,EAAE,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,gBAAgB;gBAC7B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,QAAQ,EAAE,aAAa,EAAE;aACvD,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE;gBACrB,eAAe,EAAE,QAAQ;gBACzB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC9E,0CAA0C,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,SAAS,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,WAAW,GAAG,iBAAiB,EAAE,CAAC;YAExC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,SAAS;gBACf,WAAW,EAAE,gBAAgB;gBAC7B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,QAAQ,EAAE,WAAW,EAAE;aACxD,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,SAAS,EAAE;gBACxB,eAAe,EAAE,WAAW;gBAC5B,cAAc;gBACd,WAAW;aACZ,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAEzD,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,EAAE,2BAA2B,CAAC,CAAC;YAC1E,MAAM,SAAS,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;gBAChC,eAAe,EAAE,QAAQ;aAC1B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC/D,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3C,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,SAAS,mBAAmB,CAAC,IAAY,EAAE,OAAO,GAAG,iBAAiB;YACpE,OAAO,kBAAkB,CAAC;gBACxB,IAAI;gBACJ,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;oBAC3B,OAAO;oBACP,KAAK,EAAE,GAAG,IAAI,QAAQ;oBACtB,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;oBAC7D,YAAY,EAAE,MAAM;iBACrB,CAAC,CAAC;aACJ,CAAC,CAAC;QACL,CAAC;QAED,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,SAAS,GAAG,mBAAmB,CAAC,WAAW,EAAE,6BAA6B,CAAC,CAAC;YAElF,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,0BAA0B;gBACvC,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACrC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE;aAC9C,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE;gBAChC,eAAe,EAAE,QAAQ;gBACzB,cAAc;gBACd,eAAe,EAAE;oBACf,WAAW,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,eAAe,EAAE;iBAC/D;aACF,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,0BAA0B;gBACvC,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACrC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE;aAC9C,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE;gBACrB,eAAe,EAAE,QAAQ;gBACzB,cAAc;gBACd,eAAe,EAAE;oBACf,WAAW,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE;iBACtC;aACF,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,IAAI,GAAI,MAAM,CAAC,QAAqC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9E,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iBAAiB;gBAC9B,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACnB,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;aACzC,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE;gBACrB,eAAe,EAAE,QAAQ;gBACzB,cAAc;gBACd,YAAY,EAAE,aAAa;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAExD,MAAM,IAAI,GAAI,MAAM,CAAC,QAAqC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9E,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;YAC5C,cAAc,CAAC,QAAQ,CACrB,YAAY,CAAC;gBACX,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,0BAA0B;gBACvC,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACrC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE;aAC9C,CAAC,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,eAAe,CAAC;gBAC9B,SAAS,EAAE,EAAE,MAAM,EAAE;gBACrB,eAAe,EAAE,QAAQ;gBACzB,cAAc;gBACd,YAAY,EAAE,gBAAgB;aAC/B,CAAC,CAAC;YAEH,MAAM,OAAO,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;YAE1E,MAAM,IAAI,GAAI,MAAM,CAAC,QAAqC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC9E,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ai/__tests__/chunking.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=chunking.test.d.ts.map

package/dist/ai/__tests__/chunking.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunking.test.d.ts","sourceRoot":"","sources":["../../../src/ai/__tests__/chunking.test.ts"],"names":[],"mappings":""}

package/dist/ai/__tests__/chunking.test.js

@@ -0,0 +1,44 @@
+import { describe, expect, it } from 'vitest';
+import { chunkDocument } from '../rag/chunking.js';
+describe('Document Chunking', () => {
+    it('returns single chunk for short text', () => {
+        const chunks = chunkDocument('Hello world');
+        expect(chunks).toHaveLength(1);
+        expect(chunks[0]?.content).toBe('Hello world');
+        expect(chunks[0]?.index).toBe(0);
+    });
+    it('splits long text into overlapping chunks', () => {
+        const text = 'A'.repeat(2500);
+        const chunks = chunkDocument(text, { maxChunkSize: 1000, overlap: 200 });
+        expect(chunks.length).toBeGreaterThan(1);
+        // First chunk should be maxChunkSize
+        expect(chunks[0]?.content.length).toBe(1000);
+        // Chunks should have sequential indexes
+        for (let i = 0; i < chunks.length; i++) {
+            expect(chunks[i]?.index).toBe(i);
+        }
+    });
+    it('handles paragraph strategy', () => {
+        const text = 'Paragraph one.\n\nParagraph two.\n\nParagraph three.';
+        const chunks = chunkDocument(text, { strategy: 'paragraph', maxChunkSize: 100 });
+        expect(chunks.length).toBeGreaterThanOrEqual(1);
+        // All content should be preserved somewhere
+        const combined = chunks.map((c) => c.content).join('\n\n');
+        expect(combined).toContain('Paragraph one');
+        expect(combined).toContain('Paragraph three');
+    });
+    it('merges small paragraphs up to maxChunkSize', () => {
+        const text = 'A\n\nB\n\nC';
+        const chunks = chunkDocument(text, { strategy: 'paragraph', maxChunkSize: 1000 });
+        // Should merge all into one chunk since all fit
+        expect(chunks).toHaveLength(1);
+        expect(chunks[0]?.content).toContain('A');
+        expect(chunks[0]?.content).toContain('C');
+    });
+    it('splits paragraphs that exceed maxChunkSize', () => {
+        const text = `${'A'.repeat(50)}\n\n${'B'.repeat(60)}`;
+        const chunks = chunkDocument(text, { strategy: 'paragraph', maxChunkSize: 80 });
+        expect(chunks).toHaveLength(2);
+    });
+});
+//# sourceMappingURL=chunking.test.js.map

package/dist/ai/__tests__/chunking.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunking.test.js","sourceRoot":"","sources":["../../../src/ai/__tests__/chunking.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,MAAM,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAEzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACzC,qCAAqC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,wCAAwC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,IAAI,GAAG,sDAAsD,CAAC;QACpE,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;QAEjF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAChD,4CAA4C;QAC5C,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3D,MAAM,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC5C,MAAM,CAAC,QAAQ,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,aAAa,CAAC;QAC3B,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC;QAElF,gDAAgD;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,IAAI,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACtD,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAEhF,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/ai/__tests__/cost-tracker.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=cost-tracker.test.d.ts.map

package/dist/ai/__tests__/cost-tracker.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-tracker.test.d.ts","sourceRoot":"","sources":["../../../src/ai/__tests__/cost-tracker.test.ts"],"names":[],"mappings":""}

package/dist/ai/__tests__/cost-tracker.test.js

@@ -0,0 +1,201 @@
+import { describe, expect, it, vi } from 'vitest';
+import { createCostTracker } from '../cost-tracker.js';
+import { UsageAPIError } from '../usage-client.js';
+describe('Cost Tracker', () => {
+    describe('createCostTracker', () => {
+        it('records and retrieves cost entries', () => {
+            const tracker = createCostTracker();
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: null,
+                latencyMs: 500,
+            });
+            const records = tracker.getRecords();
+            expect(records).toHaveLength(1);
+            expect(records[0]?.model).toBe('gpt-4o');
+            expect(records[0]?.cost).toBeNull();
+            expect(records[0]?.id).toBeDefined();
+            expect(records[0]?.timestamp).toBeInstanceOf(Date);
+        });
+        it('records entries with actual cost from usage API', () => {
+            const tracker = createCostTracker();
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: 0.001,
+                latencyMs: 500,
+            });
+            const records = tracker.getRecords();
+            expect(records[0]?.cost).toBe(0.001);
+        });
+        it('tracks daily usage with cost available', () => {
+            const tracker = createCostTracker();
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: 0.001,
+                latencyMs: 500,
+            });
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'extract',
+                usage: { inputTokens: 200, outputTokens: 100, totalTokens: 300 },
+                cost: 0.002,
+                latencyMs: 300,
+            });
+            const daily = tracker.getDailyUsage();
+            expect(daily.requestCount).toBe(2);
+            expect(daily.totalTokens).toBe(450);
+            expect(daily.totalCost).toBeCloseTo(0.003, 4);
+            expect(daily.costAvailable).toBe(true);
+        });
+        it('reports costAvailable=false when no cost data', () => {
+            const tracker = createCostTracker();
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: null,
+                latencyMs: 500,
+            });
+            const daily = tracker.getDailyUsage();
+            expect(daily.costAvailable).toBe(false);
+            expect(daily.totalCost).toBe(0);
+        });
+        it('filters daily usage by tenant', () => {
+            const tracker = createCostTracker();
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: null,
+                latencyMs: 500,
+                tenantId: 't1',
+            });
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: null,
+                latencyMs: 500,
+                tenantId: 't2',
+            });
+            expect(tracker.getDailyUsage('t1').requestCount).toBe(1);
+            expect(tracker.getDailyUsage('t2').requestCount).toBe(1);
+        });
+        it('enforces max tokens per request', () => {
+            const budget = { maxTokensPerRequest: 1000 };
+            const tracker = createCostTracker(budget);
+            expect(tracker.checkBudget({ estimatedTokens: 500 }).allowed).toBe(true);
+            expect(tracker.checkBudget({ estimatedTokens: 1500 }).allowed).toBe(false);
+            expect(tracker.checkBudget({ estimatedTokens: 1500 }).reason).toContain('per-request limit');
+        });
+        it('enforces daily cost limit when cost data is available', () => {
+            const budget = { dailyCostLimit: 0.01 };
+            const tracker = createCostTracker(budget);
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: 0.01,
+                latencyMs: 200,
+            });
+            const result = tracker.checkBudget({});
+            expect(result.allowed).toBe(false);
+            expect(result.reason).toContain('Daily cost limit');
+        });
+        it('allows but warns when cost data unavailable for daily budget', () => {
+            const budget = { dailyCostLimit: 0.01 };
+            const tracker = createCostTracker(budget);
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: null,
+                latencyMs: 200,
+            });
+            const result = tracker.checkBudget({});
+            expect(result.allowed).toBe(true);
+            expect(result.reason).toContain('Cost data unavailable');
+        });
+        it('enforces per-tenant daily limit', () => {
+            const budget = { perTenantDailyLimit: 0.005 };
+            const tracker = createCostTracker(budget);
+            tracker.record({
+                model: 'gpt-4o',
+                provider: 'openai',
+                operation: 'generate',
+                usage: { inputTokens: 100, outputTokens: 50, totalTokens: 150 },
+                cost: 0.005,
+                latencyMs: 200,
+                tenantId: 't1',
+            });
+            expect(tracker.checkBudget({ tenantId: 't1' }).allowed).toBe(false);
+            expect(tracker.checkBudget({ tenantId: 't2' }).allowed).toBe(true);
+        });
+        it('allows when under budget', () => {
+            const budget = {
+                maxTokensPerRequest: 5000,
+                dailyCostLimit: 100,
+                perTenantDailyLimit: 50,
+            };
+            const tracker = createCostTracker(budget);
+            const result = tracker.checkBudget({
+                tenantId: 't1',
+                estimatedTokens: 1000,
+            });
+            expect(result.allowed).toBe(true);
+        });
+    });
+    describe('syncCosts', () => {
+        it('returns error when no usage clients configured', async () => {
+            const tracker = createCostTracker();
+            const result = await tracker.syncCosts?.();
+            expect(result?.synced).toBe(false);
+            expect(result?.error).toContain('No usage API clients');
+        });
+        it('fetches costs from usage API client', async () => {
+            const mockClient = {
+                provider: 'openai',
+                fetchUsage: vi.fn(async () => ({
+                    totalCost: 1.5,
+                    currency: 'usd',
+                    entries: [
+                        { model: 'gpt-4o', inputTokens: 1000, outputTokens: 500, cost: 1.5, requestCount: 10 },
+                    ],
+                })),
+            };
+            const tracker = createCostTracker(undefined, [mockClient]);
+            const result = await tracker.syncCosts?.();
+            expect(result?.synced).toBe(true);
+            expect(result?.totalCost).toBe(1.5);
+            expect(mockClient.fetchUsage).toHaveBeenCalled();
+        });
+        it('handles usage API errors gracefully', async () => {
+            const failingClient = {
+                provider: 'openai',
+                fetchUsage: vi.fn(async () => {
+                    throw new UsageAPIError('openai', 'OpenAI Usage API returned 401: Unauthorized');
+                }),
+            };
+            const tracker = createCostTracker(undefined, [failingClient]);
+            const result = await tracker.syncCosts?.();
+            expect(result?.synced).toBe(false);
+            expect(result?.error).toContain('401');
+        });
+    });
+});
+//# sourceMappingURL=cost-tracker.test.js.map