@retrivora-ai/rag-engine 1.0.0 → 1.0.1

package/dist/server.d.mts

@@ -2,8 +2,8 @@ import { e as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, c as RagConf
 export { R as RAGConfig, U as UIConfig } from './RagConfig-DRJO4hGU.mjs';
 import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-C-sCZPhi.mjs';
 export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-C-sCZPhi.mjs';
-import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-v669iV-k.mjs';
-export { C as ConfigValidator, V as ValidationError, c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from './index-v669iV-k.mjs';
+import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-Bjy0es5a.mjs';
+export { C as ConfigValidator, V as ValidationError, c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from './index-Bjy0es5a.mjs';
 import 'next/server';
 
 /**
@@ -152,6 +152,10 @@ declare class VectorPlugin {
      * Run a chat query.
      */
     chat(message: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    /**
+     * Run a streaming chat query.
+     */
+    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, any>;
     /**
      * Ingest documents into the vector database.
      */

package/dist/server.d.ts

@@ -2,8 +2,8 @@ import { e as VectorDBConfig, L as LLMConfig, E as EmbeddingConfig, c as RagConf
 export { R as RAGConfig, U as UIConfig } from './RagConfig-DRJO4hGU.js';
 import { C as ChatMessage, I as ILLMProvider, a as ChatOptions, E as EmbedOptions } from './DocumentChunker-C-sCZPhi.js';
 export { b as Chunk, c as ChunkOptions, D as DocumentChunker } from './DocumentChunker-C-sCZPhi.js';
-import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-CrGMwXfO.js';
-export { C as ConfigValidator, V as ValidationError, c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from './index-CrGMwXfO.js';
+import { H as HealthCheckResult, I as IProviderValidator, a as IProviderHealthChecker } from './index-B2mutkgp.js';
+export { C as ConfigValidator, V as ValidationError, c as createChatHandler, b as createHealthHandler, d as createIngestHandler, e as createUploadHandler } from './index-B2mutkgp.js';
 import 'next/server';
 
 /**
@@ -152,6 +152,10 @@ declare class VectorPlugin {
      * Run a chat query.
      */
     chat(message: string, history?: ChatMessage[], namespace?: string): Promise<ChatResponse>;
+    /**
+     * Run a streaming chat query.
+     */
+    chatStream(message: string, history?: ChatMessage[], namespace?: string): AsyncGenerator<string | ChatResponse, void, any>;
     /**
      * Ingest documents into the vector database.
      */

package/dist/server.js

@@ -10,6 +10,9 @@ var __getProtoOf = Object.getPrototypeOf;
 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 = {}))
@@ -74,6 +77,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);
 
 // src/utils/templateUtils.ts
@@ -2908,6 +2939,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}"
 
@@ -3922,6 +3956,14 @@ var VectorPlugin = class {
   async chat(message, history = [], namespace) {
     return this.pipeline.ask(message, history, namespace);
   }
+  /**
+   * Run a streaming chat query.
+   */
+  chatStream(_0) {
+    return __asyncGenerator(this, arguments, function* (message, history = [], namespace) {
+      yield* __yieldStar(this.pipeline.askStream(message, history, namespace));
+    });
+  }
   /**
    * Ingest documents into the vector database.
    */

package/dist/server.mjs

@@ -1,15 +1,15 @@
 import {
   ChromaDBProvider
-} from "./chunk-WGSZNY3X.mjs";
+} from "./chunk-CFVEZTBJ.mjs";
 import {
   RedisProvider
-} from "./chunk-NXUCKY5L.mjs";
+} from "./chunk-5YGUXK7Z.mjs";
 import {
   WeaviateProvider
-} from "./chunk-OOQXNLXD.mjs";
+} from "./chunk-M6JSPGAR.mjs";
 import {
   UniversalVectorProvider
-} from "./chunk-U6KHVZLF.mjs";
+} from "./chunk-LR3VMDVK.mjs";
 import {
   AnthropicProvider,
   BatchProcessor,
@@ -34,23 +34,23 @@ import {
   createIngestHandler,
   createUploadHandler,
   getRagConfig
-} from "./chunk-P4HAQ7KB.mjs";
-import "./chunk-67AJ6SMD.mjs";
+} from "./chunk-5W2YWFT3.mjs";
+import "./chunk-YLTMFW4M.mjs";
 import {
   PineconeProvider
-} from "./chunk-ZNBKHNJ4.mjs";
+} from "./chunk-VUQJVIJT.mjs";
 import {
   PostgreSQLProvider
 } from "./chunk-FLOSGE6A.mjs";
 import {
   MongoDBProvider
-} from "./chunk-7SOSCZGS.mjs";
+} from "./chunk-CD6TSNL4.mjs";
 import {
   MilvusProvider
-} from "./chunk-TYHTZIDP.mjs";
+} from "./chunk-U55XRW3U.mjs";
 import {
   QdrantProvider
-} from "./chunk-4A47RCG2.mjs";
+} from "./chunk-3DSHW676.mjs";
 import {
   BaseVectorProvider
 } from "./chunk-IMP6FUCY.mjs";
@@ -58,7 +58,7 @@ import {
   __objRest,
   __spreadProps,
   __spreadValues
-} from "./chunk-QMIKLALV.mjs";
+} from "./chunk-X4TOT24V.mjs";
 
 // src/config/ConfigBuilder.ts
 var ConfigBuilder = class {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Retrivora AI is a plug-and-play AI engine for RAG chat experiences — generic vector DB + LLM provider, embeddable or standalone.",
   "author": "Abhinav Alkuchi",
   "license": "MIT",

package/src/core/VectorPlugin.ts

@@ -83,6 +83,13 @@ export class VectorPlugin {
     return this.pipeline.ask(message, history, namespace);
   }
 
+  /**
+   * Run a streaming chat query.
+   */
+  async *chatStream(message: string, history: ChatMessage[] = [], namespace?: string) {
+    yield* this.pipeline.askStream(message, history, namespace);
+  }
+
   /**
    * Ingest documents into the vector database.
    */

package/src/handlers/index.ts

@@ -32,6 +32,51 @@ export function createChatHandler(config?: Partial<RagConfig>) {
   };
 }
 
+/**
+ * createStreamHandler — factory that returns a streaming Next.js App Router POST handler
+ */
+export function createStreamHandler(config?: Partial<RagConfig>) {
+  const plugin = new VectorPlugin(config);
+
+  return async function POST(req: NextRequest) {
+    try {
+      const { message, history = [], namespace } = await req.json();
+      
+      const encoder = new TextEncoder();
+      const stream = new ReadableStream({
+        async start(controller) {
+          const pipelineStream = plugin.chatStream(message, history, namespace);
+          
+          for await (const chunk of pipelineStream) {
+            if (typeof chunk === 'string') {
+              controller.enqueue(encoder.encode(chunk));
+            } else {
+              // Yield metadata/sources at the end
+              controller.enqueue(encoder.encode(`\n\n__METADATA__${JSON.stringify(chunk)}`));
+            }
+          }
+          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' },
+      });
+    }
+  };
+}
+
+
 /**
  * createIngestHandler — factory for the document ingestion endpoint.
  */

package/src/rag/EntityExtractor.ts

@@ -20,6 +20,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}"