@retrivora-ai/rag-engine 1.0.4 → 1.0.5

package/dist/server.mjs

@@ -32,9 +32,15 @@ import {
   createChatHandler,
   createHealthHandler,
   createIngestHandler,
+  createStreamHandler,
   createUploadHandler,
-  getRagConfig
-} from "./chunk-6MLZHQZT.mjs";
+  getRagConfig,
+  resetConfigCache,
+  sseErrorFrame,
+  sseFrame,
+  sseMetaFrame,
+  sseTextFrame
+} from "./chunk-MWL4AGQI.mjs";
 import "./chunk-YLTMFW4M.mjs";
 import {
   PineconeProvider
@@ -62,21 +68,11 @@ import {
 
 // src/config/ConfigBuilder.ts
 var ConfigBuilder = class {
-  constructor() {
-    this.config = {
-      projectId: "default-project",
-      rag: {
-        chunkSize: 1e3,
-        chunkOverlap: 200,
-        topK: 5
-      }
-    };
-  }
   /**
    * Set the project/application ID for namespacing
    */
   projectId(id) {
-    this.config.projectId = id;
+    this._projectId = id;
     return this;
   }
   /**
@@ -85,9 +81,9 @@ var ConfigBuilder = class {
   vectorDb(provider, options) {
     var _a;
     if (provider === "auto") {
-      this.config.vectorDb = this.autoDetectVectorDb();
+      this._vectorDb = this._autoDetectVectorDb();
     } else {
-      this.config.vectorDb = {
+      this._vectorDb = {
         provider,
         indexName: (_a = options == null ? void 0 : options.indexName) != null ? _a : "default",
         options: __spreadValues({}, options)
@@ -101,9 +97,9 @@ var ConfigBuilder = class {
   llm(provider, model, apiKey, options) {
     var _a, _b;
     if (provider === "auto") {
-      this.config.llm = this.autoDetectLLM();
+      this._llm = this._autoDetectLLM();
     } else {
-      this.config.llm = {
+      this._llm = {
         provider,
         model: model != null ? model : "default-model",
         apiKey,
@@ -121,9 +117,9 @@ var ConfigBuilder = class {
    */
   embedding(provider, model, apiKey, options) {
     if (provider === "auto") {
-      this.config.embedding = this.autoDetectEmbedding();
+      this._embedding = this._autoDetectEmbedding();
     } else {
-      this.config.embedding = {
+      this._embedding = {
         provider,
         model: model != null ? model : "default-embedding",
         apiKey,
@@ -134,26 +130,46 @@ var ConfigBuilder = class {
     return this;
   }
   /**
-   * Set RAG-specific parameters
+   * Set RAG-specific pipeline parameters
    */
   rag(options) {
     var _a;
-    this.config.rag = __spreadValues(__spreadValues({}, (_a = this.config.rag) != null ? _a : {}), options);
+    this._rag = __spreadValues(__spreadValues({}, (_a = this._rag) != null ? _a : {}), options);
     return this;
   }
   /**
-   * Set UI/UX parameters
+   * Set UI branding and appearance options.
+   * Accepts the full UIConfig interface.
    */
   ui(options) {
-    this.config.ui = options;
+    this._ui = options;
     return this;
   }
   /**
-   * Build and return the configuration
+   * Build and return the final RagConfig.
+   * Throws if required fields (projectId, vectorDb, llm, embedding) are not set.
    */
   build() {
-    const finalConfig = this.config;
-    return finalConfig;
+    var _a;
+    const missing = [];
+    if (!this._projectId) missing.push('projectId (call .projectId("my-app"))');
+    if (!this._vectorDb) missing.push('vectorDb (call .vectorDb("pinecone", { ... }))');
+    if (!this._llm) missing.push('llm (call .llm("openai", "gpt-4o", apiKey))');
+    if (!this._embedding) missing.push('embedding (call .embedding("openai", "text-embedding-3-small"))');
+    if (missing.length > 0) {
+      throw new Error(
+        `[ConfigBuilder] Cannot build \u2014 required fields are missing:
+  ${missing.join("\n  ")}`
+      );
+    }
+    return __spreadValues({
+      projectId: this._projectId,
+      vectorDb: this._vectorDb,
+      llm: this._llm,
+      embedding: this._embedding,
+      ui: this._rag ? this._ui : void 0,
+      rag: (_a = this._rag) != null ? _a : { chunkSize: 1e3, chunkOverlap: 200, topK: 5 }
+    }, this._ui ? { ui: this._ui } : {});
   }
   /**
    * Build and return as JSON for serialization
@@ -164,169 +180,58 @@ var ConfigBuilder = class {
   // ============================================================================
   // Private helper methods for auto-detection
   // ============================================================================
-  autoDetectVectorDb() {
+  _autoDetectVectorDb() {
     if (process.env.PINECONE_API_KEY && process.env.PINECONE_INDEX) {
-      return {
-        provider: "pinecone",
-        indexName: process.env.PINECONE_INDEX,
-        options: { apiKey: process.env.PINECONE_API_KEY }
-      };
+      return { provider: "pinecone", indexName: process.env.PINECONE_INDEX, options: { apiKey: process.env.PINECONE_API_KEY } };
     }
     if (process.env.QDRANT_URL) {
-      return {
-        provider: "qdrant",
-        indexName: process.env.QDRANT_COLLECTION || "documents",
-        options: {
-          url: process.env.QDRANT_URL,
-          apiKey: process.env.QDRANT_API_KEY
-        }
-      };
+      return { provider: "qdrant", indexName: process.env.QDRANT_COLLECTION || "documents", options: { url: process.env.QDRANT_URL, apiKey: process.env.QDRANT_API_KEY } };
     }
     if (process.env.DATABASE_URL) {
-      return {
-        provider: "postgresql",
-        indexName: process.env.PG_TABLE || "documents",
-        options: {
-          connectionString: process.env.DATABASE_URL
-        }
-      };
+      return { provider: "postgresql", indexName: process.env.PG_TABLE || "documents", options: { connectionString: process.env.DATABASE_URL } };
     }
     if (process.env.MONGODB_URI) {
-      return {
-        provider: "mongodb",
-        indexName: process.env.MONGODB_COLLECTION || "documents",
-        options: {
-          uri: process.env.MONGODB_URI,
-          database: process.env.MONGODB_DB || "ai_db",
-          collection: process.env.MONGODB_COLLECTION || "documents"
-        }
-      };
+      return { provider: "mongodb", indexName: process.env.MONGODB_COLLECTION || "documents", options: { uri: process.env.MONGODB_URI, database: process.env.MONGODB_DB || "ai_db", collection: process.env.MONGODB_COLLECTION || "documents" } };
     }
     if (process.env.REDIS_URL) {
-      return {
-        provider: "redis",
-        indexName: process.env.REDIS_INDEX || "documents",
-        options: {
-          url: process.env.REDIS_URL
-        }
-      };
+      return { provider: "redis", indexName: process.env.REDIS_INDEX || "documents", options: { url: process.env.REDIS_URL } };
     }
-    return {
-      provider: "qdrant",
-      indexName: "documents",
-      options: {
-        url: process.env.QDRANT_URL || "http://localhost:6333"
-      }
-    };
+    return { provider: "qdrant", indexName: "documents", options: { url: process.env.QDRANT_URL || "http://localhost:6333" } };
   }
-  autoDetectLLM() {
+  _autoDetectLLM() {
     if (process.env.OPENAI_API_KEY) {
-      return {
-        provider: "openai",
-        model: process.env.OPENAI_MODEL || "gpt-4o-mini",
-        apiKey: process.env.OPENAI_API_KEY,
-        maxTokens: parseInt(process.env.OPENAI_MAX_TOKENS || "1024"),
-        temperature: parseFloat(process.env.OPENAI_TEMPERATURE || "0.7")
-      };
+      return { provider: "openai", model: process.env.OPENAI_MODEL || "gpt-4o-mini", apiKey: process.env.OPENAI_API_KEY, maxTokens: parseInt(process.env.OPENAI_MAX_TOKENS || "1024"), temperature: parseFloat(process.env.OPENAI_TEMPERATURE || "0.7") };
     }
     if (process.env.ANTHROPIC_API_KEY) {
-      return {
-        provider: "anthropic",
-        model: process.env.ANTHROPIC_MODEL || "claude-3-haiku-20240307",
-        apiKey: process.env.ANTHROPIC_API_KEY,
-        maxTokens: parseInt(process.env.ANTHROPIC_MAX_TOKENS || "1024"),
-        temperature: parseFloat(process.env.ANTHROPIC_TEMPERATURE || "0.7")
-      };
+      return { provider: "anthropic", model: process.env.ANTHROPIC_MODEL || "claude-3-haiku-20240307", apiKey: process.env.ANTHROPIC_API_KEY, maxTokens: parseInt(process.env.ANTHROPIC_MAX_TOKENS || "1024"), temperature: parseFloat(process.env.ANTHROPIC_TEMPERATURE || "0.7") };
     }
     if (process.env.OLLAMA_BASE_URL) {
-      return {
-        provider: "ollama",
-        model: process.env.OLLAMA_MODEL || "mistral",
-        baseUrl: process.env.OLLAMA_BASE_URL,
-        maxTokens: parseInt(process.env.OLLAMA_MAX_TOKENS || "1024"),
-        temperature: parseFloat(process.env.OLLAMA_TEMPERATURE || "0.7")
-      };
+      return { provider: "ollama", model: process.env.OLLAMA_MODEL || "mistral", baseUrl: process.env.OLLAMA_BASE_URL, maxTokens: parseInt(process.env.OLLAMA_MAX_TOKENS || "1024"), temperature: parseFloat(process.env.OLLAMA_TEMPERATURE || "0.7") };
     }
-    return {
-      provider: "openai",
-      model: "gpt-4o-mini",
-      apiKey: process.env.OPENAI_API_KEY
-    };
+    return { provider: "openai", model: "gpt-4o-mini", apiKey: process.env.OPENAI_API_KEY };
   }
-  autoDetectEmbedding() {
+  _autoDetectEmbedding() {
     if (process.env.EMBEDDING_PROVIDER) {
-      const provider = process.env.EMBEDDING_PROVIDER;
-      return {
-        provider,
-        model: process.env.EMBEDDING_MODEL || "default",
-        apiKey: process.env.EMBEDDING_API_KEY,
-        baseUrl: process.env.EMBEDDING_BASE_URL
-      };
+      return { provider: process.env.EMBEDDING_PROVIDER, model: process.env.EMBEDDING_MODEL || "default", apiKey: process.env.EMBEDDING_API_KEY, baseUrl: process.env.EMBEDDING_BASE_URL };
     }
-    return {
-      provider: "openai",
-      model: "text-embedding-3-small",
-      apiKey: process.env.OPENAI_API_KEY
-    };
+    return { provider: "openai", model: "text-embedding-3-small", apiKey: process.env.OPENAI_API_KEY };
   }
 };
 var PRESETS = {
-  /**
-   * OpenAI + Pinecone: Production-ready cloud setup
-   */
-  "openai-pinecone": {
-    vectorDb: "pinecone",
-    llm: "openai",
-    embedding: "openai"
-  },
-  /**
-   * Claude + Qdrant: Open-source vector DB + proprietary LLM
-   */
-  "claude-qdrant": {
-    vectorDb: "qdrant",
-    llm: "anthropic",
-    embedding: "openai"
-  },
-  /**
-   * Local development: Ollama + local Qdrant
-   */
-  "local-dev": {
-    vectorDb: "qdrant",
-    llm: "ollama",
-    embedding: "ollama"
-  },
-  /**
-   * Fully open-source: Ollama LLM + Qdrant vector DB + Ollama embeddings
-   */
-  "fully-open-source": {
-    vectorDb: "qdrant",
-    llm: "ollama",
-    embedding: "ollama"
-  },
-  /**
-   * PostgreSQL stack: pgvector + OpenAI
-   */
-  "postgres-openai": {
-    vectorDb: "postgresql",
-    llm: "openai",
-    embedding: "openai"
-  },
-  /**
-   * Enterprise MongoDB: MongoDB Atlas with OpenAI
-   */
-  "mongodb-openai": {
-    vectorDb: "mongodb",
-    llm: "openai",
-    embedding: "openai"
-  },
-  /**
-   * Redis stack for caching + search
-   */
-  "redis-openai": {
-    vectorDb: "redis",
-    llm: "openai",
-    embedding: "openai"
-  }
+  /** OpenAI + Pinecone: Production-ready cloud setup */
+  "openai-pinecone": { vectorDb: "pinecone", llm: "openai", embedding: "openai" },
+  /** Claude + Qdrant: Open-source vector DB + proprietary LLM */
+  "claude-qdrant": { vectorDb: "qdrant", llm: "anthropic", embedding: "openai" },
+  /** Local development: Ollama + local Qdrant */
+  "local-dev": { vectorDb: "qdrant", llm: "ollama", embedding: "ollama" },
+  /** Fully open-source: Ollama LLM + Qdrant + Ollama embeddings */
+  "fully-open-source": { vectorDb: "qdrant", llm: "ollama", embedding: "ollama" },
+  /** PostgreSQL stack: pgvector + OpenAI */
+  "postgres-openai": { vectorDb: "postgresql", llm: "openai", embedding: "openai" },
+  /** Enterprise MongoDB: MongoDB Atlas with OpenAI */
+  "mongodb-openai": { vectorDb: "mongodb", llm: "openai", embedding: "openai" },
+  /** Redis stack for caching + search */
+  "redis-openai": { vectorDb: "redis", llm: "openai", embedding: "openai" }
 };
 function createFromPreset(presetName) {
   const preset = PRESETS[presetName];
@@ -564,6 +469,12 @@ export {
   createFromPreset,
   createHealthHandler,
   createIngestHandler,
+  createStreamHandler,
   createUploadHandler,
-  getRagConfig
+  getRagConfig,
+  resetConfigCache,
+  sseErrorFrame,
+  sseFrame,
+  sseMetaFrame,
+  sseTextFrame
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "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",
@@ -73,7 +73,7 @@
   "scripts": {
     "dev": "next dev",
     "build": "next build",
-    "build:pkg": "tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --dts --clean --tsconfig tsconfig.build.json --external react,react-dom,next,mongodb,pg,openai,@anthropic-ai/sdk,@pinecone-database/pinecone,axios,lucide-react,mammoth,pdf-parse,react-markdown,remark-gfm,next-themes",
+    "build:pkg": "tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --dts --clean --tsconfig tsconfig.build.json --external react,react-dom,next,mongodb,pg,openai,@anthropic-ai/sdk,@pinecone-database/pinecone,axios,lucide-react,mammoth,pdf-parse,react-markdown,remark-gfm,next-themes,langchain,@langchain/core,@langchain/openai,llamaindex",
     "start": "next start",
     "lint": "eslint",
     "prepublishOnly": "npm run build:pkg"
@@ -86,25 +86,27 @@
   "dependencies": {
     "@anthropic-ai/sdk": "^0.90.0",
     "@google/genai": "^1.50.1",
-    "@langchain/core": "^1.1.42",
-    "@langchain/openai": "^1.4.5",
     "@pinecone-database/pinecone": "^7.2.0",
     "axios": "^1.15.0",
-    "langchain": "^1.3.5",
-    "llamaindex": "^0.12.1",
     "lucide-react": "^1.8.0",
-    "mammoth": "^1.8.0",
-    "mongodb": "^7.1.1",
     "next": ">=14.0.0",
     "next-themes": "^0.4.6",
     "openai": "^6.34.0",
-    "pdf-parse": "^1.1.1",
-    "pg": "^8.20.0",
     "react": ">=18.0.0",
     "react-dom": ">=18.0.0",
     "react-markdown": "^10.1.0",
     "remark-gfm": "^4.0.1"
   },
+  "optionalDependencies": {
+    "@langchain/core": "^1.1.42",
+    "@langchain/openai": "^1.4.5",
+    "langchain": "^1.3.5",
+    "llamaindex": "^0.12.1",
+    "mammoth": "^1.8.0",
+    "mongodb": "^7.1.1",
+    "pdf-parse": "^1.1.1",
+    "pg": "^8.20.0"
+  },
   "devDependencies": {
     "@tailwindcss/postcss": "^4",
     "@types/node": "^20",

package/src/components/ChatWindow.tsx

@@ -11,8 +11,8 @@ import {
 } from 'lucide-react';
 import { MessageBubble } from './MessageBubble';
 import { useConfig } from './ConfigProvider';
-import { useRagChat } from '@/hooks/useRagChat';
-import { BORDER_RADIUS_MAP, CHAT_SUGGESTIONS } from '@/app/constants';
+import { useRagChat } from '../hooks/useRagChat';
+import { BORDER_RADIUS_MAP, CHAT_SUGGESTIONS } from '../config/uiConstants';
 
 interface ChatWindowProps {
   /** Additional className for the wrapper div */

package/src/components/ConfigProvider.tsx

@@ -8,8 +8,8 @@
  */
 
 import React, { createContext, useContext, ReactNode } from 'react';
-import { UIConfig } from '@/config/RagConfig';
-import { mergeDefined } from '@/utils/templateUtils';
+import { UIConfig } from '../config/RagConfig';
+import { mergeDefined } from '../utils/templateUtils';
 
 export interface ClientConfig {
   projectId: string;

package/src/components/MessageBubble.tsx

@@ -4,8 +4,8 @@ import React from 'react';
 import { Bot, User, ChevronDown, ChevronRight } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
-import { ChatMessage } from '@/llm/ILLMProvider';
-import { VectorMatch } from '@/types';
+import { ChatMessage } from '../llm/ILLMProvider';
+import { VectorMatch } from '../types';
 import { SourceCard } from './SourceCard';
 
 interface MessageBubbleProps {

package/src/components/SourceCard.tsx

@@ -2,7 +2,7 @@
 
 import React from 'react';
 import { ChevronDown, ChevronUp } from 'lucide-react';
-import { VectorMatch } from '@/types';
+import { VectorMatch } from '../types';
 
 interface SourceCardProps {
   source: VectorMatch;

package/src/components/ThemeToggle.tsx

@@ -3,7 +3,7 @@
 import * as React from 'react';
 import { Moon, Sun } from 'lucide-react';
 import { useTheme } from 'next-themes';
-import { useHydrated } from '@/hooks/useHydrated';
+import { useHydrated } from '../hooks/useHydrated';
 
 export function ThemeToggle() {
   const { theme, setTheme } = useTheme();