@hazeljs/rag 0.2.4 → 0.3.1

package/dist/rag.service.js

@@ -3,12 +3,45 @@
  * RAG Service
  * Main service for RAG operations in HazelJS
  */
+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);
 };
@@ -20,15 +53,39 @@ const prompts_1 = require("@hazeljs/prompts");
 require("./prompts/rag-answer.prompt");
 const rag_answer_prompt_1 = require("./prompts/rag-answer.prompt");
 const types_1 = require("./types");
+const multi_query_1 = require("./retrieval/multi-query");
+const hybrid_search_1 = require("./retrieval/hybrid-search");
+const bm25_1 = require("./retrieval/bm25");
+const text_file_loader_1 = require("./loaders/text-file.loader");
+const json_file_loader_1 = require("./loaders/json-file.loader");
+const csv_file_loader_1 = require("./loaders/csv-file.loader");
+const markdown_file_loader_1 = require("./loaders/markdown-file.loader");
+const html_file_loader_1 = require("./loaders/html-file.loader");
+const directory_loader_1 = require("./loaders/directory.loader");
+const web_loader_1 = require("./loaders/web.loader");
+const debug_1 = require("./utils/debug");
+const dbg = (0, debug_1.debug)('rag');
 let RAGService = class RAGService {
     constructor(config) {
         this.config = config;
+        this.conversationHistory = new Map();
         this.pipeline = new rag_pipeline_1.RAGPipeline({
             vectorStore: config.vectorStore,
             embeddingProvider: config.embeddingProvider,
             textSplitter: config.textSplitter,
             topK: config.topK,
         }, config.llmFunction);
+        // Initialize BM25 for keyword search
+        this.bm25 = new bm25_1.BM25();
+        // Initialize multi-query retrieval
+        this.multiQueryRetrieval = new multi_query_1.MultiQueryRetrieval(config.vectorStore, {
+            customGenerator: config.queryGenerator,
+        });
+        // Initialize hybrid search
+        this.hybridRetrieval = new hybrid_search_1.HybridSearchRetrieval(config.vectorStore, {
+            vectorWeight: config.hybridWeights?.vectorWeight ?? 0.7,
+            keywordWeight: config.hybridWeights?.keywordWeight ?? 0.3,
+        });
     }
     /**
      * Initialize the RAG service
@@ -43,12 +100,99 @@ let RAGService = class RAGService {
         const docs = Array.isArray(documents) ? documents : [documents];
         return this.pipeline.addDocuments(docs);
     }
+    /**
+     * Ingest documents from a file path, URL, or directory.
+     * Auto-detects the source type and uses the appropriate loader.
+     *
+     * @example
+     * ```ts
+     * await rag.ingest('./docs/guide.pdf');
+     * await rag.ingest('./data/faq.csv');
+     * await rag.ingest('https://example.com/page');
+     * await rag.ingest('./knowledge-base/');  // loads entire directory
+     * ```
+     */
+    async ingest(source) {
+        dbg('ingest start source=%s', source);
+        const docs = await this.loadSource(source);
+        dbg('ingest loaded docs=%d', docs.length);
+        const ids = await this.index(docs);
+        dbg('ingest complete ids=%d', ids.length);
+        return ids;
+    }
+    /**
+     * Auto-detect source type and load documents.
+     */
+    async loadSource(source) {
+        // URL detection
+        if (source.startsWith('http://') || source.startsWith('https://')) {
+            const loader = new web_loader_1.WebLoader({ url: source });
+            return loader.load();
+        }
+        // Detect if source is a directory (ends with / or has no extension)
+        const ext = this.getExtension(source);
+        if (!ext || source.endsWith('/')) {
+            const loader = new directory_loader_1.DirectoryLoader({ path: source });
+            return loader.load();
+        }
+        // File-based loading by extension
+        switch (ext) {
+            case '.txt':
+            case '.log':
+                return new text_file_loader_1.TextFileLoader({ path: source }).load();
+            case '.json':
+                return new json_file_loader_1.JSONFileLoader({ path: source }).load();
+            case '.csv':
+                return new csv_file_loader_1.CSVFileLoader({ path: source }).load();
+            case '.md':
+            case '.mdx':
+                return new markdown_file_loader_1.MarkdownFileLoader({ path: source }).load();
+            case '.html':
+            case '.htm':
+                return new html_file_loader_1.HTMLFileLoader({ path: source }).load();
+            case '.pdf': {
+                // Optional dependency — dynamic import
+                try {
+                    const { PdfLoader } = await Promise.resolve().then(() => __importStar(require('./loaders/pdf.loader')));
+                    return new PdfLoader({ path: source }).load();
+                }
+                catch {
+                    throw new Error(`PDF loading requires the 'pdf-parse' package. Install it with: npm install pdf-parse`);
+                }
+            }
+            case '.docx': {
+                try {
+                    const { DocxLoader } = await Promise.resolve().then(() => __importStar(require('./loaders/docx.loader')));
+                    return new DocxLoader({ path: source }).load();
+                }
+                catch {
+                    throw new Error(`DOCX loading requires the 'mammoth' package. Install it with: npm install mammoth`);
+                }
+            }
+            default:
+                // Fallback: treat as text file
+                return new text_file_loader_1.TextFileLoader({ path: source }).load();
+        }
+    }
+    /**
+     * Extract file extension from a path.
+     */
+    getExtension(filePath) {
+        const dotIndex = filePath.lastIndexOf('.');
+        const slashIndex = Math.max(filePath.lastIndexOf('/'), filePath.lastIndexOf('\\'));
+        if (dotIndex <= slashIndex || dotIndex === -1)
+            return '';
+        return filePath.slice(dotIndex).toLowerCase();
+    }
     /**
      * Search for similar documents
      */
     async search(query, options) {
+        dbg('search query=%s strategy=%s', query, options?.strategy || 'similarity');
         const { strategy, ...queryOptions } = options || {};
-        return this.pipeline.retrieve(query, queryOptions, strategy);
+        const results = await this.pipeline.retrieve(query, queryOptions, strategy);
+        dbg('search results=%d', results.length);
+        return results;
     }
     /**
      * Retrieve relevant context for a query
@@ -76,41 +220,169 @@ let RAGService = class RAGService {
      * Full RAG pipeline: retrieve + generate
      */
     async ask(query, options) {
+        dbg('ask query=%s', query);
         const sources = await this.retrieve(query, options);
         const answer = await this.generate(query, sources);
+        dbg('ask complete answer_len=%d sources=%d', answer.length, sources.length);
         return { answer, sources };
     }
     /**
      * Multi-query RAG
-     * Generates multiple search queries and combines results
+     * Generates multiple search queries from a single question and combines
+     * deduplicated results, ranked by frequency and average score.
      */
-    async multiQuery(question, _numQueries = 3) {
-        // TODO: Implement query generation using LLM
-        // For now, just use the original query
-        return this.search(question, { topK: 10 });
+    async multiQuery(question, numQueries = 3) {
+        dbg('multiQuery question=%s numQueries=%d', question, numQueries);
+        if (!this.multiQueryRetrieval) {
+            return this.search(question, { topK: 10 });
+        }
+        // Override numQueries on the fly
+        const retrieval = new multi_query_1.MultiQueryRetrieval(this.config.vectorStore, {
+            numQueries,
+            customGenerator: this.config.queryGenerator,
+        });
+        const results = await retrieval.retrieve(question, { topK: numQueries * 5 });
+        dbg('multiQuery results=%d', results.length);
+        return results;
     }
     /**
-     * Compress retrieved context
+     * Compress retrieved context by removing low-relevance and redundant results.
+     * Uses LLM-based compression when available, otherwise applies score-based filtering
+     * with deduplication.
      */
-    async compress(documents, _query) {
-        // TODO: Implement context compression
-        // For now, return top results
-        return documents.slice(0, 5);
+    async compress(documents, query) {
+        if (documents.length === 0)
+            return [];
+        // Step 1: Remove results below a relevance threshold (adaptive: median * 0.6)
+        const scores = documents.map((d) => d.score);
+        const sortedScores = [...scores].sort((a, b) => a - b);
+        const median = sortedScores[Math.floor(sortedScores.length / 2)];
+        const threshold = median * 0.6;
+        let filtered = documents.filter((d) => d.score >= threshold);
+        // Step 2: Deduplicate near-identical content (Jaccard similarity > 0.8)
+        const deduplicated = [];
+        for (const doc of filtered) {
+            const isDuplicate = deduplicated.some((existing) => this.jaccardSimilarity(existing.content, doc.content) > 0.8);
+            if (!isDuplicate) {
+                deduplicated.push(doc);
+            }
+        }
+        filtered = deduplicated;
+        // Step 3: LLM-based compression if available
+        if (this.config.llmFunction && filtered.length > 5) {
+            try {
+                const contextBlock = filtered.map((d, i) => `[${i}] ${d.content}`).join('\n\n');
+                const compressPrompt = `Given the query: "${query}"\n\n` +
+                    `Below are retrieved passages. Return ONLY the indices (comma-separated) of the ` +
+                    `passages that are most relevant and non-redundant. Return at most 5 indices.\n\n` +
+                    contextBlock;
+                const response = await this.config.llmFunction(compressPrompt);
+                const indices = response
+                    .replace(/[^0-9,]/g, '')
+                    .split(',')
+                    .map(Number)
+                    .filter((n) => !isNaN(n) && n >= 0 && n < filtered.length);
+                if (indices.length > 0) {
+                    return indices.map((i) => filtered[i]);
+                }
+            }
+            catch {
+                // Fallback to score-based top-5 below
+            }
+        }
+        return filtered.slice(0, 5);
     }
     /**
-     * Self-query with automatic metadata extraction
+     * Jaccard similarity between two strings (token-level).
+     */
+    jaccardSimilarity(a, b) {
+        const tokensA = new Set(a.toLowerCase().split(/\s+/));
+        const tokensB = new Set(b.toLowerCase().split(/\s+/));
+        const intersection = new Set([...tokensA].filter((t) => tokensB.has(t)));
+        const union = new Set([...tokensA, ...tokensB]);
+        return union.size === 0 ? 0 : intersection.size / union.size;
+    }
+    /**
+     * Self-query with automatic metadata extraction.
+     * Extracts filter conditions from natural language when an LLM is available,
+     * then applies them as metadata filters on the vector search.
      */
     async selfQuery(naturalLanguageQuery) {
-        // TODO: Implement metadata extraction from natural language
-        // For now, just do regular search
-        return this.search(naturalLanguageQuery);
+        if (!this.config.llmFunction) {
+            return this.search(naturalLanguageQuery);
+        }
+        try {
+            const extractionPrompt = `Given the following natural language query, extract:\n` +
+                `1. A simplified search query (just the semantic intent)\n` +
+                `2. Any metadata filters expressed as JSON (e.g. {"category": "science", "year": 2024})\n\n` +
+                `Query: "${naturalLanguageQuery}"\n\n` +
+                `Respond in this exact JSON format:\n` +
+                `{"searchQuery": "...", "filters": {...}}`;
+            const response = await this.config.llmFunction(extractionPrompt);
+            const cleaned = response
+                .replace(/^```(?:json)?\s*/i, '')
+                .replace(/\s*```$/i, '')
+                .trim();
+            const parsed = JSON.parse(cleaned);
+            const searchQuery = parsed.searchQuery || naturalLanguageQuery;
+            const filter = parsed.filters && Object.keys(parsed.filters).length > 0 ? parsed.filters : undefined;
+            return this.search(searchQuery, { filter });
+        }
+        catch {
+            // Fallback to plain search on parse error
+            return this.search(naturalLanguageQuery);
+        }
     }
     /**
-     * Conversational RAG with session memory
+     * Conversational RAG with session memory.
+     * Maintains per-session conversation history and rewrites the user's
+     * follow-up question into a standalone query before retrieval.
      */
-    async chat(message, _sessionId) {
-        // TODO: Implement conversation memory
-        return this.ask(message);
+    async chat(message, sessionId) {
+        dbg('chat message=%s session=%s', message, sessionId);
+        // Get or create conversation history
+        if (!this.conversationHistory.has(sessionId)) {
+            this.conversationHistory.set(sessionId, []);
+        }
+        const history = this.conversationHistory.get(sessionId);
+        // Rewrite the follow-up into a standalone question using conversation context
+        let standaloneQuery = message;
+        if (history.length > 0 && this.config.llmFunction) {
+            try {
+                const historyText = history
+                    .slice(-6) // last 3 exchanges
+                    .map((m) => `${m.role}: ${m.content}`)
+                    .join('\n');
+                const rewritePrompt = `Given the conversation history and a follow-up question, ` +
+                    `rewrite the follow-up into a standalone question that captures the full context.\n\n` +
+                    `Conversation:\n${historyText}\n\n` +
+                    `Follow-up: ${message}\n\n` +
+                    `Standalone question:`;
+                standaloneQuery = await this.config.llmFunction(rewritePrompt);
+                dbg('chat rewritten query=%s', standaloneQuery);
+            }
+            catch {
+                standaloneQuery = message;
+            }
+        }
+        // Retrieve and generate
+        const sources = await this.retrieve(standaloneQuery);
+        const answer = await this.generate(standaloneQuery, sources);
+        // Update conversation history
+        history.push({ role: 'user', content: message });
+        history.push({ role: 'assistant', content: answer });
+        // Keep history bounded
+        while (history.length > 20) {
+            history.shift();
+        }
+        dbg('chat complete answer_len=%d sources=%d history_len=%d', answer.length, sources.length, history.length);
+        return { answer, sources };
+    }
+    /**
+     * Clear conversation history for a session.
+     */
+    clearChat(sessionId) {
+        this.conversationHistory.delete(sessionId);
     }
     /**
      * Hybrid search combining vector and keyword search
@@ -119,19 +391,109 @@ let RAGService = class RAGService {
         return this.search(query, { ...options, strategy: types_1.RetrievalStrategy.HYBRID });
     }
     /**
-     * Rerank search results
+     * Rerank search results using cross-encoder-style LLM scoring.
+     * When no LLM is configured, falls back to BM25-boosted re-scoring.
      */
-    async rerank(results, _query, topN) {
-        // TODO: Implement reranking with external model
-        // For now, return top N results
-        return results.slice(0, topN || 5);
+    async rerank(results, query, topN) {
+        const n = topN || 5;
+        if (results.length <= n)
+            return results;
+        // Strategy 1: LLM-based reranking (when available)
+        if (this.config.llmFunction) {
+            try {
+                const passages = results.map((r, i) => `[${i}] ${r.content.slice(0, 300)}`).join('\n\n');
+                const rerankPrompt = `Given the query: "${query}"\n\n` +
+                    `Rank the following passages by relevance. Return ONLY the indices ` +
+                    `in order from most to least relevant (comma-separated). ` +
+                    `Return at most ${n} indices.\n\n${passages}`;
+                const response = await this.config.llmFunction(rerankPrompt);
+                const indices = response
+                    .replace(/[^0-9,]/g, '')
+                    .split(',')
+                    .map(Number)
+                    .filter((idx) => !isNaN(idx) && idx >= 0 && idx < results.length);
+                if (indices.length > 0) {
+                    const seen = new Set();
+                    const reranked = [];
+                    for (const idx of indices) {
+                        if (!seen.has(idx)) {
+                            seen.add(idx);
+                            reranked.push(results[idx]);
+                        }
+                        if (reranked.length >= n)
+                            break;
+                    }
+                    return reranked;
+                }
+            }
+            catch {
+                // Fallback to BM25 boost below
+            }
+        }
+        // Strategy 2: BM25 keyword boost
+        const bm25 = new bm25_1.BM25();
+        const bm25Docs = results.map((r) => ({
+            id: r.id,
+            content: r.content,
+            tokens: r.content
+                .toLowerCase()
+                .replace(/[^\w\s]/g, ' ')
+                .split(/\s+/)
+                .filter(Boolean),
+        }));
+        bm25.addDocuments(bm25Docs);
+        const keywordScores = bm25.search(query, results.length);
+        const kwMap = new Map(keywordScores.map((k) => [k.id, k.score]));
+        // Normalize BM25 scores
+        const maxKw = Math.max(...keywordScores.map((k) => k.score), 1);
+        const boosted = results.map((r) => ({
+            ...r,
+            score: r.score * 0.7 + ((kwMap.get(r.id) || 0) / maxKw) * 0.3,
+        }));
+        return boosted.sort((a, b) => b.score - a.score).slice(0, n);
     }
     /**
-     * Ensemble retrieval combining multiple methods
+     * Ensemble retrieval combining multiple retrieval strategies.
+     * Runs each strategy, normalizes scores, applies weights, and fuses results.
      */
-    async ensemble(query, _methods, _weights) {
-        // TODO: Implement ensemble retrieval
-        return this.search(query);
+    async ensemble(query, methods, weights) {
+        if (methods.length === 0)
+            return this.search(query);
+        // Default: equal weights
+        const w = weights && weights.length === methods.length
+            ? weights
+            : methods.map(() => 1 / methods.length);
+        // Run all strategies in parallel
+        const allResults = await Promise.all(methods.map((strategy) => this.search(query, { strategy, topK: 20 })));
+        // Normalize and weight each result set
+        const scoreMap = new Map();
+        for (let i = 0; i < allResults.length; i++) {
+            const results = allResults[i];
+            const weight = w[i];
+            // Min-max normalize within this strategy
+            const scores = results.map((r) => r.score);
+            const min = Math.min(...scores, 0);
+            const max = Math.max(...scores, 1);
+            const range = max - min || 1;
+            for (const result of results) {
+                const normalizedScore = ((result.score - min) / range) * weight;
+                const existing = scoreMap.get(result.id);
+                if (existing) {
+                    existing.score += normalizedScore;
+                }
+                else {
+                    scoreMap.set(result.id, {
+                        result: { ...result },
+                        score: normalizedScore,
+                    });
+                }
+            }
+        }
+        // Sort by fused score
+        const fused = [...scoreMap.values()]
+            .sort((a, b) => b.score - a.score)
+            .map(({ result, score }) => ({ ...result, score }));
+        return fused.slice(0, 10);
     }
     /**
      * Time-weighted retrieval favoring recent documents

package/dist/rag.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"rag.service.js","sourceRoot":"","sources":["../src/rag.service.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;AAEH,wCAAwC;AACxC,iDAA0D;AAC1D,8CAAkD;AAClD,uCAAqC;AACrC,mEAA6D;AAC7D,mCAQiB;AAWV,IAAM,UAAU,GAAhB,MAAM,UAAU;IAGrB,YAAoB,MAAwB;QAAxB,WAAM,GAAN,MAAM,CAAkB;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,0BAAW,CAC7B;YACE,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;YAC3C,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,EACD,MAAM,CAAC,WAAW,CACnB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,SAAgC;QAC1C,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,KAAa,EACb,OAAyD;QAEzD,MAAM,EAAE,QAAQ,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;QACpD,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAsB;QAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAgC;QAC5D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GACd,OAAO,OAAO,KAAK,QAAQ;YACzB,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAqC,kCAAc,CAAC,CAAC,MAAM,CAAC;YAC3F,OAAO,EAAE,UAAU;YACnB,KAAK;SACN,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CACP,KAAa,EACb,OAAsB;QAEtB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACnD,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,cAAsB,CAAC;QACxD,6CAA6C;QAC7C,uCAAuC;QACvC,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,SAAyB,EAAE,MAAc;QACtD,sCAAsC;QACtC,8BAA8B;QAC9B,OAAO,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,oBAA4B;QAC1C,4DAA4D;QAC5D,kCAAkC;QAClC,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CACR,OAAe,EACf,UAAkB;QAElB,sCAAsC;QACtC,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,KAAa,EACb,OAA0E;QAE1E,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,yBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;IAChF,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,OAAuB,EAAE,MAAc,EAAE,IAAa;QACjE,gDAAgD;QAChD,gCAAgC;QAChC,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CACZ,KAAa,EACb,QAA6B,EAC7B,QAAmB;QAEnB,qCAAqC;QACrC,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,KAAa,EAAE,YAAoB,IAAI;QACxD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAEpE,6BAA6B;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,OAAO,OAAO;aACX,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACd,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,EAAE,SAAS,IAAI,GAAG,CAAC;YACpD,MAAM,GAAG,GAAG,CAAC,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO;YACtE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;YAE9C,OAAO;gBACL,GAAG,MAAM;gBACT,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,UAAU;aACjC,CAAC;QACJ,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,GAAa;QACxB,MAAM,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF,CAAA;AA1LY,gCAAU;qBAAV,UAAU;IADtB,IAAA,cAAO,GAAE;;GACG,UAAU,CA0LtB"}
+{"version":3,"file":"rag.service.js","sourceRoot":"","sources":["../src/rag.service.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,wCAAwC;AACxC,iDAA0D;AAC1D,8CAAkD;AAClD,uCAAqC;AACrC,mEAA6D;AAC7D,mCAQiB;AACjB,yDAA8D;AAC9D,6DAAkE;AAClE,2CAAsD;AACtD,iEAA4D;AAC5D,iEAA4D;AAC5D,+DAA0D;AAC1D,yEAAoE;AACpE,iEAA4D;AAC5D,iEAA6D;AAC7D,qDAAiD;AACjD,yCAAsC;AAEtC,MAAM,GAAG,GAAG,IAAA,aAAK,EAAC,KAAK,CAAC,CAAC;AAelB,IAAM,UAAU,GAAhB,MAAM,UAAU;IAOrB,YAAoB,MAAwB;QAAxB,WAAM,GAAN,MAAM,CAAkB;QAFpC,wBAAmB,GAA0D,IAAI,GAAG,EAAE,CAAC;QAG7F,IAAI,CAAC,QAAQ,GAAG,IAAI,0BAAW,CAC7B;YACE,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;YAC3C,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,IAAI,EAAE,MAAM,CAAC,IAAI;SAClB,EACD,MAAM,CAAC,WAAW,CACnB,CAAC;QAEF,qCAAqC;QACrC,IAAI,CAAC,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QAEvB,mCAAmC;QACnC,IAAI,CAAC,mBAAmB,GAAG,IAAI,iCAAmB,CAAC,MAAM,CAAC,WAAW,EAAE;YACrE,eAAe,EAAE,MAAM,CAAC,cAAc;SACvC,CAAC,CAAC;QAEH,2BAA2B;QAC3B,IAAI,CAAC,eAAe,GAAG,IAAI,qCAAqB,CAAC,MAAM,CAAC,WAAW,EAAE;YACnE,YAAY,EAAE,MAAM,CAAC,aAAa,EAAE,YAAY,IAAI,GAAG;YACvD,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,aAAa,IAAI,GAAG;SAC1D,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,SAAgC;QAC1C,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAChE,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,MAAM,CAAC,MAAc;QACzB,GAAG,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QACtC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC3C,GAAG,CAAC,uBAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnC,GAAG,CAAC,wBAAwB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1C,OAAO,GAAG,CAAC;IACb,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,UAAU,CAAC,MAAc;QACrC,gBAAgB;QAChB,IAAI,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,IAAI,sBAAS,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC;YAC9C,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;QACvB,CAAC;QAED,oEAAoE;QACpE,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAEtC,IAAI,CAAC,GAAG,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,IAAI,kCAAe,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YACrD,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;QACvB,CAAC;QAED,kCAAkC;QAClC,QAAQ,GAAG,EAAE,CAAC;YACZ,KAAK,MAAM,CAAC;YACZ,KAAK,MAAM;gBACT,OAAO,IAAI,iCAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACrD,KAAK,OAAO;gBACV,OAAO,IAAI,iCAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACrD,KAAK,MAAM;gBACT,OAAO,IAAI,+BAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACpD,KAAK,KAAK,CAAC;YACX,KAAK,MAAM;gBACT,OAAO,IAAI,yCAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACzD,KAAK,OAAO,CAAC;YACb,KAAK,MAAM;gBACT,OAAO,IAAI,iCAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;YACrD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,uCAAuC;gBACvC,IAAI,CAAC;oBACH,MAAM,EAAE,SAAS,EAAE,GAAG,wDAAa,sBAAsB,GAAC,CAAC;oBAC3D,OAAO,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBAChD,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,IAAI,CAAC;oBACH,MAAM,EAAE,UAAU,EAAE,GAAG,wDAAa,uBAAuB,GAAC,CAAC;oBAC7D,OAAO,IAAI,UAAU,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gBACjD,CAAC;gBAAC,MAAM,CAAC;oBACP,MAAM,IAAI,KAAK,CACb,mFAAmF,CACpF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD;gBACE,+BAA+B;gBAC/B,OAAO,IAAI,iCAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,QAAgB;QACnC,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;QACnF,IAAI,QAAQ,IAAI,UAAU,IAAI,QAAQ,KAAK,CAAC,CAAC;YAAE,OAAO,EAAE,CAAC;QACzD,OAAO,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CACV,KAAa,EACb,OAAyD;QAEzD,GAAG,CAAC,6BAA6B,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,IAAI,YAAY,CAAC,CAAC;QAC7E,MAAM,EAAE,QAAQ,EAAE,GAAG,YAAY,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5E,GAAG,CAAC,mBAAmB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAsB;QAClD,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAa,EAAE,OAAgC;QAC5D,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;QAED,MAAM,UAAU,GACd,OAAO,OAAO,KAAK,QAAQ;YACzB,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAqC,kCAAc,CAAC,CAAC,MAAM,CAAC;YAC3F,OAAO,EAAE,UAAU;YACnB,KAAK;SACN,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,GAAG,CACP,KAAa,EACb,OAAsB;QAEtB,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QAC3B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QACnD,GAAG,CAAC,uCAAuC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5E,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,UAAU,CAAC,QAAgB,EAAE,aAAqB,CAAC;QACvD,GAAG,CAAC,sCAAsC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,CAAC;QAED,iCAAiC;QACjC,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE;YACjE,UAAU;YACV,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;SAC5C,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7E,GAAG,CAAC,uBAAuB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7C,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CAAC,SAAyB,EAAE,KAAa;QACrD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEtC,8EAA8E;QAC9E,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACvD,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,SAAS,GAAG,MAAM,GAAG,GAAG,CAAC;QAC/B,IAAI,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,SAAS,CAAC,CAAC;QAE7D,wEAAwE;QACxE,MAAM,YAAY,GAAmB,EAAE,CAAC;QACxC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,WAAW,GAAG,YAAY,CAAC,IAAI,CACnC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,GAAG,CAC1E,CAAC;YACF,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;QACD,QAAQ,GAAG,YAAY,CAAC;QAExB,6CAA6C;QAC7C,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnD,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAEhF,MAAM,cAAc,GAClB,qBAAqB,KAAK,OAAO;oBACjC,iFAAiF;oBACjF,kFAAkF;oBAClF,YAAY,CAAC;gBAEf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;gBAC/D,MAAM,OAAO,GAAG,QAAQ;qBACrB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;qBACvB,KAAK,CAAC,GAAG,CAAC;qBACV,GAAG,CAAC,MAAM,CAAC;qBACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAE7D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvB,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,sCAAsC;YACxC,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,CAAS,EAAE,CAAS;QAC5C,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC;QAChD,OAAO,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;IAC/D,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,SAAS,CAAC,oBAA4B;QAC1C,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAC3C,CAAC;QAED,IAAI,CAAC;YACH,MAAM,gBAAgB,GACpB,wDAAwD;gBACxD,2DAA2D;gBAC3D,4FAA4F;gBAC5F,WAAW,oBAAoB,OAAO;gBACtC,sCAAsC;gBACtC,0CAA0C,CAAC;YAE7C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,gBAAgB,CAAC,CAAC;YACjE,MAAM,OAAO,GAAG,QAAQ;iBACrB,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC;iBAChC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;iBACvB,IAAI,EAAE,CAAC;YAEV,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAGhC,CAAC;YAEF,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,oBAAoB,CAAC;YAC/D,MAAM,MAAM,GACV,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;YAExF,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,0CAA0C;YAC1C,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,IAAI,CACR,OAAe,EACf,SAAiB;QAEjB,GAAG,CAAC,4BAA4B,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QACtD,qCAAqC;QACrC,IAAI,CAAC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;QAEzD,8EAA8E;QAC9E,IAAI,eAAe,GAAG,OAAO,CAAC;QAC9B,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAClD,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,OAAO;qBACxB,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB;qBAC7B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;qBACrC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAEd,MAAM,aAAa,GACjB,2DAA2D;oBAC3D,sFAAsF;oBACtF,kBAAkB,WAAW,MAAM;oBACnC,cAAc,OAAO,MAAM;oBAC3B,sBAAsB,CAAC;gBAEzB,eAAe,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;gBAC/D,GAAG,CAAC,yBAAyB,EAAE,eAAe,CAAC,CAAC;YAClD,CAAC;YAAC,MAAM,CAAC;gBACP,eAAe,GAAG,OAAO,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;QAE7D,8BAA8B;QAC9B,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,uBAAuB;QACvB,OAAO,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;QAED,GAAG,CACD,uDAAuD,EACvD,MAAM,CAAC,MAAM,EACb,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,MAAM,CACf,CAAC;QACF,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,SAAiB;QACzB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,KAAa,EACb,OAA0E;QAE1E,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,yBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC;IAChF,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,MAAM,CAAC,OAAuB,EAAE,KAAa,EAAE,IAAa;QAChE,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;QACpB,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC;YAAE,OAAO,OAAO,CAAC;QAExC,mDAAmD;QACnD,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAEzF,MAAM,YAAY,GAChB,qBAAqB,KAAK,OAAO;oBACjC,oEAAoE;oBACpE,0DAA0D;oBAC1D,kBAAkB,CAAC,gBAAgB,QAAQ,EAAE,CAAC;gBAEhD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;gBAC7D,MAAM,OAAO,GAAG,QAAQ;qBACrB,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;qBACvB,KAAK,CAAC,GAAG,CAAC;qBACV,GAAG,CAAC,MAAM,CAAC;qBACX,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;gBAEpE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvB,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;oBAC/B,MAAM,QAAQ,GAAmB,EAAE,CAAC;oBACpC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;wBAC1B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;4BACnB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;4BACd,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;wBAC9B,CAAC;wBACD,IAAI,QAAQ,CAAC,MAAM,IAAI,CAAC;4BAAE,MAAM;oBAClC,CAAC;oBACD,OAAO,QAAQ,CAAC;gBAClB,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QAED,iCAAiC;QACjC,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAmB,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnD,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,MAAM,EAAE,CAAC,CAAC,OAAO;iBACd,WAAW,EAAE;iBACb,OAAO,CAAC,UAAU,EAAE,GAAG,CAAC;iBACxB,KAAK,CAAC,KAAK,CAAC;iBACZ,MAAM,CAAC,OAAO,CAAC;SACnB,CAAC,CAAC,CAAC;QACJ,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAEjE,wBAAwB;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAClC,GAAG,CAAC;YACJ,KAAK,EAAE,CAAC,CAAC,KAAK,GAAG,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,GAAG;SAC9D,CAAC,CAAC,CAAC;QAEJ,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,QAAQ,CACZ,KAAa,EACb,OAA4B,EAC5B,OAAkB;QAElB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAEpD,yBAAyB;QACzB,MAAM,CAAC,GACL,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM;YAC1C,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAE5C,iCAAiC;QACjC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CACtE,CAAC;QAEF,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAmD,CAAC;QAE5E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,OAAO,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAEpB,yCAAyC;YACzC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC;YACnC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,EAAE,CAAC,CAAC,CAAC;YACnC,MAAM,KAAK,GAAG,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;YAE7B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;gBAC7B,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,MAAM,CAAC;gBAChE,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACzC,IAAI,QAAQ,EAAE,CAAC;oBACb,QAAQ,CAAC,KAAK,IAAI,eAAe,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE;wBACtB,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE;wBACrB,KAAK,EAAE,eAAe;qBACvB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,sBAAsB;QACtB,MAAM,KAAK,GAAG,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;aACjC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAEtD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,KAAa,EAAE,YAAoB,IAAI;QACxD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAEpE,6BAA6B;QAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,OAAO,OAAO;aACX,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YACd,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,EAAE,SAAS,IAAI,GAAG,CAAC;YACpD,MAAM,GAAG,GAAG,CAAC,GAAG,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO;YACtE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC;YAE9C,OAAO;gBACL,GAAG,MAAM;gBACT,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,UAAU;aACjC,CAAC;QACJ,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,GAAa;QACxB,MAAM,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF,CAAA;AApjBY,gCAAU;qBAAV,UAAU;IADtB,IAAA,cAAO,GAAE;;GACG,UAAU,CAojBtB"}

package/dist/reranking/cohere.reranker.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Cohere Reranker
+ *
+ * Uses Cohere's Rerank API to re-order search results by relevance to the query.
+ * This significantly improves RAG retrieval quality compared to pure vector search.
+ */
+import { Reranker, SearchResult } from '../types';
+export interface CohereRerankerConfig {
+    /** Cohere API Key */
+    apiKey: string;
+    /**
+     * The model to use.
+     * Default: rerank-english-v3.0
+     * Other options: rerank-multilingual-v3.0
+     */
+    model?: string;
+    /** Number of maximum chunks per document (defaults to 10) */
+    maxChunksPerDoc?: number;
+}
+export declare class CohereReranker implements Reranker {
+    private client;
+    private model;
+    private maxChunksPerDoc;
+    constructor(config: CohereRerankerConfig);
+    /**
+     * Reranks SearchResult objects based on how relevant they are to the given query.
+     *
+     * @param query The user's search query
+     * @param results The potentially relevant documents (e.g. from a vector store)
+     * @param topN How many top results to return (defaults to all returned results)
+     */
+    rerank(query: string, results: SearchResult[], topN?: number): Promise<SearchResult[]>;
+}
+//# sourceMappingURL=cohere.reranker.d.ts.map

package/dist/reranking/cohere.reranker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.reranker.d.ts","sourceRoot":"","sources":["../../src/reranking/cohere.reranker.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAKlD,MAAM,WAAW,oBAAoB;IACnC,qBAAqB;IACrB,MAAM,EAAE,MAAM,CAAC;IACf;;;;OAIG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,6DAA6D;IAC7D,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,qBAAa,cAAe,YAAW,QAAQ;IAC7C,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,eAAe,CAAS;gBAEpB,MAAM,EAAE,oBAAoB;IASxC;;;;;;OAMG;IACG,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;CAmC7F"}

package/dist/reranking/cohere.reranker.js

@@ -0,0 +1,58 @@
+"use strict";
+/**
+ * Cohere Reranker
+ *
+ * Uses Cohere's Rerank API to re-order search results by relevance to the query.
+ * This significantly improves RAG retrieval quality compared to pure vector search.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CohereReranker = void 0;
+class CohereReranker {
+    constructor(config) {
+        // Dynamically require to avoid hard peer dependency
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
+        const { CohereClient } = require('cohere-ai');
+        this.client = new CohereClient({ token: config.apiKey });
+        this.model = config.model || 'rerank-english-v3.0';
+        this.maxChunksPerDoc = config.maxChunksPerDoc || 10;
+    }
+    /**
+     * Reranks SearchResult objects based on how relevant they are to the given query.
+     *
+     * @param query The user's search query
+     * @param results The potentially relevant documents (e.g. from a vector store)
+     * @param topN How many top results to return (defaults to all returned results)
+     */
+    async rerank(query, results, topN) {
+        if (!results || results.length === 0) {
+            return [];
+        }
+        // Prepare documents for Cohere API (must be strings or objects with text)
+        const documents = results.map((r) => r.content);
+        try {
+            const response = await this.client.rerank({
+                model: this.model,
+                query,
+                documents,
+                topN: topN || results.length,
+                maxChunksPerDoc: this.maxChunksPerDoc,
+                returnDocuments: false, // We already have the documents in `results` array
+            });
+            // The response returns items with indices and relevance scores
+            // Re-map the newly scored results back to our SearchResult format
+            const reranked = response.results.map((item) => {
+                const originalResult = results[item.index];
+                return {
+                    ...originalResult,
+                    score: item.relevanceScore, // Override with the superior rerank score
+                };
+            });
+            return reranked;
+        }
+        catch (error) {
+            throw new Error(`Failed to rerank results using Cohere: ${error}`);
+        }
+    }
+}
+exports.CohereReranker = CohereReranker;
+//# sourceMappingURL=cohere.reranker.js.map

package/dist/reranking/cohere.reranker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.reranker.js","sourceRoot":"","sources":["../../src/reranking/cohere.reranker.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAoBH,MAAa,cAAc;IAKzB,YAAY,MAA4B;QACtC,oDAAoD;QACpD,iEAAiE;QACjE,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,qBAAqB,CAAC;QACnD,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC;IACtD,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAuB,EAAE,IAAa;QAChE,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,0EAA0E;QAC1E,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAEhD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACxC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK;gBACL,SAAS;gBACT,IAAI,EAAE,IAAI,IAAI,OAAO,CAAC,MAAM;gBAC5B,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,eAAe,EAAE,KAAK,EAAE,mDAAmD;aAC5E,CAAC,CAAC;YAEH,+DAA+D;YAC/D,kEAAkE;YAClE,MAAM,QAAQ,GAAmB,QAAQ,CAAC,OAAO,CAAC,GAAG,CACnD,CAAC,IAA+C,EAAE,EAAE;gBAClD,MAAM,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC3C,OAAO;oBACL,GAAG,cAAc;oBACjB,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE,0CAA0C;iBACvE,CAAC;YACJ,CAAC,CACF,CAAC;YAEF,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,0CAA0C,KAAK,EAAE,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;CACF;AAxDD,wCAwDC"}

package/dist/types/index.d.ts

@@ -112,6 +112,7 @@ export interface DocumentLoader {
 export interface RAGConfig {
     vectorStore: VectorStore;
     embeddingProvider: EmbeddingProvider;
+    reranker?: Reranker;
     textSplitter?: TextSplitter;
     chunkSize?: number;
     chunkOverlap?: number;
@@ -141,4 +142,13 @@ export interface RAGResponse {
     sources: SearchResult[];
     context: string;
 }
+/**
+ * Reranker interface
+ */
+export interface Reranker {
+    /**
+     * Rerank search results based on a query
+     */
+    rerank(query: string, results: SearchResult[], topN?: number): Promise<SearchResult[]>;
+}
 //# sourceMappingURL=index.d.ts.map