@plumbus/core 0.1.0

package/dist/ai/provider.js

@@ -0,0 +1,322 @@
+// ── AI Provider Adapter Interface ──
+// Abstract interface for AI provider adapters (OpenAI, Anthropic, etc.)
+export function createOpenAIAdapter(config) {
+    const defaultModel = config.model ?? 'gpt-4o';
+    const defaultEmbeddingModel = config.embeddingModel ?? 'text-embedding-3-small';
+    const baseUrl = config.baseUrl ?? 'https://api.openai.com/v1';
+    return {
+        name: 'openai',
+        async complete(request) {
+            const messages = [];
+            if (request.system) {
+                messages.push({ role: 'system', content: request.system });
+            }
+            messages.push({ role: 'user', content: request.prompt });
+            const body = {
+                model: request.model ?? defaultModel,
+                messages,
+                temperature: request.temperature ?? 0.7,
+            };
+            if (request.maxTokens)
+                body.max_tokens = request.maxTokens;
+            if (request.responseFormat === 'json') {
+                body.response_format = { type: 'json_object' };
+            }
+            const resp = await fetch(`${baseUrl}/chat/completions`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${config.apiKey}`,
+                },
+                body: JSON.stringify(body),
+            });
+            if (!resp.ok) {
+                const text = await resp.text();
+                throw new Error(`OpenAI API error (${resp.status}): ${text}`);
+            }
+            const data = (await resp.json());
+            const choice = data.choices[0];
+            if (!choice)
+                throw new Error('OpenAI returned no choices');
+            return {
+                content: choice.message.content,
+                model: data.model,
+                usage: {
+                    inputTokens: data.usage.prompt_tokens,
+                    outputTokens: data.usage.completion_tokens,
+                    totalTokens: data.usage.total_tokens,
+                },
+                finishReason: choice.finish_reason,
+            };
+        },
+        async *stream(request) {
+            const messages = [];
+            if (request.system) {
+                messages.push({ role: 'system', content: request.system });
+            }
+            messages.push({ role: 'user', content: request.prompt });
+            const body = {
+                model: request.model ?? defaultModel,
+                messages,
+                temperature: request.temperature ?? 0.7,
+                stream: true,
+                stream_options: { include_usage: true },
+            };
+            if (request.maxTokens)
+                body.max_tokens = request.maxTokens;
+            if (request.responseFormat === 'json') {
+                body.response_format = { type: 'json_object' };
+            }
+            const resp = await fetch(`${baseUrl}/chat/completions`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${config.apiKey}`,
+                },
+                body: JSON.stringify(body),
+            });
+            if (!resp.ok) {
+                const text = await resp.text();
+                yield { type: 'error', error: `OpenAI API error (${resp.status}): ${text}` };
+                return;
+            }
+            yield* parseSSEStream(resp, parseOpenAISSEChunk);
+        },
+        async embed(request) {
+            const body = {
+                model: request.model ?? defaultEmbeddingModel,
+                input: request.texts,
+            };
+            const resp = await fetch(`${baseUrl}/embeddings`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${config.apiKey}`,
+                },
+                body: JSON.stringify(body),
+            });
+            if (!resp.ok) {
+                const text = await resp.text();
+                throw new Error(`OpenAI Embedding API error (${resp.status}): ${text}`);
+            }
+            const data = (await resp.json());
+            return {
+                embeddings: data.data.map((d) => d.embedding),
+                model: data.model,
+                usage: { totalTokens: data.usage.total_tokens },
+            };
+        },
+    };
+}
+export function createAnthropicAdapter(config) {
+    const defaultModel = config.model ?? 'claude-sonnet-4-20250514';
+    const baseUrl = config.baseUrl ?? 'https://api.anthropic.com/v1';
+    return {
+        name: 'anthropic',
+        async complete(request) {
+            const body = {
+                model: request.model ?? defaultModel,
+                messages: [{ role: 'user', content: request.prompt }],
+                max_tokens: request.maxTokens ?? 4096,
+                temperature: request.temperature ?? 0.7,
+            };
+            if (request.system)
+                body.system = request.system;
+            const resp = await fetch(`${baseUrl}/messages`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'x-api-key': config.apiKey,
+                    'anthropic-version': '2023-06-01',
+                },
+                body: JSON.stringify(body),
+            });
+            if (!resp.ok) {
+                const text = await resp.text();
+                throw new Error(`Anthropic API error (${resp.status}): ${text}`);
+            }
+            const data = (await resp.json());
+            const text = data.content
+                .filter((c) => c.type === 'text')
+                .map((c) => c.text)
+                .join('');
+            return {
+                content: text,
+                model: data.model,
+                usage: {
+                    inputTokens: data.usage.input_tokens,
+                    outputTokens: data.usage.output_tokens,
+                    totalTokens: data.usage.input_tokens + data.usage.output_tokens,
+                },
+                finishReason: data.stop_reason,
+            };
+        },
+        async *stream(request) {
+            const body = {
+                model: request.model ?? defaultModel,
+                messages: [{ role: 'user', content: request.prompt }],
+                max_tokens: request.maxTokens ?? 4096,
+                temperature: request.temperature ?? 0.7,
+                stream: true,
+            };
+            if (request.system)
+                body.system = request.system;
+            const resp = await fetch(`${baseUrl}/messages`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    'x-api-key': config.apiKey,
+                    'anthropic-version': '2023-06-01',
+                },
+                body: JSON.stringify(body),
+            });
+            if (!resp.ok) {
+                const text = await resp.text();
+                yield { type: 'error', error: `Anthropic API error (${resp.status}): ${text}` };
+                return;
+            }
+            yield* parseSSEStream(resp, parseAnthropicSSEChunk);
+        },
+        async embed() {
+            // Anthropic doesn't have a public embedding API
+            // Users should pair with an embedding-capable provider for RAG
+            throw new Error('Anthropic does not provide an embedding API. Use an OpenAI-compatible provider for embeddings.');
+        },
+    };
+}
+async function* parseSSEStream(resp, parseChunk) {
+    if (!resp.body) {
+        yield { type: 'error', error: 'No response body for streaming' };
+        return;
+    }
+    const reader = resp.body.getReader();
+    const decoder = new TextDecoder();
+    let buffer = '';
+    let currentEvent = '';
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split('\n');
+            buffer = lines.pop() ?? ''; // Keep incomplete line in buffer
+            for (const line of lines) {
+                if (line.startsWith('event: ')) {
+                    currentEvent = line.slice(7).trim();
+                    continue;
+                }
+                if (line.startsWith('data: ')) {
+                    const data = line.slice(6);
+                    if (data === '[DONE]') {
+                        yield { type: 'done' };
+                        return;
+                    }
+                    const event = parseChunk(currentEvent, data);
+                    if (event)
+                        yield event;
+                }
+            }
+        }
+    }
+    finally {
+        reader.releaseLock();
+    }
+}
+function parseOpenAISSEChunk(_eventType, data) {
+    try {
+        const parsed = JSON.parse(data);
+        // Usage-only chunk (sent when stream_options.include_usage is true)
+        if (parsed.usage &&
+            (!parsed.choices || parsed.choices.length === 0 || !parsed.choices[0]?.delta?.content)) {
+            return {
+                type: 'usage',
+                usage: {
+                    inputTokens: parsed.usage.prompt_tokens,
+                    outputTokens: parsed.usage.completion_tokens,
+                    totalTokens: parsed.usage.total_tokens,
+                },
+            };
+        }
+        const choice = parsed.choices?.[0];
+        if (!choice)
+            return null;
+        if (choice.finish_reason) {
+            return { type: 'done', finishReason: choice.finish_reason };
+        }
+        if (choice.delta?.content) {
+            return { type: 'content_delta', delta: choice.delta.content };
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+}
+function parseAnthropicSSEChunk(eventType, data) {
+    try {
+        const parsed = JSON.parse(data);
+        switch (eventType) {
+            case 'content_block_delta': {
+                const delta = parsed.delta;
+                if (delta?.type === 'text_delta' && delta.text) {
+                    return { type: 'content_delta', delta: delta.text };
+                }
+                return null;
+            }
+            case 'message_delta': {
+                const delta = parsed.delta;
+                const usage = parsed.usage;
+                return {
+                    type: 'done',
+                    finishReason: delta?.stop_reason ?? 'end_turn',
+                    usage: usage?.output_tokens != null
+                        ? {
+                            inputTokens: 0,
+                            outputTokens: usage.output_tokens,
+                            totalTokens: usage.output_tokens,
+                        }
+                        : undefined,
+                };
+            }
+            case 'message_start': {
+                const message = parsed.message;
+                if (message?.usage) {
+                    return {
+                        type: 'usage',
+                        usage: {
+                            inputTokens: message.usage.input_tokens,
+                            outputTokens: message.usage.output_tokens,
+                            totalTokens: message.usage.input_tokens + message.usage.output_tokens,
+                        },
+                    };
+                }
+                return null;
+            }
+            default:
+                return null;
+        }
+    }
+    catch {
+        return null;
+    }
+}
+// ── Provider Factory ──
+/** Create a provider adapter by name from a provider config entry */
+export function createProviderAdapter(name, providerConfig) {
+    const cfg = {
+        apiKey: providerConfig.apiKey,
+        model: providerConfig.model,
+        baseUrl: providerConfig.baseUrl,
+    };
+    switch (name) {
+        case 'openai':
+            return createOpenAIAdapter(cfg);
+        case 'anthropic':
+            return createAnthropicAdapter(cfg);
+        default:
+            // Unknown providers: try OpenAI-compat adapter (covers Ollama, Azure, etc.)
+            return createOpenAIAdapter(cfg);
+    }
+}
+//# sourceMappingURL=provider.js.map

package/dist/ai/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../../src/ai/provider.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,wEAAwE;AAmFxE,MAAM,UAAU,mBAAmB,CAAC,MAA2B;IAC7D,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,IAAI,QAAQ,CAAC;IAC9C,MAAM,qBAAqB,GAAG,MAAM,CAAC,cAAc,IAAI,wBAAwB,CAAC;IAChF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,2BAA2B,CAAC;IAE9D,OAAO;QACL,IAAI,EAAE,QAAQ;QAEd,KAAK,CAAC,QAAQ,CAAC,OAAwB;YACrC,MAAM,QAAQ,GAA6C,EAAE,CAAC;YAC9D,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAEzD,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY;gBACpC,QAAQ;gBACR,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;aACxC,CAAC;YACF,IAAI,OAAO,CAAC,SAAS;gBAAE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;YAC3D,IAAI,OAAO,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;gBACtC,IAAI,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YACjD,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,MAAM,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/B,MAAM,IAAI,KAAK,CAAC,qBAAqB,IAAI,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;YAChE,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAI9B,CAAC;YAEF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAC/B,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAC3D,OAAO;gBACL,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;gBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE;oBACL,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;oBACrC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,iBAAiB;oBAC1C,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;iBACrC;gBACD,YAAY,EAAE,MAAM,CAAC,aAAa;aACnC,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAwB;YACpC,MAAM,QAAQ,GAA6C,EAAE,CAAC;YAC9D,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAEzD,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY;gBACpC,QAAQ;gBACR,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;gBACvC,MAAM,EAAE,IAAI;gBACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;aACxC,CAAC;YACF,IAAI,OAAO,CAAC,SAAS;gBAAE,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;YAC3D,IAAI,OAAO,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;gBACtC,IAAI,CAAC,eAAe,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;YACjD,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,MAAM,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,qBAAqB,IAAI,CAAC,MAAM,MAAM,IAAI,EAAE,EAAE,CAAC;gBAC7E,OAAO;YACT,CAAC;YAED,KAAK,CAAC,CAAC,cAAc,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;QACnD,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,OAAyB;YACnC,MAAM,IAAI,GAAG;gBACX,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,qBAAqB;gBAC7C,KAAK,EAAE,OAAO,CAAC,KAAK;aACrB,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,aAAa,EAAE;gBAChD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,MAAM,CAAC,MAAM,EAAE;iBACzC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/B,MAAM,IAAI,KAAK,CAAC,+BAA+B,IAAI,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;YAC1E,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAI9B,CAAC;YAEF,OAAO;gBACL,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE;aAChD,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AASD,MAAM,UAAU,sBAAsB,CAAC,MAA8B;IACnE,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,IAAI,0BAA0B,CAAC;IAChE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,8BAA8B,CAAC;IAEjE,OAAO;QACL,IAAI,EAAE,WAAW;QAEjB,KAAK,CAAC,QAAQ,CAAC,OAAwB;YACrC,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY;gBACpC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrD,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;gBACrC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;aACxC,CAAC;YACF,IAAI,OAAO,CAAC,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,WAAW,EAAE;gBAC9C,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/B,MAAM,IAAI,KAAK,CAAC,wBAAwB,IAAI,CAAC,MAAM,MAAM,IAAI,EAAE,CAAC,CAAC;YACnE,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAK9B,CAAC;YAEF,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO;iBACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,EAAE,CAAC,CAAC;YAEZ,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE;oBACL,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;oBACpC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,aAAa;oBACtC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa;iBAChE;gBACD,YAAY,EAAE,IAAI,CAAC,WAAW;aAC/B,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAwB;YACpC,MAAM,IAAI,GAA4B;gBACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY;gBACpC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBACrD,UAAU,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;gBACrC,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;gBACvC,MAAM,EAAE,IAAI;aACb,CAAC;YACF,IAAI,OAAO,CAAC,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,WAAW,EAAE;gBAC9C,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM,CAAC,MAAM;oBAC1B,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC/B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,wBAAwB,IAAI,CAAC,MAAM,MAAM,IAAI,EAAE,EAAE,CAAC;gBAChF,OAAO;YACT,CAAC;YAED,KAAK,CAAC,CAAC,cAAc,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;QACtD,CAAC;QAED,KAAK,CAAC,KAAK;YACT,gDAAgD;YAChD,+DAA+D;YAC/D,MAAM,IAAI,KAAK,CACb,gGAAgG,CACjG,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAMD,KAAK,SAAS,CAAC,CAAC,cAAc,CAC5B,IAAc,EACd,UAA0B;IAE1B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gCAAgC,EAAE,CAAC;QACjE,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,YAAY,GAAG,EAAE,CAAC;IAEtB,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,iCAAiC;YAE7D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC/B,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;oBACpC,SAAS;gBACX,CAAC;gBAED,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC3B,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;wBACtB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;wBACvB,OAAO;oBACT,CAAC;oBAED,MAAM,KAAK,GAAG,UAAU,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;oBAC7C,IAAI,KAAK;wBAAE,MAAM,KAAK,CAAC;gBACzB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,UAAkB,EAAE,IAAY;IAC3D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAG7B,CAAC;QAEF,oEAAoE;QACpE,IACE,MAAM,CAAC,KAAK;YACZ,CAAC,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,EACtF,CAAC;YACD,OAAO;gBACL,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa;oBACvC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,iBAAiB;oBAC5C,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;iBACvC;aACF,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAEzB,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC;QAC9D,CAAC;QAED,IAAI,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;YAC1B,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;QAChE,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,sBAAsB,CAAC,SAAiB,EAAE,IAAY;IAC7D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B,CAAC;QAE3D,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,qBAAqB,CAAC,CAAC,CAAC;gBAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAqD,CAAC;gBAC3E,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;oBAC/C,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;gBACtD,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,KAAK,GAAG,MAAM,CAAC,KAA6C,CAAC;gBACnE,MAAM,KAAK,GAAG,MAAM,CAAC,KAA+C,CAAC;gBACrE,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,YAAY,EAAE,KAAK,EAAE,WAAW,IAAI,UAAU;oBAC9C,KAAK,EACH,KAAK,EAAE,aAAa,IAAI,IAAI;wBAC1B,CAAC,CAAC;4BACE,WAAW,EAAE,CAAC;4BACd,YAAY,EAAE,KAAK,CAAC,aAAa;4BACjC,WAAW,EAAE,KAAK,CAAC,aAAa;yBACjC;wBACH,CAAC,CAAC,SAAS;iBAChB,CAAC;YACJ,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,OAAO,GAAG,MAAM,CAAC,OAEV,CAAC;gBACd,IAAI,OAAO,EAAE,KAAK,EAAE,CAAC;oBACnB,OAAO;wBACL,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE;4BACL,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,YAAY;4BACvC,YAAY,EAAE,OAAO,CAAC,KAAK,CAAC,aAAa;4BACzC,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,aAAa;yBACtE;qBACF,CAAC;gBACJ,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;YACD;gBACE,OAAO,IAAI,CAAC;QAChB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,yBAAyB;AAEzB,qEAAqE;AACrE,MAAM,UAAU,qBAAqB,CACnC,IAAY,EACZ,cAAgC;IAEhC,MAAM,GAAG,GAAG;QACV,MAAM,EAAE,cAAc,CAAC,MAAM;QAC7B,KAAK,EAAE,cAAc,CAAC,KAAK;QAC3B,OAAO,EAAE,cAAc,CAAC,OAAO;KAChC,CAAC;IAEF,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,OAAO,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAClC,KAAK,WAAW;YACd,OAAO,sBAAsB,CAAC,GAAG,CAAC,CAAC;QACrC;YACE,4EAA4E;YAC5E,OAAO,mBAAmB,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC;AACH,CAAC"}

package/dist/ai/rag/chunking.d.ts

@@ -0,0 +1,18 @@
+export interface ChunkConfig {
+    /** Maximum chunk size in characters (default: 1000) */
+    maxChunkSize?: number;
+    /** Overlap between chunks in characters (default: 200) */
+    overlap?: number;
+    /** Chunking strategy */
+    strategy?: 'size' | 'paragraph';
+}
+export interface DocumentChunk {
+    content: string;
+    index: number;
+    metadata?: Record<string, unknown>;
+}
+/**
+ * Split text into overlapping chunks
+ */
+export declare function chunkDocument(text: string, config?: ChunkConfig): DocumentChunk[];
+//# sourceMappingURL=chunking.d.ts.map

package/dist/ai/rag/chunking.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunking.d.ts","sourceRoot":"","sources":["../../../src/ai/rag/chunking.ts"],"names":[],"mappings":"AAGA,MAAM,WAAW,WAAW;IAC1B,uDAAuD;IACvD,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,0DAA0D;IAC1D,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,wBAAwB;IACxB,QAAQ,CAAC,EAAE,MAAM,GAAG,WAAW,CAAC;CACjC;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,aAAa,EAAE,CAOjF"}

package/dist/ai/rag/chunking.js

@@ -0,0 +1,58 @@
+// ── Document Chunking ──
+// Split documents into chunks for embedding and retrieval
+/**
+ * Split text into overlapping chunks
+ */
+export function chunkDocument(text, config) {
+    const strategy = config?.strategy ?? 'size';
+    if (strategy === 'paragraph') {
+        return chunkByParagraph(text, config);
+    }
+    return chunkBySize(text, config);
+}
+function chunkBySize(text, config) {
+    const maxSize = config?.maxChunkSize ?? 1000;
+    const overlap = config?.overlap ?? 200;
+    const chunks = [];
+    if (text.length <= maxSize) {
+        return [{ content: text, index: 0 }];
+    }
+    let start = 0;
+    let idx = 0;
+    while (start < text.length) {
+        const end = Math.min(start + maxSize, text.length);
+        chunks.push({ content: text.slice(start, end), index: idx });
+        idx++;
+        if (end === text.length)
+            break;
+        start = end - overlap;
+        // Avoid tiny trailing chunks
+        if (text.length - start <= overlap) {
+            chunks.push({ content: text.slice(start), index: idx });
+            break;
+        }
+    }
+    return chunks;
+}
+function chunkByParagraph(text, config) {
+    const maxSize = config?.maxChunkSize ?? 1000;
+    const paragraphs = text.split(/\n\s*\n/).filter((p) => p.trim().length > 0);
+    const chunks = [];
+    let current = '';
+    let idx = 0;
+    for (const para of paragraphs) {
+        if (current.length + para.length + 1 > maxSize && current.length > 0) {
+            chunks.push({ content: current.trim(), index: idx });
+            idx++;
+            current = para;
+        }
+        else {
+            current += (current ? '\n\n' : '') + para;
+        }
+    }
+    if (current.trim().length > 0) {
+        chunks.push({ content: current.trim(), index: idx });
+    }
+    return chunks;
+}
+//# sourceMappingURL=chunking.js.map

package/dist/ai/rag/chunking.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chunking.js","sourceRoot":"","sources":["../../../src/ai/rag/chunking.ts"],"names":[],"mappings":"AAAA,0BAA0B;AAC1B,0DAA0D;AAiB1D;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,IAAY,EAAE,MAAoB;IAC9D,MAAM,QAAQ,GAAG,MAAM,EAAE,QAAQ,IAAI,MAAM,CAAC;IAE5C,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO,gBAAgB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,WAAW,CAAC,IAAY,EAAE,MAAoB;IACrD,MAAM,OAAO,GAAG,MAAM,EAAE,YAAY,IAAI,IAAI,CAAC;IAC7C,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,IAAI,GAAG,CAAC;IACvC,MAAM,MAAM,GAAoB,EAAE,CAAC;IAEnC,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,EAAE,CAAC;QAC3B,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,OAAO,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,GAAG,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7D,GAAG,EAAE,CAAC;QACN,IAAI,GAAG,KAAK,IAAI,CAAC,MAAM;YAAE,MAAM;QAC/B,KAAK,GAAG,GAAG,GAAG,OAAO,CAAC;QACtB,6BAA6B;QAC7B,IAAI,IAAI,CAAC,MAAM,GAAG,KAAK,IAAI,OAAO,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;YACxD,MAAM;QACR,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY,EAAE,MAAoB;IAC1D,MAAM,OAAO,GAAG,MAAM,EAAE,YAAY,IAAI,IAAI,CAAC;IAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAC5E,MAAM,MAAM,GAAoB,EAAE,CAAC;IAEnC,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,GAAG,GAAG,CAAC,CAAC;IAEZ,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrE,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;YACrD,GAAG,EAAE,CAAC;YACN,OAAO,GAAG,IAAI,CAAC;QACjB,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;QAC5C,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/ai/rag/index.d.ts

@@ -0,0 +1,4 @@
+export { chunkDocument, type ChunkConfig, type DocumentChunk } from './chunking.js';
+export { createInMemoryVectorStore, createRAGPipeline, type IngestDocumentInput, type RAGPipeline, type RAGPipelineConfig, type RetrievalQuery, type StoredChunk, type VectorStore, } from './pipeline.js';
+export { documentChunksTable, documentsTable } from './schema.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/ai/rag/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/ai/rag/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,KAAK,WAAW,EAAE,KAAK,aAAa,EAAE,MAAM,eAAe,CAAC;AACpF,OAAO,EACL,yBAAyB,EACzB,iBAAiB,EACjB,KAAK,mBAAmB,EACxB,KAAK,WAAW,EAChB,KAAK,iBAAiB,EACtB,KAAK,cAAc,EACnB,KAAK,WAAW,EAChB,KAAK,WAAW,GACjB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/dist/ai/rag/index.js

@@ -0,0 +1,5 @@
+// ── RAG Barrel Export ──
+export { chunkDocument } from './chunking.js';
+export { createInMemoryVectorStore, createRAGPipeline, } from './pipeline.js';
+export { documentChunksTable, documentsTable } from './schema.js';
+//# sourceMappingURL=index.js.map

package/dist/ai/rag/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/ai/rag/index.ts"],"names":[],"mappings":"AAAA,0BAA0B;AAC1B,OAAO,EAAE,aAAa,EAAwC,MAAM,eAAe,CAAC;AACpF,OAAO,EACL,yBAAyB,EACzB,iBAAiB,GAOlB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/dist/ai/rag/pipeline.d.ts

@@ -0,0 +1,69 @@
+import type { AIDocument } from '../../types/context.js';
+import type { AIProviderAdapter } from '../provider.js';
+import { type ChunkConfig } from './chunking.js';
+export interface IngestDocumentInput {
+    /** Unique document identifier */
+    documentId: string;
+    /** Raw text content */
+    content: string;
+    /** Source reference (URL, file path, etc.) */
+    source: string;
+    /** Tenant isolation */
+    tenantId?: string;
+    /** Data classification level */
+    classification?: string;
+    /** Additional metadata */
+    metadata?: Record<string, unknown>;
+}
+export interface StoredChunk {
+    id: string;
+    documentId: string;
+    content: string;
+    chunkIndex: number;
+    embedding: number[];
+    source?: string;
+    tenantId?: string;
+    classification?: string;
+    metadata?: Record<string, unknown>;
+}
+export interface RetrievalQuery {
+    query: string;
+    tenantId?: string;
+    /** Max classification level allowed (default: no filter) */
+    maxClassification?: string;
+    /** Max results (default: 5) */
+    limit?: number;
+    /** Minimum similarity score (default: 0.0) */
+    minScore?: number;
+}
+export interface VectorStore {
+    insert(chunks: StoredChunk[]): Promise<void>;
+    search(embedding: number[], options: {
+        tenantId?: string;
+        maxClassification?: string;
+        limit: number;
+        minScore: number;
+    }): Promise<Array<StoredChunk & {
+        score: number;
+    }>>;
+    deleteByDocumentId(documentId: string): Promise<void>;
+}
+export interface RAGPipeline {
+    ingest(input: IngestDocumentInput): Promise<{
+        documentId: string;
+        chunkCount: number;
+    }>;
+    retrieve(query: RetrievalQuery): Promise<AIDocument[]>;
+    deleteDocument(documentId: string): Promise<void>;
+}
+export interface RAGPipelineConfig {
+    provider: AIProviderAdapter;
+    vectorStore: VectorStore;
+    chunkConfig?: ChunkConfig;
+    embeddingModel?: string;
+    /** Max chunks to embed in a single batch (default: 100) */
+    embeddingBatchSize?: number;
+}
+export declare function createRAGPipeline(config: RAGPipelineConfig): RAGPipeline;
+export declare function createInMemoryVectorStore(): VectorStore;
+//# sourceMappingURL=pipeline.d.ts.map

package/dist/ai/rag/pipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../../../src/ai/rag/pipeline.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAiB,KAAK,WAAW,EAAsB,MAAM,eAAe,CAAC;AAGpF,MAAM,WAAW,mBAAmB;IAClC,iCAAiC;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,uBAAuB;IACvB,OAAO,EAAE,MAAM,CAAC;IAChB,8CAA8C;IAC9C,MAAM,EAAE,MAAM,CAAC;IACf,uBAAuB;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,gCAAgC;IAChC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,0BAA0B;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAGD,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAGD,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,4DAA4D;IAC5D,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,+BAA+B;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAGD,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,CACJ,SAAS,EAAE,MAAM,EAAE,EACnB,OAAO,EAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;KAClB,GACA,OAAO,CAAC,KAAK,CAAC,WAAW,GAAG;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC,CAAC;IACnD,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD;AAGD,MAAM,WAAW,WAAW;IAC1B,MAAM,CAAC,KAAK,EAAE,mBAAmB,GAAG,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACxF,QAAQ,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;IACvD,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACnD;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,2DAA2D;IAC3D,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,iBAAiB,GAAG,WAAW,CAyFxE;AAGD,wBAAgB,yBAAyB,IAAI,WAAW,CAmDvD"}

package/dist/ai/rag/pipeline.js

@@ -0,0 +1,135 @@
+// ── RAG Pipeline ──
+// Ingestion, embedding, storage, and retrieval
+import { chunkDocument } from './chunking.js';
+export function createRAGPipeline(config) {
+    const { provider, vectorStore } = config;
+    const batchSize = config.embeddingBatchSize ?? 100;
+    return {
+        async ingest(input) {
+            // 1. Chunk the document
+            const chunks = chunkDocument(input.content, config.chunkConfig);
+            // 2. Generate embeddings in batches
+            const storedChunks = [];
+            for (let i = 0; i < chunks.length; i += batchSize) {
+                const batch = chunks.slice(i, i + batchSize);
+                const embeddingResponse = await provider.embed({
+                    texts: batch.map((c) => c.content),
+                    model: config.embeddingModel,
+                });
+                for (let j = 0; j < batch.length; j++) {
+                    const chunk = batch[j];
+                    const emb = embeddingResponse.embeddings[j];
+                    if (!chunk || !emb)
+                        continue;
+                    storedChunks.push({
+                        id: crypto.randomUUID(),
+                        documentId: input.documentId,
+                        content: chunk.content,
+                        chunkIndex: chunk.index,
+                        embedding: emb,
+                        source: input.source,
+                        tenantId: input.tenantId,
+                        classification: input.classification,
+                        metadata: input.metadata,
+                    });
+                }
+            }
+            // 3. Store chunks with embeddings
+            await vectorStore.insert(storedChunks);
+            return { documentId: input.documentId, chunkCount: storedChunks.length };
+        },
+        async retrieve(query) {
+            const limit = query.limit ?? 5;
+            const minScore = query.minScore ?? 0.0;
+            // 1. Embed the query
+            const embeddingResponse = await provider.embed({
+                texts: [query.query],
+                model: config.embeddingModel,
+            });
+            const queryEmbedding = embeddingResponse.embeddings[0];
+            if (!queryEmbedding)
+                return [];
+            // 2. Search vector store
+            const results = await vectorStore.search(queryEmbedding, {
+                tenantId: query.tenantId,
+                maxClassification: query.maxClassification,
+                limit,
+                minScore,
+            });
+            // 3. Deduplicate by documentId+chunkIndex
+            const seen = new Set();
+            const deduplicated = results.filter((r) => {
+                const key = `${r.documentId}:${r.chunkIndex}`;
+                if (seen.has(key))
+                    return false;
+                seen.add(key);
+                return true;
+            });
+            // 4. Map to AIDocument
+            return deduplicated.map((r) => ({
+                content: r.content,
+                source: r.source ?? r.documentId,
+                score: r.score,
+                metadata: {
+                    ...r.metadata,
+                    documentId: r.documentId,
+                    chunkIndex: r.chunkIndex,
+                    classification: r.classification,
+                },
+            }));
+        },
+        async deleteDocument(documentId) {
+            await vectorStore.deleteByDocumentId(documentId);
+        },
+    };
+}
+// ── In-Memory Vector Store (for testing) ──
+export function createInMemoryVectorStore() {
+    const chunks = [];
+    function cosineSimilarity(a, b) {
+        let dot = 0;
+        let magA = 0;
+        let magB = 0;
+        for (let i = 0; i < a.length; i++) {
+            const ai = a[i] ?? 0;
+            const bi = b[i] ?? 0;
+            dot += ai * bi;
+            magA += ai * ai;
+            magB += bi * bi;
+        }
+        const denom = Math.sqrt(magA) * Math.sqrt(magB);
+        return denom === 0 ? 0 : dot / denom;
+    }
+    return {
+        async insert(newChunks) {
+            chunks.push(...newChunks);
+        },
+        async search(embedding, options) {
+            const filtered = chunks.filter((c) => {
+                if (options.tenantId && c.tenantId !== options.tenantId)
+                    return false;
+                return true;
+            });
+            const scored = filtered
+                .map((c) => ({
+                ...c,
+                score: cosineSimilarity(embedding, c.embedding),
+            }))
+                .filter((c) => c.score >= options.minScore)
+                .sort((a, b) => b.score - a.score)
+                .slice(0, options.limit);
+            return scored;
+        },
+        async deleteByDocumentId(documentId) {
+            const idxs = [];
+            for (let i = chunks.length - 1; i >= 0; i--) {
+                if (chunks[i]?.documentId === documentId)
+                    idxs.push(i);
+            }
+            for (const idx of idxs) {
+                chunks.splice(idx, 1);
+            }
+        },
+    };
+}
+//# sourceMappingURL=pipeline.js.map