@rodrigocoliveira/agno-react 1.3.1 → 2.0.0

package/dist/ui.mjs

@@ -749,12 +749,20 @@ import { useCallback } from "react";
 import { StickToBottom, useStickToBottomContext } from "use-stick-to-bottom";
 import { useStickToBottomContext as useStickToBottomContext2 } from "use-stick-to-bottom";
 import { jsx as jsx13, jsxs as jsxs7, Fragment } from "react/jsx-runtime";
-var Conversation = ({ className, ...props }) => /* @__PURE__ */ jsx13(StickToBottom, {
-  className: cn("relative flex-1 overflow-y-auto", className),
-  initial: "smooth",
-  role: "log",
-  ...props
-});
+var Conversation = ({ className, scrollBehavior, ...props }) => {
+  const initial = scrollBehavior?.initial ?? "smooth";
+  const resize = scrollBehavior?.resize ?? "instant";
+  return /* @__PURE__ */ jsx13(StickToBottom, {
+    className: cn("relative flex-1 overflow-y-auto", className),
+    initial,
+    resize,
+    ...scrollBehavior?.damping !== undefined && { damping: scrollBehavior.damping },
+    ...scrollBehavior?.stiffness !== undefined && { stiffness: scrollBehavior.stiffness },
+    ...scrollBehavior?.mass !== undefined && { mass: scrollBehavior.mass },
+    role: "log",
+    ...props
+  });
+};
 var ConversationContent = ({ className, ...props }) => /* @__PURE__ */ jsx13(StickToBottom.Content, {
   className: cn("p-4", className),
   ...props
@@ -2684,582 +2692,1066 @@ var PromptInputCommandSeparator = ({ className, ...props }) => /* @__PURE__ */ j
   className: cn(className),
   ...props
 });
-// src/ui/composed/AgnoMessageItem.tsx
-import { useState as useState8 } from "react";
-import { GenerativeUIRenderer } from "@rodrigocoliveira/agno-react";
-import {
-  AlertCircle,
-  FileIcon as FileIcon3,
-  FileText,
-  Image as ImageIcon2,
-  Lightbulb,
-  Music,
-  Paperclip,
-  Video
-} from "lucide-react";
-import { jsx as jsx35, jsxs as jsxs20, Fragment as Fragment5 } from "react/jsx-runtime";
-var defaultFormatTimestamp = formatSmartTimestamp;
-var getToolState = (tool) => {
-  return tool.tool_call_error ? "output-error" : "output-available";
-};
-function AgnoMessageItem({
+// src/ui/composed/agno-message/message.tsx
+import { useMemo as useMemo5, useState as useState10 } from "react";
+import { AlertCircle as AlertCircle2, FileIcon as FileIcon4, Music as Music2 } from "lucide-react";
+
+// src/ui/composed/agno-message/context.ts
+import { createContext as createContext3, useContext as useContext3 } from "react";
+var AgnoMessageContext = createContext3(null);
+function useAgnoMessageContext() {
+  const ctx = useContext3(AgnoMessageContext);
+  if (!ctx) {
+    throw new Error("useAgnoMessageContext must be used within an <AgnoMessage> provider. " + "Wrap your slots with <AgnoMessage message={...}>.");
+  }
+  return ctx;
+}
+
+// src/ui/composed/agno-message/reasoning.tsx
+import { Lightbulb } from "lucide-react";
+import { jsx as jsx35, jsxs as jsxs20 } from "react/jsx-runtime";
+function AgnoMessageReasoning() {
+  const { message, classNames } = useAgnoMessageContext();
+  const steps = message.extra_data?.reasoning_steps;
+  if (!steps || steps.length === 0)
+    return null;
+  return /* @__PURE__ */ jsxs20("div", {
+    className: cn("space-y-2 pt-1", classNames?.assistant?.reasoning),
+    children: [
+      /* @__PURE__ */ jsxs20("div", {
+        className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+        children: [
+          /* @__PURE__ */ jsx35(Lightbulb, {
+            className: "h-3.5 w-3.5"
+          }),
+          "Reasoning (",
+          steps.length,
+          " steps)"
+        ]
+      }),
+      /* @__PURE__ */ jsx35(Accordion, {
+        type: "multiple",
+        className: "w-full",
+        children: steps.map((step, idx) => /* @__PURE__ */ jsxs20(AccordionItem, {
+          value: `reasoning-${idx}`,
+          className: "border-muted",
+          children: [
+            /* @__PURE__ */ jsx35(AccordionTrigger, {
+              className: "text-xs py-1.5 hover:no-underline",
+              children: step.title || `Step ${idx + 1}`
+            }),
+            /* @__PURE__ */ jsxs20(AccordionContent, {
+              className: "space-y-1.5 text-xs text-muted-foreground",
+              children: [
+                step.action && /* @__PURE__ */ jsxs20("div", {
+                  children: [
+                    /* @__PURE__ */ jsx35("span", {
+                      className: "font-medium text-foreground",
+                      children: "Action:"
+                    }),
+                    " ",
+                    step.action
+                  ]
+                }),
+                step.reasoning && /* @__PURE__ */ jsxs20("div", {
+                  children: [
+                    /* @__PURE__ */ jsx35("span", {
+                      className: "font-medium text-foreground",
+                      children: "Reasoning:"
+                    }),
+                    " ",
+                    step.reasoning
+                  ]
+                }),
+                step.result && /* @__PURE__ */ jsxs20("div", {
+                  children: [
+                    /* @__PURE__ */ jsx35("span", {
+                      className: "font-medium text-foreground",
+                      children: "Result:"
+                    }),
+                    " ",
+                    step.result
+                  ]
+                }),
+                step.confidence !== undefined && /* @__PURE__ */ jsxs20("div", {
+                  children: [
+                    /* @__PURE__ */ jsx35("span", {
+                      className: "font-medium text-foreground",
+                      children: "Confidence:"
+                    }),
+                    " ",
+                    (step.confidence * 100).toFixed(1),
+                    "%"
+                  ]
+                })
+              ]
+            })
+          ]
+        }, idx))
+      })
+    ]
+  });
+}
+
+// src/ui/composed/agno-message/media.tsx
+import { FileIcon as FileIcon3, Image as ImageIcon2, Music, Paperclip, Video } from "lucide-react";
+import { jsx as jsx36, jsxs as jsxs21, Fragment as Fragment5 } from "react/jsx-runtime";
+function AgnoMessageMedia() {
+  const {
+    message,
+    classNames,
+    showImageLightbox,
+    showFilePreview,
+    openImageLightbox,
+    openFilePreview
+  } = useAgnoMessageContext();
+  const mediaClassName = classNames?.assistant?.media;
+  const hasImages = !!message.images && message.images.length > 0;
+  const hasVideos = !!message.videos && message.videos.length > 0;
+  const hasAudio = !!message.audio && message.audio.length > 0;
+  const hasFiles = !!message.files && message.files.length > 0;
+  const hasResponseAudio = !!message.response_audio;
+  if (!hasImages && !hasVideos && !hasAudio && !hasFiles && !hasResponseAudio)
+    return null;
+  return /* @__PURE__ */ jsxs21(Fragment5, {
+    children: [
+      hasImages && /* @__PURE__ */ jsxs21("div", {
+        className: cn("space-y-2 pt-1", mediaClassName),
+        children: [
+          /* @__PURE__ */ jsxs21("div", {
+            className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+            children: [
+              /* @__PURE__ */ jsx36(ImageIcon2, {
+                className: "h-3.5 w-3.5"
+              }),
+              "Images (",
+              message.images.length,
+              ")"
+            ]
+          }),
+          /* @__PURE__ */ jsx36("div", {
+            className: "grid grid-cols-2 gap-2",
+            children: message.images.map((img, idx) => /* @__PURE__ */ jsxs21("div", {
+              className: "space-y-1",
+              children: [
+                showImageLightbox ? /* @__PURE__ */ jsx36("button", {
+                  type: "button",
+                  onClick: () => openImageLightbox(message.images.map((i) => ({ url: i.url, alt: i.revised_prompt })), idx),
+                  className: "group relative w-full overflow-hidden rounded-lg border border-border cursor-pointer hover:border-primary/50 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring transition-colors",
+                  children: /* @__PURE__ */ jsx36("img", {
+                    src: img.url,
+                    alt: img.revised_prompt || "Generated image",
+                    className: "w-full rounded-lg"
+                  })
+                }) : /* @__PURE__ */ jsx36("img", {
+                  src: img.url,
+                  alt: img.revised_prompt || "Generated image",
+                  className: "w-full rounded-lg border border-border"
+                }),
+                img.revised_prompt && /* @__PURE__ */ jsx36("p", {
+                  className: "text-[11px] text-muted-foreground italic px-0.5",
+                  children: img.revised_prompt
+                })
+              ]
+            }, idx))
+          })
+        ]
+      }),
+      hasVideos && /* @__PURE__ */ jsxs21("div", {
+        className: cn("space-y-2 pt-1", mediaClassName),
+        children: [
+          /* @__PURE__ */ jsxs21("div", {
+            className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+            children: [
+              /* @__PURE__ */ jsx36(Video, {
+                className: "h-3.5 w-3.5"
+              }),
+              "Videos (",
+              message.videos.length,
+              ")"
+            ]
+          }),
+          /* @__PURE__ */ jsx36("div", {
+            className: "space-y-2",
+            children: message.videos.map((video, idx) => /* @__PURE__ */ jsx36("div", {
+              children: video.url ? /* @__PURE__ */ jsx36("video", {
+                src: video.url,
+                controls: true,
+                className: "w-full rounded-lg border border-border"
+              }) : /* @__PURE__ */ jsxs21("div", {
+                className: "bg-muted/50 border border-border p-2.5 rounded-lg text-xs text-muted-foreground",
+                children: [
+                  "Video ID: ",
+                  video.id,
+                  " (ETA: ",
+                  video.eta,
+                  "s)"
+                ]
+              })
+            }, idx))
+          })
+        ]
+      }),
+      hasAudio && /* @__PURE__ */ jsxs21("div", {
+        className: cn("space-y-2 pt-1", mediaClassName),
+        children: [
+          /* @__PURE__ */ jsxs21("div", {
+            className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+            children: [
+              /* @__PURE__ */ jsx36(Music, {
+                className: "h-3.5 w-3.5"
+              }),
+              "Audio (",
+              message.audio.length,
+              ")"
+            ]
+          }),
+          /* @__PURE__ */ jsx36("div", {
+            className: "space-y-2",
+            children: message.audio.map((audio, idx) => /* @__PURE__ */ jsx36("div", {
+              children: audio.url ? /* @__PURE__ */ jsx36("audio", {
+                src: audio.url,
+                controls: true,
+                className: "w-full"
+              }) : audio.base64_audio ? /* @__PURE__ */ jsx36("audio", {
+                src: `data:${audio.mime_type || "audio/wav"};base64,${audio.base64_audio}`,
+                controls: true,
+                className: "w-full"
+              }) : /* @__PURE__ */ jsx36("div", {
+                className: "bg-muted/50 border border-border p-2.5 rounded-lg text-xs text-muted-foreground",
+                children: "Audio data unavailable"
+              })
+            }, idx))
+          })
+        ]
+      }),
+      hasFiles && /* @__PURE__ */ jsxs21("div", {
+        className: cn("space-y-2 pt-1", mediaClassName),
+        children: [
+          /* @__PURE__ */ jsxs21("div", {
+            className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+            children: [
+              /* @__PURE__ */ jsx36(Paperclip, {
+                className: "h-3.5 w-3.5"
+              }),
+              "Files (",
+              message.files.length,
+              ")"
+            ]
+          }),
+          /* @__PURE__ */ jsx36("div", {
+            className: "flex flex-wrap gap-2",
+            children: message.files.map((file, idx) => showFilePreview ? /* @__PURE__ */ jsx36(FilePreviewCard, {
+              file: { name: file.name, type: file.type, url: file.url, size: file.size },
+              onClick: () => openFilePreview({ name: file.name, type: file.type, url: file.url, size: file.size })
+            }, idx) : /* @__PURE__ */ jsxs21("div", {
+              className: "flex items-center gap-2 rounded-lg border border-border px-3 py-2 text-xs bg-muted/30 hover:bg-muted/50 transition-colors",
+              children: [
+                /* @__PURE__ */ jsx36(FileIcon3, {
+                  className: "h-3.5 w-3.5 shrink-0 text-muted-foreground"
+                }),
+                /* @__PURE__ */ jsx36("span", {
+                  className: "truncate max-w-[180px]",
+                  children: file.name
+                }),
+                file.size && /* @__PURE__ */ jsxs21("span", {
+                  className: "text-muted-foreground/70",
+                  children: [
+                    "(",
+                    (file.size / 1024).toFixed(1),
+                    "KB)"
+                  ]
+                }),
+                file.url && /^https?:\/\//i.test(file.url) && /* @__PURE__ */ jsx36("a", {
+                  href: file.url,
+                  target: "_blank",
+                  rel: "noopener noreferrer",
+                  className: "text-primary hover:underline font-medium",
+                  children: "View"
+                })
+              ]
+            }, idx))
+          })
+        ]
+      }),
+      hasResponseAudio && message.response_audio && /* @__PURE__ */ jsxs21("div", {
+        className: cn("space-y-2 pt-1", mediaClassName),
+        children: [
+          /* @__PURE__ */ jsxs21("div", {
+            className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+            children: [
+              /* @__PURE__ */ jsx36(Music, {
+                className: "h-3.5 w-3.5"
+              }),
+              "Response Audio"
+            ]
+          }),
+          message.response_audio.transcript && /* @__PURE__ */ jsxs21("div", {
+            className: "text-xs italic bg-muted/50 border border-border p-2.5 rounded-lg text-muted-foreground",
+            children: [
+              '"',
+              message.response_audio.transcript,
+              '"'
+            ]
+          }),
+          message.response_audio.content && /* @__PURE__ */ jsx36("audio", {
+            src: `data:audio/wav;base64,${message.response_audio.content}`,
+            controls: true,
+            className: "w-full"
+          })
+        ]
+      })
+    ]
+  });
+}
+
+// src/ui/composed/agno-chat/context.ts
+import { createContext as createContext4, useContext as useContext4 } from "react";
+var AgnoChatContext = createContext4(null);
+function useAgnoChatContext() {
+  const ctx = useContext4(AgnoChatContext);
+  if (!ctx) {
+    throw new Error("useAgnoChatContext must be used within an <AgnoChat> provider. " + "Wrap your component tree with <AgnoChat>.");
+  }
+  return ctx;
+}
+
+// src/components/GenerativeUIRenderer.tsx
+import React13 from "react";
+
+// src/utils/component-registry.ts
+class ComponentRegistry {
+  static instance;
+  components = new Map;
+  constructor() {}
+  static getInstance() {
+    if (!ComponentRegistry.instance) {
+      ComponentRegistry.instance = new ComponentRegistry;
+    }
+    return ComponentRegistry.instance;
+  }
+  register(type, renderer) {
+    this.components.set(type, renderer);
+  }
+  registerBatch(components) {
+    Object.entries(components).forEach(([type, renderer]) => {
+      this.register(type, renderer);
+    });
+  }
+  get(type) {
+    return this.components.get(type);
+  }
+  has(type) {
+    return this.components.has(type);
+  }
+  unregister(type) {
+    this.components.delete(type);
+  }
+  getRegisteredTypes() {
+    return Array.from(this.components.keys());
+  }
+  clear() {
+    this.components.clear();
+  }
+}
+function getComponentRegistry() {
+  return ComponentRegistry.getInstance();
+}
+
+// src/hooks/useAgnoToolExecution.ts
+import { useState as useState9, useEffect as useEffect7, useCallback as useCallback8, useMemo as useMemo4 } from "react";
+
+// src/context/AgnoContext.tsx
+import { createContext as createContext5, useContext as useContext5, useMemo as useMemo3, useEffect as useEffect6 } from "react";
+import { AgnoClient } from "@rodrigocoliveira/agno-client";
+import { jsx as jsx37 } from "react/jsx-runtime";
+var AgnoContext = createContext5(null);
+
+// src/context/ToolHandlerContext.tsx
+import { createContext as createContext6, useContext as useContext6, useState as useState8, useCallback as useCallback7 } from "react";
+import { jsx as jsx38 } from "react/jsx-runtime";
+var ToolHandlerContext = createContext6(null);
+
+// src/hooks/useAgnoToolExecution.ts
+var customRenderRegistry = new Map;
+function getCustomRender(key) {
+  return customRenderRegistry.get(key);
+}
+
+// src/components/GenerativeUIRenderer.tsx
+import { jsx as jsx39, jsxs as jsxs22 } from "react/jsx-runtime";
+
+class UIErrorBoundary extends React13.Component {
+  constructor(props) {
+    super(props);
+    this.state = { hasError: false };
+  }
+  static getDerivedStateFromError(error) {
+    return { hasError: true, error };
+  }
+  componentDidCatch(error, errorInfo) {
+    console.error("[GenerativeUIRenderer] Error rendering component:", error, errorInfo);
+    this.props.onError?.(error);
+  }
+  render() {
+    if (this.state.hasError) {
+      return this.props.fallback || /* @__PURE__ */ jsxs22("div", {
+        className: "p-4 border border-red-300 rounded-md bg-red-50 text-red-800",
+        children: [
+          /* @__PURE__ */ jsx39("p", {
+            className: "font-semibold",
+            children: "Failed to render UI component"
+          }),
+          /* @__PURE__ */ jsx39("p", {
+            className: "text-sm mt-1",
+            children: this.state.error?.message || "Unknown error"
+          })
+        ]
+      });
+    }
+    return this.props.children;
+  }
+}
+function GenerativeUIRenderer({
+  spec,
+  className,
+  onError
+}) {
+  const registry = getComponentRegistry();
+  if (spec.type === "custom") {
+    const customSpec = spec;
+    if (customSpec.renderKey) {
+      const renderFn = getCustomRender(customSpec.renderKey);
+      if (renderFn) {
+        return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+          onError,
+          children: /* @__PURE__ */ jsx39("div", {
+            className,
+            children: renderFn(customSpec.props || {})
+          })
+        });
+      }
+    }
+    return /* @__PURE__ */ jsxs22("div", {
+      className: `p-4 border border-yellow-300 rounded-md bg-yellow-50 text-yellow-800 ${className || ""}`,
+      children: [
+        /* @__PURE__ */ jsx39("p", {
+          className: "font-semibold",
+          children: "Custom component not available"
+        }),
+        /* @__PURE__ */ jsx39("p", {
+          className: "text-sm mt-1",
+          children: "The custom render function for this component is not available."
+        })
+      ]
+    });
+  }
+  if (spec.type === "chart") {
+    const chartSpec = spec;
+    const chartType = `chart:${chartSpec.component}`;
+    if (registry.has(chartType)) {
+      const ChartRenderer = registry.get(chartType);
+      return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+        onError,
+        children: /* @__PURE__ */ jsxs22("div", {
+          className,
+          children: [
+            chartSpec.title && /* @__PURE__ */ jsx39("h3", {
+              className: "font-semibold mb-2",
+              children: chartSpec.title
+            }),
+            chartSpec.description && /* @__PURE__ */ jsx39("p", {
+              className: "text-sm text-gray-600 mb-4",
+              children: chartSpec.description
+            }),
+            /* @__PURE__ */ jsx39(ChartRenderer, {
+              ...chartSpec.props
+            })
+          ]
+        })
+      });
+    }
+    return /* @__PURE__ */ jsxs22("div", {
+      className: `p-4 border border-gray-300 rounded-md ${className || ""}`,
+      children: [
+        /* @__PURE__ */ jsx39("p", {
+          className: "font-semibold mb-2",
+          children: chartSpec.title || "Chart Data"
+        }),
+        chartSpec.description && /* @__PURE__ */ jsx39("p", {
+          className: "text-sm text-gray-600 mb-2",
+          children: chartSpec.description
+        }),
+        /* @__PURE__ */ jsx39("pre", {
+          className: "text-xs bg-gray-100 p-2 rounded overflow-auto",
+          children: JSON.stringify(chartSpec.props.data, null, 2)
+        })
+      ]
+    });
+  }
+  if (spec.type === "card-grid") {
+    const cardGridSpec = spec;
+    if (registry.has("card-grid")) {
+      const CardGridRenderer = registry.get("card-grid");
+      return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+        onError,
+        children: /* @__PURE__ */ jsxs22("div", {
+          className,
+          children: [
+            cardGridSpec.title && /* @__PURE__ */ jsx39("h3", {
+              className: "font-semibold mb-2",
+              children: cardGridSpec.title
+            }),
+            cardGridSpec.description && /* @__PURE__ */ jsx39("p", {
+              className: "text-sm text-gray-600 mb-4",
+              children: cardGridSpec.description
+            }),
+            /* @__PURE__ */ jsx39(CardGridRenderer, {
+              ...cardGridSpec.props
+            })
+          ]
+        })
+      });
+    }
+  }
+  if (spec.type === "table") {
+    const tableSpec = spec;
+    if (registry.has("table")) {
+      const TableRenderer = registry.get("table");
+      return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+        onError,
+        children: /* @__PURE__ */ jsxs22("div", {
+          className,
+          children: [
+            tableSpec.title && /* @__PURE__ */ jsx39("h3", {
+              className: "font-semibold mb-2",
+              children: tableSpec.title
+            }),
+            tableSpec.description && /* @__PURE__ */ jsx39("p", {
+              className: "text-sm text-gray-600 mb-4",
+              children: tableSpec.description
+            }),
+            /* @__PURE__ */ jsx39(TableRenderer, {
+              ...tableSpec.props
+            })
+          ]
+        })
+      });
+    }
+  }
+  if (spec.type === "markdown") {
+    const markdownSpec = spec;
+    if (registry.has("markdown")) {
+      const MarkdownRenderer = registry.get("markdown");
+      return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+        onError,
+        children: /* @__PURE__ */ jsx39("div", {
+          className,
+          children: /* @__PURE__ */ jsx39(MarkdownRenderer, {
+            ...markdownSpec.props
+          })
+        })
+      });
+    }
+    return /* @__PURE__ */ jsx39("div", {
+      className,
+      children: markdownSpec.props.content
+    });
+  }
+  if (spec.type === "artifact") {
+    const artifactSpec = spec;
+    return /* @__PURE__ */ jsx39(UIErrorBoundary, {
+      onError,
+      children: /* @__PURE__ */ jsxs22("div", {
+        className: `p-4 border rounded-md ${className || ""}`,
+        children: [
+          artifactSpec.title && /* @__PURE__ */ jsx39("h3", {
+            className: "font-semibold mb-4",
+            children: artifactSpec.title
+          }),
+          artifactSpec.description && /* @__PURE__ */ jsx39("p", {
+            className: "text-sm text-gray-600 mb-4",
+            children: artifactSpec.description
+          }),
+          /* @__PURE__ */ jsx39("div", {
+            className: "space-y-4",
+            children: artifactSpec.props.content?.map((childSpec, index) => /* @__PURE__ */ jsx39(GenerativeUIRenderer, {
+              spec: childSpec,
+              onError
+            }, index))
+          })
+        ]
+      })
+    });
+  }
+  return /* @__PURE__ */ jsxs22("div", {
+    className: `p-4 border border-gray-300 rounded-md ${className || ""}`,
+    children: [
+      /* @__PURE__ */ jsx39("p", {
+        className: "font-semibold",
+        children: "Unsupported UI component"
+      }),
+      /* @__PURE__ */ jsxs22("p", {
+        className: "text-sm text-gray-600 mt-1",
+        children: [
+          "Component type: ",
+          spec.type
+        ]
+      })
+    ]
+  });
+}
+
+// src/ui/composed/agno-chat/tool-building-blocks.tsx
+import { jsx as jsx40, jsxs as jsxs23 } from "react/jsx-runtime";
+var getToolState = (tool) => tool.tool_call_error ? "output-error" : "output-available";
+function ToolDebugCard({ tool, defaultOpen }) {
+  const output = tool.result ?? tool.content;
+  return /* @__PURE__ */ jsxs23(Tool, {
+    defaultOpen,
+    children: [
+      /* @__PURE__ */ jsx40(ToolHeader, {
+        title: tool.tool_name,
+        type: "tool-use",
+        state: getToolState(tool)
+      }),
+      /* @__PURE__ */ jsxs23(ToolContent, {
+        children: [
+          /* @__PURE__ */ jsx40(ToolInput, {
+            input: tool.tool_args
+          }),
+          output ? /* @__PURE__ */ jsx40(ToolOutput, {
+            output,
+            errorText: tool.tool_call_error ? "Tool execution failed" : undefined
+          }) : null
+        ]
+      })
+    ]
+  });
+}
+function ToolGenerativeUI({ tool }) {
+  const uiComponent = tool.ui_component;
+  if (!uiComponent)
+    return null;
+  return uiComponent.layout === "artifact" ? /* @__PURE__ */ jsx40(Artifact, {
+    children: /* @__PURE__ */ jsx40(GenerativeUIRenderer, {
+      spec: uiComponent,
+      className: "w-full p-2"
+    })
+  }) : /* @__PURE__ */ jsx40(GenerativeUIRenderer, {
+    spec: uiComponent,
+    className: "w-full"
+  });
+}
+
+// src/ui/composed/agno-message/tools.tsx
+import { jsx as jsx41, jsxs as jsxs24, Fragment as Fragment6 } from "react/jsx-runtime";
+function AgnoMessageTools({ renderTool: renderToolProp } = {}) {
+  const { message, classNames, renderTool: ctxMsgRenderTool } = useAgnoMessageContext();
+  const { renderTool: ctxChatRenderTool, isDebug } = useAgnoChatContext();
+  const renderTool = renderToolProp ?? ctxMsgRenderTool ?? ctxChatRenderTool;
+  if (!message.tool_calls || message.tool_calls.length === 0)
+    return null;
+  return /* @__PURE__ */ jsx41("div", {
+    className: cn("space-y-2 pt-1", classNames?.assistant?.toolCalls),
+    children: message.tool_calls.map((tool, idx) => {
+      const defaultRender = () => /* @__PURE__ */ jsxs24(Fragment6, {
+        children: [
+          /* @__PURE__ */ jsx41(ToolGenerativeUI, {
+            tool
+          }),
+          isDebug ? /* @__PURE__ */ jsx41(ToolDebugCard, {
+            tool,
+            defaultOpen: idx === 0
+          }) : null
+        ]
+      });
+      const node = renderTool ? renderTool(tool, { index: idx, isDebug, defaultRender }) : defaultRender();
+      if (node === null || node === undefined)
+        return null;
+      return /* @__PURE__ */ jsx41("div", {
+        children: node
+      }, tool.tool_call_id || idx);
+    })
+  });
+}
+
+// src/ui/composed/agno-message/content.tsx
+import { jsx as jsx42 } from "react/jsx-runtime";
+function AgnoMessageContent() {
+  const { message } = useAgnoMessageContext();
+  if (!message.content)
+    return null;
+  return /* @__PURE__ */ jsx42("div", {
+    className: "prose prose-sm dark:prose-invert max-w-none prose-p:leading-relaxed prose-pre:bg-muted prose-pre:border prose-pre:border-border",
+    children: /* @__PURE__ */ jsx42(Response, {
+      children: message.content
+    })
+  });
+}
+
+// src/ui/composed/agno-message/references.tsx
+import { FileText } from "lucide-react";
+import { jsx as jsx43, jsxs as jsxs25 } from "react/jsx-runtime";
+function AgnoMessageReferences() {
+  const { message, classNames } = useAgnoMessageContext();
+  const references = message.extra_data?.references;
+  if (!references || references.length === 0)
+    return null;
+  return /* @__PURE__ */ jsxs25("div", {
+    className: cn("space-y-2 pt-1", classNames?.assistant?.references),
+    children: [
+      /* @__PURE__ */ jsxs25("div", {
+        className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
+        children: [
+          /* @__PURE__ */ jsx43(FileText, {
+            className: "h-3.5 w-3.5"
+          }),
+          "References (",
+          references.length,
+          ")"
+        ]
+      }),
+      /* @__PURE__ */ jsx43("div", {
+        className: "space-y-2",
+        children: references.map((refData, idx) => /* @__PURE__ */ jsxs25("div", {
+          className: "text-xs space-y-1.5",
+          children: [
+            refData.query && /* @__PURE__ */ jsxs25("div", {
+              className: "font-medium text-foreground",
+              children: [
+                "Query: ",
+                refData.query
+              ]
+            }),
+            refData.references.map((ref, refIdx) => /* @__PURE__ */ jsxs25("div", {
+              className: "bg-muted/50 border border-border p-2.5 rounded-lg",
+              children: [
+                /* @__PURE__ */ jsxs25("div", {
+                  className: "italic text-muted-foreground mb-1",
+                  children: [
+                    '"',
+                    ref.content,
+                    '"'
+                  ]
+                }),
+                /* @__PURE__ */ jsxs25("div", {
+                  className: "text-muted-foreground/70",
+                  children: [
+                    "Source: ",
+                    ref.name,
+                    " (chunk ",
+                    ref.meta_data.chunk,
+                    "/",
+                    ref.meta_data.chunk_size,
+                    ")"
+                  ]
+                })
+              ]
+            }, refIdx))
+          ]
+        }, idx))
+      })
+    ]
+  });
+}
+
+// src/ui/composed/agno-message/footer.tsx
+import { AlertCircle } from "lucide-react";
+import { jsx as jsx44, jsxs as jsxs26 } from "react/jsx-runtime";
+function AgnoMessageFooter({ showTimestamp = true } = {}) {
+  const { message, classNames, actions, isLastAssistantMessage, formatTimestamp } = useAgnoMessageContext();
+  const hasError = message.streamingError;
+  const isCustomTimestamp = !!formatTimestamp;
+  const resolvedFormatTimestamp = formatTimestamp ?? formatSmartTimestamp;
+  if (!actions?.assistant && !showTimestamp && !hasError)
+    return null;
+  return /* @__PURE__ */ jsxs26("div", {
+    className: "flex items-center gap-2 pt-1",
+    children: [
+      actions?.assistant && (() => {
+        const visibility = actions.visibility ?? "visible";
+        if (visibility === "last-assistant" && !isLastAssistantMessage)
+          return null;
+        const useHover = visibility === "hover" || visibility === "hover-last-visible" && !isLastAssistantMessage;
+        return /* @__PURE__ */ jsx44("div", {
+          className: cn("flex items-center gap-1 transition-opacity", useHover && "opacity-0 group-hover/message:opacity-100", classNames?.assistant?.actions),
+          children: actions.assistant(message)
+        });
+      })(),
+      hasError && /* @__PURE__ */ jsxs26("span", {
+        className: "flex items-center gap-1 text-[11px] text-destructive",
+        children: [
+          /* @__PURE__ */ jsx44(AlertCircle, {
+            className: "h-3 w-3"
+          }),
+          "Error"
+        ]
+      }),
+      showTimestamp && /* @__PURE__ */ jsx44(SmartTimestamp, {
+        date: new Date(message.created_at * 1000),
+        formatShort: isCustomTimestamp ? resolvedFormatTimestamp : undefined,
+        className: "text-[11px] text-muted-foreground"
+      })
+    ]
+  });
+}
+
+// src/ui/composed/agno-message/message.tsx
+import { jsx as jsx45, jsxs as jsxs27, Fragment as Fragment7 } from "react/jsx-runtime";
+function DefaultAssistantComposition({ showTimestamp }) {
+  return /* @__PURE__ */ jsxs27(Fragment7, {
+    children: [
+      /* @__PURE__ */ jsx45(AgnoMessageReasoning, {}),
+      /* @__PURE__ */ jsx45(AgnoMessageMedia, {}),
+      /* @__PURE__ */ jsx45(AgnoMessageTools, {}),
+      /* @__PURE__ */ jsx45(AgnoMessageContent, {}),
+      /* @__PURE__ */ jsx45(AgnoMessageReferences, {}),
+      /* @__PURE__ */ jsx45(AgnoMessageFooter, {
+        showTimestamp
+      })
+    ]
+  });
+}
+function AgnoMessage({
   message,
   className,
   classNames,
-  renderContent,
-  renderToolCall,
-  renderMedia,
   avatars,
   actions,
   isLastAssistantMessage = false,
-  showReasoning = true,
-  showReferences = true,
-  showTimestamp = true,
-  showGenerativeUI = true,
-  showToolCalls = true,
   showFilePreview = true,
   showImageLightbox = true,
+  showTimestamp = true,
   formatTimestamp,
-  toolResultRenderers
+  renderTool,
+  children
 }) {
   const isUser = message.role === "user";
   const hasError = message.streamingError;
-  const toolsWithUI = message.tool_calls?.filter((tool) => tool.ui_component) || [];
-  const [preview, setPreview] = useState8(null);
+  const [preview, setPreview] = useState10(null);
   const isCustomTimestamp = !!formatTimestamp;
-  const resolvedFormatTimestamp = formatTimestamp ?? defaultFormatTimestamp;
-  const openImageLightbox = (images, index) => {
-    if (!showImageLightbox)
-      return;
-    setPreview({ type: "image", images, initialIndex: index });
-  };
-  const openFilePreview = (file) => {
-    if (!showFilePreview)
-      return;
-    setPreview({ type: "file", file });
-  };
+  const resolvedFormatTimestamp = formatTimestamp ?? formatSmartTimestamp;
+  const ctx = useMemo5(() => ({
+    message,
+    isLastAssistantMessage,
+    classNames,
+    actions,
+    avatars,
+    formatTimestamp,
+    showFilePreview,
+    showImageLightbox,
+    openImageLightbox: (images, index) => {
+      if (!showImageLightbox)
+        return;
+      setPreview({ type: "image", images, initialIndex: index });
+    },
+    openFilePreview: (file) => {
+      if (!showFilePreview)
+        return;
+      setPreview({ type: "file", file });
+    },
+    renderTool
+  }), [
+    message,
+    isLastAssistantMessage,
+    classNames,
+    actions,
+    avatars,
+    formatTimestamp,
+    showFilePreview,
+    showImageLightbox,
+    renderTool
+  ]);
   const closePreview = () => setPreview(null);
-  return /* @__PURE__ */ jsxs20("div", {
-    className: cn("py-5 first:pt-2", isUser ? "flex justify-end" : "", classNames?.root, className),
+  return /* @__PURE__ */ jsx45(AgnoMessageContext.Provider, {
+    value: ctx,
+    children: /* @__PURE__ */ jsxs27("div", {
+      className: cn("py-5 first:pt-2", isUser ? "flex justify-end" : "", classNames?.root, className),
+      children: [
+        isUser ? /* @__PURE__ */ jsx45(UserMessageLayout, {
+          message,
+          classNames,
+          avatars,
+          actions,
+          isCustomTimestamp,
+          resolvedFormatTimestamp,
+          showTimestamp,
+          showImageLightbox,
+          showFilePreview,
+          openImageLightbox: ctx.openImageLightbox,
+          openFilePreview: ctx.openFilePreview,
+          hasError
+        }) : /* @__PURE__ */ jsxs27("div", {
+          className: "flex items-start gap-3 group/message",
+          children: [
+            avatars?.assistant,
+            /* @__PURE__ */ jsx45("div", {
+              className: cn("flex-1 min-w-0 space-y-3", classNames?.assistant?.container),
+              children: children ?? /* @__PURE__ */ jsx45(DefaultAssistantComposition, {
+                showTimestamp
+              })
+            })
+          ]
+        }),
+        preview?.type === "image" && /* @__PURE__ */ jsx45(ImageLightbox, {
+          open: true,
+          onOpenChange: (open) => {
+            if (!open)
+              closePreview();
+          },
+          images: preview.images,
+          initialIndex: preview.initialIndex
+        }),
+        preview?.type === "file" && /* @__PURE__ */ jsx45(FilePreviewModal, {
+          open: true,
+          onOpenChange: (open) => {
+            if (!open)
+              closePreview();
+          },
+          file: preview.file
+        })
+      ]
+    })
+  });
+}
+AgnoMessage.Reasoning = AgnoMessageReasoning;
+AgnoMessage.Media = AgnoMessageMedia;
+AgnoMessage.Tools = AgnoMessageTools;
+AgnoMessage.Content = AgnoMessageContent;
+AgnoMessage.References = AgnoMessageReferences;
+AgnoMessage.Footer = AgnoMessageFooter;
+function UserMessageLayout({
+  message,
+  classNames,
+  avatars,
+  actions,
+  isCustomTimestamp,
+  resolvedFormatTimestamp,
+  showTimestamp,
+  showImageLightbox,
+  showFilePreview,
+  openImageLightbox,
+  openFilePreview,
+  hasError
+}) {
+  return /* @__PURE__ */ jsxs27("div", {
+    className: "flex items-start gap-2.5 max-w-[80%] flex-row-reverse",
     children: [
-      isUser ? /* @__PURE__ */ jsxs20("div", {
-        className: "flex items-start gap-2.5 max-w-[80%] flex-row-reverse",
+      avatars?.user,
+      /* @__PURE__ */ jsxs27("div", {
+        className: "space-y-1.5 flex flex-col items-end min-w-0",
         children: [
-          avatars?.user,
-          /* @__PURE__ */ jsxs20("div", {
-            className: "space-y-1.5 flex flex-col items-end min-w-0",
+          (message.images && message.images.length > 0 || message.audio && message.audio.length > 0 || message.files && message.files.length > 0) && /* @__PURE__ */ jsxs27("div", {
+            className: "flex flex-wrap gap-2 justify-end",
             children: [
-              (message.images && message.images.length > 0 || message.audio && message.audio.length > 0 || message.files && message.files.length > 0) && /* @__PURE__ */ jsxs20("div", {
-                className: "flex flex-wrap gap-2 justify-end",
+              message.images?.map((img, idx) => /* @__PURE__ */ jsx45(FilePreviewCard, {
+                file: { name: img.revised_prompt || `Image ${idx + 1}`, type: "image/png", url: img.url },
+                onClick: showImageLightbox ? () => openImageLightbox(message.images.map((i) => ({ url: i.url, alt: i.revised_prompt })), idx) : undefined
+              }, `img-${idx}`)),
+              message.audio?.map((audio, idx) => /* @__PURE__ */ jsxs27("div", {
+                className: "flex items-center gap-1.5 rounded-lg border border-border bg-muted/50 px-2.5 py-1.5 text-xs text-foreground self-end",
                 children: [
-                  message.images?.map((img, idx) => /* @__PURE__ */ jsx35(FilePreviewCard, {
-                    file: { name: img.revised_prompt || `Image ${idx + 1}`, type: "image/png", url: img.url },
-                    onClick: showImageLightbox ? () => openImageLightbox(message.images.map((i) => ({ url: i.url, alt: i.revised_prompt })), idx) : undefined
-                  }, `img-${idx}`)),
-                  message.audio?.map((audio, idx) => /* @__PURE__ */ jsxs20("div", {
-                    className: "flex items-center gap-1.5 rounded-lg border border-border bg-muted/50 px-2.5 py-1.5 text-xs text-foreground self-end",
-                    children: [
-                      /* @__PURE__ */ jsx35(Music, {
-                        className: "h-3.5 w-3.5 text-muted-foreground"
-                      }),
-                      /* @__PURE__ */ jsx35("span", {
-                        className: "truncate max-w-[150px]",
-                        children: audio.id || `Audio ${idx + 1}`
-                      })
-                    ]
-                  }, `audio-${idx}`)),
-                  message.files?.map((file, idx) => showFilePreview ? /* @__PURE__ */ jsx35(FilePreviewCard, {
-                    file: { name: file.name, type: file.type, url: file.url, size: file.size },
-                    onClick: () => openFilePreview({ name: file.name, type: file.type, url: file.url, size: file.size })
-                  }, `file-${idx}`) : /* @__PURE__ */ jsxs20("div", {
-                    className: "flex items-center gap-1.5 rounded-lg border border-border bg-muted/50 px-2.5 py-1.5 text-xs text-foreground self-end",
-                    children: [
-                      /* @__PURE__ */ jsx35(FileIcon3, {
-                        className: "h-3.5 w-3.5 text-muted-foreground"
-                      }),
-                      /* @__PURE__ */ jsx35("span", {
-                        className: "truncate max-w-[150px]",
-                        children: file.name
-                      })
-                    ]
-                  }, `file-${idx}`))
-                ]
-              }),
-              message.content && /* @__PURE__ */ jsx35("div", {
-                className: cn("rounded-2xl rounded-br-md px-4 py-2.5", classNames?.user?.bubble ?? "bg-primary text-primary-foreground", hasError && "opacity-70"),
-                children: /* @__PURE__ */ jsx35("p", {
-                  className: "text-sm whitespace-pre-wrap",
-                  children: message.content
-                })
-              }),
-              (showTimestamp || actions?.user) && /* @__PURE__ */ jsxs20("div", {
-                className: "flex items-center justify-end gap-1.5 px-1",
-                children: [
-                  actions?.user && /* @__PURE__ */ jsx35("div", {
-                    className: "flex items-center gap-1",
-                    children: actions.user(message)
+                  /* @__PURE__ */ jsx45(Music2, {
+                    className: "h-3.5 w-3.5 text-muted-foreground"
                   }),
-                  /* @__PURE__ */ jsx35(SmartTimestamp, {
-                    date: new Date(message.created_at * 1000),
-                    formatShort: isCustomTimestamp ? resolvedFormatTimestamp : undefined,
-                    className: "text-[11px] text-muted-foreground"
-                  }),
-                  hasError && /* @__PURE__ */ jsx35(AlertCircle, {
-                    className: "h-3 w-3 text-destructive"
+                  /* @__PURE__ */ jsx45("span", {
+                    className: "truncate max-w-[150px]",
+                    children: audio.id || `Audio ${idx + 1}`
                   })
                 ]
-              })
-            ]
-          })
-        ]
-      }) : /* @__PURE__ */ jsxs20("div", {
-        className: "flex items-start gap-3 group/message",
-        children: [
-          avatars?.assistant,
-          /* @__PURE__ */ jsxs20("div", {
-            className: cn("flex-1 min-w-0 space-y-3", classNames?.assistant?.container),
-            children: [
-              renderContent ? renderContent(message) : /* @__PURE__ */ jsxs20(Fragment5, {
+              }, `audio-${idx}`)),
+              message.files?.map((file, idx) => showFilePreview ? /* @__PURE__ */ jsx45(FilePreviewCard, {
+                file: { name: file.name, type: file.type, url: file.url, size: file.size },
+                onClick: () => openFilePreview({ name: file.name, type: file.type, url: file.url, size: file.size })
+              }, `file-${idx}`) : /* @__PURE__ */ jsxs27("div", {
+                className: "flex items-center gap-1.5 rounded-lg border border-border bg-muted/50 px-2.5 py-1.5 text-xs text-foreground self-end",
                 children: [
-                  showReasoning && message.extra_data?.reasoning_steps && message.extra_data.reasoning_steps.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                    className: cn("space-y-2 pt-1", classNames?.assistant?.reasoning),
-                    children: [
-                      /* @__PURE__ */ jsxs20("div", {
-                        className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                        children: [
-                          /* @__PURE__ */ jsx35(Lightbulb, {
-                            className: "h-3.5 w-3.5"
-                          }),
-                          "Reasoning (",
-                          message.extra_data.reasoning_steps.length,
-                          " steps)"
-                        ]
-                      }),
-                      /* @__PURE__ */ jsx35(Accordion, {
-                        type: "multiple",
-                        className: "w-full",
-                        children: message.extra_data.reasoning_steps.map((step, idx) => /* @__PURE__ */ jsxs20(AccordionItem, {
-                          value: `reasoning-${idx}`,
-                          className: "border-muted",
-                          children: [
-                            /* @__PURE__ */ jsx35(AccordionTrigger, {
-                              className: "text-xs py-1.5 hover:no-underline",
-                              children: step.title || `Step ${idx + 1}`
-                            }),
-                            /* @__PURE__ */ jsxs20(AccordionContent, {
-                              className: "space-y-1.5 text-xs text-muted-foreground",
-                              children: [
-                                step.action && /* @__PURE__ */ jsxs20("div", {
-                                  children: [
-                                    /* @__PURE__ */ jsx35("span", {
-                                      className: "font-medium text-foreground",
-                                      children: "Action:"
-                                    }),
-                                    " ",
-                                    step.action
-                                  ]
-                                }),
-                                step.reasoning && /* @__PURE__ */ jsxs20("div", {
-                                  children: [
-                                    /* @__PURE__ */ jsx35("span", {
-                                      className: "font-medium text-foreground",
-                                      children: "Reasoning:"
-                                    }),
-                                    " ",
-                                    step.reasoning
-                                  ]
-                                }),
-                                step.result && /* @__PURE__ */ jsxs20("div", {
-                                  children: [
-                                    /* @__PURE__ */ jsx35("span", {
-                                      className: "font-medium text-foreground",
-                                      children: "Result:"
-                                    }),
-                                    " ",
-                                    step.result
-                                  ]
-                                }),
-                                step.confidence !== undefined && /* @__PURE__ */ jsxs20("div", {
-                                  children: [
-                                    /* @__PURE__ */ jsx35("span", {
-                                      className: "font-medium text-foreground",
-                                      children: "Confidence:"
-                                    }),
-                                    " ",
-                                    (step.confidence * 100).toFixed(1),
-                                    "%"
-                                  ]
-                                })
-                              ]
-                            })
-                          ]
-                        }, idx))
-                      })
-                    ]
-                  }),
-                  message.content && /* @__PURE__ */ jsx35("div", {
-                    className: "prose prose-sm dark:prose-invert max-w-none prose-p:leading-relaxed prose-pre:bg-muted prose-pre:border prose-pre:border-border",
-                    children: /* @__PURE__ */ jsx35(Response, {
-                      children: message.content
-                    })
-                  }),
-                  showGenerativeUI && toolsWithUI.length > 0 && /* @__PURE__ */ jsx35("div", {
-                    className: "space-y-3",
-                    children: toolsWithUI.map((tool) => {
-                      const uiComponent = tool.ui_component;
-                      return /* @__PURE__ */ jsx35("div", {
-                        children: uiComponent.layout === "artifact" ? /* @__PURE__ */ jsx35(Artifact, {
-                          children: /* @__PURE__ */ jsx35(GenerativeUIRenderer, {
-                            spec: uiComponent,
-                            className: "w-full p-2"
-                          })
-                        }) : /* @__PURE__ */ jsx35(GenerativeUIRenderer, {
-                          spec: uiComponent,
-                          className: "w-full"
-                        })
-                      }, tool.tool_call_id);
-                    })
-                  }),
-                  toolResultRenderers && message.tool_calls?.map((tool) => {
-                    const renderer = toolResultRenderers[tool.tool_name];
-                    const content = tool.result ?? tool.content;
-                    if (!renderer || !content)
-                      return null;
-                    return /* @__PURE__ */ jsx35("div", {
-                      children: renderer(tool.tool_args, content)
-                    }, `result-${tool.tool_call_id}`);
+                  /* @__PURE__ */ jsx45(FileIcon4, {
+                    className: "h-3.5 w-3.5 text-muted-foreground"
                   }),
-                  renderMedia ? renderMedia(message) : (() => {
-                    const mediaClassName = classNames?.assistant?.media;
-                    return /* @__PURE__ */ jsxs20(Fragment5, {
-                      children: [
-                        message.images && message.images.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                          className: cn("space-y-2 pt-1", mediaClassName),
-                          children: [
-                            /* @__PURE__ */ jsxs20("div", {
-                              className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                              children: [
-                                /* @__PURE__ */ jsx35(ImageIcon2, {
-                                  className: "h-3.5 w-3.5"
-                                }),
-                                "Images (",
-                                message.images.length,
-                                ")"
-                              ]
-                            }),
-                            /* @__PURE__ */ jsx35("div", {
-                              className: "grid grid-cols-2 gap-2",
-                              children: message.images.map((img, idx) => /* @__PURE__ */ jsxs20("div", {
-                                className: "space-y-1",
-                                children: [
-                                  showImageLightbox ? /* @__PURE__ */ jsx35("button", {
-                                    type: "button",
-                                    onClick: () => openImageLightbox(message.images.map((i) => ({ url: i.url, alt: i.revised_prompt })), idx),
-                                    className: "group relative w-full overflow-hidden rounded-lg border border-border cursor-pointer hover:border-primary/50 focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring transition-colors",
-                                    children: /* @__PURE__ */ jsx35("img", {
-                                      src: img.url,
-                                      alt: img.revised_prompt || "Generated image",
-                                      className: "w-full rounded-lg"
-                                    })
-                                  }) : /* @__PURE__ */ jsx35("img", {
-                                    src: img.url,
-                                    alt: img.revised_prompt || "Generated image",
-                                    className: "w-full rounded-lg border border-border"
-                                  }),
-                                  img.revised_prompt && /* @__PURE__ */ jsx35("p", {
-                                    className: "text-[11px] text-muted-foreground italic px-0.5",
-                                    children: img.revised_prompt
-                                  })
-                                ]
-                              }, idx))
-                            })
-                          ]
-                        }),
-                        message.videos && message.videos.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                          className: cn("space-y-2 pt-1", mediaClassName),
-                          children: [
-                            /* @__PURE__ */ jsxs20("div", {
-                              className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                              children: [
-                                /* @__PURE__ */ jsx35(Video, {
-                                  className: "h-3.5 w-3.5"
-                                }),
-                                "Videos (",
-                                message.videos.length,
-                                ")"
-                              ]
-                            }),
-                            /* @__PURE__ */ jsx35("div", {
-                              className: "space-y-2",
-                              children: message.videos.map((video, idx) => /* @__PURE__ */ jsx35("div", {
-                                children: video.url ? /* @__PURE__ */ jsx35("video", {
-                                  src: video.url,
-                                  controls: true,
-                                  className: "w-full rounded-lg border border-border"
-                                }) : /* @__PURE__ */ jsxs20("div", {
-                                  className: "bg-muted/50 border border-border p-2.5 rounded-lg text-xs text-muted-foreground",
-                                  children: [
-                                    "Video ID: ",
-                                    video.id,
-                                    " (ETA: ",
-                                    video.eta,
-                                    "s)"
-                                  ]
-                                })
-                              }, idx))
-                            })
-                          ]
-                        }),
-                        message.audio && message.audio.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                          className: cn("space-y-2 pt-1", mediaClassName),
-                          children: [
-                            /* @__PURE__ */ jsxs20("div", {
-                              className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                              children: [
-                                /* @__PURE__ */ jsx35(Music, {
-                                  className: "h-3.5 w-3.5"
-                                }),
-                                "Audio (",
-                                message.audio.length,
-                                ")"
-                              ]
-                            }),
-                            /* @__PURE__ */ jsx35("div", {
-                              className: "space-y-2",
-                              children: message.audio.map((audio, idx) => /* @__PURE__ */ jsx35("div", {
-                                children: audio.url ? /* @__PURE__ */ jsx35("audio", {
-                                  src: audio.url,
-                                  controls: true,
-                                  className: "w-full"
-                                }) : audio.base64_audio ? /* @__PURE__ */ jsx35("audio", {
-                                  src: `data:${audio.mime_type || "audio/wav"};base64,${audio.base64_audio}`,
-                                  controls: true,
-                                  className: "w-full"
-                                }) : /* @__PURE__ */ jsx35("div", {
-                                  className: "bg-muted/50 border border-border p-2.5 rounded-lg text-xs text-muted-foreground",
-                                  children: "Audio data unavailable"
-                                })
-                              }, idx))
-                            })
-                          ]
-                        }),
-                        message.files && message.files.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                          className: cn("space-y-2 pt-1", mediaClassName),
-                          children: [
-                            /* @__PURE__ */ jsxs20("div", {
-                              className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                              children: [
-                                /* @__PURE__ */ jsx35(Paperclip, {
-                                  className: "h-3.5 w-3.5"
-                                }),
-                                "Files (",
-                                message.files.length,
-                                ")"
-                              ]
-                            }),
-                            /* @__PURE__ */ jsx35("div", {
-                              className: "flex flex-wrap gap-2",
-                              children: message.files.map((file, idx) => showFilePreview ? /* @__PURE__ */ jsx35(FilePreviewCard, {
-                                file: { name: file.name, type: file.type, url: file.url, size: file.size },
-                                onClick: () => openFilePreview({ name: file.name, type: file.type, url: file.url, size: file.size })
-                              }, idx) : /* @__PURE__ */ jsxs20("div", {
-                                className: "flex items-center gap-2 rounded-lg border border-border px-3 py-2 text-xs bg-muted/30 hover:bg-muted/50 transition-colors",
-                                children: [
-                                  /* @__PURE__ */ jsx35(FileIcon3, {
-                                    className: "h-3.5 w-3.5 shrink-0 text-muted-foreground"
-                                  }),
-                                  /* @__PURE__ */ jsx35("span", {
-                                    className: "truncate max-w-[180px]",
-                                    children: file.name
-                                  }),
-                                  file.size && /* @__PURE__ */ jsxs20("span", {
-                                    className: "text-muted-foreground/70",
-                                    children: [
-                                      "(",
-                                      (file.size / 1024).toFixed(1),
-                                      "KB)"
-                                    ]
-                                  }),
-                                  file.url && /^https?:\/\//i.test(file.url) && /* @__PURE__ */ jsx35("a", {
-                                    href: file.url,
-                                    target: "_blank",
-                                    rel: "noopener noreferrer",
-                                    className: "text-primary hover:underline font-medium",
-                                    children: "View"
-                                  })
-                                ]
-                              }, idx))
-                            })
-                          ]
-                        }),
-                        message.response_audio && /* @__PURE__ */ jsxs20("div", {
-                          className: cn("space-y-2 pt-1", mediaClassName),
-                          children: [
-                            /* @__PURE__ */ jsxs20("div", {
-                              className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                              children: [
-                                /* @__PURE__ */ jsx35(Music, {
-                                  className: "h-3.5 w-3.5"
-                                }),
-                                "Response Audio"
-                              ]
-                            }),
-                            message.response_audio.transcript && /* @__PURE__ */ jsxs20("div", {
-                              className: "text-xs italic bg-muted/50 border border-border p-2.5 rounded-lg text-muted-foreground",
-                              children: [
-                                '"',
-                                message.response_audio.transcript,
-                                '"'
-                              ]
-                            }),
-                            message.response_audio.content && /* @__PURE__ */ jsx35("audio", {
-                              src: `data:audio/wav;base64,${message.response_audio.content}`,
-                              controls: true,
-                              className: "w-full"
-                            })
-                          ]
-                        })
-                      ]
-                    });
-                  })(),
-                  showToolCalls && message.tool_calls && message.tool_calls.length > 0 && /* @__PURE__ */ jsx35("div", {
-                    className: cn("space-y-2 pt-1", classNames?.assistant?.toolCalls),
-                    children: message.tool_calls.map((tool, idx) => renderToolCall ? renderToolCall(tool, idx) : /* @__PURE__ */ jsxs20(Tool, {
-                      defaultOpen: idx === 0,
-                      children: [
-                        /* @__PURE__ */ jsx35(ToolHeader, {
-                          title: tool.tool_name,
-                          type: "tool-use",
-                          state: getToolState(tool)
-                        }),
-                        /* @__PURE__ */ jsxs20(ToolContent, {
-                          children: [
-                            /* @__PURE__ */ jsx35(ToolInput, {
-                              input: tool.tool_args
-                            }),
-                            tool.content && /* @__PURE__ */ jsx35(ToolOutput, {
-                              output: tool.content,
-                              errorText: tool.tool_call_error ? "Tool execution failed" : undefined
-                            })
-                          ]
-                        })
-                      ]
-                    }, tool.tool_call_id || idx))
-                  }),
-                  showReferences && message.extra_data?.references && message.extra_data.references.length > 0 && /* @__PURE__ */ jsxs20("div", {
-                    className: cn("space-y-2 pt-1", classNames?.assistant?.references),
-                    children: [
-                      /* @__PURE__ */ jsxs20("div", {
-                        className: "flex items-center gap-2 text-xs font-medium text-muted-foreground",
-                        children: [
-                          /* @__PURE__ */ jsx35(FileText, {
-                            className: "h-3.5 w-3.5"
-                          }),
-                          "References (",
-                          message.extra_data.references.length,
-                          ")"
-                        ]
-                      }),
-                      /* @__PURE__ */ jsx35("div", {
-                        className: "space-y-2",
-                        children: message.extra_data.references.map((refData, idx) => /* @__PURE__ */ jsxs20("div", {
-                          className: "text-xs space-y-1.5",
-                          children: [
-                            refData.query && /* @__PURE__ */ jsxs20("div", {
-                              className: "font-medium text-foreground",
-                              children: [
-                                "Query: ",
-                                refData.query
-                              ]
-                            }),
-                            refData.references.map((ref, refIdx) => /* @__PURE__ */ jsxs20("div", {
-                              className: "bg-muted/50 border border-border p-2.5 rounded-lg",
-                              children: [
-                                /* @__PURE__ */ jsxs20("div", {
-                                  className: "italic text-muted-foreground mb-1",
-                                  children: [
-                                    '"',
-                                    ref.content,
-                                    '"'
-                                  ]
-                                }),
-                                /* @__PURE__ */ jsxs20("div", {
-                                  className: "text-muted-foreground/70",
-                                  children: [
-                                    "Source: ",
-                                    ref.name,
-                                    " (chunk ",
-                                    ref.meta_data.chunk,
-                                    "/",
-                                    ref.meta_data.chunk_size,
-                                    ")"
-                                  ]
-                                })
-                              ]
-                            }, refIdx))
-                          ]
-                        }, idx))
-                      })
-                    ]
+                  /* @__PURE__ */ jsx45("span", {
+                    className: "truncate max-w-[150px]",
+                    children: file.name
                   })
                 ]
+              }, `file-${idx}`))
+            ]
+          }),
+          message.content && /* @__PURE__ */ jsx45("div", {
+            className: cn("rounded-2xl rounded-br-md px-4 py-2.5", classNames?.user?.bubble ?? "bg-primary text-primary-foreground", hasError && "opacity-70"),
+            children: /* @__PURE__ */ jsx45("p", {
+              className: "text-sm whitespace-pre-wrap",
+              children: message.content
+            })
+          }),
+          (showTimestamp || actions?.user) && /* @__PURE__ */ jsxs27("div", {
+            className: "flex items-center justify-end gap-1.5 px-1",
+            children: [
+              actions?.user && /* @__PURE__ */ jsx45("div", {
+                className: "flex items-center gap-1",
+                children: actions.user(message)
               }),
-              (actions?.assistant || showTimestamp || hasError) && /* @__PURE__ */ jsxs20("div", {
-                className: "flex items-center gap-2 pt-1",
-                children: [
-                  actions?.assistant && (() => {
-                    const visibility = actions.visibility ?? "visible";
-                    if (visibility === "last-assistant" && !isLastAssistantMessage)
-                      return null;
-                    const useHover = visibility === "hover" || visibility === "hover-last-visible" && !isLastAssistantMessage;
-                    return /* @__PURE__ */ jsx35("div", {
-                      className: cn("flex items-center gap-1 transition-opacity", useHover && "opacity-0 group-hover/message:opacity-100", classNames?.assistant?.actions),
-                      children: actions.assistant(message)
-                    });
-                  })(),
-                  hasError && /* @__PURE__ */ jsxs20("span", {
-                    className: "flex items-center gap-1 text-[11px] text-destructive",
-                    children: [
-                      /* @__PURE__ */ jsx35(AlertCircle, {
-                        className: "h-3 w-3"
-                      }),
-                      "Error"
-                    ]
-                  }),
-                  showTimestamp && /* @__PURE__ */ jsx35(SmartTimestamp, {
-                    date: new Date(message.created_at * 1000),
-                    formatShort: isCustomTimestamp ? resolvedFormatTimestamp : undefined,
-                    className: "text-[11px] text-muted-foreground"
-                  })
-                ]
+              /* @__PURE__ */ jsx45(SmartTimestamp, {
+                date: new Date(message.created_at * 1000),
+                formatShort: isCustomTimestamp ? resolvedFormatTimestamp : undefined,
+                className: "text-[11px] text-muted-foreground"
+              }),
+              hasError && /* @__PURE__ */ jsx45(AlertCircle2, {
+                className: "h-3 w-3 text-destructive"
               })
             ]
           })
         ]
-      }),
-      preview?.type === "image" && /* @__PURE__ */ jsx35(ImageLightbox, {
-        open: true,
-        onOpenChange: (open) => {
-          if (!open)
-            closePreview();
-        },
-        images: preview.images,
-        initialIndex: preview.initialIndex
-      }),
-      preview?.type === "file" && /* @__PURE__ */ jsx35(FilePreviewModal, {
-        open: true,
-        onOpenChange: (open) => {
-          if (!open)
-            closePreview();
-        },
-        file: preview.file
+      })
+    ]
+  });
+}
+
+// src/ui/composed/AgnoMessageItem.tsx
+import { jsx as jsx46, jsxs as jsxs28 } from "react/jsx-runtime";
+function AgnoMessageItem({
+  message,
+  className,
+  classNames,
+  avatars,
+  actions,
+  isLastAssistantMessage = false,
+  showReasoning = true,
+  showReferences = true,
+  showTimestamp = true,
+  showFilePreview = true,
+  showImageLightbox = true,
+  formatTimestamp,
+  renderTool
+}) {
+  return /* @__PURE__ */ jsxs28(AgnoMessage, {
+    message,
+    className,
+    classNames,
+    avatars,
+    actions,
+    isLastAssistantMessage,
+    showTimestamp,
+    showFilePreview,
+    showImageLightbox,
+    formatTimestamp,
+    renderTool,
+    children: [
+      showReasoning && /* @__PURE__ */ jsx46(AgnoMessage.Reasoning, {}),
+      /* @__PURE__ */ jsx46(AgnoMessage.Media, {}),
+      /* @__PURE__ */ jsx46(AgnoMessage.Tools, {}),
+      /* @__PURE__ */ jsx46(AgnoMessage.Content, {}),
+      showReferences && /* @__PURE__ */ jsx46(AgnoMessage.References, {}),
+      /* @__PURE__ */ jsx46(AgnoMessage.Footer, {
+        showTimestamp
       })
     ]
   });
 }
 // src/ui/composed/AgnoChatInput.tsx
 import { CircleStop } from "lucide-react";
-import { jsx as jsx36, jsxs as jsxs21 } from "react/jsx-runtime";
+import { jsx as jsx47, jsxs as jsxs29 } from "react/jsx-runtime";
 var DEFAULT_ACCEPTED_FILE_TYPES = "image/*,audio/*,.pdf,.doc,.docx,.txt,.csv,.xlsx,.xls,.ppt,.pptx,.md,.json,.xml";
 function normalizeAudio(audio) {
   if (audio === true)
@@ -3279,14 +3771,14 @@ function dataUrlToBlob(dataUrl) {
   return new Blob([buf], { type: mime });
 }
 function CancelButton({ onCancel }) {
-  return /* @__PURE__ */ jsx36(Button, {
+  return /* @__PURE__ */ jsx47(Button, {
     type: "button",
     variant: "destructive",
     size: "icon",
     className: "h-8 w-8 rounded-lg",
     onClick: onCancel,
     "aria-label": "Stop",
-    children: /* @__PURE__ */ jsx36(CircleStop, {
+    children: /* @__PURE__ */ jsx47(CircleStop, {
       className: "size-4"
     })
   });
@@ -3294,7 +3786,7 @@ function CancelButton({ onCancel }) {
 function SubmitButton({ disabled, status }) {
   const { textInput } = usePromptInputController();
   const hasText = textInput.value.trim().length > 0;
-  return /* @__PURE__ */ jsx36(PromptInputSubmit, {
+  return /* @__PURE__ */ jsx47(PromptInputSubmit, {
     disabled: disabled || !hasText,
     status
   });
@@ -3303,9 +3795,9 @@ function AttachmentHeader() {
   const { files } = usePromptInputAttachments();
   if (files.length === 0)
     return null;
-  return /* @__PURE__ */ jsx36(PromptInputHeader, {
-    children: /* @__PURE__ */ jsx36(PromptInputAttachments, {
-      children: (attachment) => /* @__PURE__ */ jsx36(PromptInputAttachment, {
+  return /* @__PURE__ */ jsx47(PromptInputHeader, {
+    children: /* @__PURE__ */ jsx47(PromptInputAttachments, {
+      children: (attachment) => /* @__PURE__ */ jsx47(PromptInputAttachment, {
         data: attachment
       })
     })
@@ -3320,7 +3812,7 @@ function TranscribeAudioRecorder({
   labels
 }) {
   const { textInput } = usePromptInputController();
-  return /* @__PURE__ */ jsx36(AudioRecorder, {
+  return /* @__PURE__ */ jsx47(AudioRecorder, {
     mode: "transcribe",
     transcriptionEndpoint: endpoint,
     transcriptionHeaders: headers,
@@ -3384,8 +3876,8 @@ function AgnoChatInput({
   };
   const computedStatus = status ?? (disabled ? "submitted" : undefined);
   const showCancelButton = allowCancelRun && isStreaming && onCancel;
-  return /* @__PURE__ */ jsx36(PromptInputProvider, {
-    children: /* @__PURE__ */ jsxs21(PromptInput, {
+  return /* @__PURE__ */ jsx47(PromptInputProvider, {
+    children: /* @__PURE__ */ jsxs29(PromptInput, {
       onSubmit: handleSubmit,
       accept: fileUpload?.accept ?? DEFAULT_ACCEPTED_FILE_TYPES,
       multiple: fileUpload?.multiple ?? true,
@@ -3395,35 +3887,35 @@ function AgnoChatInput({
       dragListenerTarget: dropZoneContainerRef,
       className: cn("w-full", className),
       children: [
-        /* @__PURE__ */ jsx36(AttachmentHeader, {}),
-        /* @__PURE__ */ jsx36(PromptInputBody, {
-          children: /* @__PURE__ */ jsx36(PromptInputTextarea, {
+        /* @__PURE__ */ jsx47(AttachmentHeader, {}),
+        /* @__PURE__ */ jsx47(PromptInputBody, {
+          children: /* @__PURE__ */ jsx47(PromptInputTextarea, {
             placeholder: placeholder || "Type your message... (Enter to send, Shift+Enter for new line)",
             disabled
           })
         }),
-        /* @__PURE__ */ jsxs21(PromptInputFooter, {
+        /* @__PURE__ */ jsxs29(PromptInputFooter, {
           children: [
-            /* @__PURE__ */ jsxs21(PromptInputTools, {
+            /* @__PURE__ */ jsxs29(PromptInputTools, {
               children: [
-                showAttachments && /* @__PURE__ */ jsxs21(PromptInputActionMenu, {
+                showAttachments && /* @__PURE__ */ jsxs29(PromptInputActionMenu, {
                   children: [
-                    /* @__PURE__ */ jsx36(PromptInputActionMenuTrigger, {}),
-                    /* @__PURE__ */ jsx36(PromptInputActionMenuContent, {
-                      children: /* @__PURE__ */ jsx36(PromptInputActionAddAttachments, {
+                    /* @__PURE__ */ jsx47(PromptInputActionMenuTrigger, {}),
+                    /* @__PURE__ */ jsx47(PromptInputActionMenuContent, {
+                      children: /* @__PURE__ */ jsx47(PromptInputActionAddAttachments, {
                         label: "Add files"
                       })
                     })
                   ]
                 }),
-                audioEnabled && (audioMode === "transcribe" && resolvedAudio?.endpoint ? /* @__PURE__ */ jsx36(TranscribeAudioRecorder, {
+                audioEnabled && (audioMode === "transcribe" && resolvedAudio?.endpoint ? /* @__PURE__ */ jsx47(TranscribeAudioRecorder, {
                   endpoint: resolvedAudio.endpoint,
                   headers: resolvedAudio.headers,
                   disabled,
                   parseResponse: resolvedAudio.parseResponse,
                   onRequestPermission: resolvedAudio.requestPermission,
                   labels: resolvedAudio.labels
-                }) : /* @__PURE__ */ jsx36(AudioRecorder, {
+                }) : /* @__PURE__ */ jsx47(AudioRecorder, {
                   onRecordingComplete: handleAudioRecording,
                   disabled,
                   onRequestPermission: resolvedAudio?.requestPermission,
@@ -3432,15 +3924,15 @@ function AgnoChatInput({
                 extraTools
               ]
             }),
-            showCancelButton ? /* @__PURE__ */ jsx36(CancelButton, {
+            showCancelButton ? /* @__PURE__ */ jsx47(CancelButton, {
               onCancel
-            }) : /* @__PURE__ */ jsx36(SubmitButton, {
+            }) : /* @__PURE__ */ jsx47(SubmitButton, {
               disabled,
               status: computedStatus
             })
           ]
         }),
-        showAttachments && /* @__PURE__ */ jsx36(PromptInputDropZone, {
+        showAttachments && /* @__PURE__ */ jsx47(PromptInputDropZone, {
           ...dropZoneProps,
           container: dropZoneContainerRef
         })
@@ -3448,39 +3940,46 @@ function AgnoChatInput({
     })
   });
 }
+// src/ui/composed/agno-chat/render-tool.ts
+function byToolName(map, fallback) {
+  return (tool, args) => {
+    if (!Object.prototype.hasOwnProperty.call(map, tool.tool_name)) {
+      return fallback ? fallback(tool, args) : args.defaultRender();
+    }
+    const entry = map[tool.tool_name];
+    if (entry === false)
+      return null;
+    return entry(tool, args);
+  };
+}
 // src/ui/composed/agno-chat/agno-chat.tsx
-import { useCallback as useCallback7, useMemo as useMemo3, useRef as useRef6 } from "react";
+import { useCallback as useCallback9, useMemo as useMemo6, useRef as useRef6 } from "react";
 import { useAgnoChat, useAgnoToolExecution } from "@rodrigocoliveira/agno-react";
-
-// src/ui/composed/agno-chat/context.ts
-import { createContext as createContext3, useContext as useContext3 } from "react";
-var AgnoChatContext = createContext3(null);
-function useAgnoChatContext() {
-  const ctx = useContext3(AgnoChatContext);
-  if (!ctx) {
-    throw new Error("useAgnoChatContext must be used within an <AgnoChat> provider. " + "Wrap your component tree with <AgnoChat>.");
-  }
-  return ctx;
+import { jsx as jsx48 } from "react/jsx-runtime";
+function resolveDebug(debug) {
+  if (typeof debug === "boolean")
+    return debug;
+  return typeof process !== "undefined" && process.env?.NODE_ENV === "development";
 }
-
-// src/ui/composed/agno-chat/agno-chat.tsx
-import { jsx as jsx37 } from "react/jsx-runtime";
 function AgnoChatRoot({
   children,
   toolHandlers = {},
   autoExecuteTools = true,
-  toolResultRenderers,
+  renderTool,
+  debug,
+  skipHydration,
   className,
   ...divProps
 }) {
   const chat = useAgnoChat();
   const toolExec = useAgnoToolExecution(toolHandlers, autoExecuteTools, {
-    skipHydration: toolResultRenderers ? Object.keys(toolResultRenderers) : undefined
+    skipHydration
   });
   const containerRef = useRef6(null);
+  const isDebug = resolveDebug(debug);
   const sendRef = useRef6(chat.sendMessage);
   sendRef.current = chat.sendMessage;
-  const handleSend = useCallback7(async (message) => {
+  const handleSend = useCallback9(async (message) => {
     try {
       await sendRef.current(message);
     } catch {}
@@ -3506,7 +4005,7 @@ function AgnoChatRoot({
     continueWithResults,
     executionError
   } = toolExec;
-  const contextValue = useMemo3(() => ({
+  const contextValue = useMemo6(() => ({
     messages,
     sendMessage,
     clearMessages,
@@ -3527,7 +4026,8 @@ function AgnoChatRoot({
     handleSend,
     inputDisabled: isStreaming || isPaused,
     dropZoneContainerRef: containerRef,
-    toolResultRenderers
+    renderTool,
+    isDebug
   }), [
     messages,
     sendMessage,
@@ -3547,11 +4047,12 @@ function AgnoChatRoot({
     continueWithResults,
     executionError,
     handleSend,
-    toolResultRenderers
+    renderTool,
+    isDebug
   ]);
-  return /* @__PURE__ */ jsx37(AgnoChatContext.Provider, {
+  return /* @__PURE__ */ jsx48(AgnoChatContext.Provider, {
     value: contextValue,
-    children: /* @__PURE__ */ jsx37("div", {
+    children: /* @__PURE__ */ jsx48("div", {
       ref: containerRef,
       className: cn("relative h-full flex flex-col", className),
       ...divProps,
@@ -3561,25 +4062,25 @@ function AgnoChatRoot({
 }
 
 // src/ui/composed/agno-chat/messages.tsx
-import { useEffect as useEffect6, useRef as useRef7 } from "react";
+import { isValidElement as isValidElement2, useEffect as useEffect8, useRef as useRef7 } from "react";
 
 // src/ui/composed/agno-chat/suggested-prompts.tsx
-import { jsx as jsx38, jsxs as jsxs22 } from "react/jsx-runtime";
+import { jsx as jsx49, jsxs as jsxs30 } from "react/jsx-runtime";
 function AgnoChatSuggestedPrompts({ className, prompts }) {
   const { handleSend } = useAgnoChatContext();
   if (prompts.length === 0)
     return null;
-  return /* @__PURE__ */ jsx38("div", {
+  return /* @__PURE__ */ jsx49("div", {
     className: cn("grid grid-cols-2 gap-2 w-full max-w-md", className),
-    children: prompts.map((prompt, idx) => /* @__PURE__ */ jsxs22("button", {
+    children: prompts.map((prompt, idx) => /* @__PURE__ */ jsxs30("button", {
       onClick: () => handleSend(prompt.text),
       className: "flex items-center gap-2 px-3 py-2.5 rounded-xl border border-border bg-card hover:bg-accent/50 hover:border-primary/20 transition-all duration-200 text-left text-sm group",
       children: [
-        prompt.icon && /* @__PURE__ */ jsx38("span", {
+        prompt.icon && /* @__PURE__ */ jsx49("span", {
           className: "text-muted-foreground group-hover:text-primary transition-colors",
           children: prompt.icon
         }),
-        /* @__PURE__ */ jsx38("span", {
+        /* @__PURE__ */ jsx49("span", {
           className: "text-muted-foreground group-hover:text-foreground transition-colors text-xs leading-snug",
           children: prompt.text
         })
@@ -3590,11 +4091,11 @@ function AgnoChatSuggestedPrompts({ className, prompts }) {
 
 // src/ui/composed/agno-chat/messages.tsx
 import { Bot as Bot2 } from "lucide-react";
-import { jsx as jsx39, jsxs as jsxs23 } from "react/jsx-runtime";
+import { jsx as jsx50, jsxs as jsxs31 } from "react/jsx-runtime";
 function ScrollOnNewUserMessage({ messageCount }) {
   const { scrollToBottom } = useStickToBottomContext2();
   const prevCount = useRef7(messageCount);
-  useEffect6(() => {
+  useEffect8(() => {
     if (messageCount > prevCount.current) {
       scrollToBottom("smooth");
     }
@@ -3614,13 +4115,8 @@ function AgnoChatMessages({
   showReasoning,
   showReferences,
   showTimestamp,
-  showGenerativeUI,
-  showToolCalls,
   showFilePreview,
   showImageLightbox,
-  renderToolCall,
-  renderContent,
-  renderMedia,
   formatTimestamp,
   messageClassNames,
   emptyState,
@@ -3628,12 +4124,28 @@ function AgnoChatMessages({
   children,
   showThinkingIndicator = true,
   renderThinkingIndicator,
-  toolResultRenderers: propToolResultRenderers
+  renderTool: propRenderTool,
+  scrollBehavior,
+  scrollToBottomButton
 }) {
-  const { messages, isStreaming, toolResultRenderers: contextToolResultRenderers } = useAgnoChatContext();
-  const toolResultRenderers = propToolResultRenderers ?? contextToolResultRenderers;
+  const { messages, isStreaming } = useAgnoChatContext();
   const lastMessage = messages[messages.length - 1];
   const isThinking = showThinkingIndicator && isStreaming && (!lastMessage || lastMessage.role !== "user") && !lastMessage?.content;
+  const resolvedScrollButton = (() => {
+    if (scrollToBottomButton === false)
+      return null;
+    if (scrollToBottomButton === undefined || scrollToBottomButton === true) {
+      return /* @__PURE__ */ jsx50(ConversationScrollButton, {});
+    }
+    if (isValidElement2(scrollToBottomButton))
+      return scrollToBottomButton;
+    if (typeof scrollToBottomButton === "object" && scrollToBottomButton !== null && "className" in scrollToBottomButton) {
+      return /* @__PURE__ */ jsx50(ConversationScrollButton, {
+        className: scrollToBottomButton.className
+      });
+    }
+    return scrollToBottomButton;
+  })();
   let lastAssistantIndex = -1;
   for (let i = messages.length - 1;i >= 0; i--) {
     if (messages[i].role !== "user") {
@@ -3645,70 +4157,66 @@ function AgnoChatMessages({
     ...showReasoning !== undefined && { showReasoning },
     ...showReferences !== undefined && { showReferences },
     ...showTimestamp !== undefined && { showTimestamp },
-    ...showGenerativeUI !== undefined && { showGenerativeUI },
-    ...showToolCalls !== undefined && { showToolCalls },
     ...showFilePreview !== undefined && { showFilePreview },
     ...showImageLightbox !== undefined && { showImageLightbox },
-    ...renderToolCall !== undefined && { renderToolCall },
-    ...renderContent !== undefined && { renderContent },
-    ...renderMedia !== undefined && { renderMedia },
     ...formatTimestamp !== undefined && { formatTimestamp },
     ...messageClassNames !== undefined && { classNames: messageClassNames },
-    ...toolResultRenderers !== undefined && { toolResultRenderers }
+    ...propRenderTool !== undefined && { renderTool: propRenderTool }
   };
-  const resolvedEmptyState = children ?? emptyState ?? /* @__PURE__ */ jsxs23("div", {
+  const resolvedEmptyState = children ?? emptyState ?? /* @__PURE__ */ jsxs31("div", {
     className: "flex flex-col items-center gap-6",
     children: [
-      /* @__PURE__ */ jsxs23("div", {
+      /* @__PURE__ */ jsxs31("div", {
         className: "relative",
         children: [
-          /* @__PURE__ */ jsx39("div", {
+          /* @__PURE__ */ jsx50("div", {
             className: "h-16 w-16 rounded-2xl bg-primary/10 flex items-center justify-center",
-            children: /* @__PURE__ */ jsx39(Bot2, {
+            children: /* @__PURE__ */ jsx50(Bot2, {
               className: "h-8 w-8 text-primary"
             })
           }),
-          /* @__PURE__ */ jsx39("div", {
+          /* @__PURE__ */ jsx50("div", {
             className: "absolute -bottom-1 -right-1 h-5 w-5 rounded-full bg-green-500 border-2 border-background flex items-center justify-center",
-            children: /* @__PURE__ */ jsx39("span", {
+            children: /* @__PURE__ */ jsx50("span", {
               className: "h-2 w-2 rounded-full bg-white animate-pulse"
             })
           })
         ]
       }),
-      /* @__PURE__ */ jsxs23("div", {
+      /* @__PURE__ */ jsxs31("div", {
         className: "space-y-2 text-center",
         children: [
-          /* @__PURE__ */ jsx39("h3", {
+          /* @__PURE__ */ jsx50("h3", {
             className: "text-xl font-semibold tracking-tight",
             children: "Welcome to Agno Chat"
           }),
-          /* @__PURE__ */ jsx39("p", {
+          /* @__PURE__ */ jsx50("p", {
             className: "text-muted-foreground text-sm max-w-sm",
             children: "Start a conversation with your AI agent. Ask questions, explore ideas, or run tools."
           })
         ]
       }),
-      suggestedPrompts.length > 0 && /* @__PURE__ */ jsx39(AgnoChatSuggestedPrompts, {
+      suggestedPrompts.length > 0 && /* @__PURE__ */ jsx50(AgnoChatSuggestedPrompts, {
         prompts: suggestedPrompts
       })
     ]
   });
-  return /* @__PURE__ */ jsxs23(Conversation, {
+  return /* @__PURE__ */ jsxs31(Conversation, {
     className: cn("relative flex-1 w-full", className),
+    scrollBehavior,
     children: [
-      /* @__PURE__ */ jsx39(ScrollOnNewUserMessage, {
+      /* @__PURE__ */ jsx50(ScrollOnNewUserMessage, {
         messageCount: messages.length
       }),
-      /* @__PURE__ */ jsxs23(ConversationContent, {
+      /* @__PURE__ */ jsxs31(ConversationContent, {
         className: "max-w-3xl mx-auto",
         children: [
-          messages.length === 0 ? /* @__PURE__ */ jsx39(ConversationEmptyState, {
+          messages.length === 0 ? /* @__PURE__ */ jsx50(ConversationEmptyState, {
             children: resolvedEmptyState
           }) : messages.map((message, index) => {
             if (isThinking && index === messages.length - 1 && message === lastMessage)
               return null;
-            return renderMessage ? renderMessage(message, index) : /* @__PURE__ */ jsx39(AgnoMessageItem, {
+            return renderMessage ? renderMessage(message, index) : /* @__PURE__ */ jsx50(AgnoMessageItem, {
               message,
               avatars,
               actions,
@@ -3716,23 +4224,23 @@ function AgnoChatMessages({
               ...messageItemProps
             }, `msg-${index}-${message.created_at}`);
           }),
-          isThinking && /* @__PURE__ */ jsx39("div", {
+          isThinking && /* @__PURE__ */ jsx50("div", {
             className: "py-2",
-            children: renderThinkingIndicator ?? /* @__PURE__ */ jsx39(StreamingIndicator, {
+            children: renderThinkingIndicator ?? /* @__PURE__ */ jsx50(StreamingIndicator, {
               avatar: avatars?.assistant
             })
           })
         ]
       }),
-      /* @__PURE__ */ jsx39(ConversationScrollButton, {})
+      resolvedScrollButton
     ]
   });
 }
 
 // src/ui/composed/agno-chat/empty-state.tsx
-import { jsx as jsx40 } from "react/jsx-runtime";
+import { jsx as jsx51 } from "react/jsx-runtime";
 function AgnoChatEmptyState({ children, className, ...props }) {
-  return /* @__PURE__ */ jsx40("div", {
+  return /* @__PURE__ */ jsx51("div", {
     className: cn("flex flex-col items-center gap-6", className),
     ...props,
     children
@@ -3740,8 +4248,8 @@ function AgnoChatEmptyState({ children, className, ...props }) {
 }
 
 // src/ui/composed/agno-chat/error-bar.tsx
-import { useState as useState9, useEffect as useEffect7, useRef as useRef8 } from "react";
-import { jsx as jsx41, jsxs as jsxs24 } from "react/jsx-runtime";
+import { useState as useState11, useEffect as useEffect9, useRef as useRef8 } from "react";
+import { jsx as jsx52, jsxs as jsxs32 } from "react/jsx-runtime";
 function AgnoChatErrorBar({
   className,
   text,
@@ -3751,10 +4259,10 @@ function AgnoChatErrorBar({
   children
 }) {
   const { error, executionError } = useAgnoChatContext();
-  const [hidden, setHidden] = useState9(false);
+  const [hidden, setHidden] = useState11(false);
   const timerRef = useRef8(null);
   const rawMessage = error || executionError;
-  useEffect7(() => {
+  useEffect9(() => {
     if (!rawMessage)
       return;
     setHidden(false);
@@ -3781,23 +4289,23 @@ function AgnoChatErrorBar({
     if (children) {
       return typeof children === "function" ? children(rawMessage) : children;
     }
-    return /* @__PURE__ */ jsxs24("div", {
+    return /* @__PURE__ */ jsxs32("div", {
       className: "flex items-center gap-2 max-w-3xl mx-auto",
       children: [
-        icon && /* @__PURE__ */ jsx41("span", {
+        icon && /* @__PURE__ */ jsx52("span", {
           className: "shrink-0",
           children: icon
         }),
-        /* @__PURE__ */ jsx41("p", {
+        /* @__PURE__ */ jsx52("p", {
           className: "text-sm text-destructive flex-1",
           children: displayMessage
         }),
-        dismissible && /* @__PURE__ */ jsx41("button", {
+        dismissible && /* @__PURE__ */ jsx52("button", {
           type: "button",
           onClick: () => setHidden(true),
           className: "shrink-0 text-destructive/60 hover:text-destructive transition-colors",
           "aria-label": "Dismiss error",
-          children: /* @__PURE__ */ jsxs24("svg", {
+          children: /* @__PURE__ */ jsxs32("svg", {
             xmlns: "http://www.w3.org/2000/svg",
             width: "16",
             height: "16",
@@ -3808,10 +4316,10 @@ function AgnoChatErrorBar({
             strokeLinecap: "round",
             strokeLinejoin: "round",
             children: [
-              /* @__PURE__ */ jsx41("path", {
+              /* @__PURE__ */ jsx52("path", {
                 d: "M18 6 6 18"
               }),
-              /* @__PURE__ */ jsx41("path", {
+              /* @__PURE__ */ jsx52("path", {
                 d: "m6 6 12 12"
               })
             ]
@@ -3820,14 +4328,14 @@ function AgnoChatErrorBar({
       ]
     });
   };
-  return /* @__PURE__ */ jsx41("div", {
+  return /* @__PURE__ */ jsx52("div", {
     className: cn("px-4 py-2.5 bg-destructive/5 border-t border-destructive/20", className),
     children: renderContent()
   });
 }
 
 // src/ui/composed/agno-chat/input.tsx
-import { jsx as jsx42 } from "react/jsx-runtime";
+import { jsx as jsx53 } from "react/jsx-runtime";
 function AgnoChatInputArea({
   className,
   children,
@@ -3840,11 +4348,11 @@ function AgnoChatInputArea({
   dropZoneProps
 }) {
   const { handleSend, inputDisabled, isStreaming, isPaused, cancelRun, dropZoneContainerRef } = useAgnoChatContext();
-  return /* @__PURE__ */ jsx42("div", {
+  return /* @__PURE__ */ jsx53("div", {
     className: cn("border-t border-border bg-background/80 backdrop-blur-sm", className),
-    children: /* @__PURE__ */ jsx42("div", {
+    children: /* @__PURE__ */ jsx53("div", {
       className: "mx-auto px-4 py-2",
-      children: children ? children({ onSend: handleSend, disabled: inputDisabled, isStreaming, isPaused }) : /* @__PURE__ */ jsx42(AgnoChatInput, {
+      children: children ? children({ onSend: handleSend, disabled: inputDisabled, isStreaming, isPaused }) : /* @__PURE__ */ jsx53(AgnoChatInput, {
         onSend: handleSend,
         disabled: inputDisabled,
         isStreaming,
@@ -3875,6 +4383,7 @@ export {
   usePromptInputDropZone,
   usePromptInputController,
   usePromptInputAttachments,
+  useAgnoMessageContext,
   useAgnoChatContext,
   isPreviewable,
   getFilePreviewType,
@@ -3883,6 +4392,7 @@ export {
   formatFullTimestamp,
   formatFileSize,
   cn,
+  byToolName,
   buttonVariants,
   badgeVariants,
   TooltipTrigger,
@@ -3892,6 +4402,8 @@ export {
   ToolOutput,
   ToolInput,
   ToolHeader,
+  ToolGenerativeUI,
+  ToolDebugCard,
   ToolContent,
   Tool,
   StreamingIndicator,
@@ -4013,7 +4525,15 @@ export {
   ArtifactActions,
   ArtifactAction,
   Artifact,
+  AgnoMessageTools,
+  AgnoMessageReferences,
+  AgnoMessageReasoning,
+  AgnoMessageMedia,
   AgnoMessageItem,
+  AgnoMessageFooter,
+  AgnoMessageContext,
+  AgnoMessageContent,
+  AgnoMessage,
   AgnoChatSuggestedPrompts,
   AgnoChatRoot,
   AgnoChatMessages,
@@ -4028,4 +4548,4 @@ export {
   Accordion
 };
 
-//# debugId=B7CF327D3F6FBB7F64756E2164756E21
+//# debugId=849670CE2678388A64756E2164756E21