@antipopp/agno-react 0.9.0 → 0.10.0

package/dist/index.mjs

@@ -1,56 +1,12 @@
-// src/context/AgnoContext.tsx
-import { createContext, useContext, useMemo, useEffect } from "react";
-import { AgnoClient } from "@antipopp/agno-client";
-import { jsx } from "react/jsx-runtime";
-var AgnoContext = createContext(null);
-function AgnoProvider({ config, children }) {
-  const client = useMemo(() => new AgnoClient(config), []);
-  useEffect(() => {
-    client.updateConfig(config);
-  }, [client, config]);
-  useEffect(() => {
-    return () => {
-      client.removeAllListeners();
-    };
-  }, [client]);
-  return /* @__PURE__ */ jsx(AgnoContext.Provider, { value: client, children });
-}
-function useAgnoClient() {
-  const client = useContext(AgnoContext);
-  if (!client) {
-    throw new Error("useAgnoClient must be used within an AgnoProvider");
-  }
-  return client;
-}
+// src/components/GenerativeUIRenderer.tsx
+import React from "react";
 
-// src/context/ToolHandlerContext.tsx
-import { createContext as createContext2, useContext as useContext2, useState, useCallback } from "react";
-import { jsx as jsx2 } from "react/jsx-runtime";
-var ToolHandlerContext = createContext2(null);
-function ToolHandlerProvider({ handlers: initialHandlers = {}, children }) {
-  const [handlers, setHandlers] = useState(initialHandlers);
-  const registerHandler = useCallback((name, handler) => {
-    setHandlers((prev) => ({ ...prev, [name]: handler }));
-  }, []);
-  const unregisterHandler = useCallback((name) => {
-    setHandlers((prev) => {
-      const { [name]: _, ...rest } = prev;
-      return rest;
-    });
-  }, []);
-  const value = {
-    handlers,
-    registerHandler,
-    unregisterHandler
-  };
-  return /* @__PURE__ */ jsx2(ToolHandlerContext.Provider, { value, children });
-}
-function useToolHandlers() {
-  return useContext2(ToolHandlerContext);
-}
+// src/hooks/useAgnoToolExecution.ts
+import { useCallback as useCallback2, useEffect as useEffect2, useMemo, useState as useState2 } from "react";
 
-// src/components/GenerativeUIRenderer.tsx
-import React3 from "react";
+// src/context/AgnoContext.tsx
+import { AgnoClient } from "@antipopp/agno-client";
+import { createContext, useContext, useEffect, useRef } from "react";
 
 // src/utils/component-registry.ts
 var ComponentRegistry = class _ComponentRegistry {
@@ -110,6 +66,17 @@ var ComponentRegistry = class _ComponentRegistry {
   clear() {
     this.components.clear();
   }
+  /**
+   * Reset the singleton instance.
+   * Call this during cleanup (e.g., when AgnoProvider unmounts) to prevent memory leaks.
+   * After calling this, getInstance() will create a fresh instance.
+   */
+  static resetInstance() {
+    if (_ComponentRegistry.instance) {
+      _ComponentRegistry.instance.clear();
+      _ComponentRegistry.instance = void 0;
+    }
+  }
 };
 function getComponentRegistry() {
   return ComponentRegistry.getInstance();
@@ -121,8 +88,65 @@ function getChartComponent(name) {
   return getComponentRegistry().get(`chart:${name}`);
 }
 
+// src/context/AgnoContext.tsx
+import { jsx } from "react/jsx-runtime";
+var AgnoContext = createContext(null);
+function AgnoProvider({ config, children }) {
+  const clientRef = useRef(null);
+  if (clientRef.current === null) {
+    clientRef.current = new AgnoClient(config);
+  }
+  const client = clientRef.current;
+  useEffect(() => {
+    client.updateConfig(config);
+  }, [client, config]);
+  useEffect(() => {
+    return () => {
+      client.dispose();
+      clearCustomRenderRegistry();
+      ComponentRegistry.resetInstance();
+    };
+  }, [client]);
+  return /* @__PURE__ */ jsx(AgnoContext.Provider, { value: client, children });
+}
+function useAgnoClient() {
+  const client = useContext(AgnoContext);
+  if (!client) {
+    throw new Error("useAgnoClient must be used within an AgnoProvider");
+  }
+  return client;
+}
+
+// src/context/ToolHandlerContext.tsx
+import { createContext as createContext2, useCallback, useContext as useContext2, useState } from "react";
+import { jsx as jsx2 } from "react/jsx-runtime";
+var ToolHandlerContext = createContext2(null);
+function ToolHandlerProvider({
+  handlers: initialHandlers = {},
+  children
+}) {
+  const [handlers, setHandlers] = useState(initialHandlers);
+  const registerHandler = useCallback((name, handler) => {
+    setHandlers((prev) => ({ ...prev, [name]: handler }));
+  }, []);
+  const unregisterHandler = useCallback((name) => {
+    setHandlers((prev) => {
+      const { [name]: _, ...rest } = prev;
+      return rest;
+    });
+  }, []);
+  const value = {
+    handlers,
+    registerHandler,
+    unregisterHandler
+  };
+  return /* @__PURE__ */ jsx2(ToolHandlerContext.Provider, { value, children });
+}
+function useToolHandlers() {
+  return useContext2(ToolHandlerContext);
+}
+
 // src/hooks/useAgnoToolExecution.ts
-import { useState as useState2, useEffect as useEffect2, useCallback as useCallback2, useMemo as useMemo2 } from "react";
 var customRenderRegistry = /* @__PURE__ */ new Map();
 function registerCustomRender(renderFn) {
   const key = `custom-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
@@ -132,6 +156,9 @@ function registerCustomRender(renderFn) {
 function getCustomRender(key) {
   return customRenderRegistry.get(key);
 }
+function clearCustomRenderRegistry() {
+  customRenderRegistry.clear();
+}
 function isToolHandlerResult(value) {
   return value && typeof value === "object" && ("data" in value || "ui" in value);
 }
@@ -141,7 +168,7 @@ function isUIComponentSpec(value) {
 function processToolResult(result, _tool) {
   if (isToolHandlerResult(result)) {
     const { data, ui } = result;
-    let uiComponent = void 0;
+    let uiComponent;
     if (ui) {
       if (ui.type === "custom" && typeof ui.render === "function") {
         const renderKey = registerCustomRender(ui.render);
@@ -193,7 +220,7 @@ function useAgnoToolExecution(handlers = {}, autoExecute = true) {
       );
     }
   }, [isTeamMode]);
-  const mergedHandlers = useMemo2(() => {
+  const mergedHandlers = useMemo(() => {
     const globalHandlers = toolHandlerContext?.handlers || {};
     return { ...globalHandlers, ...handlers };
   }, [toolHandlerContext?.handlers, handlers]);
@@ -280,14 +307,17 @@ function useAgnoToolExecution(handlers = {}, autoExecute = true) {
     const handleSessionLoaded = async (_sessionId) => {
       const messages = client.getMessages();
       for (const message of messages) {
-        if (!message.tool_calls)
+        if (!message.tool_calls) {
           continue;
+        }
         for (const tool of message.tool_calls) {
-          if (tool.ui_component)
+          if (tool.ui_component) {
             continue;
+          }
           const handler = mergedHandlers[tool.tool_name];
-          if (!handler)
+          if (!handler) {
             continue;
+          }
           try {
             const result = await handler(tool.tool_args);
             const { uiComponent } = processToolResult(result, tool);
@@ -310,8 +340,9 @@ function useAgnoToolExecution(handlers = {}, autoExecute = true) {
       return Promise.all(
         tools.map(async (tool) => {
           const handler = mergedHandlers[tool.tool_name];
-          if (!handler)
+          if (!handler) {
             return tool;
+          }
           try {
             const result = await handler(tool.tool_args);
             const { resultData, uiComponent } = processToolResult(result, tool);
@@ -352,7 +383,13 @@ function useAgnoToolExecution(handlers = {}, autoExecute = true) {
     if (autoExecute && isPaused && !isExecuting && pendingTools.length > 0) {
       executeAndContinue();
     }
-  }, [autoExecute, isPaused, isExecuting, pendingTools.length, executeAndContinue]);
+  }, [
+    autoExecute,
+    isPaused,
+    isExecuting,
+    pendingTools.length,
+    executeAndContinue
+  ]);
   return {
     /** Whether the run is currently paused awaiting tool execution */
     isPaused,
@@ -373,7 +410,7 @@ function useAgnoToolExecution(handlers = {}, autoExecute = true) {
 
 // src/components/GenerativeUIRenderer.tsx
 import { jsx as jsx3, jsxs } from "react/jsx-runtime";
-var UIErrorBoundary = class extends React3.Component {
+var UIErrorBoundary = class extends React.Component {
   constructor(props) {
     super(props);
     this.state = { hasError: false };
@@ -382,14 +419,18 @@ var UIErrorBoundary = class extends React3.Component {
     return { hasError: true, error };
   }
   componentDidCatch(error, errorInfo) {
-    console.error("[GenerativeUIRenderer] Error rendering component:", error, errorInfo);
+    console.error(
+      "[GenerativeUIRenderer] Error rendering component:",
+      error,
+      errorInfo
+    );
     this.props.onError?.(error);
   }
   render() {
     if (this.state.hasError) {
-      return this.props.fallback || /* @__PURE__ */ jsxs("div", { className: "p-4 border border-red-300 rounded-md bg-red-50 text-red-800", children: [
+      return this.props.fallback || /* @__PURE__ */ jsxs("div", { className: "rounded-md border border-red-300 bg-red-50 p-4 text-red-800", children: [
         /* @__PURE__ */ jsx3("p", { className: "font-semibold", children: "Failed to render UI component" }),
-        /* @__PURE__ */ jsx3("p", { className: "text-sm mt-1", children: this.state.error?.message || "Unknown error" })
+        /* @__PURE__ */ jsx3("p", { className: "mt-1 text-sm", children: this.state.error?.message || "Unknown error" })
       ] });
     }
     return this.props.children;
@@ -409,10 +450,16 @@ function GenerativeUIRenderer({
         return /* @__PURE__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsx3("div", { className, children: renderFn(customSpec.props || {}) }) });
       }
     }
-    return /* @__PURE__ */ jsxs("div", { className: `p-4 border border-yellow-300 rounded-md bg-yellow-50 text-yellow-800 ${className || ""}`, children: [
-      /* @__PURE__ */ jsx3("p", { className: "font-semibold", children: "Custom component not available" }),
-      /* @__PURE__ */ jsx3("p", { className: "text-sm mt-1", children: "The custom render function for this component is not available." })
-    ] });
+    return /* @__PURE__ */ jsxs(
+      "div",
+      {
+        className: `rounded-md border border-yellow-300 bg-yellow-50 p-4 text-yellow-800 ${className || ""}`,
+        children: [
+          /* @__PURE__ */ jsx3("p", { className: "font-semibold", children: "Custom component not available" }),
+          /* @__PURE__ */ jsx3("p", { className: "mt-1 text-sm", children: "The custom render function for this component is not available." })
+        ]
+      }
+    );
   }
   if (spec.type === "chart") {
     const chartSpec = spec;
@@ -420,24 +467,30 @@ function GenerativeUIRenderer({
     if (registry.has(chartType)) {
       const ChartRenderer = registry.get(chartType);
       return /* @__PURE__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsxs("div", { className, children: [
-        chartSpec.title && /* @__PURE__ */ jsx3("h3", { className: "font-semibold mb-2", children: chartSpec.title }),
-        chartSpec.description && /* @__PURE__ */ jsx3("p", { className: "text-sm text-gray-600 mb-4", children: chartSpec.description }),
+        chartSpec.title && /* @__PURE__ */ jsx3("h3", { className: "mb-2 font-semibold", children: chartSpec.title }),
+        chartSpec.description && /* @__PURE__ */ jsx3("p", { className: "mb-4 text-gray-600 text-sm", children: chartSpec.description }),
         /* @__PURE__ */ jsx3(ChartRenderer, { ...chartSpec.props })
       ] }) });
     }
-    return /* @__PURE__ */ jsxs("div", { className: `p-4 border border-gray-300 rounded-md ${className || ""}`, children: [
-      /* @__PURE__ */ jsx3("p", { className: "font-semibold mb-2", children: chartSpec.title || "Chart Data" }),
-      chartSpec.description && /* @__PURE__ */ jsx3("p", { className: "text-sm text-gray-600 mb-2", children: chartSpec.description }),
-      /* @__PURE__ */ jsx3("pre", { className: "text-xs bg-gray-100 p-2 rounded overflow-auto", children: JSON.stringify(chartSpec.props.data, null, 2) })
-    ] });
+    return /* @__PURE__ */ jsxs(
+      "div",
+      {
+        className: `rounded-md border border-gray-300 p-4 ${className || ""}`,
+        children: [
+          /* @__PURE__ */ jsx3("p", { className: "mb-2 font-semibold", children: chartSpec.title || "Chart Data" }),
+          chartSpec.description && /* @__PURE__ */ jsx3("p", { className: "mb-2 text-gray-600 text-sm", children: chartSpec.description }),
+          /* @__PURE__ */ jsx3("pre", { className: "overflow-auto rounded bg-gray-100 p-2 text-xs", 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__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsxs("div", { className, children: [
-        cardGridSpec.title && /* @__PURE__ */ jsx3("h3", { className: "font-semibold mb-2", children: cardGridSpec.title }),
-        cardGridSpec.description && /* @__PURE__ */ jsx3("p", { className: "text-sm text-gray-600 mb-4", children: cardGridSpec.description }),
+        cardGridSpec.title && /* @__PURE__ */ jsx3("h3", { className: "mb-2 font-semibold", children: cardGridSpec.title }),
+        cardGridSpec.description && /* @__PURE__ */ jsx3("p", { className: "mb-4 text-gray-600 text-sm", children: cardGridSpec.description }),
         /* @__PURE__ */ jsx3(CardGridRenderer, { ...cardGridSpec.props })
       ] }) });
     }
@@ -447,8 +500,8 @@ function GenerativeUIRenderer({
     if (registry.has("table")) {
       const TableRenderer = registry.get("table");
       return /* @__PURE__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsxs("div", { className, children: [
-        tableSpec.title && /* @__PURE__ */ jsx3("h3", { className: "font-semibold mb-2", children: tableSpec.title }),
-        tableSpec.description && /* @__PURE__ */ jsx3("p", { className: "text-sm text-gray-600 mb-4", children: tableSpec.description }),
+        tableSpec.title && /* @__PURE__ */ jsx3("h3", { className: "mb-2 font-semibold", children: tableSpec.title }),
+        tableSpec.description && /* @__PURE__ */ jsx3("p", { className: "mb-4 text-gray-600 text-sm", children: tableSpec.description }),
         /* @__PURE__ */ jsx3(TableRenderer, { ...tableSpec.props })
       ] }) });
     }
@@ -463,109 +516,370 @@ function GenerativeUIRenderer({
   }
   if (spec.type === "artifact") {
     const artifactSpec = spec;
-    return /* @__PURE__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsxs("div", { className: `p-4 border rounded-md ${className || ""}`, children: [
-      artifactSpec.title && /* @__PURE__ */ jsx3("h3", { className: "font-semibold mb-4", children: artifactSpec.title }),
-      artifactSpec.description && /* @__PURE__ */ jsx3("p", { className: "text-sm text-gray-600 mb-4", children: artifactSpec.description }),
-      /* @__PURE__ */ jsx3("div", { className: "space-y-4", children: artifactSpec.props.content?.map((childSpec, index) => /* @__PURE__ */ jsx3(GenerativeUIRenderer, { spec: childSpec, onError }, index)) })
+    return /* @__PURE__ */ jsx3(UIErrorBoundary, { onError, children: /* @__PURE__ */ jsxs("div", { className: `rounded-md border p-4 ${className || ""}`, children: [
+      artifactSpec.title && /* @__PURE__ */ jsx3("h3", { className: "mb-4 font-semibold", children: artifactSpec.title }),
+      artifactSpec.description && /* @__PURE__ */ jsx3("p", { className: "mb-4 text-gray-600 text-sm", children: artifactSpec.description }),
+      /* @__PURE__ */ jsx3("div", { className: "space-y-4", children: artifactSpec.props.content?.map(
+        (childSpec, index) => /* @__PURE__ */ jsx3(
+          GenerativeUIRenderer,
+          {
+            onError,
+            spec: childSpec
+          },
+          index
+        )
+      ) })
     ] }) });
   }
-  return /* @__PURE__ */ jsxs("div", { className: `p-4 border border-gray-300 rounded-md ${className || ""}`, children: [
+  return /* @__PURE__ */ jsxs("div", { className: `rounded-md border border-gray-300 p-4 ${className || ""}`, children: [
     /* @__PURE__ */ jsx3("p", { className: "font-semibold", children: "Unsupported UI component" }),
-    /* @__PURE__ */ jsxs("p", { className: "text-sm text-gray-600 mt-1", children: [
+    /* @__PURE__ */ jsxs("p", { className: "mt-1 text-gray-600 text-sm", children: [
       "Component type: ",
       spec.type
     ] })
   ] });
 }
 
-// src/utils/ui-helpers.ts
-function createBarChart(data, xKey, bars, options) {
-  return {
-    type: "chart",
-    component: "BarChart",
-    layout: options?.layout,
-    title: options?.title,
-    description: options?.description,
-    props: {
-      data,
-      xKey,
-      bars: bars.map((bar) => ({
-        key: bar.key,
-        label: bar.label || bar.key,
-        color: bar.color
-      })),
-      showLegend: options?.showLegend ?? true,
-      showGrid: options?.showGrid ?? true,
-      height: options?.height,
-      width: options?.width
-    }
-  };
-}
-function createLineChart(data, xKey, lines, options) {
-  return {
-    type: "chart",
-    component: "LineChart",
-    layout: options?.layout,
-    title: options?.title,
-    description: options?.description,
-    props: {
-      data,
-      xKey,
-      lines: lines.map((line) => ({
-        key: line.key,
-        label: line.label || line.key,
-        color: line.color
-      })),
-      showLegend: options?.showLegend ?? true,
-      showGrid: options?.showGrid ?? true,
-      height: options?.height,
-      width: options?.width
-    }
-  };
-}
-function createPieChart(data, dataKey, nameKey, options) {
-  return {
-    type: "chart",
-    component: "PieChart",
-    layout: options?.layout,
-    title: options?.title,
-    description: options?.description,
-    props: {
-      data,
-      pie: {
-        dataKey,
-        nameKey,
-        label: options?.showLabel ?? true
-      },
-      showLegend: options?.showLegend ?? true,
-      height: options?.height || 400,
-      width: options?.width
-    }
-  };
-}
-function createAreaChart(data, xKey, areas, options) {
+// src/hooks/useAgnoActions.ts
+import { useCallback as useCallback3, useState as useState3 } from "react";
+function useAgnoActions() {
+  const client = useAgnoClient();
+  const [isInitializing, setIsInitializing] = useState3(false);
+  const [error, setError] = useState3();
+  const initialize = useCallback3(
+    async (options) => {
+      setIsInitializing(true);
+      setError(void 0);
+      try {
+        const result = await client.initialize(options);
+        return result;
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      } finally {
+        setIsInitializing(false);
+      }
+    },
+    [client]
+  );
+  const checkStatus = useCallback3(
+    async (options) => {
+      setError(void 0);
+      try {
+        return await client.checkStatus(options);
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        return false;
+      }
+    },
+    [client]
+  );
+  const fetchAgents = useCallback3(
+    async (options) => {
+      setError(void 0);
+      try {
+        return await client.fetchAgents(options);
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      }
+    },
+    [client]
+  );
+  const fetchTeams = useCallback3(
+    async (options) => {
+      setError(void 0);
+      try {
+        return await client.fetchTeams(options);
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      }
+    },
+    [client]
+  );
+  const updateConfig = useCallback3(
+    (updates) => {
+      client.updateConfig(updates);
+    },
+    [client]
+  );
   return {
-    type: "chart",
-    component: "AreaChart",
-    layout: options?.layout,
-    title: options?.title,
-    description: options?.description,
-    props: {
-      data,
-      xKey,
-      areas: areas.map((area) => ({
-        key: area.key,
-        label: area.label || area.key,
-        color: area.color
-      })),
-      showLegend: options?.showLegend ?? true,
-      showGrid: options?.showGrid ?? true,
-      height: options?.height,
-      width: options?.width
-    }
+    initialize,
+    checkStatus,
+    fetchAgents,
+    fetchTeams,
+    updateConfig,
+    isInitializing,
+    error
   };
 }
-function createCardGrid(cards, options) {
+
+// src/hooks/useAgnoChat.ts
+import { useCallback as useCallback4, useEffect as useEffect3, useState as useState4 } from "react";
+function useAgnoChat() {
+  const client = useAgnoClient();
+  const [messages, setMessages] = useState4(client.getMessages());
+  const [state, setState] = useState4(client.getState());
+  const [error, setError] = useState4();
+  useEffect3(() => {
+    const handleMessageUpdate = (updatedMessages) => {
+      setMessages(updatedMessages);
+    };
+    const handleMessageComplete = (updatedMessages) => {
+      setMessages(updatedMessages);
+    };
+    const handleMessageRefreshed = (updatedMessages) => {
+      setMessages(updatedMessages);
+    };
+    const handleMessageError = (errorMessage) => {
+      setError(errorMessage);
+    };
+    const handleStateChange = (newState) => {
+      setState(newState);
+    };
+    const handleUIRender = (event) => {
+      const { tools } = event;
+      for (const tool of tools) {
+        if (tool.ui_component) {
+          client.hydrateToolCallUI(
+            tool.tool_call_id,
+            tool.ui_component
+          );
+        }
+      }
+    };
+    client.on("message:update", handleMessageUpdate);
+    client.on("message:complete", handleMessageComplete);
+    client.on("message:refreshed", handleMessageRefreshed);
+    client.on("message:error", handleMessageError);
+    client.on("state:change", handleStateChange);
+    client.on("ui:render", handleUIRender);
+    setMessages(client.getMessages());
+    setState(client.getState());
+    return () => {
+      client.off("message:update", handleMessageUpdate);
+      client.off("message:complete", handleMessageComplete);
+      client.off("message:refreshed", handleMessageRefreshed);
+      client.off("message:error", handleMessageError);
+      client.off("state:change", handleStateChange);
+      client.off("ui:render", handleUIRender);
+    };
+  }, [client]);
+  const sendMessage = useCallback4(
+    async (message, options) => {
+      setError(void 0);
+      try {
+        await client.sendMessage(message, options);
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      }
+    },
+    [client]
+  );
+  const cancelRun = useCallback4(async () => {
+    try {
+      await client.cancelRun();
+    } catch (err) {
+      const errorMessage = err instanceof Error ? err.message : String(err);
+      setError(errorMessage);
+      throw err;
+    }
+  }, [client]);
+  const clearMessages = useCallback4(() => {
+    client.clearMessages();
+    setMessages([]);
+    setError(void 0);
+  }, [client]);
+  return {
+    messages,
+    sendMessage,
+    clearMessages,
+    cancelRun,
+    isStreaming: state.isStreaming,
+    isRefreshing: state.isRefreshing,
+    isPaused: state.isPaused,
+    isCancelling: state.isCancelling,
+    currentRunId: state.currentRunId,
+    error,
+    state
+  };
+}
+
+// src/hooks/useAgnoSession.ts
+import { useCallback as useCallback5, useEffect as useEffect4, useState as useState5 } from "react";
+function useAgnoSession() {
+  const client = useAgnoClient();
+  const [sessions, setSessions] = useState5([]);
+  const [currentSessionId, setCurrentSessionId] = useState5(
+    client.getConfig().sessionId
+  );
+  const [isLoading, setIsLoading] = useState5(false);
+  const [error, setError] = useState5();
+  useEffect4(() => {
+    const handleSessionLoaded = (sessionId) => {
+      setCurrentSessionId(sessionId);
+    };
+    const handleSessionCreated = (session) => {
+      setSessions((prev) => [session, ...prev]);
+      setCurrentSessionId(session.session_id);
+    };
+    const handleStateChange = () => {
+      const config = client.getConfig();
+      setCurrentSessionId(config.sessionId);
+      setSessions(client.getState().sessions);
+    };
+    client.on("session:loaded", handleSessionLoaded);
+    client.on("session:created", handleSessionCreated);
+    client.on("state:change", handleStateChange);
+    setSessions(client.getState().sessions);
+    setCurrentSessionId(client.getConfig().sessionId);
+    return () => {
+      client.off("session:loaded", handleSessionLoaded);
+      client.off("session:created", handleSessionCreated);
+      client.off("state:change", handleStateChange);
+    };
+  }, [client]);
+  const loadSession = useCallback5(
+    async (sessionId, options) => {
+      setIsLoading(true);
+      setError(void 0);
+      try {
+        const messages = await client.loadSession(sessionId, options);
+        setCurrentSessionId(sessionId);
+        return messages;
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      } finally {
+        setIsLoading(false);
+      }
+    },
+    [client]
+  );
+  const fetchSessions = useCallback5(
+    async (options) => {
+      setIsLoading(true);
+      setError(void 0);
+      try {
+        const fetchedSessions = await client.fetchSessions(options);
+        setSessions(fetchedSessions);
+        return fetchedSessions;
+      } catch (err) {
+        const errorMessage = err instanceof Error ? err.message : String(err);
+        setError(errorMessage);
+        throw err;
+      } finally {
+        setIsLoading(false);
+      }
+    },
+    [client]
+  );
+  return {
+    sessions,
+    currentSessionId,
+    loadSession,
+    fetchSessions,
+    isLoading,
+    error
+  };
+}
+
+// src/utils/ui-helpers.ts
+function createBarChart(data, xKey, bars, options) {
+  return {
+    type: "chart",
+    component: "BarChart",
+    layout: options?.layout,
+    title: options?.title,
+    description: options?.description,
+    props: {
+      data,
+      xKey,
+      bars: bars.map((bar) => ({
+        key: bar.key,
+        label: bar.label || bar.key,
+        color: bar.color
+      })),
+      showLegend: options?.showLegend ?? true,
+      showGrid: options?.showGrid ?? true,
+      height: options?.height,
+      width: options?.width
+    }
+  };
+}
+function createLineChart(data, xKey, lines, options) {
+  return {
+    type: "chart",
+    component: "LineChart",
+    layout: options?.layout,
+    title: options?.title,
+    description: options?.description,
+    props: {
+      data,
+      xKey,
+      lines: lines.map((line) => ({
+        key: line.key,
+        label: line.label || line.key,
+        color: line.color
+      })),
+      showLegend: options?.showLegend ?? true,
+      showGrid: options?.showGrid ?? true,
+      height: options?.height,
+      width: options?.width
+    }
+  };
+}
+function createPieChart(data, dataKey, nameKey, options) {
+  return {
+    type: "chart",
+    component: "PieChart",
+    layout: options?.layout,
+    title: options?.title,
+    description: options?.description,
+    props: {
+      data,
+      pie: {
+        dataKey,
+        nameKey,
+        label: options?.showLabel ?? true
+      },
+      showLegend: options?.showLegend ?? true,
+      height: options?.height || 400,
+      width: options?.width
+    }
+  };
+}
+function createAreaChart(data, xKey, areas, options) {
+  return {
+    type: "chart",
+    component: "AreaChart",
+    layout: options?.layout,
+    title: options?.title,
+    description: options?.description,
+    props: {
+      data,
+      xKey,
+      areas: areas.map((area) => ({
+        key: area.key,
+        label: area.label || area.key,
+        color: area.color
+      })),
+      showLegend: options?.showLegend ?? true,
+      showGrid: options?.showGrid ?? true,
+      height: options?.height,
+      width: options?.width
+    }
+  };
+}
+function createCardGrid(cards, options) {
   return {
     type: "card-grid",
     layout: options?.layout,
@@ -644,9 +958,13 @@ function createSmartChart(data, options) {
   const firstItem = data[0];
   const keys = Object.keys(firstItem);
   const xKey = options?.xKey || keys.find(
-    (k) => ["name", "label", "category", "date", "time", "month", "year"].includes(k.toLowerCase())
+    (k) => ["name", "label", "category", "date", "time", "month", "year"].includes(
+      k.toLowerCase()
+    )
   ) || keys[0];
-  const numericKeys = keys.filter((k) => k !== xKey && typeof firstItem[k] === "number");
+  const numericKeys = keys.filter(
+    (k) => k !== xKey && typeof firstItem[k] === "number"
+  );
   const yKeys = options?.yKeys || numericKeys;
   if (options?.preferredType) {
     switch (options.preferredType) {
@@ -708,233 +1026,12 @@ function resultWithCardGrid(cards, options) {
 function resultWithTable(data, columns, options) {
   return createToolResult(data, createTable(data, columns, options));
 }
-
-// src/hooks/useAgnoChat.ts
-import { useState as useState3, useEffect as useEffect3, useCallback as useCallback3 } from "react";
-function useAgnoChat() {
-  const client = useAgnoClient();
-  const [messages, setMessages] = useState3(client.getMessages());
-  const [state, setState] = useState3(client.getState());
-  const [error, setError] = useState3();
-  useEffect3(() => {
-    const handleMessageUpdate = (updatedMessages) => {
-      setMessages(updatedMessages);
-    };
-    const handleMessageComplete = (updatedMessages) => {
-      setMessages(updatedMessages);
-    };
-    const handleMessageRefreshed = (updatedMessages) => {
-      setMessages(updatedMessages);
-    };
-    const handleMessageError = (errorMessage) => {
-      setError(errorMessage);
-    };
-    const handleStateChange = (newState) => {
-      setState(newState);
-    };
-    const handleUIRender = (event) => {
-      const { tools } = event;
-      for (const tool of tools) {
-        if (tool.ui_component) {
-          client.hydrateToolCallUI(tool.tool_call_id, tool.ui_component);
-        }
-      }
-    };
-    client.on("message:update", handleMessageUpdate);
-    client.on("message:complete", handleMessageComplete);
-    client.on("message:refreshed", handleMessageRefreshed);
-    client.on("message:error", handleMessageError);
-    client.on("state:change", handleStateChange);
-    client.on("ui:render", handleUIRender);
-    setMessages(client.getMessages());
-    setState(client.getState());
-    return () => {
-      client.off("message:update", handleMessageUpdate);
-      client.off("message:complete", handleMessageComplete);
-      client.off("message:refreshed", handleMessageRefreshed);
-      client.off("message:error", handleMessageError);
-      client.off("state:change", handleStateChange);
-      client.off("ui:render", handleUIRender);
-    };
-  }, [client]);
-  const sendMessage = useCallback3(
-    async (message, options) => {
-      setError(void 0);
-      try {
-        await client.sendMessage(message, options);
-      } catch (err) {
-        const errorMessage = err instanceof Error ? err.message : String(err);
-        setError(errorMessage);
-        throw err;
-      }
-    },
-    [client]
-  );
-  const clearMessages = useCallback3(() => {
-    client.clearMessages();
-    setMessages([]);
-    setError(void 0);
-  }, [client]);
-  return {
-    messages,
-    sendMessage,
-    clearMessages,
-    isStreaming: state.isStreaming,
-    isRefreshing: state.isRefreshing,
-    isPaused: state.isPaused,
-    error,
-    state
-  };
-}
-
-// src/hooks/useAgnoSession.ts
-import { useState as useState4, useEffect as useEffect4, useCallback as useCallback4 } from "react";
-function useAgnoSession() {
-  const client = useAgnoClient();
-  const [sessions, setSessions] = useState4([]);
-  const [currentSessionId, setCurrentSessionId] = useState4(
-    client.getConfig().sessionId
-  );
-  const [isLoading, setIsLoading] = useState4(false);
-  const [error, setError] = useState4();
-  useEffect4(() => {
-    const handleSessionLoaded = (sessionId) => {
-      setCurrentSessionId(sessionId);
-    };
-    const handleSessionCreated = (session) => {
-      setSessions((prev) => [session, ...prev]);
-      setCurrentSessionId(session.session_id);
-    };
-    const handleStateChange = () => {
-      const config = client.getConfig();
-      setCurrentSessionId(config.sessionId);
-      setSessions(client.getState().sessions);
-    };
-    client.on("session:loaded", handleSessionLoaded);
-    client.on("session:created", handleSessionCreated);
-    client.on("state:change", handleStateChange);
-    setSessions(client.getState().sessions);
-    setCurrentSessionId(client.getConfig().sessionId);
-    return () => {
-      client.off("session:loaded", handleSessionLoaded);
-      client.off("session:created", handleSessionCreated);
-      client.off("state:change", handleStateChange);
-    };
-  }, [client]);
-  const loadSession = useCallback4(
-    async (sessionId, options) => {
-      setIsLoading(true);
-      setError(void 0);
-      try {
-        const messages = await client.loadSession(sessionId, options);
-        setCurrentSessionId(sessionId);
-        return messages;
-      } catch (err) {
-        const errorMessage = err instanceof Error ? err.message : String(err);
-        setError(errorMessage);
-        throw err;
-      } finally {
-        setIsLoading(false);
-      }
-    },
-    [client]
-  );
-  const fetchSessions = useCallback4(async (options) => {
-    setIsLoading(true);
-    setError(void 0);
-    try {
-      const fetchedSessions = await client.fetchSessions(options);
-      setSessions(fetchedSessions);
-      return fetchedSessions;
-    } catch (err) {
-      const errorMessage = err instanceof Error ? err.message : String(err);
-      setError(errorMessage);
-      throw err;
-    } finally {
-      setIsLoading(false);
-    }
-  }, [client]);
-  return {
-    sessions,
-    currentSessionId,
-    loadSession,
-    fetchSessions,
-    isLoading,
-    error
-  };
-}
-
-// src/hooks/useAgnoActions.ts
-import { useState as useState5, useCallback as useCallback5 } from "react";
-function useAgnoActions() {
-  const client = useAgnoClient();
-  const [isInitializing, setIsInitializing] = useState5(false);
-  const [error, setError] = useState5();
-  const initialize = useCallback5(async (options) => {
-    setIsInitializing(true);
-    setError(void 0);
-    try {
-      const result = await client.initialize(options);
-      return result;
-    } catch (err) {
-      const errorMessage = err instanceof Error ? err.message : String(err);
-      setError(errorMessage);
-      throw err;
-    } finally {
-      setIsInitializing(false);
-    }
-  }, [client]);
-  const checkStatus = useCallback5(async (options) => {
-    setError(void 0);
-    try {
-      return await client.checkStatus(options);
-    } catch (err) {
-      const errorMessage = err instanceof Error ? err.message : String(err);
-      setError(errorMessage);
-      return false;
-    }
-  }, [client]);
-  const fetchAgents = useCallback5(async (options) => {
-    setError(void 0);
-    try {
-      return await client.fetchAgents(options);
-    } catch (err) {
-      const errorMessage = err instanceof Error ? err.message : String(err);
-      setError(errorMessage);
-      throw err;
-    }
-  }, [client]);
-  const fetchTeams = useCallback5(async (options) => {
-    setError(void 0);
-    try {
-      return await client.fetchTeams(options);
-    } catch (err) {
-      const errorMessage = err instanceof Error ? err.message : String(err);
-      setError(errorMessage);
-      throw err;
-    }
-  }, [client]);
-  const updateConfig = useCallback5(
-    (updates) => {
-      client.updateConfig(updates);
-    },
-    [client]
-  );
-  return {
-    initialize,
-    checkStatus,
-    fetchAgents,
-    fetchTeams,
-    updateConfig,
-    isInitializing,
-    error
-  };
-}
 export {
   AgnoProvider,
   ComponentRegistry,
   GenerativeUIRenderer,
   ToolHandlerProvider,
+  clearCustomRenderRegistry,
   createAreaChart,
   createArtifact,
   createBarChart,