@retrivora-ai/rag-engine 1.7.6 → 1.7.8

package/dist/handlers/index.mjs

@@ -10,7 +10,7 @@ import {
   sseMetaFrame,
   sseTextFrame,
   sseUIFrame
-} from "../chunk-XGVKIPNY.mjs";
+} from "../chunk-UHGYLTTY.mjs";
 import "../chunk-YLTMFW4M.mjs";
 import "../chunk-X4TOT24V.mjs";
 export {

package/dist/{index-BejNscWZ.d.mts → index-B1u4loP6.d.mts}

@@ -1,4 +1,4 @@
-import { h as RagConfig, C as ChatMessage, d as ChatResponse, I as IngestDocument } from './index-Cti1u0y1.mjs';
+import { h as RagConfig, C as ChatMessage, d as ChatResponse, I as IngestDocument } from './index-DfsVx0a4.mjs';
 import { NextRequest, NextResponse } from 'next/server';
 
 interface ValidationError {

package/dist/{index-CbkMJvu5.d.ts → index-D9w9fLjh.d.ts}

@@ -1,4 +1,4 @@
-import { h as RagConfig, C as ChatMessage, d as ChatResponse, I as IngestDocument } from './index-Cti1u0y1.js';
+import { h as RagConfig, C as ChatMessage, d as ChatResponse, I as IngestDocument } from './index-DfsVx0a4.js';
 import { NextRequest, NextResponse } from 'next/server';
 
 interface ValidationError {

package/dist/{index-Cti1u0y1.d.mts → index-DfsVx0a4.d.mts}

@@ -6,6 +6,7 @@ interface ChatMessage {
 interface RagMessage extends ChatMessage {
     id: string;
     sources?: VectorMatch[];
+    uiTransformation?: any;
     createdAt: string;
 }
 interface ChatOptions {

package/dist/{index-Cti1u0y1.d.ts → index-DfsVx0a4.d.ts}

@@ -6,6 +6,7 @@ interface ChatMessage {
 interface RagMessage extends ChatMessage {
     id: string;
     sources?: VectorMatch[];
+    uiTransformation?: any;
     createdAt: string;
 }
 interface ChatOptions {

package/dist/index.d.mts

@@ -1,7 +1,7 @@
 import React, { CSSProperties, MouseEvent, ReactNode } from 'react';
-import { P as Product, U as UIConfig, R as RagMessage, V as VectorMatch, a as UseRagChatOptions, b as UseRagChatReturn } from './index-Cti1u0y1.mjs';
-export { C as ChatMessage, c as ChatOptions, d as ChatResponse, E as EmbeddingConfig, e as EmbeddingProvider, I as IngestDocument, L as LLMConfig, f as LLMProvider, g as RAGConfig, h as RagConfig, i as UpsertDocument, j as VectorDBConfig, k as VectorDBProvider } from './index-Cti1u0y1.mjs';
-export { C as Chunk, a as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-D1dg5iCi.mjs';
+import { P as Product, U as UIConfig, R as RagMessage, V as VectorMatch, a as UseRagChatOptions, b as UseRagChatReturn } from './index-DfsVx0a4.mjs';
+export { C as ChatMessage, c as ChatOptions, d as ChatResponse, E as EmbeddingConfig, e as EmbeddingProvider, I as IngestDocument, L as LLMConfig, f as LLMProvider, g as RAGConfig, h as RagConfig, i as UpsertDocument, j as VectorDBConfig, k as VectorDBProvider } from './index-DfsVx0a4.mjs';
+export { C as Chunk, a as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-wKE98F_G.mjs';
 
 type ChatViewportSize = 'compact' | 'medium' | 'large';
 interface ChatWindowProps {

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import React, { CSSProperties, MouseEvent, ReactNode } from 'react';
-import { P as Product, U as UIConfig, R as RagMessage, V as VectorMatch, a as UseRagChatOptions, b as UseRagChatReturn } from './index-Cti1u0y1.js';
-export { C as ChatMessage, c as ChatOptions, d as ChatResponse, E as EmbeddingConfig, e as EmbeddingProvider, I as IngestDocument, L as LLMConfig, f as LLMProvider, g as RAGConfig, h as RagConfig, i as UpsertDocument, j as VectorDBConfig, k as VectorDBProvider } from './index-Cti1u0y1.js';
-export { C as Chunk, a as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-BXOUMKoP.js';
+import { P as Product, U as UIConfig, R as RagMessage, V as VectorMatch, a as UseRagChatOptions, b as UseRagChatReturn } from './index-DfsVx0a4.js';
+export { C as ChatMessage, c as ChatOptions, d as ChatResponse, E as EmbeddingConfig, e as EmbeddingProvider, I as IngestDocument, L as LLMConfig, f as LLMProvider, g as RAGConfig, h as RagConfig, i as UpsertDocument, j as VectorDBConfig, k as VectorDBProvider } from './index-DfsVx0a4.js';
+export { C as Chunk, a as ChunkOptions, E as EmbedOptions, I as ILLMProvider } from './DocumentChunker-DMZVv6hi.js';
 
 type ChatViewportSize = 'compact' | 'medium' | 'large';
 interface ChatWindowProps {

package/dist/index.js

@@ -393,6 +393,178 @@ function DynamicChart({
   ))))));
 }
 
+// src/components/VisualizationRenderer.tsx
+var import_image2 = __toESM(require("next/image"));
+function VisualizationRenderer({
+  data,
+  className = ""
+}) {
+  if (!data) {
+    return /* @__PURE__ */ React.createElement("div", { className: `p-4 text-gray-500 ${className}` }, "No data available for visualization");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: `space-y-4 ${className}` }, /* @__PURE__ */ React.createElement("div", { className: "space-y-2" }, /* @__PURE__ */ React.createElement("h3", { className: "text-lg font-semibold text-gray-900 dark:text-white" }, data.title), data.description && /* @__PURE__ */ React.createElement("p", { className: "text-sm text-gray-600 dark:text-gray-400" }, data.description)), renderVisualization(data.type, data.data));
+}
+function renderVisualization(type, data) {
+  switch (type) {
+    case "pie_chart":
+      return renderPieChart(data);
+    case "bar_chart":
+      return renderBarChart(data);
+    case "line_chart":
+      return renderLineChart(data);
+    case "table":
+      return renderTable(data);
+    case "product_carousel":
+      return renderProductCarousel(data);
+    case "text":
+    default:
+      return renderText(data);
+  }
+}
+function renderPieChart(data) {
+  if (!Array.isArray(data)) {
+    return /* @__PURE__ */ React.createElement("div", { className: "text-red-500" }, "Invalid pie chart data");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: "bg-white dark:bg-gray-800 p-4 rounded-lg shadow" }, /* @__PURE__ */ React.createElement("div", { className: "flex flex-wrap gap-2" }, data.map((item, idx) => /* @__PURE__ */ React.createElement(
+    "div",
+    {
+      key: idx,
+      className: "flex items-center gap-2 p-2 bg-gray-50 dark:bg-gray-700 rounded"
+    },
+    /* @__PURE__ */ React.createElement(
+      "div",
+      {
+        className: "w-3 h-3 rounded-full",
+        style: { backgroundColor: generateColor(idx) }
+      }
+    ),
+    /* @__PURE__ */ React.createElement("span", { className: "text-sm font-medium text-gray-700 dark:text-gray-300" }, item.label, ": ", item.value),
+    item.inStock !== void 0 && /* @__PURE__ */ React.createElement(
+      "span",
+      {
+        className: `text-xs px-2 py-1 rounded ${item.inStock ? "bg-green-100 text-green-800" : "bg-red-100 text-red-800"}`
+      },
+      item.inStock ? "In Stock" : "Out of Stock"
+    )
+  ))));
+}
+function renderBarChart(data) {
+  if (!Array.isArray(data)) {
+    return /* @__PURE__ */ React.createElement("div", { className: "text-red-500" }, "Invalid bar chart data");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: "bg-white dark:bg-gray-800 p-4 rounded-lg shadow overflow-x-auto" }, /* @__PURE__ */ React.createElement("div", { className: "space-y-3" }, data.map((item, idx) => /* @__PURE__ */ React.createElement("div", { key: idx, className: "space-y-1" }, /* @__PURE__ */ React.createElement("div", { className: "flex justify-between items-center" }, /* @__PURE__ */ React.createElement("span", { className: "text-sm font-medium text-gray-700 dark:text-gray-300" }, item.category), /* @__PURE__ */ React.createElement("span", { className: "text-sm text-gray-600 dark:text-gray-400" }, item.value)), /* @__PURE__ */ React.createElement("div", { className: "w-full bg-gray-200 dark:bg-gray-700 rounded-full h-2" }, /* @__PURE__ */ React.createElement(
+    "div",
+    {
+      className: "h-full rounded-full transition-all",
+      style: {
+        width: `${item.value / Math.max(...data.map((d) => d.value)) * 100}%`,
+        backgroundColor: generateColor(idx)
+      }
+    }
+  )), item.inStock !== void 0 && /* @__PURE__ */ React.createElement(
+    "span",
+    {
+      className: `text-xs px-2 py-1 rounded inline-block ${item.inStock ? "bg-green-100 text-green-800" : "bg-red-100 text-red-800"}`
+    },
+    item.inStock ? "In Stock" : "Out of Stock"
+  )))));
+}
+function renderLineChart(data) {
+  if (!Array.isArray(data)) {
+    return /* @__PURE__ */ React.createElement("div", { className: "text-red-500" }, "Invalid line chart data");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: "bg-white dark:bg-gray-800 p-4 rounded-lg shadow overflow-x-auto" }, /* @__PURE__ */ React.createElement("div", { className: "space-y-2" }, data.map((point, idx) => /* @__PURE__ */ React.createElement(
+    "div",
+    {
+      key: idx,
+      className: "flex justify-between items-center p-2 bg-gray-50 dark:bg-gray-700 rounded"
+    },
+    /* @__PURE__ */ React.createElement("span", { className: "text-sm text-gray-600 dark:text-gray-400" }, point.label || point.timestamp),
+    /* @__PURE__ */ React.createElement("span", { className: "text-sm font-semibold text-gray-900 dark:text-white" }, typeof point.value === "number" ? point.value.toFixed(2) : point.value)
+  ))));
+}
+function renderTable(data) {
+  if (!data.columns || !data.rows) {
+    return /* @__PURE__ */ React.createElement("div", { className: "text-red-500" }, "Invalid table data");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: "bg-white dark:bg-gray-800 rounded-lg shadow overflow-x-auto" }, /* @__PURE__ */ React.createElement("table", { className: "w-full text-sm" }, /* @__PURE__ */ React.createElement("thead", { className: "bg-gray-100 dark:bg-gray-700 border-b border-gray-200 dark:border-gray-600" }, /* @__PURE__ */ React.createElement("tr", null, data.columns.map((col, idx) => /* @__PURE__ */ React.createElement(
+    "th",
+    {
+      key: idx,
+      className: "px-4 py-2 text-left font-semibold text-gray-900 dark:text-white"
+    },
+    col
+  )))), /* @__PURE__ */ React.createElement("tbody", { className: "divide-y divide-gray-200 dark:divide-gray-600" }, data.rows.map((row, rowIdx) => /* @__PURE__ */ React.createElement(
+    "tr",
+    {
+      key: rowIdx,
+      className: "hover:bg-gray-50 dark:hover:bg-gray-700 transition-colors"
+    },
+    row.map((cell, cellIdx) => /* @__PURE__ */ React.createElement(
+      "td",
+      {
+        key: cellIdx,
+        className: "px-4 py-2 text-gray-700 dark:text-gray-300"
+      },
+      String(cell)
+    ))
+  )))));
+}
+function renderProductCarousel(data) {
+  if (!Array.isArray(data)) {
+    return /* @__PURE__ */ React.createElement("div", { className: "text-red-500" }, "Invalid carousel data");
+  }
+  return /* @__PURE__ */ React.createElement("div", { className: "overflow-x-auto" }, /* @__PURE__ */ React.createElement("div", { className: "flex gap-4 pb-4" }, data.map((product) => /* @__PURE__ */ React.createElement(
+    "div",
+    {
+      key: product.id,
+      className: "flex-shrink-0 w-64 bg-white dark:bg-gray-800 rounded-lg shadow-md overflow-hidden hover:shadow-lg transition-shadow"
+    },
+    product.image && /* @__PURE__ */ React.createElement("div", { className: "relative w-full h-40 bg-gray-200" }, /* @__PURE__ */ React.createElement(
+      import_image2.default,
+      {
+        src: product.image,
+        alt: product.name,
+        fill: true,
+        className: "object-cover",
+        sizes: "(max-width: 768px) 100vw, (max-width: 1200px) 50vw, 33vw"
+      }
+    )),
+    /* @__PURE__ */ React.createElement("div", { className: "p-4 space-y-2" }, /* @__PURE__ */ React.createElement("h4", { className: "font-semibold text-gray-900 dark:text-white line-clamp-2" }, product.name), product.brand && /* @__PURE__ */ React.createElement("p", { className: "text-xs text-gray-500 dark:text-gray-400" }, "Brand: ", product.brand), product.price && /* @__PURE__ */ React.createElement("p", { className: "text-lg font-bold text-gray-900 dark:text-white" }, "$", typeof product.price === "number" ? product.price.toFixed(2) : product.price), product.description && /* @__PURE__ */ React.createElement("p", { className: "text-sm text-gray-600 dark:text-gray-400 line-clamp-2" }, product.description), /* @__PURE__ */ React.createElement("div", { className: "flex gap-2 pt-2" }, /* @__PURE__ */ React.createElement(
+      "span",
+      {
+        className: `text-xs px-2 py-1 rounded flex-1 text-center ${product.inStock ? "bg-green-100 text-green-800" : "bg-red-100 text-red-800"}`
+      },
+      product.inStock ? "In Stock" : "Out of Stock"
+    )))
+  ))));
+}
+function renderText(data) {
+  const content = (data == null ? void 0 : data.content) || String(data);
+  return /* @__PURE__ */ React.createElement("div", { className: "bg-white dark:bg-gray-800 p-4 rounded-lg shadow prose dark:prose-invert max-w-none" }, /* @__PURE__ */ React.createElement("p", { className: "text-gray-700 dark:text-gray-300 whitespace-pre-wrap" }, content));
+}
+function generateColor(index) {
+  const colors = [
+    "#3B82F6",
+    // blue
+    "#10B981",
+    // green
+    "#F59E0B",
+    // amber
+    "#EF4444",
+    // red
+    "#8B5CF6",
+    // purple
+    "#EC4899",
+    // pink
+    "#06B6D4",
+    // cyan
+    "#F97316"
+    // orange
+  ];
+  return colors[index % colors.length];
+}
+
 // src/components/MessageBubble.tsx
 function sanitizeJson(raw) {
   let s = raw.trim();
@@ -1031,7 +1203,7 @@ ${match.trim()}
         onAddToCart,
         viewportSize
       }
-    ), !shouldRenderStructuredOnly && /* @__PURE__ */ import_react5.default.createElement(import_react_markdown.default, { remarkPlugins: [import_remark_gfm.default], components: markdownComponents }, processedMarkdown), isStreaming && message.content && /* @__PURE__ */ import_react5.default.createElement("span", { className: "inline-block w-1.5 h-3.5 ml-1 bg-emerald-500 animate-pulse align-middle" }))
+    ), !shouldRenderStructuredOnly && /* @__PURE__ */ import_react5.default.createElement(import_react_markdown.default, { remarkPlugins: [import_remark_gfm.default], components: markdownComponents }, processedMarkdown), !isUser && !structuredContent.payload && message.uiTransformation && /* @__PURE__ */ import_react5.default.createElement("div", { className: "mt-4 border-t border-slate-100 dark:border-white/5 pt-4" }, /* @__PURE__ */ import_react5.default.createElement(VisualizationRenderer, { data: message.uiTransformation })), isStreaming && message.content && /* @__PURE__ */ import_react5.default.createElement("span", { className: "inline-block w-1.5 h-3.5 ml-1 bg-emerald-500 animate-pulse align-middle" }))
   ), !isUser && !hasStructuredProductBlock && allProducts.length > 0 && /* @__PURE__ */ import_react5.default.createElement("div", { className: "w-full mt-1" }, /* @__PURE__ */ import_react5.default.createElement(
     ProductCarousel,
     {
@@ -1125,7 +1297,7 @@ function useConfig() {
 var import_react7 = require("react");
 
 // src/hooks/useStoredMessages.ts
-var React7 = __toESM(require("react"));
+var React8 = __toESM(require("react"));
 function readStoredMessages(storageKey) {
   if (typeof window === "undefined") {
     return [];
@@ -1142,8 +1314,8 @@ function readStoredMessages(storageKey) {
   }
 }
 function useStoredMessages(storageKey, persist) {
-  const [messages, setMessages] = React7.useState(() => persist ? readStoredMessages(storageKey) : []);
-  React7.useEffect(() => {
+  const [messages, setMessages] = React8.useState(() => persist ? readStoredMessages(storageKey) : []);
+  React8.useEffect(() => {
     if (!persist || typeof window === "undefined") {
       return;
     }
@@ -1229,6 +1401,7 @@ function useRagChat(projectId, options = {}) {
         const decoder = new TextDecoder();
         let assistantContent = "";
         let sources = [];
+        let uiTransformation = null;
         let buffer = "";
         const assistantMessageId = `assistant_${Date.now()}`;
         setMessages((prev) => [
@@ -1253,13 +1426,19 @@ function useRagChat(projectId, options = {}) {
               assistantContent += frame.text;
             } else if (frame.type === "metadata") {
               sources = (_a = frame.sources) != null ? _a : [];
+            } else if (frame.type === "ui_transformation") {
+              uiTransformation = frame.data;
             } else if (frame.type === "error") {
               setError(frame.error || "Stream error");
             }
           }
           setMessages(
             (prev) => prev.map(
-              (msg) => msg.id === assistantMessageId ? __spreadProps(__spreadValues({}, msg), { content: assistantContent, sources: sources.length > 0 ? sources : msg.sources }) : msg
+              (msg) => msg.id === assistantMessageId ? __spreadProps(__spreadValues({}, msg), {
+                content: assistantContent,
+                sources: sources.length > 0 ? sources : msg.sources,
+                uiTransformation: uiTransformation || msg.uiTransformation
+              }) : msg
             )
           );
         }
@@ -1274,6 +1453,7 @@ function useRagChat(projectId, options = {}) {
           role: "assistant",
           content: assistantContent,
           sources: sources.length > 0 ? sources : void 0,
+          uiTransformation: uiTransformation || void 0,
           createdAt: (/* @__PURE__ */ new Date()).toISOString()
         };
         onReply == null ? void 0 : onReply(finalAssistantMessage);