@retrivora-ai/rag-engine 1.9.6 → 1.9.8

package/src/config/RagConfig.ts

@@ -157,6 +157,19 @@ export interface UIConfig {
 // RAG Pipeline Tuning
 // ---------------------------------------------------------------------------
 
+export interface CAGConfig {
+  /** Enable Cold RAG (Cache Augmented Generation) */
+  enabled?: boolean;
+  /** Namespace containing the cold data (e.g., static manuals/guides) */
+  coldNamespace?: string;
+  /** Limit of cold documents to retrieve at initialization and keep in cache/context */
+  coldLimit?: number;
+  /** Specific search query to retrieve the cold documents (e.g., "manual", "documentation") */
+  coldQuery?: string;
+  /** Specific document IDs to load as cold context */
+  coldDocumentIds?: string[];
+}
+
 export interface RAGConfig {
   /** Number of top-K chunks retrieved per query */
   topK?: number;
@@ -197,21 +210,33 @@ export interface RAGConfig {
    * Defaults to built-in list (find, search, tell me about, etc.).
    */
   vectorKeywords?: string[];
+  /** Cold RAG (Cache Augmented Generation) configuration */
+  cag?: CAGConfig;
+  /** Chat history configuration */
+  history?: {
+    enabled?: boolean;
+    tableName?: string;
+  };
+  /** User feedback configuration */
+  feedback?: {
+    enabled?: boolean;
+    tableName?: string;
+  };
 }
 
 export interface RetrievalConfig {
   /** Retrieval strategy selected by the host app. */
   strategy?:
-    | 'vector'
-    | 'keyword'
-    | 'hybrid'
-    | 'multi-query'
-    | 'self-query'
-    | 'parent-document'
-    | 'contextual-compression'
-    | 'ensemble'
-    | 'recursive'
-    | 'agentic';
+  | 'vector'
+  | 'keyword'
+  | 'hybrid'
+  | 'multi-query'
+  | 'self-query'
+  | 'parent-document'
+  | 'contextual-compression'
+  | 'ensemble'
+  | 'recursive'
+  | 'agentic';
   /** Number of documents/chunks retrieved per query. */
   topK?: number;
   /** Minimum similarity score to include a retrieved chunk. */
@@ -229,11 +254,23 @@ export interface WorkflowConfig {
   options?: Record<string, unknown>;
 }
 
+export interface MCPServerConfig {
+  name: string;
+  transport: 'stdio' | 'sse';
+  url?: string;      // Required for SSE transport
+  command?: string;  // Required for Stdio transport
+  args?: string[];   // Optional for Stdio transport
+  env?: Record<string, string>; // Optional env variables for Stdio child process
+}
+
 // ---------------------------------------------------------------------------
 // Root Config
 // ---------------------------------------------------------------------------
 
 export interface RagConfig {
+  /** License key for commercial production validation */
+  licenseKey?: string;
+
   /**
    * Unique identifier for the consuming project.
    * Used as the vector DB namespace to achieve multi-tenancy.
@@ -257,6 +294,15 @@ export interface RagConfig {
 
   /** Optional Graph database configuration */
   graphDb?: GraphDBConfig;
+
+  /** Optional Telemetry configuration for observability logging */
+  telemetry?: {
+    enabled?: boolean;
+    url?: string;
+  };
+
+  /** Optional Model Context Protocol (MCP) server configurations */
+  mcpServers?: MCPServerConfig[];
 }
 
 /**

package/src/config/constants.ts

@@ -4,6 +4,7 @@
  * This file serves as the single source of truth for all statically supported
  * providers and stylistic options in the Retrivora AI RAG Engine.
  */
+export const VERSION = '1.9.6';
 
 export const VECTOR_DB_PROVIDERS = [
   'pinecone',
@@ -25,6 +26,8 @@ export const LLM_PROVIDERS = [
   'anthropic',
   'ollama',
   'gemini',
+  'groq',
+  'qwen',
   'rest',
   'universal_rest',
   'custom',
@@ -34,6 +37,7 @@ export const EMBEDDING_PROVIDERS = [
   'openai',
   'ollama',
   'gemini',
+  'qwen',
   'rest',
   'universal_rest',
   'custom',
@@ -50,6 +54,7 @@ export const PROVIDERS_WITH_EMBEDDINGS = [
   'openai',
   'ollama',
   'gemini',
+  'qwen',
   'rest',
   'universal_rest',
 ] as const;

package/src/config/serverConfig.ts

@@ -149,6 +149,7 @@ export function getEnvConfig(env: Record<string, string | undefined> = process.e
 
   return {
     projectId,
+    licenseKey: readString(env, 'RETRIVORA_LICENSE_KEY') ?? readString(env, 'LICENSE_KEY') ?? base?.licenseKey,
     vectorDb: {
       provider: vectorProvider,
       indexName:
@@ -217,6 +218,10 @@ export function getEnvConfig(env: Record<string, string | undefined> = process.e
         try { return JSON.parse(raw) as Record<string, string>; } catch { return undefined; }
       })(),
     },
+    telemetry: {
+      enabled: readString(env, 'TELEMETRY_ENABLED') === 'true' || readString(env, 'NEXT_PUBLIC_TELEMETRY_ENABLED') === 'true' || base?.telemetry?.enabled || false,
+      url: readString(env, 'TELEMETRY_URL') ?? readString(env, 'NEXT_PUBLIC_TELEMETRY_URL') ?? base?.telemetry?.url ?? '/api/telemetry',
+    },
     // Optional graph DB — driven by GRAPH_DB_PROVIDER env var
     ...(readString(env, 'GRAPH_DB_PROVIDER') ? {
       graphDb: {
@@ -228,5 +233,15 @@ export function getEnvConfig(env: Record<string, string | undefined> = process.e
         },
       },
     } : {}),
+    mcpServers: (() => {
+      const raw = readString(env, 'MCP_SERVERS') ?? readString(env, 'NEXT_PUBLIC_MCP_SERVERS');
+      if (!raw) return undefined;
+      try {
+        return JSON.parse(raw);
+      } catch (e) {
+        console.warn('[serverConfig] Failed to parse MCP_SERVERS env:', e);
+        return undefined;
+      }
+    })(),
   };
 }

package/src/core/ConfigResolver.ts

@@ -9,7 +9,6 @@
 import { RagConfig, UniversalRagConfig, RAGConfig, RetrievalConfig, WorkflowConfig } from '../config/RagConfig';
 import { mergeDefined } from '../utils/templateUtils';
 import { getEnvConfig } from '../config/serverConfig';
-import { ConfigurationException } from '../exceptions';
 
 
 
@@ -32,36 +31,42 @@ export class ConfigResolver {
       projectId: hostConfig.projectId || envConfig.projectId,
       vectorDb: hostConfig.vectorDb
         ? {
-            ...envConfig.vectorDb,
-            ...hostConfig.vectorDb,
-            options: {
-              ...envConfig.vectorDb.options,
-              ...(hostConfig.vectorDb.options || {}),
-            },
-          }
+          ...envConfig.vectorDb,
+          ...hostConfig.vectorDb,
+          options: {
+            ...envConfig.vectorDb.options,
+            ...(hostConfig.vectorDb.options || {}),
+          },
+        }
         : envConfig.vectorDb,
       llm: hostConfig.llm
         ? {
-            ...envConfig.llm,
-            ...hostConfig.llm,
-            options: {
-              ...(envConfig.llm.options || {}),
-              ...(hostConfig.llm.options || {}),
-            },
-          }
+          ...envConfig.llm,
+          ...hostConfig.llm,
+          options: {
+            ...(envConfig.llm.options || {}),
+            ...(hostConfig.llm.options || {}),
+          },
+        }
         : envConfig.llm,
       embedding: hostConfig.embedding
         ? {
-            ...envConfig.embedding,
-            ...hostConfig.embedding,
-            options: {
-              ...(envConfig.embedding.options || {}),
-              ...(hostConfig.embedding.options || {}),
-            },
-          }
+          ...envConfig.embedding,
+          ...hostConfig.embedding,
+          options: {
+            ...(envConfig.embedding.options || {}),
+            ...(hostConfig.embedding.options || {}),
+          },
+        }
         : envConfig.embedding,
       ui: hostConfig.ui ? mergeDefined(envConfig.ui, hostConfig.ui) : envConfig.ui,
       rag: hostConfig.rag ? { ...envConfig.rag, ...hostConfig.rag } : envConfig.rag,
+      telemetry: hostConfig.telemetry
+        ? {
+          ...envConfig.telemetry,
+          ...hostConfig.telemetry,
+        }
+        : envConfig.telemetry,
     };
   }
 
@@ -87,13 +92,13 @@ export class ConfigResolver {
    */
   static validate(config: RagConfig): void {
     if (!config.projectId) {
-      throw new ConfigurationException('[ConfigResolver] projectId is required');
+      throw new Error('[ConfigResolver] projectId is required');
     }
     if (!config.vectorDb.provider) {
-      throw new ConfigurationException('[ConfigResolver] vectorDb.provider is required');
+      throw new Error('[ConfigResolver] vectorDb.provider is required');
     }
     if (!config.llm.provider) {
-      throw new ConfigurationException('[ConfigResolver] llm.provider is required');
+      throw new Error('[ConfigResolver] llm.provider is required');
     }
   }