@retrivora-ai/rag-engine 1.0.0 → 1.0.2

package/dist/handlers/index.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 = {}))
@@ -62,6 +65,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
@@ -515,7 +546,7 @@ var init_MongoDBProvider = __esm({
         if (!opts.uri || !opts.database || !opts.collection) {
           throw new Error("[MongoDBProvider] options uri, database, and collection are required.");
         }
-        this.client = new import_mongodb.MongoClient(opts.uri);
+        this.client = new import_mongodb.MongoClient(opts.uri, { serverSelectionTimeoutMS: 5e3 });
         this.dbName = opts.database;
         this.collectionName = opts.collection;
         this.embeddingKey = opts.embeddingKey || "embedding";
@@ -1476,6 +1507,7 @@ __export(handlers_exports, {
   createChatHandler: () => createChatHandler,
   createHealthHandler: () => createHealthHandler,
   createIngestHandler: () => createIngestHandler,
+  createStreamHandler: () => createStreamHandler,
   createUploadHandler: () => createUploadHandler
 });
 module.exports = __toCommonJS(handlers_exports);
@@ -2825,6 +2857,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}"
 
@@ -3399,6 +3434,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,
@@ -3783,29 +3819,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);
   }
   /**
@@ -3831,12 +3847,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);
   }
 };
@@ -3918,6 +3945,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) {
@@ -3990,5 +4076,6 @@ function createUploadHandler(config) {
   createChatHandler,
   createHealthHandler,
   createIngestHandler,
+  createStreamHandler,
   createUploadHandler
 });

package/dist/handlers/index.mjs

@@ -2,13 +2,15 @@ import {
   createChatHandler,
   createHealthHandler,
   createIngestHandler,
+  createStreamHandler,
   createUploadHandler
-} from "../chunk-P4HAQ7KB.mjs";
-import "../chunk-67AJ6SMD.mjs";
-import "../chunk-QMIKLALV.mjs";
+} from "../chunk-OJNAKSQ2.mjs";
+import "../chunk-YLTMFW4M.mjs";
+import "../chunk-X4TOT24V.mjs";
 export {
   createChatHandler,
   createHealthHandler,
   createIngestHandler,
+  createStreamHandler,
   createUploadHandler
 };

package/dist/{index-CrGMwXfO.d.ts → index-B2mutkgp.d.ts}

@@ -71,6 +71,10 @@ interface IProviderHealthChecker {
 declare function createChatHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
     error: string;
 }> | NextResponse<ChatResponse>>;
+/**
+ * createStreamHandler — factory that returns a streaming Next.js App Router POST handler
+ */
+declare function createStreamHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<Response>;
 /**
  * createIngestHandler — factory for the document ingestion endpoint.
  */
@@ -109,4 +113,4 @@ declare function createUploadHandler(config?: Partial<RagConfig>): (req: NextReq
     }[];
 }>>;
 
-export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, createHealthHandler as b, createChatHandler as c, createIngestHandler as d, createUploadHandler as e };
+export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, createHealthHandler as b, createChatHandler as c, createIngestHandler as d, createUploadHandler as e, createStreamHandler as f };

package/dist/{index-v669iV-k.d.mts → index-Bjy0es5a.d.mts}

@@ -71,6 +71,10 @@ interface IProviderHealthChecker {
 declare function createChatHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<NextResponse<{
     error: string;
 }> | NextResponse<ChatResponse>>;
+/**
+ * createStreamHandler — factory that returns a streaming Next.js App Router POST handler
+ */
+declare function createStreamHandler(config?: Partial<RagConfig>): (req: NextRequest) => Promise<Response>;
 /**
  * createIngestHandler — factory for the document ingestion endpoint.
  */
@@ -109,4 +113,4 @@ declare function createUploadHandler(config?: Partial<RagConfig>): (req: NextReq
     }[];
 }>>;
 
-export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, createHealthHandler as b, createChatHandler as c, createIngestHandler as d, createUploadHandler as e };
+export { ConfigValidator as C, type HealthCheckResult as H, type IProviderValidator as I, type ValidationError as V, type IProviderHealthChecker as a, createHealthHandler as b, createChatHandler as c, createIngestHandler as d, createUploadHandler as e, createStreamHandler as f };

package/dist/index.d.mts

@@ -1,4 +1,3 @@
-import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
 import { C as ChatMessage } from './DocumentChunker-C-sCZPhi.mjs';
 export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-C-sCZPhi.mjs';
@@ -9,7 +8,7 @@ interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */
     position?: 'bottom-right' | 'bottom-left';
 }
-declare function ChatWidget({ position }: ChatWidgetProps): react_jsx_runtime.JSX.Element | null;
+declare function ChatWidget({ position }: ChatWidgetProps): React$1.JSX.Element | null;
 
 interface ChatWindowProps {
     /** Additional className for the wrapper div */
@@ -21,7 +20,7 @@ interface ChatWindowProps {
     /** Whether to show a close (X) button in the header */
     showClose?: boolean;
 }
-declare function ChatWindow({ className, style, onClose, showClose }: ChatWindowProps): react_jsx_runtime.JSX.Element;
+declare function ChatWindow({ className, style, onClose, showClose }: ChatWindowProps): React$1.JSX.Element;
 
 interface DocumentUploadProps {
     /** Optional namespace for the upload */
@@ -31,7 +30,7 @@ interface DocumentUploadProps {
     /** Additional className */
     className?: string;
 }
-declare function DocumentUpload({ namespace, onUploadComplete, className }: DocumentUploadProps): react_jsx_runtime.JSX.Element;
+declare function DocumentUpload({ namespace, onUploadComplete, className }: DocumentUploadProps): React$1.JSX.Element;
 
 interface MessageBubbleProps {
     message: ChatMessage;
@@ -40,13 +39,20 @@ interface MessageBubbleProps {
     primaryColor?: string;
     accentColor?: string;
 }
-declare function MessageBubble({ message, sources, isStreaming, primaryColor, accentColor, }: MessageBubbleProps): react_jsx_runtime.JSX.Element;
+declare function MessageBubble({ message, sources, isStreaming, primaryColor, accentColor, }: MessageBubbleProps): React$1.JSX.Element;
 
 interface SourceCardProps {
     source: VectorMatch;
     index: number;
 }
-declare function SourceCard({ source, index }: SourceCardProps): react_jsx_runtime.JSX.Element;
+declare function SourceCard({ source, index }: SourceCardProps): React$1.JSX.Element;
+
+/**
+ * ConfigProvider — React Context that makes RagConfig (UI subset) available
+ * to all child components without prop drilling.
+ *
+ * Only the UI-safe portions of RagConfig are exposed to the client.
+ */
 
 interface ClientConfig {
     projectId: string;
@@ -58,7 +64,7 @@ declare function ConfigProvider({ config, children, }: {
         ui?: Partial<UIConfig>;
     };
     children: ReactNode;
-}): react_jsx_runtime.JSX.Element;
+}): React$1.JSX.Element;
 declare function useConfig(): ClientConfig;
 
 type MessageRole = 'user' | 'assistant';

package/dist/index.d.ts

@@ -1,4 +1,3 @@
-import * as react_jsx_runtime from 'react/jsx-runtime';
 import React$1, { ReactNode } from 'react';
 import { C as ChatMessage } from './DocumentChunker-C-sCZPhi.js';
 export { a as ChatOptions, b as Chunk, c as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-C-sCZPhi.js';
@@ -9,7 +8,7 @@ interface ChatWidgetProps {
     /** Position of the floating button. Defaults to bottom-right. */
     position?: 'bottom-right' | 'bottom-left';
 }
-declare function ChatWidget({ position }: ChatWidgetProps): react_jsx_runtime.JSX.Element | null;
+declare function ChatWidget({ position }: ChatWidgetProps): React$1.JSX.Element | null;
 
 interface ChatWindowProps {
     /** Additional className for the wrapper div */
@@ -21,7 +20,7 @@ interface ChatWindowProps {
     /** Whether to show a close (X) button in the header */
     showClose?: boolean;
 }
-declare function ChatWindow({ className, style, onClose, showClose }: ChatWindowProps): react_jsx_runtime.JSX.Element;
+declare function ChatWindow({ className, style, onClose, showClose }: ChatWindowProps): React$1.JSX.Element;
 
 interface DocumentUploadProps {
     /** Optional namespace for the upload */
@@ -31,7 +30,7 @@ interface DocumentUploadProps {
     /** Additional className */
     className?: string;
 }
-declare function DocumentUpload({ namespace, onUploadComplete, className }: DocumentUploadProps): react_jsx_runtime.JSX.Element;
+declare function DocumentUpload({ namespace, onUploadComplete, className }: DocumentUploadProps): React$1.JSX.Element;
 
 interface MessageBubbleProps {
     message: ChatMessage;
@@ -40,13 +39,20 @@ interface MessageBubbleProps {
     primaryColor?: string;
     accentColor?: string;
 }
-declare function MessageBubble({ message, sources, isStreaming, primaryColor, accentColor, }: MessageBubbleProps): react_jsx_runtime.JSX.Element;
+declare function MessageBubble({ message, sources, isStreaming, primaryColor, accentColor, }: MessageBubbleProps): React$1.JSX.Element;
 
 interface SourceCardProps {
     source: VectorMatch;
     index: number;
 }
-declare function SourceCard({ source, index }: SourceCardProps): react_jsx_runtime.JSX.Element;
+declare function SourceCard({ source, index }: SourceCardProps): React$1.JSX.Element;
+
+/**
+ * ConfigProvider — React Context that makes RagConfig (UI subset) available
+ * to all child components without prop drilling.
+ *
+ * Only the UI-safe portions of RagConfig are exposed to the client.
+ */
 
 interface ClientConfig {
     projectId: string;
@@ -58,7 +64,7 @@ declare function ConfigProvider({ config, children, }: {
         ui?: Partial<UIConfig>;
     };
     children: ReactNode;
-}): react_jsx_runtime.JSX.Element;
+}): React$1.JSX.Element;
 declare function useConfig(): ClientConfig;
 
 type MessageRole = 'user' | 'assistant';