@hazeljs/rag 0.2.0-beta.1

package/dist/decorators/semantic-search.decorator.d.ts

@@ -0,0 +1,69 @@
+/**
+ * Semantic Search Decorators
+ * Enable semantic search capabilities on methods
+ */
+import 'reflect-metadata';
+export interface SemanticSearchOptions {
+    topK?: number;
+    minScore?: number;
+    includeMetadata?: boolean;
+    filter?: Record<string, unknown>;
+}
+export interface HybridSearchOptions extends SemanticSearchOptions {
+    vectorWeight?: number;
+    keywordWeight?: number;
+    algorithm?: 'rrf' | 'weighted' | 'linear';
+}
+/**
+ * Enables semantic search on a method
+ */
+export declare function SemanticSearch(options?: SemanticSearchOptions): MethodDecorator;
+/**
+ * Enables hybrid search (vector + keyword) on a method
+ */
+export declare function HybridSearch(options?: HybridSearchOptions): MethodDecorator;
+/**
+ * Auto-embed decorator for automatic embedding generation
+ */
+export declare function AutoEmbed(_fields?: string[]): MethodDecorator;
+/**
+ * Multi-query RAG decorator
+ */
+export declare function MultiQueryRAG(options?: {
+    queries?: number;
+}): MethodDecorator;
+/**
+ * Compress context decorator
+ */
+export declare function CompressContext(options?: {
+    maxTokens?: number;
+}): MethodDecorator;
+/**
+ * Self-query RAG with automatic metadata filtering
+ */
+export declare function SelfQueryRAG(): MethodDecorator;
+/**
+ * Rerank results decorator
+ */
+export declare function Rerank(options?: {
+    model?: string;
+    topN?: number;
+}): MethodDecorator;
+/**
+ * Parent-child document retrieval
+ */
+export declare function ParentChildRetrieval(): MethodDecorator;
+/**
+ * Ensemble retrieval combining multiple methods
+ */
+export declare function EnsembleRetrieval(options?: {
+    methods?: string[];
+    weights?: number[];
+}): MethodDecorator;
+/**
+ * Time-weighted retrieval favoring recent documents
+ */
+export declare function TimeWeightedRetrieval(options?: {
+    decayRate?: number;
+}): MethodDecorator;
+//# sourceMappingURL=semantic-search.decorator.d.ts.map

package/dist/decorators/semantic-search.decorator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"semantic-search.decorator.d.ts","sourceRoot":"","sources":["../../src/decorators/semantic-search.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAE1B,MAAM,WAAW,qBAAqB;IACpC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,mBAAoB,SAAQ,qBAAqB;IAChE,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,KAAK,GAAG,UAAU,GAAG,QAAQ,CAAC;CAC3C;AAKD;;GAEG;AACH,wBAAgB,cAAc,CAAC,OAAO,GAAE,qBAA0B,GAAG,eAAe,CAKnF;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,GAAE,mBAAwB,GAAG,eAAe,CAK/E;AAED;;GAEG;AACH,wBAAgB,SAAS,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,eAAe,CAe7D;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,OAAO,GAAE;IAAE,OAAO,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,eAAe,CAKjF;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,GAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,eAAe,CAKrF;AAED;;GAEG;AACH,wBAAgB,YAAY,IAAI,eAAe,CAK9C;AAED;;GAEG;AACH,wBAAgB,MAAM,CAAC,OAAO,GAAE;IAAE,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,eAAe,CAKvF;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,eAAe,CAKtD;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,GAAE;IACP,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;CACf,GACL,eAAe,CAKjB;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,GAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAO,GAAG,eAAe,CAK3F"}

package/dist/decorators/semantic-search.decorator.js

@@ -0,0 +1,116 @@
+"use strict";
+/**
+ * Semantic Search Decorators
+ * Enable semantic search capabilities on methods
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SemanticSearch = SemanticSearch;
+exports.HybridSearch = HybridSearch;
+exports.AutoEmbed = AutoEmbed;
+exports.MultiQueryRAG = MultiQueryRAG;
+exports.CompressContext = CompressContext;
+exports.SelfQueryRAG = SelfQueryRAG;
+exports.Rerank = Rerank;
+exports.ParentChildRetrieval = ParentChildRetrieval;
+exports.EnsembleRetrieval = EnsembleRetrieval;
+exports.TimeWeightedRetrieval = TimeWeightedRetrieval;
+require("reflect-metadata");
+const SEMANTIC_SEARCH_KEY = Symbol('semanticSearch');
+const HYBRID_SEARCH_KEY = Symbol('hybridSearch');
+/**
+ * Enables semantic search on a method
+ */
+function SemanticSearch(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata(SEMANTIC_SEARCH_KEY, options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Enables hybrid search (vector + keyword) on a method
+ */
+function HybridSearch(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata(HYBRID_SEARCH_KEY, options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Auto-embed decorator for automatic embedding generation
+ */
+function AutoEmbed(_fields) {
+    return (target, propertyKey, descriptor) => {
+        const originalMethod = descriptor.value;
+        descriptor.value = async function (...args) {
+            const result = await originalMethod.apply(this, args);
+            // TODO: Implement automatic embedding generation
+            // This would integrate with the RAG service to generate embeddings
+            return result;
+        };
+        return descriptor;
+    };
+}
+/**
+ * Multi-query RAG decorator
+ */
+function MultiQueryRAG(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('multiQueryRAG', options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Compress context decorator
+ */
+function CompressContext(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('compressContext', options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Self-query RAG with automatic metadata filtering
+ */
+function SelfQueryRAG() {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('selfQueryRAG', true, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Rerank results decorator
+ */
+function Rerank(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('rerank', options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Parent-child document retrieval
+ */
+function ParentChildRetrieval() {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('parentChildRetrieval', true, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Ensemble retrieval combining multiple methods
+ */
+function EnsembleRetrieval(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('ensembleRetrieval', options, target, propertyKey);
+        return descriptor;
+    };
+}
+/**
+ * Time-weighted retrieval favoring recent documents
+ */
+function TimeWeightedRetrieval(options = {}) {
+    return (target, propertyKey, descriptor) => {
+        Reflect.defineMetadata('timeWeightedRetrieval', options, target, propertyKey);
+        return descriptor;
+    };
+}
+//# sourceMappingURL=semantic-search.decorator.js.map

package/dist/decorators/semantic-search.decorator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"semantic-search.decorator.js","sourceRoot":"","sources":["../../src/decorators/semantic-search.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAuBH,wCAKC;AAKD,oCAKC;AAKD,8BAeC;AAKD,sCAKC;AAKD,0CAKC;AAKD,oCAKC;AAKD,wBAKC;AAKD,oDAKC;AAKD,8CAUC;AAKD,sDAKC;AAnID,4BAA0B;AAe1B,MAAM,mBAAmB,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;AACrD,MAAM,iBAAiB,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AAEjD;;GAEG;AACH,SAAgB,cAAc,CAAC,UAAiC,EAAE;IAChE,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAC1E,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,YAAY,CAAC,UAA+B,EAAE;IAC5D,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,iBAAiB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QACxE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,SAAS,CAAC,OAAkB;IAC1C,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAEtD,iDAAiD;YACjD,mEAAmE;YAEnE,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,aAAa,CAAC,UAAgC,EAAE;IAC9D,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,eAAe,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QACtE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,eAAe,CAAC,UAAkC,EAAE;IAClE,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,iBAAiB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QACxE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,YAAY;IAC1B,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAClE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,MAAM,CAAC,UAA6C,EAAE;IACpE,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAC/D,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB;IAClC,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,sBAAsB,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAC1E,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,iBAAiB,CAC/B,UAGI,EAAE;IAEN,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,mBAAmB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAC1E,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAgB,qBAAqB,CAAC,UAAkC,EAAE;IACxE,OAAO,CAAC,MAAc,EAAE,WAA4B,EAAE,UAA8B,EAAE,EAAE;QACtF,OAAO,CAAC,cAAc,CAAC,uBAAuB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QAC9E,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC"}

package/dist/embeddings/cohere-embeddings.d.ts

@@ -0,0 +1,33 @@
+/**
+ * Cohere Embeddings Provider
+ * High-quality embeddings with multilingual support
+ */
+import { EmbeddingProvider } from '../types';
+export interface CohereEmbeddingsConfig {
+    apiKey: string;
+    model?: string;
+    inputType?: 'search_document' | 'search_query' | 'classification' | 'clustering';
+    truncate?: 'NONE' | 'START' | 'END';
+    batchSize?: number;
+}
+export declare class CohereEmbeddings implements EmbeddingProvider {
+    private client;
+    private model;
+    private inputType;
+    private truncate;
+    private batchSize;
+    private dimension;
+    constructor(config: CohereEmbeddingsConfig);
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+    getDimension(): number;
+    /**
+     * Get embedding dimension for a given model
+     */
+    private getModelDimension;
+    /**
+     * Create embeddings for search queries (optimized input type)
+     */
+    embedQuery(query: string): Promise<number[]>;
+}
+//# sourceMappingURL=cohere-embeddings.d.ts.map

package/dist/embeddings/cohere-embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere-embeddings.d.ts","sourceRoot":"","sources":["../../src/embeddings/cohere-embeddings.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAQ7C,MAAM,WAAW,sBAAsB;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,iBAAiB,GAAG,cAAc,GAAG,gBAAgB,GAAG,YAAY,CAAC;IACjF,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,GAAG,KAAK,CAAC;IACpC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,sBAAsB;IAapC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAetC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAwBtD,YAAY,IAAI,MAAM;IAItB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAczB;;OAEG;IACG,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CAcnD"}

package/dist/embeddings/cohere-embeddings.js

@@ -0,0 +1,91 @@
+"use strict";
+/**
+ * Cohere Embeddings Provider
+ * High-quality embeddings with multilingual support
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CohereEmbeddings = void 0;
+class CohereEmbeddings {
+    constructor(config) {
+        // 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 || 'embed-english-v3.0';
+        this.inputType = config.inputType || 'search_document';
+        this.truncate = config.truncate || 'END';
+        this.batchSize = config.batchSize || 96;
+        // Set dimension based on model
+        this.dimension = this.getModelDimension(this.model);
+    }
+    async embed(text) {
+        try {
+            const response = await this.client.embed({
+                texts: [text],
+                model: this.model,
+                inputType: this.inputType,
+                truncate: this.truncate,
+            });
+            return response.embeddings[0];
+        }
+        catch (error) {
+            throw new Error(`Failed to generate Cohere embedding: ${error}`);
+        }
+    }
+    async embedBatch(texts) {
+        const embeddings = [];
+        // Process in batches
+        for (let i = 0; i < texts.length; i += this.batchSize) {
+            const batch = texts.slice(i, i + this.batchSize);
+            try {
+                const response = await this.client.embed({
+                    texts: batch,
+                    model: this.model,
+                    inputType: this.inputType,
+                    truncate: this.truncate,
+                });
+                embeddings.push(...response.embeddings);
+            }
+            catch (error) {
+                throw new Error(`Failed to generate Cohere batch embeddings: ${error}`);
+            }
+        }
+        return embeddings;
+    }
+    getDimension() {
+        return this.dimension;
+    }
+    /**
+     * Get embedding dimension for a given model
+     */
+    getModelDimension(model) {
+        const dimensions = {
+            'embed-english-v3.0': 1024,
+            'embed-english-light-v3.0': 384,
+            'embed-multilingual-v3.0': 1024,
+            'embed-multilingual-light-v3.0': 384,
+            'embed-english-v2.0': 4096,
+            'embed-english-light-v2.0': 1024,
+            'embed-multilingual-v2.0': 768,
+        };
+        return dimensions[model] || 1024;
+    }
+    /**
+     * Create embeddings for search queries (optimized input type)
+     */
+    async embedQuery(query) {
+        try {
+            const response = await this.client.embed({
+                texts: [query],
+                model: this.model,
+                inputType: 'search_query',
+                truncate: this.truncate,
+            });
+            return response.embeddings[0];
+        }
+        catch (error) {
+            throw new Error(`Failed to generate Cohere query embedding: ${error}`);
+        }
+    }
+}
+exports.CohereEmbeddings = CohereEmbeddings;
+//# sourceMappingURL=cohere-embeddings.js.map

package/dist/embeddings/cohere-embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere-embeddings.js","sourceRoot":"","sources":["../../src/embeddings/cohere-embeddings.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAkBH,MAAa,gBAAgB;IAQ3B,YAAY,MAA8B;QACxC,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,oBAAoB,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,iBAAiB,CAAC;QACvD,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC;QACzC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;QAExC,+BAA+B;QAC/B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACvC,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,qBAAqB;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YAEjD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;oBACvC,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;iBACxB,CAAC,CAAC;gBAEH,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,+CAA+C,KAAK,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAa;QACrC,MAAM,UAAU,GAA2B;YACzC,oBAAoB,EAAE,IAAI;YAC1B,0BAA0B,EAAE,GAAG;YAC/B,yBAAyB,EAAE,IAAI;YAC/B,+BAA+B,EAAE,GAAG;YACpC,oBAAoB,EAAE,IAAI;YAC1B,0BAA0B,EAAE,IAAI;YAChC,yBAAyB,EAAE,GAAG;SAC/B,CAAC;QAEF,OAAO,UAAU,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACvC,KAAK,EAAE,CAAC,KAAK,CAAC;gBACd,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,cAAc;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,8CAA8C,KAAK,EAAE,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;CACF;AAlGD,4CAkGC"}

package/dist/embeddings/openai-embeddings.d.ts

@@ -0,0 +1,21 @@
+/**
+ * OpenAI Embeddings Provider
+ */
+import { EmbeddingProvider } from '../types';
+export interface OpenAIEmbeddingsConfig {
+    apiKey: string;
+    model?: string;
+    dimensions?: number;
+    batchSize?: number;
+}
+export declare class OpenAIEmbeddings implements EmbeddingProvider {
+    private client;
+    private model;
+    private dimensions;
+    private batchSize;
+    constructor(config: OpenAIEmbeddingsConfig);
+    embed(text: string): Promise<number[]>;
+    embedBatch(texts: string[]): Promise<number[][]>;
+    getDimension(): number;
+}
+//# sourceMappingURL=openai-embeddings.d.ts.map

package/dist/embeddings/openai-embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embeddings.d.ts","sourceRoot":"","sources":["../../src/embeddings/openai-embeddings.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAM7C,MAAM,WAAW,sBAAsB;IACrC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,sBAAsB;IASpC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IActC,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAuBtD,YAAY,IAAI,MAAM;CAGvB"}

package/dist/embeddings/openai-embeddings.js

@@ -0,0 +1,53 @@
+"use strict";
+/**
+ * OpenAI Embeddings Provider
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIEmbeddings = void 0;
+class OpenAIEmbeddings {
+    constructor(config) {
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
+        const { OpenAI: OpenAIClass } = require('openai');
+        this.client = new OpenAIClass({ apiKey: config.apiKey });
+        this.model = config.model || 'text-embedding-3-small';
+        this.dimensions = config.dimensions || 1536;
+        this.batchSize = config.batchSize || 100;
+    }
+    async embed(text) {
+        try {
+            const response = await this.client.embeddings.create({
+                model: this.model,
+                input: text,
+                dimensions: this.dimensions,
+            });
+            return response.data[0].embedding;
+        }
+        catch (error) {
+            throw new Error(`Failed to generate embedding: ${error}`);
+        }
+    }
+    async embedBatch(texts) {
+        const embeddings = [];
+        // Process in batches
+        for (let i = 0; i < texts.length; i += this.batchSize) {
+            const batch = texts.slice(i, i + this.batchSize);
+            try {
+                const response = await this.client.embeddings.create({
+                    model: this.model,
+                    input: batch,
+                    dimensions: this.dimensions,
+                });
+                embeddings.push(...response.data.map((item) => item.embedding));
+            }
+            catch (error) {
+                throw new Error(`Failed to generate batch embeddings: ${error}`);
+            }
+        }
+        return embeddings;
+    }
+    getDimension() {
+        return this.dimensions;
+    }
+}
+exports.OpenAIEmbeddings = OpenAIEmbeddings;
+//# sourceMappingURL=openai-embeddings.js.map

package/dist/embeddings/openai-embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-embeddings.js","sourceRoot":"","sources":["../../src/embeddings/openai-embeddings.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAeH,MAAa,gBAAgB;IAM3B,YAAY,MAA8B;QACxC,iEAAiE;QACjE,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,wBAAwB,CAAC;QACtD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,GAAG,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;gBACnD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,qBAAqB;QACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;YAEjD,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;oBACnD,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,KAAK,EAAE,KAAK;oBACZ,UAAU,EAAE,IAAI,CAAC,UAAU;iBAC5B,CAAC,CAAC;gBAEH,UAAU,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAA6B,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YAC3F,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;YACnE,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAvDD,4CAuDC"}

package/dist/index.d.ts

@@ -0,0 +1,27 @@
+/**
+ * @hazeljs/rag - Retrieval-Augmented Generation
+ * Vector search and RAG capabilities for HazelJS
+ */
+export * from './types';
+export * from './rag-pipeline';
+export * from './rag.service';
+export * from './rag.module';
+export * from './decorators/rag.decorator';
+export * from './decorators/embeddable.decorator';
+export * from './decorators/semantic-search.decorator';
+export * from './embeddings/openai-embeddings';
+export * from './embeddings/cohere-embeddings';
+export * from './vector-stores/memory-vector-store';
+export * from './vector-stores/pinecone.store';
+export * from './vector-stores/qdrant.store';
+export * from './vector-stores/weaviate.store';
+export * from './vector-stores/chroma.store';
+export * from './text-splitters/recursive-text-splitter';
+export * from './retrieval/bm25';
+export * from './retrieval/hybrid-search';
+export * from './retrieval/multi-query';
+export * from './utils/similarity';
+export * from './memory';
+export * from './rag-pipeline-with-memory';
+export * from './agentic';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,cAAc,SAAS,CAAC;AAGxB,cAAc,gBAAgB,CAAC;AAG/B,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAG7B,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mCAAmC,CAAC;AAClD,cAAc,wCAAwC,CAAC;AAGvD,cAAc,gCAAgC,CAAC;AAC/C,cAAc,gCAAgC,CAAC;AAG/C,cAAc,qCAAqC,CAAC;AACpD,cAAc,gCAAgC,CAAC;AAC/C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,8BAA8B,CAAC;AAG7C,cAAc,0CAA0C,CAAC;AAGzD,cAAc,kBAAkB,CAAC;AACjC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,yBAAyB,CAAC;AAGxC,cAAc,oBAAoB,CAAC;AAGnC,cAAc,UAAU,CAAC;AACzB,cAAc,4BAA4B,CAAC;AAG3C,cAAc,WAAW,CAAC"}

package/dist/index.js

@@ -0,0 +1,54 @@
+"use strict";
+/**
+ * @hazeljs/rag - Retrieval-Augmented Generation
+ * Vector search and RAG capabilities for 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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+// Core types
+__exportStar(require("./types"), exports);
+// RAG Pipeline
+__exportStar(require("./rag-pipeline"), exports);
+// RAG Service & Module
+__exportStar(require("./rag.service"), exports);
+__exportStar(require("./rag.module"), exports);
+// Decorators
+__exportStar(require("./decorators/rag.decorator"), exports);
+__exportStar(require("./decorators/embeddable.decorator"), exports);
+__exportStar(require("./decorators/semantic-search.decorator"), exports);
+// Embeddings
+__exportStar(require("./embeddings/openai-embeddings"), exports);
+__exportStar(require("./embeddings/cohere-embeddings"), exports);
+// Vector Stores
+__exportStar(require("./vector-stores/memory-vector-store"), exports);
+__exportStar(require("./vector-stores/pinecone.store"), exports);
+__exportStar(require("./vector-stores/qdrant.store"), exports);
+__exportStar(require("./vector-stores/weaviate.store"), exports);
+__exportStar(require("./vector-stores/chroma.store"), exports);
+// Text Splitters
+__exportStar(require("./text-splitters/recursive-text-splitter"), exports);
+// Retrieval Strategies
+__exportStar(require("./retrieval/bm25"), exports);
+__exportStar(require("./retrieval/hybrid-search"), exports);
+__exportStar(require("./retrieval/multi-query"), exports);
+// Utils
+__exportStar(require("./utils/similarity"), exports);
+// Memory
+__exportStar(require("./memory"), exports);
+__exportStar(require("./rag-pipeline-with-memory"), exports);
+// Agentic RAG - Advanced autonomous retrieval capabilities
+__exportStar(require("./agentic"), exports);
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;;;;;AAEH,aAAa;AACb,0CAAwB;AAExB,eAAe;AACf,iDAA+B;AAE/B,uBAAuB;AACvB,gDAA8B;AAC9B,+CAA6B;AAE7B,aAAa;AACb,6DAA2C;AAC3C,oEAAkD;AAClD,yEAAuD;AAEvD,aAAa;AACb,iEAA+C;AAC/C,iEAA+C;AAE/C,gBAAgB;AAChB,sEAAoD;AACpD,iEAA+C;AAC/C,+DAA6C;AAC7C,iEAA+C;AAC/C,+DAA6C;AAE7C,iBAAiB;AACjB,2EAAyD;AAEzD,uBAAuB;AACvB,mDAAiC;AACjC,4DAA0C;AAC1C,0DAAwC;AAExC,QAAQ;AACR,qDAAmC;AAEnC,SAAS;AACT,2CAAyB;AACzB,6DAA2C;AAE3C,2DAA2D;AAC3D,4CAA0B"}

package/dist/memory/index.d.ts

@@ -0,0 +1,11 @@
+/**
+ * @hazeljs/rag - Memory Module
+ * Persistent context and conversation management
+ */
+export * from './types';
+export * from './memory-store.interface';
+export * from './stores/buffer-memory';
+export * from './stores/vector-memory';
+export * from './stores/hybrid-memory';
+export * from './memory-manager';
+//# sourceMappingURL=index.d.ts.map

package/dist/memory/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,cAAc,SAAS,CAAC;AAGxB,cAAc,0BAA0B,CAAC;AAGzC,cAAc,wBAAwB,CAAC;AACvC,cAAc,wBAAwB,CAAC;AACvC,cAAc,wBAAwB,CAAC;AAGvC,cAAc,kBAAkB,CAAC"}

package/dist/memory/index.js

@@ -0,0 +1,31 @@
+"use strict";
+/**
+ * @hazeljs/rag - Memory Module
+ * Persistent context and conversation management
+ */
+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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+// Types
+__exportStar(require("./types"), exports);
+// Core interfaces
+__exportStar(require("./memory-store.interface"), exports);
+// Memory stores
+__exportStar(require("./stores/buffer-memory"), exports);
+__exportStar(require("./stores/vector-memory"), exports);
+__exportStar(require("./stores/hybrid-memory"), exports);
+// Memory manager
+__exportStar(require("./memory-manager"), exports);
+//# sourceMappingURL=index.js.map

package/dist/memory/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;;;;;AAEH,QAAQ;AACR,0CAAwB;AAExB,kBAAkB;AAClB,2DAAyC;AAEzC,gBAAgB;AAChB,yDAAuC;AACvC,yDAAuC;AACvC,yDAAuC;AAEvC,iBAAiB;AACjB,mDAAiC"}

package/dist/memory/memory-manager.d.ts

@@ -0,0 +1,96 @@
+/**
+ * Memory Manager - High-level interface for managing different types of memory
+ */
+import { MemoryStore } from './memory-store.interface';
+import { Memory, MemoryConfig, Message, Entity, MemoryQuery, MemorySearchOptions } from './types';
+/**
+ * Central manager for all memory operations
+ * Provides high-level APIs for conversation, entity, fact, and working memory
+ */
+export declare class MemoryManager {
+    private store;
+    private config;
+    constructor(store: MemoryStore, config?: MemoryConfig);
+    initialize(): Promise<void>;
+    /**
+     * Add a message to conversation history
+     */
+    addMessage(message: Message, sessionId: string, userId?: string): Promise<string>;
+    /**
+     * Get conversation history for a session
+     */
+    getConversationHistory(sessionId: string, limit?: number): Promise<Message[]>;
+    /**
+     * Summarize conversation history
+     */
+    summarizeConversation(sessionId: string): Promise<string>;
+    /**
+     * Clear conversation history for a session
+     */
+    clearConversation(sessionId: string): Promise<void>;
+    /**
+     * Track an entity mentioned in conversation
+     */
+    trackEntity(entity: Entity, sessionId?: string): Promise<void>;
+    /**
+     * Get entity by name
+     */
+    getEntity(name: string): Promise<Entity | null>;
+    /**
+     * Update entity information
+     */
+    updateEntity(name: string, updates: Partial<Entity>): Promise<void>;
+    /**
+     * Get all tracked entities
+     */
+    getAllEntities(sessionId?: string): Promise<Entity[]>;
+    /**
+     * Store a fact or piece of knowledge
+     */
+    storeFact(fact: string, metadata?: Record<string, unknown>): Promise<string>;
+    /**
+     * Recall facts related to a query
+     */
+    recallFacts(query: string, options?: MemorySearchOptions): Promise<string[]>;
+    /**
+     * Update a fact
+     */
+    updateFact(id: string, newContent: string): Promise<void>;
+    /**
+     * Set a value in working memory (temporary context)
+     */
+    setContext(key: string, value: unknown, sessionId: string): Promise<void>;
+    /**
+     * Get a value from working memory
+     */
+    getContext(key: string, sessionId: string): Promise<unknown>;
+    /**
+     * Clear working memory for a session
+     */
+    clearContext(sessionId: string): Promise<void>;
+    /**
+     * Get relevant memories for a query
+     */
+    relevantMemories(query: string, options: MemorySearchOptions): Promise<Memory[]>;
+    /**
+     * Get memories by query parameters
+     */
+    getMemories(query: MemoryQuery): Promise<Memory[]>;
+    /**
+     * Get memory statistics
+     */
+    getStats(sessionId?: string): Promise<import('./types').MemoryStats>;
+    /**
+     * Calculate importance score for content
+     */
+    private calculateImportance;
+    /**
+     * Check if conversation needs summarization
+     */
+    private checkAndSummarize;
+    /**
+     * Prune working memory to max size
+     */
+    private pruneWorkingMemory;
+}
+//# sourceMappingURL=memory-manager.d.ts.map