@retrivora-ai/rag-engine 0.4.5 → 1.0.1

package/dist/{chunk-H6RKMU7W.mjs → chunk-5YGUXK7Z.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/RedisProvider.ts
 import axios from "axios";

package/dist/{chunk-PQKTC73Y.mjs → chunk-CD6TSNL4.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/MongoDBProvider.ts
 import { MongoClient } from "mongodb";
@@ -21,6 +21,71 @@ var MongoDBProvider = class extends BaseVectorProvider {
     this.contentKey = opts.contentKey || "content";
     this.metadataKey = opts.metadataKey || "metadata";
   }
+  static getValidator() {
+    return {
+      validate(config) {
+        const errors = [];
+        const opts = config.options || {};
+        if (!opts.uri) {
+          errors.push({
+            field: "vectorDb.options.uri",
+            message: "MongoDB connection URI is required",
+            suggestion: "Set MONGODB_URI environment variable",
+            severity: "error"
+          });
+        }
+        if (!opts.database) {
+          errors.push({
+            field: "vectorDb.options.database",
+            message: "MongoDB database name is required",
+            severity: "error"
+          });
+        }
+        if (!opts.collection) {
+          errors.push({
+            field: "vectorDb.options.collection",
+            message: "MongoDB collection name is required",
+            severity: "error"
+          });
+        }
+        return errors;
+      }
+    };
+  }
+  static getHealthChecker() {
+    return {
+      async check(config) {
+        const opts = config.options || {};
+        const timestamp = Date.now();
+        try {
+          const { MongoClient: MongoClient2 } = await import("mongodb");
+          const client = new MongoClient2(opts.uri);
+          await client.connect();
+          const db = client.db(opts.database);
+          await db.command({ ping: 1 });
+          const collections = await db.listCollections({ name: opts.collection }).toArray();
+          await client.close();
+          return {
+            healthy: true,
+            provider: "mongodb",
+            capabilities: {
+              database: opts.database,
+              collection: opts.collection,
+              exists: collections.length > 0
+            },
+            timestamp
+          };
+        } catch (error) {
+          return {
+            healthy: false,
+            provider: "mongodb",
+            error: `Connection failed: ${error instanceof Error ? error.message : String(error)}`,
+            timestamp
+          };
+        }
+      }
+    };
+  }
   async initialize() {
     await this.client.connect();
     this.db = this.client.db(this.dbName);
@@ -59,7 +124,7 @@ var MongoDBProvider = class extends BaseVectorProvider {
           index: this.config.indexName || "vector_index",
           path: this.embeddingKey,
           queryVector: vector,
-          numCandidates: Math.max(topK * 10, 100),
+          numCandidates: this.config.options.numCandidates || Math.max(topK * 20, 200),
           limit: topK
         }, Object.keys(publicFilter).length > 0 || namespace ? { filter: __spreadValues(__spreadValues({}, publicFilter), namespace ? { namespace } : {}) } : {})
       },
@@ -87,10 +152,6 @@ var MongoDBProvider = class extends BaseVectorProvider {
   async deleteNamespace(namespace) {
     await this.collection.deleteMany({ namespace });
   }
-  /**
-   * Sanitise and flatten filter for MongoDB.
-   * Strips internal engine fields and keywords.
-   */
   sanitizeFilter(filter) {
     const sanitized = super.sanitizeFilter(filter);
     const mongoFilter = {};

package/dist/{chunk-RK2UDJA2.mjs → chunk-CFVEZTBJ.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/ChromaDBProvider.ts
 import axios from "axios";

package/dist/{chunk-LJWWPTWE.mjs → chunk-FLOSGE6A.mjs}

@@ -14,6 +14,61 @@ var PostgreSQLProvider = class extends BaseVectorProvider {
     this.connectionString = opts.connectionString;
     this.dimensions = (_a = opts.dimensions) != null ? _a : 1536;
   }
+  static getValidator() {
+    return {
+      validate(config) {
+        const errors = [];
+        const opts = config.options || {};
+        if (!opts.connectionString) {
+          errors.push({
+            field: "vectorDb.options.connectionString",
+            message: "PostgreSQL connection string is required",
+            suggestion: "Set PGVECTOR_CONNECTION_STRING environment variable",
+            severity: "error"
+          });
+        }
+        if (opts.tables && typeof opts.tables !== "string" && !Array.isArray(opts.tables)) {
+          errors.push({
+            field: "vectorDb.options.tables",
+            message: "PostgreSQL tables must be a string or a string array",
+            severity: "error"
+          });
+        }
+        return errors;
+      }
+    };
+  }
+  static getHealthChecker() {
+    return {
+      async check(config) {
+        const opts = config.options || {};
+        const timestamp = Date.now();
+        try {
+          const { Client } = await import("pg");
+          const client = new Client({ connectionString: opts.connectionString });
+          await client.connect();
+          const result = await client.query(`
+            SELECT EXISTS(SELECT 1 FROM pg_extension WHERE extname = 'vector');
+          `);
+          const hasVector = result.rows[0].exists;
+          await client.end();
+          return {
+            healthy: true,
+            provider: "postgresql",
+            capabilities: { pgvectorInstalled: hasVector },
+            timestamp
+          };
+        } catch (error) {
+          return {
+            healthy: false,
+            provider: "postgresql",
+            error: `Connection failed: ${error instanceof Error ? error.message : String(error)}`,
+            timestamp
+          };
+        }
+      }
+    };
+  }
   async initialize() {
     this.pool = new Pool({ connectionString: this.connectionString });
     const client = await this.pool.connect();
@@ -99,20 +154,27 @@ var PostgreSQLProvider = class extends BaseVectorProvider {
       }).join(" AND ");
       whereClause = whereClause ? `${whereClause} AND ${filterConditions}` : `WHERE ${filterConditions}`;
     }
-    const result = await this.pool.query(
-      `SELECT id, content, metadata, 1 - (embedding <=> $1::vector) AS score
-       FROM ${this.tableName}
-       ${whereClause}
-       ORDER BY embedding <=> $1::vector
-       LIMIT $2`,
-      params
-    );
-    return result.rows.map((row) => ({
-      id: String(row["id"]),
-      score: parseFloat(String(row["score"])),
-      content: String(row["content"]),
-      metadata: row["metadata"]
-    }));
+    const client = await this.pool.connect();
+    try {
+      const efSearch = this.config.options.efSearch || Math.max(topK * 10, 40);
+      await client.query(`SET LOCAL hnsw.ef_search = ${efSearch}`);
+      const result = await client.query(
+        `SELECT id, content, metadata, 1 - (embedding <=> $1::vector) AS score
+         FROM ${this.tableName}
+         ${whereClause}
+         ORDER BY embedding <=> $1::vector
+         LIMIT $2`,
+        params
+      );
+      return result.rows.map((row) => ({
+        id: String(row["id"]),
+        score: parseFloat(String(row["score"])),
+        content: String(row["content"]),
+        metadata: row["metadata"]
+      }));
+    } finally {
+      client.release();
+    }
   }
   async delete(id, namespace) {
     const where = namespace ? "WHERE id = $1 AND namespace = $2" : "WHERE id = $1";

package/dist/{chunk-GQT5LF4G.mjs → chunk-LR3VMDVK.mjs}

@@ -1,13 +1,13 @@
 import {
   buildPayload,
   resolvePath
-} from "./chunk-EDLTMSNY.mjs";
+} from "./chunk-YLTMFW4M.mjs";
 import {
   BaseVectorProvider
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/UniversalVectorProvider.ts
 import axios from "axios";

package/dist/{chunk-KTS3LLHY.mjs → chunk-M6JSPGAR.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/WeaviateProvider.ts
 import axios from "axios";
@@ -54,17 +54,17 @@ var WeaviateProvider = class extends BaseVectorProvider {
     };
     await this.http.post("/v1/batch/objects", payload);
   }
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async query(vector, topK, namespace, _filter) {
     var _a, _b;
+    const sanitizedFilter = this.sanitizeFilter(_filter);
+    const queryText = sanitizedFilter.queryText;
+    const searchParams = queryText ? `hybrid: { query: ${JSON.stringify(queryText)}, alpha: 0.5 }` : `nearVector: { vector: ${JSON.stringify(vector)} }`;
     const graphqlQuery = {
       query: `
         {
           Get {
             ${this.indexName}(
-              nearVector: {
-                vector: ${JSON.stringify(vector)}
-              }
+              ${searchParams}
               limit: ${topK}
               ${namespace ? `where: { path: ["namespace"], operator: Equal, valueString: "${namespace}" }` : ""}
             ) {

package/dist/{chunk-3QWAK3RZ.mjs → chunk-U55XRW3U.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/MilvusProvider.ts
 import axios from "axios";
@@ -52,7 +52,11 @@ var MilvusProvider = class extends BaseVectorProvider {
       vector,
       limit: topK,
       filter: namespace ? `namespace == "${namespace}"` : void 0,
-      outputFields: ["content", "metadata"]
+      outputFields: ["content", "metadata"],
+      searchParams: {
+        nprobe: this.config.options.nprobe || 16,
+        ef: this.config.options.efSearch || Math.max(topK * 10, 64)
+      }
     };
     const { data } = await this.http.post("/v1/vector/search", payload);
     return (data.data || []).map((res) => ({

package/dist/{chunk-XCNXPECE.mjs → chunk-VUQJVIJT.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/PineconeProvider.ts
 import { Pinecone } from "@pinecone-database/pinecone";
@@ -14,6 +14,60 @@ var PineconeProvider = class extends BaseVectorProvider {
     if (!opts.apiKey) throw new Error("[PineconeProvider] options.apiKey is required");
     this.apiKey = opts.apiKey;
   }
+  static getValidator() {
+    return {
+      validate(config) {
+        const errors = [];
+        const opts = config.options || {};
+        if (!opts.apiKey) {
+          errors.push({
+            field: "vectorDb.options.apiKey",
+            message: "Pinecone API key is required",
+            suggestion: "Set PINECONE_API_KEY environment variable",
+            severity: "error"
+          });
+        }
+        return errors;
+      }
+    };
+  }
+  static getHealthChecker() {
+    return {
+      async check(config) {
+        var _a, _b;
+        const opts = config.options || {};
+        const indexName = config.indexName;
+        const timestamp = Date.now();
+        try {
+          const { Pinecone: Pinecone2 } = await import("@pinecone-database/pinecone");
+          const client = new Pinecone2({ apiKey: opts.apiKey });
+          const indexes = await client.listIndexes();
+          const indexNames = (_b = (_a = indexes.indexes) == null ? void 0 : _a.map((i) => i.name)) != null ? _b : [];
+          if (!indexNames.includes(indexName)) {
+            return {
+              healthy: false,
+              provider: "pinecone",
+              error: `Index "${indexName}" not found. Available: ${indexNames.join(", ")}`,
+              timestamp
+            };
+          }
+          return {
+            healthy: true,
+            provider: "pinecone",
+            capabilities: { indexes: indexNames.length, targetIndex: indexName },
+            timestamp
+          };
+        } catch (error) {
+          return {
+            healthy: false,
+            provider: "pinecone",
+            error: `Connection failed: ${error instanceof Error ? error.message : String(error)}`,
+            timestamp
+          };
+        }
+      }
+    };
+  }
   async initialize() {
     var _a, _b;
     this.client = new Pinecone({ apiKey: this.apiKey });

package/dist/chunk-X4TOT24V.mjs

@@ -0,0 +1,89 @@
+var __defProp = Object.defineProperty;
+var __defProps = Object.defineProperties;
+var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __knownSymbol = (name, symbol) => (symbol = Symbol[name]) ? symbol : /* @__PURE__ */ Symbol.for("Symbol." + name);
+var __typeError = (msg) => {
+  throw TypeError(msg);
+};
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+var __objRest = (source, exclude) => {
+  var target = {};
+  for (var prop in source)
+    if (__hasOwnProp.call(source, prop) && exclude.indexOf(prop) < 0)
+      target[prop] = source[prop];
+  if (source != null && __getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(source)) {
+      if (exclude.indexOf(prop) < 0 && __propIsEnum.call(source, prop))
+        target[prop] = source[prop];
+    }
+  return target;
+};
+var __await = function(promise, isYieldStar) {
+  this[0] = promise;
+  this[1] = isYieldStar;
+};
+var __asyncGenerator = (__this, __arguments, generator) => {
+  var resume = (k, v, yes, no) => {
+    try {
+      var x = generator[k](v), isAwait = (v = x.value) instanceof __await, done = x.done;
+      Promise.resolve(isAwait ? v[0] : v).then((y) => isAwait ? resume(k === "return" ? k : "next", v[1] ? { done: y.done, value: y.value } : y, yes, no) : yes({ value: y, done })).catch((e) => resume("throw", e, yes, no));
+    } catch (e) {
+      no(e);
+    }
+  }, method = (k, call, wait, clear) => it[k] = (x) => (call = new Promise((yes, no, run) => (run = () => resume(k, x, yes, no), q ? q.then(run) : run())), clear = () => q === wait && (q = 0), q = wait = call.then(clear, clear), call), q, it = {};
+  return generator = generator.apply(__this, __arguments), it[__knownSymbol("asyncIterator")] = () => it, method("next"), method("throw"), method("return"), it;
+};
+var __yieldStar = (value) => {
+  var obj = value[__knownSymbol("asyncIterator")], isAwait = false, method, it = {};
+  if (obj == null) {
+    obj = value[__knownSymbol("iterator")]();
+    method = (k) => it[k] = (x) => obj[k](x);
+  } else {
+    obj = obj.call(value);
+    method = (k) => it[k] = (v) => {
+      if (isAwait) {
+        isAwait = false;
+        if (k === "throw") throw v;
+        return v;
+      }
+      isAwait = true;
+      return {
+        done: false,
+        value: new __await(new Promise((resolve) => {
+          var x = obj[k](v);
+          if (!(x instanceof Object)) __typeError("Object expected");
+          resolve(x);
+        }), 1)
+      };
+    };
+  }
+  return it[__knownSymbol("iterator")] = () => it, method("next"), "throw" in obj ? method("throw") : it.throw = (x) => {
+    throw x;
+  }, "return" in obj && method("return"), it;
+};
+var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")]) ? it.call(obj) : (obj = obj[__knownSymbol("iterator")](), it = {}, method = (key, fn) => (fn = obj[key]) && (it[key] = (arg) => new Promise((yes, no, done) => (arg = fn.call(obj, arg), done = arg.done, Promise.resolve(arg.value).then((value) => yes({ value, done }), no)))), method("next"), method("return"), it);
+
+export {
+  __spreadValues,
+  __spreadProps,
+  __objRest,
+  __await,
+  __asyncGenerator,
+  __yieldStar,
+  __forAwait
+};

package/dist/{chunk-EDLTMSNY.mjs → chunk-YLTMFW4M.mjs}

@@ -1,6 +1,6 @@
 import {
   __spreadValues
-} from "./chunk-FWCSY2DS.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/utils/templateUtils.ts
 function isRecord(value) {

package/dist/handlers/index.d.mts

@@ -1,3 +1,3 @@
-import '../RagConfig-BgRDL9Vy.mjs';
-export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-7qeLTPBL.mjs';
+import '../RagConfig-DRJO4hGU.mjs';
+export { c as createChatHandler, b as createHealthHandler, d as createIngestHandler, f as createStreamHandler, e as createUploadHandler } from '../index-Bjy0es5a.mjs';
 import 'next/server';

package/dist/handlers/index.d.ts

@@ -1,3 +1,3 @@
-import '../RagConfig-BgRDL9Vy.js';
-export { c as createChatHandler, a as createHealthHandler, b as createIngestHandler, d as createUploadHandler } from '../index-DowY4_K0.js';
+import '../RagConfig-DRJO4hGU.js';
+export { c as createChatHandler, b as createHealthHandler, d as createIngestHandler, f as createStreamHandler, e as createUploadHandler } from '../index-B2mutkgp.js';
 import 'next/server';