@retrivora-ai/rag-engine 2.0.8 → 2.1.2

package/dist/server.mjs

@@ -579,6 +579,7 @@ function cleanApiKeyOverride(apiKeyOverride) {
 async function dispatchChatCompletion(req, apiKeyOverride) {
   const effectiveKey = cleanApiKeyOverride(apiKeyOverride);
   const provider = resolveProvider(req.model);
+  console.log(`[LLM Gateway Router] Dispatching chat request: model=${req.model}, provider=${provider}, hasGroqKey=${Boolean(process.env.GROQ_API_KEY)}, hasGeminiKey=${Boolean(process.env.GROQ_API_KEY || process.env.GEMINI_API_KEY)}, hasHFToken=${Boolean(process.env.HF_TOKEN)}`);
   try {
     switch (provider) {
       case "groq":
@@ -597,6 +598,10 @@ async function dispatchChatCompletion(req, apiKeyOverride) {
         throw new Error(`Unsupported LLM provider for model: ${req.model}`);
     }
   } catch (error) {
+    console.error(`[LLM Gateway Router] Provider '${provider}' failed for model '${req.model}':`, {
+      message: error.message,
+      stack: error.stack
+    });
     if (/rate[- ]?limit|429|exhausted/i.test(error.message)) {
       const isGroqKey = process.env.GROQ_API_KEY || effectiveKey && effectiveKey.startsWith("gsk_");
       const reqModelLower = req.model.toLowerCase();
@@ -604,7 +609,8 @@ async function dispatchChatCompletion(req, apiKeyOverride) {
         console.warn(`[LLM Gateway] Provider ${provider} rate limited (${error.message}). Falling back to secondary Groq model (llama-3.3-70b-versatile)...`);
         try {
           return await handleGroqRequest(__spreadProps(__spreadValues({}, req), { model: "llama-3.3-70b-versatile" }), effectiveKey);
-        } catch (e) {
+        } catch (fErr) {
+          console.error("[LLM Gateway] Fallback groq/llama-3.3-70b-versatile failed:", fErr.message);
         }
       }
       const geminiKey = process.env.GEMINI_API_KEY || effectiveKey;
@@ -612,14 +618,16 @@ async function dispatchChatCompletion(req, apiKeyOverride) {
         console.warn(`[LLM Gateway] Provider ${provider} rate limited (${error.message}). Falling back to Google Gemini (gemini-2.5-flash)...`);
         try {
           return await handleGeminiRequest(__spreadProps(__spreadValues({}, req), { model: "gemini-2.5-flash" }), effectiveKey);
-        } catch (e) {
+        } catch (fErr) {
+          console.error("[LLM Gateway] Fallback gemini-2.5-flash failed:", fErr.message);
         }
       }
       if (provider !== "huggingface" && (process.env.HF_TOKEN || process.env.HUGGINGFACE_API_KEY)) {
         console.warn(`[LLM Gateway] Provider ${provider} rate limited (${error.message}). Falling back to Hugging Face (Qwen/Qwen2.5-Coder-32B-Instruct)...`);
         try {
           return await handleHuggingFaceChatRequest(__spreadProps(__spreadValues({}, req), { model: "Qwen/Qwen2.5-Coder-32B-Instruct" }), effectiveKey);
-        } catch (e) {
+        } catch (fErr) {
+          console.error("[LLM Gateway] Fallback HuggingFace Qwen failed:", fErr.message);
         }
       }
       if (provider !== "openai" && process.env.OPENAI_API_KEY) {
@@ -3393,12 +3401,13 @@ var OpenAIProvider = class {
       role: "system",
       content: buildSystemContent(basePrompt, context)
     };
+    const safeMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role,
-        content: m.content
-      }))
+      ...safeMessages
     ];
     const completion = await this.client.chat.completions.create({
       model: this.llmConfig.model,
@@ -3417,12 +3426,13 @@ var OpenAIProvider = class {
         role: "system",
         content: buildSystemContent(basePrompt, context)
       };
+      const safeStreamMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
         systemMessage,
-        ...messages.map((m) => ({
-          role: m.role,
-          content: m.content
-        }))
+        ...safeStreamMessages
       ];
       const stream = yield new __await(this.client.chat.completions.create({
         model: this.llmConfig.model,
@@ -3953,9 +3963,9 @@ var GeminiProvider = class {
    *   - Messages must strictly alternate between `user` and `model`.
    */
   buildGeminiContents(messages) {
-    return messages.filter((m) => m.role !== "system").map((m) => ({
+    return (messages || []).filter((m) => Boolean(m && typeof m === "object" && m.role !== "system")).map((m) => ({
       role: m.role === "assistant" ? "model" : "user",
-      parts: [{ text: m.content }]
+      parts: [{ text: typeof m.content === "string" ? m.content : m.content != null ? String(m.content) : "" }]
     }));
   }
   // -------------------------------------------------------------------------
@@ -4158,12 +4168,13 @@ var GroqProvider = class {
       role: "system",
       content: buildSystemContent(basePrompt, context)
     };
+    const safeMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role,
-        content: m.content
-      }))
+      ...safeMessages
     ];
     const completion = await this.client.chat.completions.create({
       model: this.llmConfig.model,
@@ -4182,12 +4193,13 @@ var GroqProvider = class {
         role: "system",
         content: buildSystemContent(basePrompt, context)
       };
+      const safeStreamMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
         systemMessage,
-        ...messages.map((m) => ({
-          role: m.role,
-          content: m.content
-        }))
+        ...safeStreamMessages
       ];
       const stream = yield new __await(this.client.chat.completions.create({
         model: this.llmConfig.model,
@@ -4316,12 +4328,13 @@ var QwenProvider = class {
       role: "system",
       content: buildSystemContent(basePrompt, context)
     };
+    const safeMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
       systemMessage,
-      ...messages.map((m) => ({
-        role: m.role,
-        content: m.content
-      }))
+      ...safeMessages
     ];
     const completion = await this.client.chat.completions.create({
       model: this.llmConfig.model,
@@ -4340,12 +4353,13 @@ var QwenProvider = class {
         role: "system",
         content: buildSystemContent(basePrompt, context)
       };
+      const safeStreamMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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 = [
         systemMessage,
-        ...messages.map((m) => ({
-          role: m.role,
-          content: m.content
-        }))
+        ...safeStreamMessages
       ];
       const stream = yield new __await(this.client.chat.completions.create({
         model: this.llmConfig.model,
@@ -4525,14 +4539,18 @@ var UniversalLLMAdapter = class {
     });
   }
   async chat(messages, context) {
-    var _a2, _b, _c;
+    var _a2, _b, _c, _d, _e;
     const path2 = (_a2 = this.opts.chatPath) != null ? _a2 : "/chat/completions";
+    const safeMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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}
 
 Context:
 ${context != null ? context : "None"}` },
-      ...messages
+      ...safeMessages
     ];
     let payload;
     if (this.opts.chatPayloadTemplate) {
@@ -4550,8 +4568,15 @@ ${context != null ? context : "None"}` },
         temperature: this.temperature
       };
     }
-    const isSelfHost = this.baseUrl.includes("retrivora.com") || this.baseUrl.includes("localhost");
-    if (isSelfHost || Boolean(process.env.VERCEL)) {
+    try {
+      const { data: data2 } = await this.http.post(path2, payload);
+      const extractPath2 = (_b = this.opts.responseExtractPath) != null ? _b : "choices[0].message.content";
+      const result2 = (_c = resolvePath(data2, extractPath2)) != null ? _c : extractContent(data2);
+      if (result2 !== void 0) {
+        return String(result2);
+      }
+    } catch (httpErr) {
+      console.warn(`[UniversalLLMAdapter] Direct HTTP POST to ${this.baseUrl}${path2} failed (${httpErr.message}). Attempting in-process gateway dispatch fallback...`);
       const _g2 = globalThis;
       let dispatch = _g2.__retrivoraDispatchChat;
       if (typeof dispatch !== "function") {
@@ -4566,28 +4591,22 @@ ${context != null ? context : "None"}` },
         }
       }
       if (typeof dispatch === "function") {
-        try {
-          const res = await dispatch({
-            model: this.model,
-            messages: formattedMessages,
-            max_tokens: this.maxTokens,
-            temperature: this.temperature
-          }, this.apiKey);
-          const content = extractContent(res == null ? void 0 : res.response);
-          if (content !== void 0) {
-            return content;
-          }
-          throw new Error(`[UniversalLLMAdapter] In-process dispatch returned empty content for model: ${this.model}`);
-        } catch (dispatchErr) {
-          throw dispatchErr;
+        const res = await dispatch({
+          model: this.model,
+          messages: formattedMessages,
+          max_tokens: this.maxTokens,
+          temperature: this.temperature
+        }, this.apiKey);
+        const content = extractContent(res == null ? void 0 : res.response);
+        if (content !== void 0) {
+          return content;
         }
-      } else {
-        throw new Error(`[UniversalLLMAdapter] In-process gateway dispatch not registered. Direct self-referential HTTP calls to ${this.baseUrl} are disabled on Vercel to prevent HTTP 508 Loop Detected.`);
       }
+      throw httpErr;
     }
     const { data } = await this.http.post(path2, payload);
-    const extractPath = (_b = this.opts.responseExtractPath) != null ? _b : "choices[0].message.content";
-    const result = (_c = resolvePath(data, extractPath)) != null ? _c : extractContent(data);
+    const extractPath = (_d = this.opts.responseExtractPath) != null ? _d : "choices[0].message.content";
+    const result = (_e = resolvePath(data, extractPath)) != null ? _e : extractContent(data);
     if (result === void 0) {
       throw new Error(`[UniversalLLMAdapter] Could not extract text from path '${extractPath}' in response.`);
     }
@@ -4604,12 +4623,16 @@ ${context != null ? context : "None"}` },
       const path2 = (_a2 = this.opts.chatPath) != null ? _a2 : "/chat/completions";
       const url = `${this.baseUrl.replace(/\/$/, "")}${path2}`;
       const extractPath = ((_b = this.opts.responseExtractPath) != null ? _b : "choices[0].message.content").replace("message.content", "delta.content");
+      const safeMessages = (Array.isArray(messages) ? messages : []).filter((m) => 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}
 
 Context:
 ${context != null ? context : "None"}` },
-        ...messages
+        ...safeMessages
       ];
       let payload;
       if (this.opts.chatPayloadTemplate) {
@@ -4725,7 +4748,7 @@ ${context != null ? context : "None"}` },
     });
   }
   async embed(text) {
-    var _a2, _b, _c, _d;
+    var _a2, _b, _c, _d, _e;
     const path2 = (_a2 = this.opts.embedPath) != null ? _a2 : "/embeddings";
     let payload;
     if (this.opts.embedPayloadTemplate) {
@@ -4739,8 +4762,15 @@ ${context != null ? context : "None"}` },
         input: text
       };
     }
-    const isSelfHost = this.baseUrl.includes("retrivora.com") || this.baseUrl.includes("localhost");
-    if (isSelfHost || Boolean(process.env.VERCEL)) {
+    try {
+      const { data: data2 } = await this.http.post(path2, payload);
+      const extractPath2 = (_b = this.opts.embedExtractPath) != null ? _b : "data[0].embedding";
+      const vector2 = resolvePath(data2, extractPath2);
+      if (Array.isArray(vector2)) {
+        return vector2;
+      }
+    } catch (httpErr) {
+      console.warn(`[UniversalLLMAdapter] Direct HTTP embedding POST to ${this.baseUrl}${path2} failed (${httpErr.message}). Attempting in-process gateway dispatch fallback...`);
       const _g2 = globalThis;
       let dispatch = _g2.__retrivoraDispatchEmbedding;
       if (typeof dispatch !== "function") {
@@ -4755,24 +4785,18 @@ ${context != null ? context : "None"}` },
         }
       }
       if (typeof dispatch === "function") {
-        try {
-          const res = await dispatch({
-            model: this.model,
-            input: text
-          }, this.apiKey);
-          if ((_c = (_b = res == null ? void 0 : res.data) == null ? void 0 : _b[0]) == null ? void 0 : _c.embedding) {
-            return res.data[0].embedding;
-          }
-          throw new Error(`[UniversalLLMAdapter] In-process embedding dispatch returned invalid vector for model: ${this.model}`);
-        } catch (dispatchErr) {
-          throw dispatchErr;
+        const res = await dispatch({
+          model: this.model,
+          input: text
+        }, this.apiKey);
+        if ((_d = (_c = res == null ? void 0 : res.data) == null ? void 0 : _c[0]) == null ? void 0 : _d.embedding) {
+          return res.data[0].embedding;
         }
-      } else {
-        throw new Error(`[UniversalLLMAdapter] In-process gateway dispatch not registered. Direct self-referential HTTP calls to ${this.baseUrl} are disabled on Vercel to prevent HTTP 508 Loop Detected.`);
       }
+      throw httpErr;
     }
     const { data } = await this.http.post(path2, payload);
-    const extractPath = (_d = this.opts.embedExtractPath) != null ? _d : "data[0].embedding";
+    const extractPath = (_e = this.opts.embedExtractPath) != null ? _e : "data[0].embedding";
     const vector = resolvePath(data, extractPath);
     if (!Array.isArray(vector)) {
       throw new Error(`[UniversalLLMAdapter] Expected a number array at '${extractPath}' for embeddings.`);
@@ -6753,8 +6777,8 @@ var IntentClassifier = class {
   }
   // ─── Intent Heuristic Checkers ─────────────────────────────────────────────
   static isInformationLookup(query) {
-    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)) {
-      if (!/\b(compare|versus|vs\.?|trend|monthly|revenue|top selling|best)\b/i.test(query)) {
+    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)) {
+      if (!/\b(compare|versus|vs\.?|trend|monthly|revenue|top 10|top selling|best|chart|graph|plot|breakdown|analytics)\b/i.test(query)) {
         return true;
       }
     }
@@ -6791,7 +6815,7 @@ var Rule1SpecificInfoRule = class {
   }
   evaluate(context, intent) {
     const q = (context.userQuery || "").toLowerCase().trim();
-    const isFactQuery = intent === "information_lookup" || /^(what|who|explain|define|where|when|why|meaning of)\s+/i.test(q) || /^(what is the price of|price of|cost of)\b/i.test(q);
+    const isFactQuery = intent === "information_lookup" || /^(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);
     if (isFactQuery && isNonVisual) {
       return {
@@ -8078,6 +8102,11 @@ ${schemaProfileText}` : ""}`;
   }
   static chooseAutomaticVisualization(data, profile, query) {
     if (profile.records.length === 0) return null;
+    const q = query.toLowerCase().trim();
+    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);
     }
@@ -8193,6 +8222,7 @@ ${schemaProfileText}` : ""}`;
     return Array.from(categories);
   }
   static getProductCategory(item) {
+    if (!item) return null;
     const meta = item.metadata || {};
     const metadataCategory = resolveMetadataValue(meta, "category");
     if (this.isUsableCategory(metadataCategory)) return String(metadataCategory).trim();
@@ -8309,6 +8339,7 @@ ${schemaProfileText}` : ""}`;
       if (!fields.has(normalized)) fields.set(normalized, clean);
     };
     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]);
@@ -8425,6 +8456,7 @@ ${schemaProfileText}` : ""}`;
     return true;
   }
   static extractStockQuantity(item) {
+    if (!item) return null;
     const meta = item.metadata || {};
     const stockValue = resolveMetadataValue(meta, "stock");
     const numericStock = this.toFiniteNumber(stockValue);
@@ -9685,7 +9717,10 @@ ${context}`;
 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) => {
+      var _a2;
+      return `${(m == null ? void 0 : m.role) || "user"}: ${(_a2 = m == null ? void 0 : m.content) != null ? _a2 : ""}`;
+    }).join("\n")}
 
 New Question: ${question}
 
@@ -9709,8 +9744,11 @@ Optimized Search Query:`;
     const ns = formatNamespace(namespace != null ? namespace : this.config.projectId);
     try {
       const { sources } = await this.retrieve(query, { namespace: ns, topK: 5 });
-      if (sources.length === 0) return [];
-      const context = sources.map((s) => s.content).join("\n\n---\n\n");
+      if (!sources || sources.length === 0) return [];
+      const context = sources.map((s) => {
+        var _a2;
+        return (_a2 = s == null ? void 0 : s.content) != null ? _a2 : "";
+      }).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/package.json

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