@retrivora-ai/rag-engine 2.0.2 → 2.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@retrivora-ai/rag-engine",
-  "version": "2.0.2",
+  "version": "2.0.4",
   "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",
@@ -69,19 +69,20 @@
     "dist",
     "src",
     ".env.example",
-    "README.md"
+    "README.md",
+    "FREE_TIER_ARCHITECTURE.md"
   ],
   "scripts": {
-    "dev": "npx @tailwindcss/cli -i src/tailwind.css -o src/index.css && tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --watch --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 --inject-style",
+    "dev": "npx @tailwindcss/cli -i src/tailwind.css -o src/index.css && tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --watch --tsconfig tsconfig.build.json --external react,react-dom,next,mongodb,pg,openai,@anthropic-ai/sdk,@pinecone-database/pinecone,axios,lucide-react,mammoth,pdf-parse,xlsx,react-markdown,remark-gfm,next-themes,langchain,@langchain/core,@langchain/openai,llamaindex --inject-style",
     "build": "npm run build:pkg",
-    "build:pkg": "npx @tailwindcss/cli -i src/tailwind.css -o src/index.css && tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --dts --clean --no-splitting --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 --inject-style && npx @tailwindcss/cli -i src/tailwind.css -o dist/index.css && rm src/index.css",
+    "build:pkg": "npx @tailwindcss/cli -i src/tailwind.css -o src/index.css && tsup src/index.ts src/handlers/index.ts src/server.ts --format cjs,esm --dts --clean --no-splitting --tsconfig tsconfig.build.json --external react,react-dom,next,mongodb,pg,openai,@anthropic-ai/sdk,@pinecone-database/pinecone,axios,lucide-react,mammoth,pdf-parse,xlsx,react-markdown,remark-gfm,next-themes,langchain,@langchain/core,@langchain/openai,llamaindex --inject-style && npx @tailwindcss/cli -i src/tailwind.css -o dist/index.css && rm src/index.css",
     "lint": "eslint",
     "clean": "rm -rf dist"
   },
   "peerDependencies": {
+    "next": ">=15.0.0",
     "react": ">=18.0.0",
-    "react-dom": ">=18.0.0",
-    "next": ">=15.0.0"
+    "react-dom": ">=18.0.0"
   },
   "peerDependenciesMeta": {
     "next": {
@@ -102,7 +103,8 @@
     "react-is": "^18.3.1",
     "react-markdown": "^10.1.0",
     "recharts": "^3.8.1",
-    "remark-gfm": "^4.0.1"
+    "remark-gfm": "^4.0.1",
+    "xlsx": "^0.18.5"
   },
   "optionalDependencies": {
     "@langchain/core": "^1.1.42",

package/src/app/page.tsx

@@ -1,28 +1,64 @@
 'use client';
 
-import React from 'react';
-import { AmbientBackground } from '@/components/AmbientBackground';
-import { Navbar } from '@/components/Navbar';
-import { Hero } from '@/components/Hero';
-import { Lifecycle } from '@/components/Lifecycle';
-import { ArchitectureCardsSection } from '@/components/ArchitectureCardsSection';
-import { RagPatternsSection } from '@/components/RagPatternsSection';
-import { Documentation } from '@/components/Documentation';
+import React, { useState } from 'react';
 import { ChatWidget } from '@/components/ChatWidget';
+import { ChatWindow } from '@/components/ChatWindow';
 
 export default function HomePage() {
+  const [showChatWindow, setShowChatWindow] = useState(true);
+  const [showChatWidget, setShowChatWidget] = useState(true);
+
   return (
-    <main className="min-h-screen bg-slate-50 dark:bg-[#080811] relative transition-colors duration-300">
-      <AmbientBackground />
-      <Navbar />
-      <div className="relative z-10 container mx-auto px-6 py-12">
-        <Hero />
-        <Lifecycle />
-        <RagPatternsSection />
-        <ArchitectureCardsSection />
-        <Documentation />
+    <main className="min-h-screen bg-slate-50 dark:bg-[#080811] text-slate-900 dark:text-slate-100 p-6 md:p-12 transition-colors duration-300">
+      <div className="max-w-4xl mx-auto space-y-8">
+        {/* Header / Intro */}
+        <div className="border-b border-slate-200 dark:border-slate-800 pb-6">
+          <h1 className="text-3xl font-bold tracking-tight">Retrivora SDK Preview</h1>
+          <p className="text-sm text-slate-500 dark:text-slate-400 mt-2">
+            Test and preview the SDK interactive components: Embedded <code className="text-xs bg-slate-200 dark:bg-slate-800 px-1.5 py-0.5 rounded">ChatWindow</code> and Floating <code className="text-xs bg-slate-200 dark:bg-slate-800 px-1.5 py-0.5 rounded">ChatWidget</code>.
+          </p>
+        </div>
+
+        {/* Controls */}
+        <div className="flex flex-wrap items-center gap-4 bg-white dark:bg-slate-900 p-4 rounded-xl border border-slate-200 dark:border-slate-800 shadow-sm">
+          <label className="flex items-center gap-2 cursor-pointer text-sm font-medium">
+            <input
+              type="checkbox"
+              checked={showChatWindow}
+              onChange={(e) => setShowChatWindow(e.target.checked)}
+              className="w-4 h-4 rounded text-blue-600 focus:ring-blue-500 text-sm"
+            />
+            Show Embedded ChatWindow
+          </label>
+
+          <label className="flex items-center gap-2 cursor-pointer text-sm font-medium">
+            <input
+              type="checkbox"
+              checked={showChatWidget}
+              onChange={(e) => setShowChatWidget(e.target.checked)}
+              className="w-4 h-4 rounded text-blue-600 focus:ring-blue-500 text-sm"
+            />
+            Enable Floating ChatWidget
+          </label>
+        </div>
+
+        {/* Embedded ChatWindow Section */}
+        {showChatWindow ? (
+          <div className="bg-white dark:bg-slate-900 border border-slate-200 dark:border-slate-800 rounded-2xl p-4 shadow-lg h-[600px]">
+            <ChatWindow
+              className="h-full"
+              showClose={false}
+            />
+          </div>
+        ) : (
+          <div className="border border-dashed border-slate-300 dark:border-slate-700 rounded-2xl p-8 text-center text-slate-400">
+            Embedded ChatWindow is disabled. Check the toggle above to enable it.
+          </div>
+        )}
       </div>
-      <ChatWidget position="bottom-right" />
+
+      {/* Floating ChatWidget */}
+      {showChatWidget && <ChatWidget position="bottom-right" />}
     </main>
   );
 }

package/src/components/ChatWindow.tsx

@@ -250,7 +250,7 @@ export function ChatWindow({
   return (
     <div
       ref={windowRef}
-      className={`relative flex flex-col border border-slate-200 dark:border-white/10 shadow-2xl transition-all duration-300 ${currentRadius} ${
+      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]'
@@ -346,7 +346,7 @@ export function ChatWindow({
       <div className="flex-1 overflow-y-auto px-4 py-4 space-y-5 scrollbar-thin scrollbar-thumb-slate-200 dark:scrollbar-thumb-white/10 scrollbar-track-transparent min-h-0 bg-slate-50 dark:bg-transparent">
         {!mounted || isEmpty ? (
           /* Welcome state */
-          <div className="h-full flex flex-col items-center justify-center text-center gap-4 px-6 py-12">
+          <div className="h-full flex flex-col items-center justify-center text-center gap-4 px-6 py-4">
             <div
               className={`w-16 h-16 flex items-center justify-center shadow-lg ${ui.borderRadius === 'full' ? 'rounded-full' : 'rounded-2xl'}`}
               style={{ background: `linear-gradient(135deg, ${ui.primaryColor}, ${ui.accentColor})` }}

package/src/components/DocumentUpload.tsx

@@ -13,7 +13,7 @@ interface FileState {
 }
 
 export function DocumentUpload({ namespace, onUploadComplete, className = '' }: DocumentUploadProps) {
-  const { ui, embedding } = useConfig();
+  const { ui, embedding, projectId } = useConfig();
   const [fileStates, setFileStates] = useState<FileState[]>([]);
   const [isDragging, setIsDragging] = useState(false);
   const fileInputRef = useRef<HTMLInputElement>(null);
@@ -81,7 +81,8 @@ export function DocumentUpload({ namespace, onUploadComplete, className = '' }:
 
     const formData = new FormData();
     idleFiles.forEach(s => formData.append('files', s.file));
-    if (namespace) formData.append('namespace', namespace);
+    const targetNs = namespace || projectId;
+    if (targetNs) formData.append('namespace', targetNs);
     formData.append('dimension', dimension.toString());
 
     try {
@@ -125,7 +126,7 @@ export function DocumentUpload({ namespace, onUploadComplete, className = '' }:
         </div>
         <h3 className="text-slate-900 dark:text-white font-semibold mb-1">Upload Documents</h3>
         <p className="text-slate-500 dark:text-white/40 text-sm mb-6 max-w-xs">
-          Drag and drop PDF, DOCX, TXT, or JSON files to train your AI on your own data.
+          Drag and drop PDF, DOCX, Excel (.xlsx, .xls), CSV, TXT, or JSON files to train your AI on your own data.
         </p>
 
         <button
@@ -146,7 +147,7 @@ export function DocumentUpload({ namespace, onUploadComplete, className = '' }:
           multiple 
           onChange={onFileSelect} 
           className="hidden" 
-          accept=".pdf,.docx,.txt,.md,.json,.csv"
+          accept=".pdf,.docx,.doc,.xlsx,.xls,.csv,.txt,.md,.json"
         />
 
         <div className="mt-6 flex flex-col items-center gap-2 text-sm">

package/src/components/MessageBubble.tsx

@@ -402,10 +402,8 @@ export function MessageBubble({
 
           const shouldShow =
             !isNegativeOrFallbackText &&
-            (ui.type === 'table' ||
-             !['text', 'table'].includes(ui.type) ||
-             (ui.type === 'text' && !hasRichUI && allProducts.length === 0 && textContent &&
-              !processedMarkdown.toLowerCase().includes(textContent.trim().toLowerCase())));
+            ui.type !== 'text' &&
+            (ui.type === 'table' || !['text', 'table'].includes(ui.type));
 
           if (!shouldShow) return null;
           return (

package/src/core/DatabaseStorage.ts

@@ -76,7 +76,11 @@ export class DatabaseStorage {
         tier !== 'enterprise' &&
         tier !== 'growth' &&
         tier !== 'pro' &&
-        tier !== 'developer_pro'
+        tier !== 'developer_pro' &&
+        tier !== 'free_trial' &&
+        tier !== 'free_tier' &&
+        tier !== 'free' &&
+        tier !== 'hobby'
       ) {
         throw new Error(
           `[Retrivora SDK] Chat History and Feedback features are not available on the "${tier}" tier. ` +

package/src/core/LicenseVerifier.ts

@@ -149,7 +149,7 @@ MwIDAQAB
               `Please upgrade to a Developer Pro or Enterprise plan.`
             );
           }
-        } else if (tier === 'pro' || tier === 'developer_pro' || tier === 'premium' || tier === 'growth') {
+        } else if (tier === 'pro' || tier === 'developer_pro' || tier === 'premium' || tier === 'growth' || tier === 'free_trial' || tier === 'free_tier' || tier === 'free') {
           // Developer Pro supports: PostgreSQL, Pinecone, Qdrant, MongoDB, Milvus, ChromaDB
           const allowedPro = [
             'postgresql', 'pgvector', 'supabase',

package/src/core/Pipeline.ts

@@ -24,6 +24,24 @@ import {
 import { UITransformer } from '../utils/UITransformer';
 import { SchemaMapper, SchemaMap } from '../utils/SchemaMapper';
 
+// ─── Namespace Formatter ───────────────────────────────────────────────────────
+
+/**
+ * Clean & format namespace string to satisfy vector DB identifier rules
+ * and enforce the 'retrivora-' namespace prefix convention for Free Tier isolation.
+ */
+export function formatNamespace(raw?: string): string {
+  if (!raw || !raw.trim()) return 'retrivora-default';
+  const trimmed = raw.trim();
+  if (trimmed.startsWith('retrivora-')) return trimmed;
+  const sanitized = trimmed
+    .toLowerCase()
+    .replace(/[^a-z0-9_-]/g, '_')
+    .replace(/_+/g, '_')
+    .replace(/^_+|_+$/g, '');
+  return `retrivora-${sanitized}`;
+}
+
 // ─── LRU Embedding Cache ───────────────────────────────────────────────────────
 
 /**
@@ -310,7 +328,7 @@ You are a helpful assistant. Use the provided context to answer questions accura
     namespace?: string
   ): Promise<Array<{ docId: string | number; chunksIngested: number }>> {
     await this.initialize();
-    const ns = namespace ?? this.config.projectId;
+    const ns = formatNamespace(namespace ?? this.config.projectId);
     const results = [];
 
     for (const doc of documents) {
@@ -500,7 +518,7 @@ You are a helpful assistant. Use the provided context to answer questions accura
    */
   async *askStreamInternal(question: string, history: ChatMessage[] = [], namespace?: string): AsyncIterable<string | ChatResponse> {
     await this.initialize();
-    const ns = namespace ?? this.config.projectId;
+    const ns = formatNamespace(namespace ?? this.config.projectId);
     const topK = this.config.rag?.topK ?? 5;
     // Default score threshold of 0.3 filters random matches while keeping
     // genuinely relevant chunks. Users can override via rag.scoreThreshold.
@@ -696,7 +714,7 @@ You are a helpful assistant. Use the provided context to answer questions accura
       const isSimulatedThinking = !isNativeThinking && (
         this.config.llm.options?.thinking === true ||
         this.config.llm.options?.thinking === 'enabled' ||
-        (isReasoningModel && this.config.llm.options?.thinking !== false)
+        this.config.llm.options?.thinking !== false
       );
 
       let finalRestrictionSuffix = restrictionSuffix;
@@ -1082,7 +1100,7 @@ You are a helpful assistant. Use the provided context to answer questions accura
   }
 
   async retrieve(query: string, options: { namespace?: string; topK?: number; filter?: Record<string, unknown> }): Promise<RetrievalResult> {
-    const ns = options.namespace ?? this.config.projectId;
+    const ns = formatNamespace(options.namespace ?? this.config.projectId);
     const topK = options.topK ?? 5;
 
     const cacheKey = `${ns}::${query}`;
@@ -1175,7 +1193,7 @@ Optimized Search Query:`;
     if (!query || query.trim().length < 3) return [];
 
     await this.initialize();
-    const ns = namespace ?? this.config.projectId;
+    const ns = formatNamespace(namespace ?? this.config.projectId);
 
     try {
       const { sources } = await this.retrieve(query, { namespace: ns, topK: 5 });

package/src/handlers/index.ts

@@ -576,6 +576,12 @@ export function createUploadHandler(
       const documents: { docId: string; content: string; metadata?: Record<string, unknown> }[] = [];
 
       for (const file of files) {
+        const isExcel =
+          file.name.toLowerCase().endsWith('.xlsx') ||
+          file.name.toLowerCase().endsWith('.xls') ||
+          file.type.includes('spreadsheet') ||
+          file.type.includes('excel');
+
         if (file.name.toLowerCase().endsWith('.csv') || file.type === 'text/csv') {
           const text = await file.text();
           const Papa = await import('papaparse');
@@ -613,7 +619,7 @@ export function createUploadHandler(
                 metadata: {
                   fileName: file.name,
                   fileSize: file.size,
-                  fileType: file.type,
+                  fileType: file.type || 'text/csv',
                   uploadedAt: new Date().toISOString(),
                   ...(dimension ? { dimension } : {}),
                   csvHeaders: csvCols,
@@ -622,6 +628,47 @@ export function createUploadHandler(
               });
             }
           }
+        } else if (isExcel) {
+          const xlsx = await import('xlsx');
+          const buffer = Buffer.isBuffer(file) ? file : Buffer.from(await file.arrayBuffer());
+          const workbook = xlsx.read(buffer, { type: 'buffer' });
+
+          for (const sheetName of workbook.SheetNames) {
+            const sheet = workbook.Sheets[sheetName];
+            const jsonRows = xlsx.utils.sheet_to_json<Record<string, any>>(sheet, { defval: '' });
+
+            if (jsonRows && jsonRows.length > 0) {
+              let i = 0;
+              const headers = Object.keys(jsonRows[0] || {});
+
+              for (const rowData of jsonRows) {
+                i++;
+                const contentParts: string[] = [];
+                for (const [key, val] of Object.entries(rowData)) {
+                  if (key && val !== undefined && val !== null && String(val).trim()) {
+                    contentParts.push(`${key}: ${val}`);
+                  }
+                }
+
+                if (contentParts.length > 0) {
+                  documents.push({
+                    docId: `${file.name}-${sheetName}-row-${i}`,
+                    content: contentParts.join(', '),
+                    metadata: {
+                      fileName: file.name,
+                      sheetName,
+                      fileSize: file.size,
+                      fileType: file.type || 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+                      uploadedAt: new Date().toISOString(),
+                      ...(dimension ? { dimension } : {}),
+                      headers,
+                      ...rowData
+                    }
+                  });
+                }
+              }
+            }
+          }
         } else {
           const content = await DocumentParser.parse(file, file.name, file.type);
           documents.push({

package/src/index.ts

@@ -25,6 +25,24 @@ export { useRagChat } from './hooks/useRagChat';
 // ── Utils ─────────────────────────────────────────────────────
 export { addSynonyms } from './utils/synonyms';
 
+// ── Visualization Decision Engine & Rendering ──────────────────
+export {
+  VisualizationDecisionEngine,
+  decideVisualization,
+  RendererRegistry,
+  IntentClassifier,
+  RuleEngine,
+} from './rendering';
+export type {
+  RenderDecision,
+  RenderType,
+  ChartType,
+  IntentCategory,
+  DecisionContext,
+  IRenderRule,
+  IRendererStrategy,
+} from './rendering';
+
 // ── Types (Interfaces/Types only, no Node.js deps) ─────────────
 export type {
   RagConfig,
@@ -53,10 +71,10 @@ export {
   ConfigurationException,
   AuthenticationException,
 } from './exceptions';
-export type { 
-  VectorMatch, 
-  UpsertDocument, 
-  IngestDocument, 
+export type {
+  VectorMatch,
+  UpsertDocument,
+  IngestDocument,
   ChatResponse,
   ChatMessage,
   ChatOptions,

package/src/rendering/IntentClassifier.ts

@@ -0,0 +1,167 @@
+import { DecisionContext, IntentCategory } from './types';
+
+export interface DataSignals {
+  rowCount: number;
+  hasNumericFields: boolean;
+  hasDateFields: boolean;
+  hasCategoricalFields: boolean;
+  isProductLike: boolean;
+  numericFieldCount: number;
+  categoricalFieldCount: number;
+}
+
+export class IntentClassifier {
+  /**
+   * Fast, zero-latency classifier that maps a user query and optional data context
+   * into a high-level IntentCategory.
+   */
+  static classify(context: DecisionContext): { intent: IntentCategory; signals: DataSignals } {
+    const q = (context.userQuery || '').toLowerCase().trim();
+    const signals = this.extractDataSignals(context);
+
+    // Rule A: Specific Information Lookup
+    if (this.isInformationLookup(q)) {
+      return { intent: 'information_lookup', signals };
+    }
+
+    // Rule B: Comparison Query
+    if (this.isComparisonQuery(q)) {
+      return { intent: 'comparison', signals };
+    }
+
+    // Rule C: Product Search / Recommendation
+    if (this.isProductQuery(q) || (signals.isProductLike && this.isRecommendationQuery(q))) {
+      return { intent: 'product_search', signals };
+    }
+
+    // Rule D: Trend Analysis
+    if (this.isTrendQuery(q) || (signals.hasDateFields && signals.hasNumericFields && /\b(growth|trend|over time|monthly|yearly|weekly)\b/.test(q))) {
+      return { intent: 'trend_analysis', signals };
+    }
+
+    // Rule E: Ranking Query
+    if (this.isRankingQuery(q)) {
+      return { intent: 'ranking', signals };
+    }
+
+    // Rule F: Analytics Query
+    if (this.isAnalyticsQuery(q) || (signals.hasNumericFields && signals.hasCategoricalFields && /\b(sales|revenue|performance|count|total|average|breakdown|share|percentage|correlation)\b/.test(q))) {
+      return { intent: 'analytics', signals };
+    }
+
+    // Rule G: Recommendation fallback
+    if (this.isRecommendationQuery(q)) {
+      return { intent: 'recommendation', signals };
+    }
+
+    // Fallback: Default to information lookup for generic queries
+    return { intent: 'information_lookup', signals };
+  }
+
+  /**
+   * Extract key data structure signals from retrieved documents.
+   */
+  static extractDataSignals(context: DecisionContext): DataSignals {
+    const docs = context.retrievedDocuments || [];
+    const rowCount = docs.length;
+
+    let hasNumericFields = false;
+    let hasDateFields = false;
+    let hasCategoricalFields = false;
+    let isProductLike = false;
+    let numericFieldCount = 0;
+    let categoricalFieldCount = 0;
+
+    if (docs.length > 0) {
+      const keys = new Set<string>();
+      let numericKeys = new Set<string>();
+      let dateKeys = new Set<string>();
+      let catKeys = new Set<string>();
+
+      let productKeyMatches = 0;
+
+      docs.forEach(doc => {
+        const meta = doc.metadata || {};
+        Object.entries(meta).forEach(([key, val]) => {
+          keys.add(key);
+          const kLower = key.toLowerCase();
+          if (['price', 'image', 'brand', 'in_stock', 'rating', 'sku', 'product'].some(p => kLower.includes(p))) {
+            productKeyMatches++;
+          }
+
+          if (typeof val === 'number') {
+            numericKeys.add(key);
+          } else if (typeof val === 'string') {
+            if (/^\d{4}-\d{2}-\d{2}/.test(val) || !isNaN(Date.parse(val)) && val.length > 5) {
+              dateKeys.add(key);
+            } else if (val.length < 50 && !val.includes('\n')) {
+              catKeys.add(key);
+            }
+          }
+        });
+      });
+
+      hasNumericFields = numericKeys.size > 0;
+      hasDateFields = dateKeys.size > 0;
+      hasCategoricalFields = catKeys.size > 0;
+      numericFieldCount = numericKeys.size;
+      categoricalFieldCount = catKeys.size;
+
+      if (productKeyMatches >= 2 || docs.some(d => d.content.toLowerCase().includes('price') || d.content.toLowerCase().includes('brand'))) {
+        isProductLike = true;
+      }
+    }
+
+    return {
+      rowCount,
+      hasNumericFields,
+      hasDateFields,
+      hasCategoricalFields,
+      isProductLike,
+      numericFieldCount,
+      categoricalFieldCount,
+    };
+  }
+
+  // ─── 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)) {
+      // 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)) {
+        return true;
+      }
+    }
+    // "price of X" / "cost of Y"
+    if (/^(what is the )?(price|cost|fee|rate) of\b/i.test(query)) {
+      return true;
+    }
+    return false;
+  }
+
+  private static isComparisonQuery(query: string): boolean {
+    return /\b(compare|comparison|vs\.?|versus|difference between|differ|contrast|against)\b/i.test(query);
+  }
+
+  private static isProductQuery(query: string): boolean {
+    return /\b(best|recommend|top|buy|shopping|under \$|cheap|shoes|laptops|chairs|headphones|phone|sneakers|apparel|products)\b/i.test(query)
+      && !/\b(compare|chart|sales|revenue|trend)\b/i.test(query);
+  }
+
+  private static isRecommendationQuery(query: string): boolean {
+    return /\b(recommend|suggest|what should i (buy|get|choose)|best options|top picks)\b/i.test(query);
+  }
+
+  private static isTrendQuery(query: string): boolean {
+    return /\b(trend|trends|growth|over time|historical|timeline|monthly|yearly|weekly|daily|revenue growth|sales growth)\b/i.test(query);
+  }
+
+  private static isRankingQuery(query: string): boolean {
+    return /\b(highest|lowest|top\s*\d+|bottom\s*\d+|rank|ranking|top performing|best performing)\b/i.test(query);
+  }
+
+  private static isAnalyticsQuery(query: string): boolean {
+    return /\b(sales|revenue|analytics|insights|distribution|percentage|share|breakdown|breakup|scatter|correlation|metrics|performance|by region|by category)\b/i.test(query);
+  }
+}