@retrivora-ai/rag-engine 1.2.1 → 1.2.3

package/src/config/RagConfig.ts

@@ -149,6 +149,8 @@ export interface UIConfig {
   allowUpload?: boolean;
   /** Whether to allow manual resizing of the chat window. Defaults to true. */
   allowResize?: boolean;
+  /** Whether to enable voice search input. Defaults to true. */
+  enableVoiceInput?: boolean;
 }
 
 // ---------------------------------------------------------------------------

package/src/config/uiConstants.ts

@@ -5,6 +5,29 @@
  * is consumed via npm without Next.js path-alias configuration.
  */
 
+import { ClientConfig } from '../types';
+
+export const DEFAULT_CONFIG: ClientConfig = {
+  projectId: 'default',
+  ui: {
+    title: 'AI Assistant',
+    subtitle: 'Powered by RAG',
+    primaryColor: '#6366f1',
+    accentColor: '#8b5cf6',
+    logoUrl: '',
+    placeholder: 'Ask me anything…',
+    showSources: true,
+    welcomeMessage: "Hello! I'm your AI assistant. Ask me anything about your documents.",
+    showWidget: true,
+    poweredBy: 'RAG',
+    visualStyle: 'glass',
+    borderRadius: 'xl',
+    allowUpload: true,
+    allowResize: true,
+    enableVoiceInput: true,
+  },
+};
+
 /** Maps the `borderRadius` config value to a Tailwind class. */
 export const BORDER_RADIUS_MAP: Record<string, string> = {
   none: 'rounded-none',

package/src/core/LangChainAgent.ts

@@ -47,12 +47,30 @@ export class LangChainAgent {
       // 2. Create the ReAct agent (langchain v1.x API)
       //    - `model` replaces the old `llm` parameter
       //    - `createAgent` replaces `createOpenAIFunctionsAgent` + `AgentExecutor`
+      const defaultAgentPrompt = 'You are a helpful AI assistant with access to a document search tool.';
+      let finalSystemPrompt = this.config.llm.systemPrompt || defaultAgentPrompt;
+
+      const chartInstruction = `\n\nWhen presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
+- For tables: use standard Markdown tables.
+- For charts (bar, line, pie): Output a JSON code block with language 'chart' matching this schema:
+\`\`\`chart
+{
+  "type": "bar" | "line" | "pie",
+  "xAxisKey": "name",
+  "dataKeys": ["value"],
+  "data": [{"name": "A", "value": 10}]
+}
+\`\`\``;
+      
+      if (!finalSystemPrompt.includes('chart')) {
+        finalSystemPrompt += chartInstruction;
+      }
+
       this.agent = createAgent({
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         model: chatModel as any,
         tools: [searchTool],
-        systemPrompt: this.config.llm.systemPrompt ||
-          'You are a helpful AI assistant with access to a document search tool.',
+        systemPrompt: finalSystemPrompt,
       });
 
       // Suppress unused import warning — HumanMessage is used in run()

package/src/core/Pipeline.ts

@@ -1,6 +1,7 @@
 import { BaseVectorProvider } from '../providers/vectordb/BaseVectorProvider';
 import { BaseGraphProvider } from '../providers/graphdb/BaseGraphProvider';
-import { ILLMProvider, ChatMessage } from '../llm/ILLMProvider';
+import { ILLMProvider } from '../llm/ILLMProvider';
+import { ChatMessage } from '../types';
 import { DocumentChunker, Chunk } from '../rag/DocumentChunker';
 import { EntityExtractor } from '../rag/EntityExtractor';
 import { Reranker } from '../rag/Reranker';
@@ -98,6 +99,23 @@ export class Pipeline {
   async initialize(): Promise<void> {
     if (this.initialised) return;
 
+    // Augment system prompt with Chart/Table instructions if not already present
+    const chartInstruction = `\n\nWhen presenting structured data, statistics, or comparisons, decide if it is best presented as a table or a chart.
+- For tables: use standard Markdown tables.
+- For charts (bar, line, pie): Output a JSON code block with language 'chart' matching this schema:
+\`\`\`chart
+{
+  "type": "bar" | "line" | "pie",
+  "xAxisKey": "name",
+  "dataKeys": ["value"],
+  "data": [{"name": "A", "value": 10}]
+}
+\`\`\``;
+
+    if (!this.config.llm.systemPrompt?.includes('chart')) {
+      this.config.llm.systemPrompt = (this.config.llm.systemPrompt || '') + chartInstruction;
+    }
+
     console.log(`[Pipeline] Initializing with provider: ${this.config.vectorDb.provider}...`);
     // Resolve vector DB provider
     this.vectorDB = await ProviderRegistry.createVectorProvider(this.config.vectorDb);

package/src/core/VectorPlugin.ts

@@ -4,7 +4,7 @@ import { ConfigValidator } from './ConfigValidator';
 import { Pipeline } from './Pipeline';
 import { ProviderHealthCheck, HealthCheckResult } from './ProviderHealthCheck';
 import { IngestDocument, ChatResponse } from '../types';
-import { ChatMessage } from '../llm/ILLMProvider';
+import { ChatMessage } from '../types';
 
 /**
  * VectorPlugin — main orchestrator class.

package/src/handlers/index.ts

@@ -1,7 +1,7 @@
 import { NextRequest, NextResponse } from 'next/server';
 import { VectorPlugin } from '../core/VectorPlugin';
 import { RagConfig } from '../config/RagConfig';
-import { ChatMessage } from '../llm/ILLMProvider';
+import { ChatMessage } from '../types';
 import { DocumentParser } from '../utils/DocumentParser';
 
 // ─── SSE helpers ──────────────────────────────────────────────────────────────

package/src/hooks/useRagChat.ts

@@ -19,53 +19,9 @@
  */
 
 import { useState, useCallback, useRef, useEffect } from 'react';
-import { VectorMatch } from '../types';
+import { VectorMatch, RagMessage, UseRagChatOptions, UseRagChatReturn } from '../types';
 import { useStoredMessages } from './useStoredMessages';
 
-// ─── Types ────────────────────────────────────────────────────────────────────
-
-export type MessageRole = 'user' | 'assistant';
-
-export interface RagMessage {
-  id: string;
-  role: MessageRole;
-  content: string;
-  /** Retrieved source chunks (assistant messages only) */
-  sources?: VectorMatch[];
-  /** ISO timestamp */
-  createdAt: string;
-}
-
-export interface UseRagChatOptions {
-  /** Override the chat API endpoint (default: /api/chat) */
-  apiUrl?: string;
-  /** Override project namespace */
-  namespace?: string;
-  /** Persist chat history to localStorage (default: true) */
-  persist?: boolean;
-  /** Called after each successful assistant reply */
-  onReply?: (message: RagMessage) => void;
-  /** Called on error */
-  onError?: (error: string) => void;
-}
-
-export interface UseRagChatReturn {
-  /** All messages in the current conversation */
-  messages: RagMessage[];
-  /** Whether a response is in flight */
-  isLoading: boolean;
-  /** Current error message, or null */
-  error: string | null;
-  /** Send a user message */
-  send: (text: string) => Promise<void>;
-  /** Clear the conversation (and localStorage if persist=true) */
-  clear: () => void;
-  /** Retry the last failed send */
-  retry: () => Promise<void>;
-  /** Programmatically set the conversation (e.g. to restore from a DB) */
-  setMessages: React.Dispatch<React.SetStateAction<RagMessage[]>>;
-}
-
 // ─── SSE Frame Parser ──────────────────────────────────────────────────────────
 
 interface SseTextFrame  { type: 'text';     text: string }

package/src/hooks/useStoredMessages.ts

@@ -4,7 +4,7 @@ import * as React from 'react';
 
 export interface StoredMessage {
   id: string;
-  role: 'user' | 'assistant';
+  role: 'user' | 'assistant' | 'system';
   content: string;
   createdAt?: string;
   sources?: unknown;

package/src/index.ts

@@ -16,9 +16,24 @@ export { useRagChat } from './hooks/useRagChat';
 
 // ── Types (Interfaces/Types only, no Node.js deps) ─────────────
 export type { RagConfig, VectorDBConfig, VectorDBProvider, LLMConfig, LLMProvider, EmbeddingConfig, EmbeddingProvider, UIConfig, RAGConfig } from './config/RagConfig';
-export type { VectorMatch, UpsertDocument } from './types';
-export type { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from './llm/ILLMProvider';
-export type { IngestDocument, ChatResponse } from './types';
+export type { ILLMProvider, EmbedOptions } from './llm/ILLMProvider';
 export type { Chunk, ChunkOptions } from './rag/DocumentChunker';
-export type { ClientConfig } from './components/ConfigProvider';
-export type { RagMessage, UseRagChatOptions, UseRagChatReturn } from './hooks/useRagChat';
+export type { 
+  VectorMatch, 
+  UpsertDocument, 
+  IngestDocument, 
+  ChatResponse,
+  ChatMessage,
+  ChatOptions,
+  RagMessage,
+  ClientConfig,
+  UseRagChatOptions,
+  UseRagChatReturn,
+  ChatWindowProps,
+  ChatWidgetProps,
+  MessageBubbleProps,
+  ProductCardProps,
+  ProductCarouselProps,
+  DocumentUploadProps,
+  SourceCardProps
+} from './types';

package/src/llm/ILLMProvider.ts

@@ -4,19 +4,7 @@
  * provider can handle both responsibilities when appropriate.
  */
 
-export interface ChatMessage {
-  role: 'user' | 'assistant' | 'system';
-  content: string;
-}
-
-export interface ChatOptions {
-  /** Override max tokens for this specific call */
-  maxTokens?: number;
-  /** Override temperature for this specific call */
-  temperature?: number;
-  /** Stop sequences */
-  stop?: string[];
-}
+import { ChatMessage, ChatOptions } from '../types';
 
 export interface EmbedOptions {
   /** Override model for this specific embed call */

package/src/llm/providers/AnthropicProvider.ts

@@ -3,7 +3,8 @@
  */
 
 import Anthropic from '@anthropic-ai/sdk';
-import { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from '../ILLMProvider';
+import { ILLMProvider, EmbedOptions } from '../ILLMProvider';
+import { ChatMessage, ChatOptions } from '../../types';
 import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
 import { IProviderValidator, IProviderHealthChecker, HealthCheckResult } from '../../core/ProviderInterfaces';
 import { ValidationError } from '../../core/ConfigValidator';

package/src/llm/providers/GeminiProvider.ts

@@ -15,7 +15,8 @@
  */
 
 import { GoogleGenAI } from '@google/genai';
-import { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from '../ILLMProvider';
+import { ILLMProvider, EmbedOptions } from '../ILLMProvider';
+import { ChatMessage, ChatOptions } from '../../types';
 import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
 import { IProviderValidator, IProviderHealthChecker, HealthCheckResult } from '../../core/ProviderInterfaces';
 import { ValidationError } from '../../core/ConfigValidator';

package/src/llm/providers/OllamaProvider.ts

@@ -3,7 +3,8 @@
  */
 
 import axios, { AxiosInstance } from 'axios';
-import { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from '../ILLMProvider';
+import { ILLMProvider, EmbedOptions } from '../ILLMProvider';
+import { ChatMessage, ChatOptions } from '../../types';
 import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
 import { IProviderValidator, IProviderHealthChecker, HealthCheckResult } from '../../core/ProviderInterfaces';
 import { ValidationError } from '../../core/ConfigValidator';

package/src/llm/providers/OpenAIProvider.ts

@@ -3,7 +3,8 @@
  */
 
 import OpenAI from 'openai';
-import { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from '../ILLMProvider';
+import { ILLMProvider, EmbedOptions } from '../ILLMProvider';
+import { ChatMessage, ChatOptions } from '../../types';
 import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
 import { IProviderValidator, IProviderHealthChecker, HealthCheckResult } from '../../core/ProviderInterfaces';
 import { ValidationError } from '../../core/ConfigValidator';

package/src/llm/providers/UniversalLLMAdapter.ts

@@ -1,5 +1,6 @@
 import axios, { AxiosInstance } from 'axios';
-import { ILLMProvider, ChatMessage } from '../ILLMProvider';
+import { ILLMProvider } from '../ILLMProvider';
+import { ChatMessage } from '../../types';
 import { LLMConfig, EmbeddingConfig } from '../../config/RagConfig';
 import { resolvePath, buildPayload } from '../../utils/templateUtils';
 import { LLM_PROFILES } from '../../config/UniversalProfiles';

package/src/server.ts

@@ -7,8 +7,8 @@
 
 // ── Shared Types ──────────────────────────────────────────────
 export type { RagConfig, VectorDBConfig, LLMConfig, EmbeddingConfig, UIConfig, RAGConfig, VectorDBProvider, LLMProvider, EmbeddingProvider } from './config/RagConfig';
-export type { VectorMatch, UpsertDocument, IngestDocument, ChatResponse } from './types';
-export type { ILLMProvider, ChatMessage, ChatOptions, EmbedOptions } from './llm/ILLMProvider';
+export type { VectorMatch, UpsertDocument, IngestDocument, ChatResponse, ChatMessage, ChatOptions } from './types';
+export type { ILLMProvider, EmbedOptions } from './llm/ILLMProvider';
 export type { Chunk, ChunkOptions } from './rag/DocumentChunker';
 export type { HealthCheckResult, IProviderValidator, IProviderHealthChecker } from './core/ProviderInterfaces';
 export type { ValidationError } from './core/ConfigValidator';

package/src/types/chat.ts

@@ -0,0 +1,53 @@
+import { VectorMatch } from './index';
+
+export type MessageRole = 'user' | 'assistant' | 'system';
+
+export interface ChatMessage {
+  role: MessageRole;
+  content: string;
+}
+
+export interface RagMessage extends ChatMessage {
+  id: string;
+  sources?: VectorMatch[];
+  createdAt: string;
+}
+
+export interface ChatOptions {
+  /** Override max tokens for this specific call */
+  maxTokens?: number;
+  /** Override temperature for this specific call */
+  temperature?: number;
+  /** Stop sequences */
+  stop?: string[];
+}
+
+export interface UseRagChatOptions {
+  /** Override the chat API endpoint (default: /api/chat) */
+  apiUrl?: string;
+  /** Override project namespace */
+  namespace?: string;
+  /** Persist chat history to localStorage (default: true) */
+  persist?: boolean;
+  /** Called after each successful assistant reply */
+  onReply?: (message: RagMessage) => void;
+  /** Called on error */
+  onError?: (error: string) => void;
+}
+
+export interface UseRagChatReturn {
+  /** All messages in the current conversation */
+  messages: RagMessage[];
+  /** Whether a response is in flight */
+  isLoading: boolean;
+  /** Current error message, or null */
+  error: string | null;
+  /** Send a user message */
+  send: (text: string) => Promise<void>;
+  /** Clear the conversation (and localStorage if persist=true) */
+  clear: () => void;
+  /** Retry the last failed send */
+  retry: () => Promise<void>;
+  /** Programmatically set the conversation (e.g. to restore from a DB) */
+  setMessages: React.Dispatch<React.SetStateAction<RagMessage[]>>;
+}

package/src/types/index.ts

@@ -64,3 +64,6 @@ export interface Product {
   link?: string;
   description?: string;
 }
+
+export * from './props';
+export * from './chat';

package/src/types/props.ts

@@ -0,0 +1,79 @@
+import { ReactNode, CSSProperties, MouseEvent } from 'react';
+import { Product, VectorMatch } from './index';
+import { RagMessage } from './chat';
+import { UIConfig } from '../config/RagConfig';
+
+export interface ChatWindowProps {
+  /** Additional className for the wrapper div */
+  className?: string;
+  /** Inline styles for the wrapper div */
+  style?: CSSProperties;
+  /** Called when the close button is clicked (for widget mode) */
+  onClose?: () => void;
+  /** Whether to show a close (X) button in the header */
+  showClose?: boolean;
+  /** Called when the user starts dragging the resize handle */
+  onResizeStart?: (e: MouseEvent) => void;
+  /** Called when the user clicks the reset size button */
+  onResetResize?: () => void;
+  /** Whether the window has been resized from its default */
+  isResized?: boolean;
+  /** Called when the user clicks the maximize/minimize button */
+  onMaximize?: () => void;
+  /** Whether the window is currently maximized */
+  isMaximized?: boolean;
+  /** Called when the user clicks 'Add to Cart' on a product */
+  onAddToCart?: (product: Product) => void;
+}
+
+export interface ChatWidgetProps {
+  /** Position of the floating button. Defaults to bottom-right. */
+  position?: 'bottom-right' | 'bottom-left';
+  /** Called when the user clicks 'Add to Cart' on a product */
+  onAddToCart?: (product: Product) => void;
+}
+
+export interface MessageBubbleProps {
+  message: RagMessage;
+  sources?: VectorMatch[];
+  isStreaming?: boolean;
+  primaryColor: string;
+  accentColor: string;
+  onAddToCart?: (product: Product) => void;
+}
+
+export interface ProductCardProps {
+  product: Product;
+  primaryColor?: string;
+  onAddToCart?: (product: Product) => void;
+}
+
+export interface ProductCarouselProps {
+  products: Product[];
+  primaryColor?: string;
+  onAddToCart?: (product: Product) => void;
+}
+
+export interface DocumentUploadProps {
+  /** Optional namespace for the upload */
+  namespace?: string;
+  /** Callback when upload completes */
+  onUploadComplete?: (results: unknown) => void;
+  /** Additional className */
+  className?: string;
+}
+
+export interface SourceCardProps {
+  source: VectorMatch;
+  index: number;
+}
+
+export interface ClientConfig {
+  projectId: string;
+  ui: Required<UIConfig>;
+}
+
+export interface ConfigProviderProps {
+  config?: { projectId?: string; ui?: Partial<UIConfig> };
+  children: ReactNode;
+}

package/dist/DocumentChunker-Dh9TvmGG.d.mts

@@ -1,45 +0,0 @@
-/**
- * DocumentChunker — splits long text into overlapping chunks
- * suitable for vector embedding and retrieval.
- *
- * Strategy: sentence-aware sliding window
- *   1. Split on sentence boundaries
- *   2. Accumulate sentences into chunks up to `chunkSize` characters
- *   3. Carry over `chunkOverlap` characters from the previous chunk
- */
-interface Chunk {
-    id: string;
-    content: string;
-    metadata?: Record<string, unknown>;
-}
-interface ChunkOptions {
-    /** Target chunk size in characters (default 1000) */
-    chunkSize?: number;
-    /** Overlap between consecutive chunks in characters (default 200) */
-    chunkOverlap?: number;
-    /** Source document identifier used as ID prefix */
-    docId?: string | number;
-    /** Extra metadata to attach to every chunk */
-    metadata?: Record<string, unknown>;
-    /** Characters used to split text, in order of priority */
-    separators?: string[];
-}
-declare class DocumentChunker {
-    private readonly chunkSize;
-    private readonly chunkOverlap;
-    private readonly separators;
-    constructor(chunkSize?: number, chunkOverlap?: number, separators?: string[]);
-    /**
-     * Split a single text string into overlapping chunks using a recursive strategy.
-     * Preserves structural boundaries (Markdown headers) where possible.
-     */
-    chunk(text: string, options?: ChunkOptions): Chunk[];
-    private recursiveSplit;
-    chunkMany(documents: Array<{
-        content: string;
-        docId?: string | number;
-        metadata?: Record<string, unknown>;
-    }>): Chunk[];
-}
-
-export { type Chunk as C, DocumentChunker as D, type ChunkOptions as a };

package/dist/DocumentChunker-Dh9TvmGG.d.ts

@@ -1,45 +0,0 @@
-/**
- * DocumentChunker — splits long text into overlapping chunks
- * suitable for vector embedding and retrieval.
- *
- * Strategy: sentence-aware sliding window
- *   1. Split on sentence boundaries
- *   2. Accumulate sentences into chunks up to `chunkSize` characters
- *   3. Carry over `chunkOverlap` characters from the previous chunk
- */
-interface Chunk {
-    id: string;
-    content: string;
-    metadata?: Record<string, unknown>;
-}
-interface ChunkOptions {
-    /** Target chunk size in characters (default 1000) */
-    chunkSize?: number;
-    /** Overlap between consecutive chunks in characters (default 200) */
-    chunkOverlap?: number;
-    /** Source document identifier used as ID prefix */
-    docId?: string | number;
-    /** Extra metadata to attach to every chunk */
-    metadata?: Record<string, unknown>;
-    /** Characters used to split text, in order of priority */
-    separators?: string[];
-}
-declare class DocumentChunker {
-    private readonly chunkSize;
-    private readonly chunkOverlap;
-    private readonly separators;
-    constructor(chunkSize?: number, chunkOverlap?: number, separators?: string[]);
-    /**
-     * Split a single text string into overlapping chunks using a recursive strategy.
-     * Preserves structural boundaries (Markdown headers) where possible.
-     */
-    chunk(text: string, options?: ChunkOptions): Chunk[];
-    private recursiveSplit;
-    chunkMany(documents: Array<{
-        content: string;
-        docId?: string | number;
-        metadata?: Record<string, unknown>;
-    }>): Chunk[];
-}
-
-export { type Chunk as C, DocumentChunker as D, type ChunkOptions as a };