@retrivora-ai/rag-engine 1.0.0 → 1.0.2

package/dist/{ChromaDBProvider-APQVJ5F7.mjs → ChromaDBProvider-MIDOR4FW.mjs}

@@ -1,8 +1,8 @@
 import {
   ChromaDBProvider
-} from "./chunk-WGSZNY3X.mjs";
+} from "./chunk-CFVEZTBJ.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   ChromaDBProvider
 };

package/dist/{MilvusProvider-35US67MS.mjs → MilvusProvider-U7SKC27V.mjs}

@@ -1,8 +1,8 @@
 import {
   MilvusProvider
-} from "./chunk-TYHTZIDP.mjs";
+} from "./chunk-U55XRW3U.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   MilvusProvider
 };

package/dist/{MongoDBProvider-COVYZDP6.mjs → MongoDBProvider-BO2Y5DRR.mjs}

@@ -1,8 +1,8 @@
 import {
   MongoDBProvider
-} from "./chunk-7SOSCZGS.mjs";
+} from "./chunk-73I6VWU3.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   MongoDBProvider
 };

package/dist/{PineconeProvider-AWFJQDZL.mjs → PineconeProvider-QZNRKTN2.mjs}

@@ -1,8 +1,8 @@
 import {
   PineconeProvider
-} from "./chunk-ZNBKHNJ4.mjs";
+} from "./chunk-VUQJVIJT.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   PineconeProvider
 };

package/dist/{PostgreSQLProvider-IEYRJ7XJ.mjs → PostgreSQLProvider-BMOETDZA.mjs}

@@ -2,7 +2,7 @@ import {
   PostgreSQLProvider
 } from "./chunk-FLOSGE6A.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   PostgreSQLProvider
 };

package/dist/{QdrantProvider-M6TQYZRO.mjs → QdrantProvider-YNUNEOZH.mjs}

@@ -1,8 +1,8 @@
 import {
   QdrantProvider
-} from "./chunk-4A47RCG2.mjs";
+} from "./chunk-3DSHW676.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   QdrantProvider
 };

package/dist/{RedisProvider-3G5PBLZ4.mjs → RedisProvider-SR65SCKV.mjs}

@@ -1,8 +1,8 @@
 import {
   RedisProvider
-} from "./chunk-NXUCKY5L.mjs";
+} from "./chunk-5YGUXK7Z.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   RedisProvider
 };

package/dist/{SimpleGraphProvider-UK7DJW37.mjs → SimpleGraphProvider-SLOXO4M7.mjs}

@@ -1,4 +1,4 @@
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 
 // src/providers/graphdb/BaseGraphProvider.ts
 var BaseGraphProvider = class {

package/dist/{UniversalVectorProvider-FYQ3B2PW.mjs → UniversalVectorProvider-IN67OS56.mjs}

@@ -1,9 +1,9 @@
 import {
   UniversalVectorProvider
-} from "./chunk-U6KHVZLF.mjs";
-import "./chunk-67AJ6SMD.mjs";
+} from "./chunk-LR3VMDVK.mjs";
+import "./chunk-YLTMFW4M.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   UniversalVectorProvider
 };

package/dist/{WeaviateProvider-ITHO36IL.mjs → WeaviateProvider-5FWDFITI.mjs}

@@ -1,8 +1,8 @@
 import {
   WeaviateProvider
-} from "./chunk-OOQXNLXD.mjs";
+} from "./chunk-M6JSPGAR.mjs";
 import "./chunk-IMP6FUCY.mjs";
-import "./chunk-QMIKLALV.mjs";
+import "./chunk-X4TOT24V.mjs";
 export {
   WeaviateProvider
 };

package/dist/{chunk-4A47RCG2.mjs → chunk-3DSHW676.mjs}

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

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

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

package/dist/{chunk-7SOSCZGS.mjs → chunk-73I6VWU3.mjs}

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-QMIKLALV.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/MongoDBProvider.ts
 import { MongoClient } from "mongodb";
@@ -14,7 +14,7 @@ var MongoDBProvider = class extends BaseVectorProvider {
     if (!opts.uri || !opts.database || !opts.collection) {
       throw new Error("[MongoDBProvider] options uri, database, and collection are required.");
     }
-    this.client = new MongoClient(opts.uri);
+    this.client = new MongoClient(opts.uri, { serverSelectionTimeoutMS: 5e3 });
     this.dbName = opts.database;
     this.collectionName = opts.collection;
     this.embeddingKey = opts.embeddingKey || "embedding";

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

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

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

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

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

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

package/dist/{chunk-P4HAQ7KB.mjs → chunk-OJNAKSQ2.mjs}

@@ -2,14 +2,15 @@ import {
   buildPayload,
   mergeDefined,
   resolvePath
-} from "./chunk-67AJ6SMD.mjs";
+} from "./chunk-YLTMFW4M.mjs";
 import {
   __asyncGenerator,
   __await,
   __forAwait,
   __spreadProps,
-  __spreadValues
-} from "./chunk-QMIKLALV.mjs";
+  __spreadValues,
+  __yieldStar
+} from "./chunk-X4TOT24V.mjs";
 
 // src/handlers/index.ts
 import { NextResponse } from "next/server";
@@ -1078,41 +1079,41 @@ var ProviderRegistry = class {
     if (this.vectorProviders[provider]) return this.vectorProviders[provider];
     switch (provider) {
       case "pinecone": {
-        const { PineconeProvider } = await import("./PineconeProvider-AWFJQDZL.mjs");
+        const { PineconeProvider } = await import("./PineconeProvider-QZNRKTN2.mjs");
         return PineconeProvider;
       }
       case "pgvector":
       case "postgresql": {
-        const { PostgreSQLProvider } = await import("./PostgreSQLProvider-IEYRJ7XJ.mjs");
+        const { PostgreSQLProvider } = await import("./PostgreSQLProvider-BMOETDZA.mjs");
         return PostgreSQLProvider;
       }
       case "mongodb": {
-        const { MongoDBProvider } = await import("./MongoDBProvider-COVYZDP6.mjs");
+        const { MongoDBProvider } = await import("./MongoDBProvider-BO2Y5DRR.mjs");
         return MongoDBProvider;
       }
       case "milvus": {
-        const { MilvusProvider } = await import("./MilvusProvider-35US67MS.mjs");
+        const { MilvusProvider } = await import("./MilvusProvider-U7SKC27V.mjs");
         return MilvusProvider;
       }
       case "qdrant": {
-        const { QdrantProvider } = await import("./QdrantProvider-M6TQYZRO.mjs");
+        const { QdrantProvider } = await import("./QdrantProvider-YNUNEOZH.mjs");
         return QdrantProvider;
       }
       case "chromadb": {
-        const { ChromaDBProvider } = await import("./ChromaDBProvider-APQVJ5F7.mjs");
+        const { ChromaDBProvider } = await import("./ChromaDBProvider-MIDOR4FW.mjs");
         return ChromaDBProvider;
       }
       case "redis": {
-        const { RedisProvider } = await import("./RedisProvider-3G5PBLZ4.mjs");
+        const { RedisProvider } = await import("./RedisProvider-SR65SCKV.mjs");
         return RedisProvider;
       }
       case "weaviate": {
-        const { WeaviateProvider } = await import("./WeaviateProvider-ITHO36IL.mjs");
+        const { WeaviateProvider } = await import("./WeaviateProvider-5FWDFITI.mjs");
         return WeaviateProvider;
       }
       case "universal_rest":
       case "rest": {
-        const { UniversalVectorProvider } = await import("./UniversalVectorProvider-FYQ3B2PW.mjs");
+        const { UniversalVectorProvider } = await import("./UniversalVectorProvider-IN67OS56.mjs");
         return UniversalVectorProvider;
       }
       default:
@@ -1130,7 +1131,7 @@ var ProviderRegistry = class {
     }
     switch (provider) {
       case "simple": {
-        const { SimpleGraphProvider } = await import("./SimpleGraphProvider-UK7DJW37.mjs");
+        const { SimpleGraphProvider } = await import("./SimpleGraphProvider-SLOXO4M7.mjs");
         return new SimpleGraphProvider(config);
       }
       default:
@@ -1393,6 +1394,9 @@ Extract entities and relationships from the following text.
 Format the output as JSON with two keys: "nodes" (array of {id, label, properties}) and "edges" (array of {source, target, type, properties}).
 Use the same ID for the same entity.
 
+IMPORTANT: Ensure all property values are valid JSON types (strings, numbers, booleans, or null). 
+DO NOT include mathematical expressions like "4.5/5" as numbers; use strings instead.
+
 Text:
 "${text}"
 
@@ -1973,6 +1977,7 @@ var Pipeline = class {
   async initialize() {
     var _a;
     if (this.initialised) return;
+    console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);
     const { llmProvider, embeddingProvider } = await EmbeddingStrategyResolver.resolve(
       this.config.llm,
@@ -2357,29 +2362,9 @@ var ProviderHealthCheck = class {
 
 // src/core/VectorPlugin.ts
 var VectorPlugin = class {
-  /**
-   * Initializes the plugin with the host configuration.
-   * @param hostConfig - Configuration object passed from the host application.
-   * @throws Error if configuration is invalid or providers are unhealthy
-   *
-   * @example
-   * const plugin = new VectorPlugin({
-   *   projectId: 'my-app',
-   *   vectorDb: {
-   *     provider: 'pinecone',
-   *     indexName: 'my-index',
-   *     options: { apiKey: process.env.PINECONE_API_KEY }
-   *   },
-   *   llm: {
-   *     provider: 'openai',
-   *     model: 'gpt-4o',
-   *     apiKey: process.env.OPENAI_API_KEY
-   *   }
-   * });
-   */
   constructor(hostConfig) {
     this.config = ConfigResolver.resolve(hostConfig);
-    ConfigValidator.validateAndThrow(this.config);
+    this.validationPromise = ConfigValidator.validateAndThrow(this.config);
     this.pipeline = new Pipeline(this.config);
   }
   /**
@@ -2405,12 +2390,23 @@ var VectorPlugin = class {
    * Run a chat query.
    */
   async chat(message, history = [], namespace) {
+    await this.validationPromise;
     return this.pipeline.ask(message, history, namespace);
   }
+  /**
+   * Run a streaming chat query.
+   */
+  chatStream(_0) {
+    return __asyncGenerator(this, arguments, function* (message, history = [], namespace) {
+      yield new __await(this.validationPromise);
+      yield* __yieldStar(this.pipeline.askStream(message, history, namespace));
+    });
+  }
   /**
    * Ingest documents into the vector database.
    */
   async ingest(documents, namespace) {
+    await this.validationPromise;
     return this.pipeline.ingest(documents, namespace);
   }
 };
@@ -2492,6 +2488,65 @@ function createChatHandler(config) {
     }
   };
 }
+function createStreamHandler(config) {
+  const plugin = new VectorPlugin(config);
+  return async function POST(req) {
+    try {
+      const body = await req.json();
+      const { message, history = [], namespace } = body;
+      const encoder = new TextEncoder();
+      const stream = new ReadableStream({
+        async start(controller) {
+          try {
+            const pipelineStream = plugin.chatStream(message, history, namespace);
+            try {
+              for (var iter = __forAwait(pipelineStream), more, temp, error; more = !(temp = await iter.next()).done; more = false) {
+                const chunk = temp.value;
+                if (typeof chunk === "string") {
+                  controller.enqueue(encoder.encode(chunk));
+                } else {
+                  controller.enqueue(encoder.encode(`
+
+__METADATA__${JSON.stringify(chunk)}`));
+                }
+              }
+            } catch (temp) {
+              error = [temp];
+            } finally {
+              try {
+                more && (temp = iter.return) && await temp.call(iter);
+              } finally {
+                if (error)
+                  throw error[0];
+              }
+            }
+            controller.close();
+          } catch (streamError) {
+            console.error("[createStreamHandler] Stream processing error:", streamError);
+            const errorMessage = streamError instanceof Error ? streamError.message : String(streamError);
+            controller.enqueue(encoder.encode(`
+
+__ERROR__${JSON.stringify({ error: errorMessage })}`));
+            controller.close();
+          }
+        }
+      });
+      return new Response(stream, {
+        headers: {
+          "Content-Type": "text/event-stream",
+          "Cache-Control": "no-cache",
+          "Connection": "keep-alive"
+        }
+      });
+    } catch (err) {
+      const message = err instanceof Error ? err.message : "Internal server error";
+      return new Response(JSON.stringify({ error: message }), {
+        status: 500,
+        headers: { "Content-Type": "application/json" }
+      });
+    }
+  };
+}
 function createIngestHandler(config) {
   const plugin = new VectorPlugin(config);
   return async function POST(req) {
@@ -2581,6 +2636,7 @@ export {
   VectorPlugin,
   DocumentParser,
   createChatHandler,
+  createStreamHandler,
   createIngestHandler,
   createHealthHandler,
   createUploadHandler

package/dist/{chunk-TYHTZIDP.mjs → chunk-U55XRW3U.mjs}

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

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

@@ -3,7 +3,7 @@ import {
 } from "./chunk-IMP6FUCY.mjs";
 import {
   __spreadValues
-} from "./chunk-QMIKLALV.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/providers/vectordb/PineconeProvider.ts
 import { Pinecone } from "@pinecone-database/pinecone";

package/dist/{chunk-QMIKLALV.mjs → chunk-X4TOT24V.mjs}

@@ -5,6 +5,9 @@ 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 = {}))
@@ -45,6 +48,34 @@ var __asyncGenerator = (__this, __arguments, generator) => {
   }, 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 {
@@ -53,5 +84,6 @@ export {
   __objRest,
   __await,
   __asyncGenerator,
+  __yieldStar,
   __forAwait
 };

package/dist/{chunk-67AJ6SMD.mjs → chunk-YLTMFW4M.mjs}

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

package/dist/handlers/index.d.mts

@@ -1,3 +1,3 @@
 import '../RagConfig-DRJO4hGU.mjs';
-export { c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from '../index-v669iV-k.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-DRJO4hGU.js';
-export { c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from '../index-CrGMwXfO.js';
+export { c as createChatHandler, b as createHealthHandler, d as createIngestHandler, f as createStreamHandler, e as createUploadHandler } from '../index-B2mutkgp.js';
 import 'next/server';