@retrivora-ai/rag-engine 2.0.8 → 2.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "2.0.8",
+  "version": "2.1.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": "UNLICENSED",

package/src/components/ChatWindow.tsx

@@ -52,10 +52,10 @@ interface WindowWithSpeech extends Window {
   webkitSpeechRecognition?: new () => ISpeechRecognition;
 }
 
-export function ChatWindow({ 
-  className = '', 
-  style, 
-  onClose, 
+export function ChatWindow({
+  className = '',
+  style,
+  onClose,
   showClose = false,
   onResizeStart,
   onResetResize,
@@ -129,7 +129,7 @@ export function ChatWindow({
       alert('Speech recognition is not supported in your browser.');
       return;
     }
-    
+
     if (isListening) {
       recognitionRef.current.stop();
     } else {
@@ -178,7 +178,7 @@ export function ChatWindow({
   const sendMessage = useCallback(async () => {
     const text = input.trim();
     if (!text || isLoading) return;
-    
+
     if (isListening) {
       recognitionRef.current?.stop();
     }
@@ -250,11 +250,10 @@ export function ChatWindow({
   return (
     <div
       ref={windowRef}
-      className={`relative flex flex-col h-full flex-1 border border-slate-200 dark:border-white/10 shadow-2xl transition-all duration-300 ${currentRadius} ${
-        isGlass 
-          ? 'bg-white/90 dark:bg-[#0f0f1a]/90 backdrop-blur-xl' 
+      className={`relative flex flex-col h-full flex-1 border border-slate-200 dark:border-white/10 shadow-2xl transition-all duration-300 ${currentRadius} ${isGlass
+          ? 'bg-white/90 dark:bg-[#0f0f1a]/90 backdrop-blur-xl'
           : 'bg-white dark:bg-[#0f0f1a]'
-      } ${className}`}
+        } ${className}`}
       style={{ '--primary': ui.primaryColor, '--accent': ui.accentColor, ...style } as React.CSSProperties}
     >
       {/* Resize Handle (Top-Left) */}
@@ -293,11 +292,13 @@ export function ChatWindow({
         </div>
 
         <div className="flex items-center gap-1.5">
-          {/* Online indicator */}
-          <span className="flex items-center gap-1 text-[10px] text-emerald-500 dark:text-emerald-400 mr-2">
-            <span className="w-1.5 h-1.5 rounded-full bg-emerald-500 dark:bg-emerald-400 animate-pulse" />
-            Online
-          </span>
+          {/* Online indicator & Deployment ID badge */}
+          <div className="flex items-center gap-1.5 mr-2">
+            <span className="flex items-center gap-1 text-[10px] text-emerald-500 dark:text-emerald-400">
+              <span className="w-1.5 h-1.5 rounded-full bg-emerald-500 dark:bg-emerald-400 animate-pulse" />
+              Online
+            </span>
+          </div>
 
           <div className="flex items-center bg-slate-100/50 dark:bg-white/5 rounded-lg p-0.5 border border-slate-200/50 dark:border-white/5">
             {mounted && messages.length > 0 && (
@@ -427,9 +428,9 @@ export function ChatWindow({
             >
               <X className="w-5 h-5" />
             </button>
-            <DocumentUpload 
-              namespace={projectId} 
-              onUploadComplete={() => setIsUploadModalOpen(false)} 
+            <DocumentUpload
+              namespace={projectId}
+              onUploadComplete={() => setIsUploadModalOpen(false)}
             />
           </div>
         </div>
@@ -437,7 +438,7 @@ export function ChatWindow({
 
       {/* ── Input ── */}
       <div className={`px-4 pb-4 pt-2 border-t border-slate-200 dark:border-white/10 ${isGlass ? 'bg-transparent' : 'bg-white dark:bg-[#0f0f1a]'}`}>
-        
+
         {/* Dynamic Suggestions */}
         {(suggestions.length > 0 || isSuggesting) && !isLoading && (
           <div className="mb-3 flex flex-wrap gap-2 animate-in fade-in slide-in-from-bottom-2 duration-300">
@@ -488,11 +489,10 @@ export function ChatWindow({
               type="button"
               onClick={toggleListening}
               disabled={isLoading}
-              className={`flex-shrink-0 w-9 h-9 rounded-xl flex items-center justify-center transition-all hover:scale-105 active:scale-95 disabled:opacity-50 disabled:cursor-not-allowed disabled:hover:scale-100 disabled:active:scale-100 ${
-                isListening
+              className={`flex-shrink-0 w-9 h-9 rounded-xl flex items-center justify-center transition-all hover:scale-105 active:scale-95 disabled:opacity-50 disabled:cursor-not-allowed disabled:hover:scale-100 disabled:active:scale-100 ${isListening
                   ? 'bg-rose-500 text-white animate-pulse shadow-md shadow-rose-500/20'
                   : 'text-slate-400 dark:text-white/40 hover:bg-slate-200 dark:hover:bg-white/10'
-              }`}
+                }`}
               title={isListening ? 'Stop listening' : 'Start voice input'}
             >
               {isListening ? <MicOff className="w-4 h-4" /> : <Mic className="w-4 h-4" />}

package/src/core/Pipeline.ts

@@ -1193,7 +1193,7 @@ You are a helpful assistant. Use the provided context to answer questions accura
 Focus on extracting the core intent and entities. Do not answer the question, just rewrite it.
 
 History:
-${history.map(m => `${m.role}: ${m.content}`).join('\n')}
+${(history || []).map(m => `${m?.role || 'user'}: ${m?.content ?? ''}`).join('\n')}
 
 New Question: ${question}
 
@@ -1222,9 +1222,9 @@ Optimized Search Query:`;
 
     try {
       const { sources } = await this.retrieve(query, { namespace: ns, topK: 5 });
-      if (sources.length === 0) return [];
+      if (!sources || sources.length === 0) return [];
 
-      const context = sources.map((s) => s.content).join('\n\n---\n\n');
+      const context = sources.map((s) => s?.content ?? '').join('\n\n---\n\n');
       const prompt = `Based on the following snippets from a document, what are at least 5 short, relevant questions a user might ask? 
 Focus on questions that can be answered by the context.
 Keep each question under 10 words and make them very specific to the content.

package/src/llm/providers/GeminiProvider.ts

@@ -175,11 +175,11 @@ export class GeminiProvider implements ILLMProvider {
    *   - Messages must strictly alternate between `user` and `model`.
    */
   private buildGeminiContents(messages: ChatMessage[]): Array<{ role: 'user' | 'model'; parts: [{ text: string }] }> {
-    return messages
-      .filter((m) => m.role !== 'system')
+    return (messages || [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object' && m.role !== 'system'))
       .map((m) => ({
-        role: m.role === 'assistant' ? ('model' as const) : ('user' as const),
-        parts: [{ text: m.content }] as [{ text: string }],
+        role: m.role === 'assistant' ? 'model' : 'user',
+        parts: [{ text: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : '') }],
       }));
   }
 

package/src/llm/providers/GroqProvider.ts

@@ -95,12 +95,16 @@ export class GroqProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeMessages,
     ];
 
     const completion = await this.client.chat.completions.create({
@@ -125,12 +129,16 @@ export class GroqProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeStreamMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeStreamMessages,
     ];
 
     const stream = await this.client.chat.completions.create({

package/src/llm/providers/OpenAIProvider.ts

@@ -91,12 +91,16 @@ export class OpenAIProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeMessages,
     ];
 
     const completion = await this.client.chat.completions.create({
@@ -121,12 +125,16 @@ export class OpenAIProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeStreamMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeStreamMessages,
     ];
 
     const stream = await this.client.chat.completions.create({

package/src/llm/providers/QwenProvider.ts

@@ -99,12 +99,16 @@ export class QwenProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeMessages,
     ];
 
     const completion = await this.client.chat.completions.create({
@@ -129,12 +133,16 @@ export class QwenProvider implements ILLMProvider {
       content: buildSystemContent(basePrompt, context),
     };
 
+    const safeStreamMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: (m.role || 'user') as 'user' | 'assistant',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role as 'user' | 'assistant',
-        content: m.content,
-      })),
+      ...safeStreamMessages,
     ];
 
     const stream = await this.client.chat.completions.create({

package/src/llm/providers/UniversalLLMAdapter.ts

@@ -83,9 +83,16 @@ export class UniversalLLMAdapter implements ILLMProvider {
   async chat(messages: ChatMessage[], context?: string): Promise<string> {
     const path = this.opts.chatPath ?? '/chat/completions';
 
+    const safeMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: m.role || 'user',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages = [
       { role: 'system', content: `${this.systemPrompt}\n\nContext:\n${context ?? 'None'}` },
-      ...messages,
+      ...safeMessages,
     ];
 
     let payload: unknown;
@@ -166,9 +173,16 @@ export class UniversalLLMAdapter implements ILLMProvider {
     const extractPath = (this.opts.responseExtractPath ?? 'choices[0].message.content')
       .replace('message.content', 'delta.content');
 
+    const safeMessages = (Array.isArray(messages) ? messages : [])
+      .filter((m): m is ChatMessage => Boolean(m && typeof m === 'object'))
+      .map((m) => ({
+        role: m.role || 'user',
+        content: typeof m.content === 'string' ? m.content : (m.content != null ? String(m.content) : ''),
+      }));
+
     const formattedMessages = [
       { role: 'system', content: `${this.systemPrompt}\n\nContext:\n${context ?? 'None'}` },
-      ...messages,
+      ...safeMessages,
     ];
 
     let payload: unknown;

package/src/rendering/IntentClassifier.ts

@@ -126,10 +126,10 @@ export class IntentClassifier {
   // ─── Intent Heuristic Checkers ─────────────────────────────────────────────
 
   private static isInformationLookup(query: string): boolean {
-    // Exact single-fact pattern matches e.g. "What is X?", "Explain Y", "Price of Z", "Can Auvira be assigned..."
-    if (/^(what|who|explain|define|where|when|why|how|how does|can|could|does|do|is|are|has|have|will|should|would|meaning of)\s+/i.test(query)) {
+    // Exact single-fact pattern matches e.g. "What is X?", "Explain Y", "Explore Z", "Tell me about A"
+    if (/^(what|who|explain|explore|define|where|when|why|how|how does|can|could|does|do|is|are|has|have|will|should|would|meaning of|tell me about|show details of|overview of|summarize)\s+/i.test(query)) {
       // Exclude comparison or analytical phrasing inside "what is the trend..."
-      if (!/\b(compare|versus|vs\.?|trend|monthly|revenue|top selling|best)\b/i.test(query)) {
+      if (!/\b(compare|versus|vs\.?|trend|monthly|revenue|top 10|top selling|best|chart|graph|plot|breakdown|analytics)\b/i.test(query)) {
         return true;
       }
     }

package/src/rendering/rules/Rule1SpecificInfoRule.ts

@@ -10,7 +10,7 @@ export class Rule1SpecificInfoRule implements IRenderRule {
     // Check specific fact/definition/explanation indicators
     const isFactQuery =
       intent === 'information_lookup' ||
-      /^(what|who|explain|define|where|when|why|meaning of)\s+/i.test(q) ||
+      /^(what|who|explain|explore|define|where|when|why|meaning of|tell me about|show details of|overview of|summarize)\s+/i.test(q) ||
       /^(what is the price of|price of|cost of)\b/i.test(q);
 
     const isNonVisual = !/\b(compare|versus|vs\.?|trend|monthly|revenue|best|top 10|chart|table|list|grid)\b/i.test(q);

package/src/utils/ProductExtractor.ts

@@ -196,6 +196,7 @@ export function isLikelyUiOnlyMessage(text: string): boolean {
 export function extractProductsFromSources(sources: VectorMatch[] | undefined, isUser: boolean): Product[] {
   if (isUser || !sources) return [];
   return sources
+    .filter((s): s is VectorMatch => Boolean(s && typeof s === 'object'))
     .filter(s => {
       const m = s.metadata ?? {};
       const keys = Object.keys(m).map(k => k.toLowerCase());
@@ -204,7 +205,7 @@ export function extractProductsFromSources(sources: VectorMatch[] | undefined, i
       );
       const hasProductIdentity = keys.some(k => ['brand', 'model'].includes(k)) &&
         keys.some(k => ['name', 'title', 'product_name', 'product'].includes(k));
-      const hasPricePattern = /\$\s*\d+/.test(s.content);
+      const hasPricePattern = /\$\s*\d+/.test(s?.content ?? '');
       return hasStrongProductKey || hasProductIdentity || hasPricePattern || m.type === 'product';
     })
     .map(s => {
@@ -215,13 +216,13 @@ export function extractProductsFromSources(sources: VectorMatch[] | undefined, i
       const description = resolveMetadataValue(m, 'description') as string;
 
       return {
-        id: s.id,
+        id: s?.id ?? `prod_${Math.random().toString(36).substring(2, 8)}`,
         name: name ?? (s?.content ?? '').split('\n')[0] ?? 'Unknown Product',
         brand: brand,
         price: price,
         image: resolveImage(m),
         link: resolveMetadataValue(m, 'link') as string,
-        description: description ? description : '' //description ?? s.content,
+        description: description ? description : '',
       };
     });
 }

package/src/utils/UITransformer.ts

@@ -1121,6 +1121,15 @@ RULES:
     query: string,
   ): UITransformationResponse | null {
     if (profile.records.length === 0) return null;
+
+    const q = query.toLowerCase().trim();
+    // Do not automatically force charts unless query contains visual or analytical intent
+    const isExplicitVisualOrAnalytic = /\b(chart|graph|plot|visualize|trend|monthly|revenue|top\s*\d+|breakdown|analytics|compare|versus|vs\.?|histogram|pie|bar|line|distribution|percentage|share|summary of metrics)\b/i.test(q);
+
+    if (!isExplicitVisualOrAnalytic) {
+      return null;
+    }
+
     if (profile.dateFields.length > 0 && profile.numericFields.length > 0) {
       return this.transformToLineChart(profile);
     }
@@ -1265,6 +1274,7 @@ RULES:
   }
 
   private static getProductCategory(item: VectorMatch): string | null {
+    if (!item) return null;
     const meta = item.metadata || {};
     const metadataCategory = resolveMetadataValue(meta, 'category');
     if (this.isUsableCategory(metadataCategory)) return String(metadataCategory).trim();
@@ -1401,6 +1411,7 @@ RULES:
     };
 
     data.forEach(item => {
+      if (!item) return;
       Object.keys(item.metadata || {}).forEach(addField);
       for (const match of (item.content || '').matchAll(/(?:^|\n)\s*([A-Za-z][A-Za-z0-9_ /-]{1,50})\s*[:\-–]\s*([^\n]+)/g)) {
         addField(match[1]);
@@ -1551,6 +1562,7 @@ RULES:
   }
 
   private static extractStockQuantity(item: VectorMatch): number | null {
+    if (!item) return null;
     const meta = item.metadata || {};
     const stockValue = resolveMetadataValue(meta, 'stock');
     const numericStock = this.toFiniteNumber(stockValue);