dp-widgets-framework 1.1.2 → 1.1.4

package/dist/index.esm.js

@@ -7,7 +7,7 @@ import * as LabelPrimitive from '@radix-ui/react-label';
 import { cva } from 'class-variance-authority';
 import * as SwitchPrimitives from '@radix-ui/react-switch';
 import * as SelectPrimitive from '@radix-ui/react-select';
-import { ChevronDown, ChevronUp, Check, AlertCircle, MoveUp, MoveDown, Trash2, Plus, Bot, Type, Layout, LayoutGrid, BarChart as BarChart$1, Filter, Search, ArrowUp, ArrowDown, RefreshCw, X, LineChart as LineChart$1, PieChart as PieChart$1, Table, FileText, Loader2, GripHorizontal, Edit, MessageCircleX, Edit2 } from 'lucide-react';
+import { ChevronDown, ChevronUp, Check, AlertCircle, MoveUp, MoveDown, Trash2, Plus, Bot, Type, Layout, LayoutGrid, BarChart, Filter, Search, ArrowUp, ArrowDown, RefreshCw, X, LineChart, PieChart as PieChart$1, Table, FileText, Loader2, GripHorizontal, Edit, MessageCircleX, Edit2 } from 'lucide-react';
 import * as ScrollAreaPrimitive from '@radix-ui/react-scroll-area';
 import { Slot } from '@radix-ui/react-slot';
 import { debounce as debounce$1 } from 'lodash';
@@ -1912,7 +1912,7 @@ const IconMap$1 = {
     search: Search,
     filter: Filter,
     results: LayoutGrid,
-    analytics: BarChart$1,
+    analytics: BarChart,
     custom: LayoutGrid,
     header: Layout,
     footer: Layout,
@@ -15048,78 +15048,6 @@ const Line = /* #__PURE__ */ createTypedChart('line', LineController);
 const Bar = /* #__PURE__ */ createTypedChart('bar', BarController);
 const Pie = /* #__PURE__ */ createTypedChart('pie', PieController);
 
-Chart$1.register(CategoryScale, LinearScale, BarElement, plugin_title, plugin_tooltip, plugin_legend);
-function BarChart({ orientation, title, data, options, className, units }) {
-    const formatValue = (value) => {
-        const numValue = typeof value === 'string' ? parseFloat(value) : value;
-        if (units === '$' && !isNaN(numValue)) {
-            return numValue.toLocaleString();
-        }
-        return value;
-    };
-    const defaultOptions = {
-        responsive: true,
-        maintainAspectRatio: false,
-        indexAxis: orientation === "horizontal" ? "y" : "x",
-        layout: {
-            padding: {
-                bottom: 0,
-                top: 0,
-                left: 0,
-                right: 0,
-            },
-        },
-        plugins: {
-            legend: {
-                display: false,
-                labels: {
-                    boxWidth: 5,
-                    boxHeight: 5,
-                },
-                position: "bottom"
-            },
-            title: {
-                display: true,
-                text: title,
-                font: {
-                    size: 18,
-                    weight: "bold",
-                },
-                color: "#0F172A"
-            },
-            tooltip: {
-                callbacks: {
-                    label: function (context) {
-                        const value = context.parsed.y || context.parsed.x;
-                        return units ? `${context.dataset.label}: ${formatValue(value)} ${units}` : `${context.dataset.label}: ${value}`;
-                    }
-                }
-            },
-        },
-        scales: {
-            x: {
-                ticks: {
-                    autoSkip: false,
-                    maxRotation: 45,
-                    minRotation: 45,
-                    font: {
-                        size: 10
-                    }
-                },
-            },
-            y: {
-                beginAtZero: true,
-                ticks: {
-                    callback: function (value) {
-                        return units ? `${formatValue(value)} ${units}` : value;
-                    }
-                }
-            },
-        },
-    };
-    return (jsxRuntimeExports.jsx(Bar, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions }));
-}
-
 Chart$1.register(CategoryScale, LinearScale, BarElement, plugin_title, plugin_tooltip, plugin_legend);
 function SeriesChart({ orientation, title, data, options, className, units }) {
     const formatValue = (value) => {
@@ -15308,70 +15236,6 @@ const PieChart = (props) => {
     return jsxRuntimeExports.jsx(Pie, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions });
 };
 
-Chart$1.register(CategoryScale, LinearScale, PointElement, LineElement, plugin_title, plugin_tooltip, plugin_legend);
-function LineChart({ title, data, options, className, units }) {
-    const formatValue = (value) => {
-        const numValue = typeof value === 'string' ? parseFloat(value) : value;
-        if (units === '$' && !isNaN(numValue)) {
-            return numValue.toLocaleString();
-        }
-        return value;
-    };
-    const defaultOptions = {
-        responsive: true,
-        maintainAspectRatio: false,
-        layout: {
-            padding: {
-                bottom: 0,
-                top: 0,
-                left: 0,
-                right: 0,
-            },
-        },
-        plugins: {
-            legend: {
-                display: false,
-                labels: {
-                    boxWidth: 5,
-                    boxHeight: 5,
-                },
-                position: "bottom"
-            },
-            title: {
-                display: true,
-                text: title,
-                font: {
-                    size: 18,
-                    weight: "bold",
-                },
-                color: "#0F172A"
-            },
-            tooltip: {
-                callbacks: {
-                    label: function (context) {
-                        const value = context.parsed.y;
-                        return units ? `${context.dataset.label}: ${formatValue(value)} ${units}` : `${context.dataset.label}: ${value}`;
-                    }
-                }
-            },
-        },
-        scales: {
-            x: {
-                ticks: { autoSkip: true, maxTicksLimit: 8, maxRotation: 45, minRotation: 0 },
-            },
-            y: {
-                beginAtZero: true,
-                ticks: {
-                    callback: function (value) {
-                        return units ? `${formatValue(value)} ${units}` : value;
-                    }
-                }
-            },
-        },
-    };
-    return (jsxRuntimeExports.jsx(Line, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions }));
-}
-
 function DataGrid({ title, data, className }) {
     const { columns, rows } = data;
     if (!columns || columns.length === 0 || !rows || rows.length === 0) {
@@ -36072,24 +35936,12 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
     const baseState = { dataset_id: datasetId || "home_generation_dataset" };
     switch (agentType) {
         case "Pie Chart Agent":
-            return {
-                pie_chart_data: {
+            return Object.assign({ pie_chart_data: {
                     title: "",
                     type: "pie",
                     chart_type: "financial",
                     data: { labels: [], values: [], percentages: [], total: 0 },
                     metadata: { categories: 0, largest_category: "", largest_value: 0, largest_percentage: 0 }
-                }
-            };
-        case "Line Chart Agent":
-        case "Bar Chart Agent":
-            return Object.assign({ bar_chart_data: {
-                    title: "",
-                    type: agentType === "Line Chart Agent" ? "line" : "bar",
-                    chart_type: "financial",
-                    orientation: agentType === "Line Chart Agent" ? "horizontal" : "vertical",
-                    data: { labels: [], values: [], total: 0, average: 0 },
-                    metadata: { categories: 0, highest_category: "", highest_value: 0, lowest_category: "", lowest_value: 0, range: 0 }
                 } }, baseState);
         case "Series Bar Chart Agent":
         case "Series Line Chart Agent":
@@ -36098,6 +35950,7 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
                     type: "series_bar",
                     chart_type: "financial",
                     orientation: "vertical",
+                    units: "",
                     data: { labels: [], series: [], total: 0, average: 0 },
                     metadata: {
                         categories: 0, series_count: 0, highest_category: "", highest_value: 0, highest_series: "",
@@ -36120,7 +35973,7 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
                     metadata: { created_at: "", content_type: "text", is_multiline: false }
                 } }, baseState);
         default:
-            return { widget_ids: widgetIds };
+            return Object.assign({ widget_ids: widgetIds }, baseState);
     }
 };
 const loadAgentState = async (widgetBackendUrl, threadId, agentName) => {
@@ -36218,58 +36071,6 @@ const getStyleValues = (styles = {}) => {
         textAlign: alignment,
     };
 };
-function BarChartComponent({ orientation, barChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
-    const hasCalledRef = useRef(false);
-    const labels = ((_b = (_a = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.labels) || ((_e = (_d = (_c = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _c === void 0 ? void 0 : _c.bar_chart_data) === null || _d === void 0 ? void 0 : _d.data) === null || _e === void 0 ? void 0 : _e.labels) || [];
-    const values = ((_g = (_f = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.values) || ((_k = (_j = (_h = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _h === void 0 ? void 0 : _h.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) || [];
-    const chartTitle = ((_l = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _l === void 0 ? void 0 : _l.title) || ((_o = (_m = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _m === void 0 ? void 0 : _m.bar_chart_data) === null || _o === void 0 ? void 0 : _o.title) || "";
-    const units = ((_p = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _p === void 0 ? void 0 : _p.units) || ((_r = (_q = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _q === void 0 ? void 0 : _q.bar_chart_data) === null || _r === void 0 ? void 0 : _r.units) || "";
-    values.length > 0 ? Math.max(...values) : 0;
-    const isEmpty = labels.length === 0 || values.length === 0;
-    const handleRefresh = async () => {
-        if (query && widgetBackendUrl && widgetId) {
-            await clearChat(widgetBackendUrl, widgetId);
-            appendMessage(new TextMessage({
-                content: `${query} and render data on a bar chart`,
-                role: Role.User,
-            }));
-        }
-    };
-    useEffect(() => {
-        if (isEmpty && query && isFirstLoad && !(barChartState === null || barChartState === void 0 ? void 0 : barChartState.agent_message) && !hasCalledRef.current) {
-            // console.log('calling append message')
-            hasCalledRef.current = true;
-            handleRefresh();
-        }
-    }, [isEmpty, query, isFirstLoad, barChartState]);
-    // Start timeout when chart is empty and loading
-    useEffect(() => {
-        if (isEmpty && startLoadingTimeout && !(barChartState === null || barChartState === void 0 ? void 0 : barChartState.agent_message)) {
-            startLoadingTimeout();
-        }
-        else if (!isEmpty && clearLoadingTimeout) {
-            clearLoadingTimeout();
-        }
-    }, [isEmpty, startLoadingTimeout, clearLoadingTimeout]);
-    const finalColors = useMemo(() => {
-        return shuffleColors(DEFAULT_COLORS, labels === null || labels === void 0 ? void 0 : labels.length);
-    }, [labels.length]);
-    if (isEmpty) {
-        return createLoadingComponent();
-    }
-    const transformedData = {
-        labels: labels,
-        datasets: [
-            {
-                label: chartTitle || "Data",
-                data: values,
-                backgroundColor: finalColors,
-            },
-        ],
-    };
-    return (jsxRuntimeExports.jsx(BarChart, { orientation: orientation, title: chartTitle, data: transformedData, units: units }));
-}
 function SeriesBarChartComponent({ orientation, seriesBarChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
     const hasCalledRef = useRef(false);
@@ -36280,7 +36081,7 @@ function SeriesBarChartComponent({ orientation, seriesBarChartState, styles, app
     const isEmpty = labels.length === 0 || series.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36331,7 +36132,7 @@ function SeriesLineChartComponent({ orientation, seriesLineChartState, styles, a
     const isEmpty = labels.length === 0 || series.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36390,7 +36191,7 @@ function PieChartComponent({ pieChartState, styles, appendMessage, query, isFirs
     const isEmpty = labels.length === 0 || values.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36434,63 +36235,6 @@ function PieChartComponent({ pieChartState, styles, appendMessage, query, isFirs
     };
     return (jsxRuntimeExports.jsx(PieChart, { title: chartTitle, data: transformedData, units: units }));
 }
-function LineChartComponent({ lineChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
-    const hasCalledRef = useRef(false);
-    const labels = ((_b = (_a = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.labels) || ((_e = (_d = (_c = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _c === void 0 ? void 0 : _c.bar_chart_data) === null || _d === void 0 ? void 0 : _d.data) === null || _e === void 0 ? void 0 : _e.labels) || [];
-    const values = ((_g = (_f = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.values) || ((_k = (_j = (_h = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _h === void 0 ? void 0 : _h.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) || [];
-    const chartTitle = ((_l = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _l === void 0 ? void 0 : _l.title) || ((_o = (_m = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _m === void 0 ? void 0 : _m.bar_chart_data) === null || _o === void 0 ? void 0 : _o.title) || "";
-    const units = ((_p = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _p === void 0 ? void 0 : _p.units) || ((_r = (_q = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _q === void 0 ? void 0 : _q.bar_chart_data) === null || _r === void 0 ? void 0 : _r.units) || "";
-    values.length > 0 ? Math.max(...values) : 0;
-    const isEmpty = labels.length === 0 || values.length === 0;
-    const handleRefresh = async () => {
-        if (query && widgetBackendUrl && widgetId) {
-            await clearChat(widgetBackendUrl, widgetId);
-            appendMessage(new TextMessage({
-                content: `${query} and render data on a bar chart`,
-                role: Role.User,
-            }));
-        }
-    };
-    useEffect(() => {
-        if (isEmpty && query && isFirstLoad && !(lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.agent_message) && !hasCalledRef.current) {
-            hasCalledRef.current = true;
-            handleRefresh();
-        }
-    }, [isEmpty, query, isFirstLoad, lineChartState]);
-    // Start timeout when chart is empty and loading
-    useEffect(() => {
-        if (isEmpty && startLoadingTimeout && !(lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.agent_message)) {
-            startLoadingTimeout();
-        }
-        else if (!isEmpty && clearLoadingTimeout) {
-            clearLoadingTimeout();
-        }
-    }, [isEmpty, startLoadingTimeout, clearLoadingTimeout]);
-    const lineColor = useMemo(() => {
-        return LINE_COLORS[Math.floor(Math.random() * LINE_COLORS.length)];
-    }, [labels.length]);
-    if (isEmpty) {
-        return createLoadingComponent();
-    }
-    const transformedData = {
-        labels: labels,
-        datasets: [
-            {
-                label: chartTitle || "Data",
-                data: values,
-                borderColor: lineColor,
-                backgroundColor: "rgba(96, 165, 250, 0.1)",
-                borderWidth: 2,
-                pointBackgroundColor: lineColor,
-                pointBorderColor: lineColor,
-                pointRadius: 4,
-                fill: true,
-            },
-        ],
-    };
-    return (jsxRuntimeExports.jsx(LineChart, { title: chartTitle, data: transformedData, units: units }));
-}
 function DataGridComponent({ dataGridState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o;
     const hasCalledRef = useRef(false);
@@ -36500,7 +36244,7 @@ function DataGridComponent({ dataGridState, styles, appendMessage, query, isFirs
     const isEmpty = columns.length === 0 || rows.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36544,7 +36288,7 @@ function SummaryComponent({ summaryState, styles, appendMessage, query, isFirstL
     const isEmpty = !content || content.trim() === "";
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36578,7 +36322,7 @@ function SummaryComponent({ summaryState, styles, appendMessage, query, isFirstL
     return (jsxRuntimeExports.jsx(SummaryWidget, { title: title, data: data, metadata: metadata, className: "" }));
 }
 function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds, widgetBackendUrl, datasetId }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18, _19, _20, _21, _22, _23, _24, _25, _26, _27, _28, _29, _30, _31;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18;
     const agentType = (_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentType;
     const orientation = (_b = widget.config) === null || _b === void 0 ? void 0 : _b.orientation;
     const isFirstLoad = (_c = widget.config) === null || _c === void 0 ? void 0 : _c.isFirstLoad;
@@ -36587,6 +36331,8 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
     const [isTimeoutTriggered, setIsTimeoutTriggered] = useState(false);
     const [apiCallCount, setApiCallCount] = useState(0);
     const [hasTimeoutError, setHasTimeoutError] = useState(false);
+    const [isAgentTriggered, setIsAgentTriggered] = useState(false);
+    const pollingIntervalRef = useRef(null);
     useEffect(() => {
         setThreadId(widget.id);
     }, [widget.id, setThreadId]);
@@ -36633,38 +36379,75 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
             timeoutRef.current = null;
         }
     }, []);
-    // Function to clear chat and reset chart state
-    useCallback(async (widgetId) => {
-        if (!widgetBackendUrl)
+    // Function to start polling for data availability
+    const startPollingForData = useCallback(() => {
+        var _a;
+        if (!widgetBackendUrl || !((_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentName))
             return;
-        try {
-            await fetch(`${widgetBackendUrl}/api/clear-chat`, {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({
-                    session_id: widgetId,
-                    delete_state: true
-                }),
-            });
-            setChartState(createInitialChartState(agentType || '', widgetIds, datasetId));
-            setApiCallCount(0);
-            setHasTimeoutError(false);
-        }
-        catch (error) {
-            console.error('Failed to clear chat and state:', error);
-        }
-    }, [widgetBackendUrl, setChartState, agentType, widgetIds]);
-    // Clean up timeout on unmount
+        setIsAgentTriggered(true);
+        setApiCallCount(0);
+        setHasTimeoutError(false);
+        // Clear any existing polling interval
+        if (pollingIntervalRef.current) {
+            clearInterval(pollingIntervalRef.current);
+        }
+        // Start polling immediately, then every CHART_REFRESH_TIMEOUT
+        const pollForData = async () => {
+            var _a, _b, _c;
+            if (apiCallCount >= 10) {
+                setHasTimeoutError(true);
+                setIsAgentTriggered(false);
+                if (pollingIntervalRef.current) {
+                    clearInterval(pollingIntervalRef.current);
+                    pollingIntervalRef.current = null;
+                }
+                return;
+            }
+            setApiCallCount(prev => prev + 1);
+            const agentName = ((_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentName) || "adk-construction-project-agent";
+            const apiResponse = await loadAgentState(widgetBackendUrl, widget.id, agentName);
+            if (apiResponse && ((_c = (_b = apiResponse.data) === null || _b === void 0 ? void 0 : _b.loadAgentState) === null || _c === void 0 ? void 0 : _c.state)) {
+                parseAndUpdateChartState(apiResponse, setChartState);
+                setIsAgentTriggered(false);
+                if (pollingIntervalRef.current) {
+                    clearInterval(pollingIntervalRef.current);
+                    pollingIntervalRef.current = null;
+                }
+            }
+        };
+        // Poll immediately
+        pollForData();
+        // Then poll every CHART_REFRESH_TIMEOUT
+        pollingIntervalRef.current = setInterval(pollForData, CHART_REFRESH_TIMEOUT);
+        // Set a maximum timeout to stop polling after 10 attempts
+        setTimeout(() => {
+            if (pollingIntervalRef.current) {
+                clearInterval(pollingIntervalRef.current);
+                pollingIntervalRef.current = null;
+                setIsAgentTriggered(false);
+                if (apiCallCount >= 10) {
+                    setHasTimeoutError(true);
+                }
+            }
+        }, CHART_REFRESH_TIMEOUT * 10);
+    }, [widgetBackendUrl, (_f = widget.config) === null || _f === void 0 ? void 0 : _f.agentName, widget.id, apiCallCount, setChartState]);
+    // Clean up timeout and polling on unmount
     useEffect(() => {
         return () => {
             if (timeoutRef.current) {
                 clearTimeout(timeoutRef.current);
             }
+            if (pollingIntervalRef.current) {
+                clearInterval(pollingIntervalRef.current);
+            }
         };
     }, []);
-    const { appendMessage, reset } = useCopilotChat();
+    const { appendMessage: originalAppendMessage, reset } = useCopilotChat();
+    // Wrapper function for appendMessage that triggers polling
+    const appendMessage = useCallback((message) => {
+        startPollingForData();
+        return originalAppendMessage(message);
+    }, [originalAppendMessage, startPollingForData]);
     // Register the reset function with the parent component
     useEffect(() => {
         if (onResetReady && reset && widget.id) {
@@ -36726,17 +36509,16 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
     }, [widget.id, appendMessage, agentType, setChartState]);
     return (jsxRuntimeExports.jsxs("div", { className: cn("flex flex-col h-full"), children: [showHeader && (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-between pb-2 border-b", children: jsxRuntimeExports.jsxs("div", { className: "flex items-center gap-2", children: [jsxRuntimeExports.jsx(Bot, { className: "h-4 w-4" }), jsxRuntimeExports.jsx("h3", { className: "text-sm font-medium", children: widget.title })] }) })), jsxRuntimeExports.jsx("div", { className: "flex-1 h-full", children: (chartState === null || chartState === void 0 ? void 0 : chartState.agent_message) && agentType !== "chatbot" && (
                 // Check if data is empty based on agent type
-                (agentType === "Bar Chart Agent" && (!((_h = (_g = (_f = chartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.labels) === null || _h === void 0 ? void 0 : _h.length) || !((_l = (_k = (_j = chartState.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) === null || _l === void 0 ? void 0 : _l.length))) ||
-                    (agentType === "Series Bar Chart Agent" && (!((_p = (_o = (_m = chartState.series_bar_chart_data) === null || _m === void 0 ? void 0 : _m.data) === null || _o === void 0 ? void 0 : _o.labels) === null || _p === void 0 ? void 0 : _p.length) || !((_s = (_r = (_q = chartState.series_bar_chart_data) === null || _q === void 0 ? void 0 : _q.data) === null || _r === void 0 ? void 0 : _r.series) === null || _s === void 0 ? void 0 : _s.length))) ||
-                    (agentType === "Series Line Chart Agent" && (!((_v = (_u = (_t = chartState.series_bar_chart_data) === null || _t === void 0 ? void 0 : _t.data) === null || _u === void 0 ? void 0 : _u.labels) === null || _v === void 0 ? void 0 : _v.length) || !((_y = (_x = (_w = chartState.series_bar_chart_data) === null || _w === void 0 ? void 0 : _w.data) === null || _x === void 0 ? void 0 : _x.series) === null || _y === void 0 ? void 0 : _y.length))) ||
-                    (agentType === "Pie Chart Agent" && (!((_1 = (_0 = (_z = chartState.pie_chart_data) === null || _z === void 0 ? void 0 : _z.data) === null || _0 === void 0 ? void 0 : _0.labels) === null || _1 === void 0 ? void 0 : _1.length) || !((_4 = (_3 = (_2 = chartState.pie_chart_data) === null || _2 === void 0 ? void 0 : _2.data) === null || _3 === void 0 ? void 0 : _3.values) === null || _4 === void 0 ? void 0 : _4.length))) ||
-                    (agentType === "Line Chart Agent" && (!((_7 = (_6 = (_5 = chartState.bar_chart_data) === null || _5 === void 0 ? void 0 : _5.data) === null || _6 === void 0 ? void 0 : _6.labels) === null || _7 === void 0 ? void 0 : _7.length) || !((_10 = (_9 = (_8 = chartState.bar_chart_data) === null || _8 === void 0 ? void 0 : _8.data) === null || _9 === void 0 ? void 0 : _9.values) === null || _10 === void 0 ? void 0 : _10.length))) ||
-                    (agentType === "Data Grid Agent" && (!((_13 = (_12 = (_11 = chartState.matrix_grid_data) === null || _11 === void 0 ? void 0 : _11.data) === null || _12 === void 0 ? void 0 : _12.headers) === null || _13 === void 0 ? void 0 : _13.length) || !((_16 = (_15 = (_14 = chartState.matrix_grid_data) === null || _14 === void 0 ? void 0 : _14.data) === null || _15 === void 0 ? void 0 : _15.rows) === null || _16 === void 0 ? void 0 : _16.length))) ||
-                    (agentType === "Summary Agent" && (!((_18 = (_17 = chartState.summary_data) === null || _17 === void 0 ? void 0 : _17.data) === null || _18 === void 0 ? void 0 : _18.content) || ((_21 = (_20 = (_19 = chartState.summary_data) === null || _19 === void 0 ? void 0 : _19.data) === null || _20 === void 0 ? void 0 : _20.content) === null || _21 === void 0 ? void 0 : _21.trim()) === ""))) ? (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-center h-full p-4", children: jsxRuntimeExports.jsx("div", { className: "text-center max-w-md", children: jsxRuntimeExports.jsx("p", { className: "text-sm text-gray-700", children: chartState.agent_message }) }) })) : agentType === "Bar Chart Agent" ? (jsxRuntimeExports.jsx(BarChartComponent, { barChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_22 = widget.config) === null || _22 === void 0 ? void 0 : _22.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Bar Chart Agent" ? (jsxRuntimeExports.jsx(SeriesBarChartComponent, { seriesBarChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_23 = widget.config) === null || _23 === void 0 ? void 0 : _23.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Line Chart Agent" ? (jsxRuntimeExports.jsx(SeriesLineChartComponent, { seriesLineChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_24 = widget.config) === null || _24 === void 0 ? void 0 : _24.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Pie Chart Agent" ? (jsxRuntimeExports.jsx(PieChartComponent, { pieChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_25 = widget.config) === null || _25 === void 0 ? void 0 : _25.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Line Chart Agent" ? (jsxRuntimeExports.jsx(LineChartComponent, { lineChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_26 = widget.config) === null || _26 === void 0 ? void 0 : _26.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Data Grid Agent" ? (jsxRuntimeExports.jsx(DataGridComponent, { dataGridState: chartState, styles: styles, appendMessage: appendMessage, query: (_27 = widget.config) === null || _27 === void 0 ? void 0 : _27.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Summary Agent" ? (jsxRuntimeExports.jsx(SummaryComponent, { summaryState: chartState, styles: styles, appendMessage: appendMessage, query: (_28 = widget.config) === null || _28 === void 0 ? void 0 : _28.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, widget_ids: widgetIds, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout, setChartState: setChartState })) : (jsxRuntimeExports.jsx(CopilotChat, { className: "h-full text-xs [&_.copilot-chat-message]:text-xs [&_.copilot-chat-input]:text-xs", labels: {
-                        title: ((_29 = widget.config) === null || _29 === void 0 ? void 0 : _29.copilotTitle) || widget.title,
-                        initial: ((_30 = widget.config) === null || _30 === void 0 ? void 0 : _30.copilotInitialMessage) || ((_31 = widget.config) === null || _31 === void 0 ? void 0 : _31.placeholder) || "How can I help you today?"
+                (agentType === "Series Bar Chart Agent" && (!((_j = (_h = (_g = chartState.series_bar_chart_data) === null || _g === void 0 ? void 0 : _g.data) === null || _h === void 0 ? void 0 : _h.labels) === null || _j === void 0 ? void 0 : _j.length) || !((_m = (_l = (_k = chartState.series_bar_chart_data) === null || _k === void 0 ? void 0 : _k.data) === null || _l === void 0 ? void 0 : _l.series) === null || _m === void 0 ? void 0 : _m.length))) ||
+                    (agentType === "Series Line Chart Agent" && (!((_q = (_p = (_o = chartState.series_bar_chart_data) === null || _o === void 0 ? void 0 : _o.data) === null || _p === void 0 ? void 0 : _p.labels) === null || _q === void 0 ? void 0 : _q.length) || !((_t = (_s = (_r = chartState.series_bar_chart_data) === null || _r === void 0 ? void 0 : _r.data) === null || _s === void 0 ? void 0 : _s.series) === null || _t === void 0 ? void 0 : _t.length))) ||
+                    (agentType === "Pie Chart Agent" && (!((_w = (_v = (_u = chartState.pie_chart_data) === null || _u === void 0 ? void 0 : _u.data) === null || _v === void 0 ? void 0 : _v.labels) === null || _w === void 0 ? void 0 : _w.length) || !((_z = (_y = (_x = chartState.pie_chart_data) === null || _x === void 0 ? void 0 : _x.data) === null || _y === void 0 ? void 0 : _y.values) === null || _z === void 0 ? void 0 : _z.length))) ||
+                    (agentType === "Data Grid Agent" && (!((_2 = (_1 = (_0 = chartState.matrix_grid_data) === null || _0 === void 0 ? void 0 : _0.data) === null || _1 === void 0 ? void 0 : _1.headers) === null || _2 === void 0 ? void 0 : _2.length) || !((_5 = (_4 = (_3 = chartState.matrix_grid_data) === null || _3 === void 0 ? void 0 : _3.data) === null || _4 === void 0 ? void 0 : _4.rows) === null || _5 === void 0 ? void 0 : _5.length))) ||
+                    (agentType === "Summary Agent" && (!((_7 = (_6 = chartState.summary_data) === null || _6 === void 0 ? void 0 : _6.data) === null || _7 === void 0 ? void 0 : _7.content) || ((_10 = (_9 = (_8 = chartState.summary_data) === null || _8 === void 0 ? void 0 : _8.data) === null || _9 === void 0 ? void 0 : _9.content) === null || _10 === void 0 ? void 0 : _10.trim()) === ""))) ? (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-center h-full p-4", children: jsxRuntimeExports.jsx("div", { className: "text-center max-w-md", children: jsxRuntimeExports.jsx("p", { className: "text-sm text-gray-700", children: chartState.agent_message }) }) })) : isAgentTriggered ? (createLoadingComponent("Agent processing your request...")) : agentType === "Series Bar Chart Agent" ? (jsxRuntimeExports.jsx(SeriesBarChartComponent, { seriesBarChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_11 = widget.config) === null || _11 === void 0 ? void 0 : _11.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Line Chart Agent" ? (jsxRuntimeExports.jsx(SeriesLineChartComponent, { seriesLineChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_12 = widget.config) === null || _12 === void 0 ? void 0 : _12.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Pie Chart Agent" ? (jsxRuntimeExports.jsx(PieChartComponent, { pieChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_13 = widget.config) === null || _13 === void 0 ? void 0 : _13.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Data Grid Agent" ? (jsxRuntimeExports.jsx(DataGridComponent, { dataGridState: chartState, styles: styles, appendMessage: appendMessage, query: (_14 = widget.config) === null || _14 === void 0 ? void 0 : _14.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Summary Agent" ? (jsxRuntimeExports.jsx(SummaryComponent, { summaryState: chartState, styles: styles, appendMessage: appendMessage, query: (_15 = widget.config) === null || _15 === void 0 ? void 0 : _15.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, widget_ids: widgetIds, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout, setChartState: setChartState })) : (jsxRuntimeExports.jsx(CopilotChat, { className: "h-full text-xs [&_.copilot-chat-message]:text-xs [&_.copilot-chat-input]:text-xs", labels: {
+                        title: ((_16 = widget.config) === null || _16 === void 0 ? void 0 : _16.copilotTitle) || widget.title,
+                        initial: ((_17 = widget.config) === null || _17 === void 0 ? void 0 : _17.copilotInitialMessage) || ((_18 = widget.config) === null || _18 === void 0 ? void 0 : _18.placeholder) || "How can I help you today?"
                     }, onSubmitMessage: () => {
                         setChartState(prevState => (Object.assign(Object.assign({}, prevState), { widget_ids: widgetIds, dataset_id: datasetId })));
+                        startPollingForData();
                     } })) })] }));
 }
 function AgentWidget({ widget, showHeader = true, widgetBackendUrl, onResetReady, widgetIds, datasetId, }) {
@@ -36820,14 +36602,14 @@ function AddWidgetDialog({ isOpen, onClose, addWidgetFn, defaultAgentName = "adk
         {
             type: 'series-bar-chart',
             title: 'Bar Chart Widget',
-            icon: BarChart$1,
+            icon: BarChart,
             description: 'Create interactive series bar charts with custom queries',
             configRequired: true
         },
         {
             type: 'series-line-chart',
             title: 'Line Chart Widget',
-            icon: LineChart$1,
+            icon: LineChart,
             description: 'Create interactive series line charts with custom queries',
             configRequired: true
         },
@@ -37003,13 +36785,13 @@ const IconMap = {
     search: Search,
     filter: Filter,
     results: LayoutGrid,
-    analytics: BarChart$1,
+    analytics: BarChart,
     custom: LayoutGrid,
     header: Layout,
     footer: Layout,
     text: Type,
     agent: Bot,
-    'bar-chart': BarChart$1,
+    'bar-chart': BarChart,
     'pie-chart': PieChart$1,
     'chatbot': Bot,
 };

package/dist/index.js

@@ -15075,78 +15075,6 @@ const Line = /* #__PURE__ */ createTypedChart('line', LineController);
 const Bar = /* #__PURE__ */ createTypedChart('bar', BarController);
 const Pie = /* #__PURE__ */ createTypedChart('pie', PieController);
 
-Chart$1.register(CategoryScale, LinearScale, BarElement, plugin_title, plugin_tooltip, plugin_legend);
-function BarChart({ orientation, title, data, options, className, units }) {
-    const formatValue = (value) => {
-        const numValue = typeof value === 'string' ? parseFloat(value) : value;
-        if (units === '$' && !isNaN(numValue)) {
-            return numValue.toLocaleString();
-        }
-        return value;
-    };
-    const defaultOptions = {
-        responsive: true,
-        maintainAspectRatio: false,
-        indexAxis: orientation === "horizontal" ? "y" : "x",
-        layout: {
-            padding: {
-                bottom: 0,
-                top: 0,
-                left: 0,
-                right: 0,
-            },
-        },
-        plugins: {
-            legend: {
-                display: false,
-                labels: {
-                    boxWidth: 5,
-                    boxHeight: 5,
-                },
-                position: "bottom"
-            },
-            title: {
-                display: true,
-                text: title,
-                font: {
-                    size: 18,
-                    weight: "bold",
-                },
-                color: "#0F172A"
-            },
-            tooltip: {
-                callbacks: {
-                    label: function (context) {
-                        const value = context.parsed.y || context.parsed.x;
-                        return units ? `${context.dataset.label}: ${formatValue(value)} ${units}` : `${context.dataset.label}: ${value}`;
-                    }
-                }
-            },
-        },
-        scales: {
-            x: {
-                ticks: {
-                    autoSkip: false,
-                    maxRotation: 45,
-                    minRotation: 45,
-                    font: {
-                        size: 10
-                    }
-                },
-            },
-            y: {
-                beginAtZero: true,
-                ticks: {
-                    callback: function (value) {
-                        return units ? `${formatValue(value)} ${units}` : value;
-                    }
-                }
-            },
-        },
-    };
-    return (jsxRuntimeExports.jsx(Bar, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions }));
-}
-
 Chart$1.register(CategoryScale, LinearScale, BarElement, plugin_title, plugin_tooltip, plugin_legend);
 function SeriesChart({ orientation, title, data, options, className, units }) {
     const formatValue = (value) => {
@@ -15335,70 +15263,6 @@ const PieChart = (props) => {
     return jsxRuntimeExports.jsx(Pie, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions });
 };
 
-Chart$1.register(CategoryScale, LinearScale, PointElement, LineElement, plugin_title, plugin_tooltip, plugin_legend);
-function LineChart({ title, data, options, className, units }) {
-    const formatValue = (value) => {
-        const numValue = typeof value === 'string' ? parseFloat(value) : value;
-        if (units === '$' && !isNaN(numValue)) {
-            return numValue.toLocaleString();
-        }
-        return value;
-    };
-    const defaultOptions = {
-        responsive: true,
-        maintainAspectRatio: false,
-        layout: {
-            padding: {
-                bottom: 0,
-                top: 0,
-                left: 0,
-                right: 0,
-            },
-        },
-        plugins: {
-            legend: {
-                display: false,
-                labels: {
-                    boxWidth: 5,
-                    boxHeight: 5,
-                },
-                position: "bottom"
-            },
-            title: {
-                display: true,
-                text: title,
-                font: {
-                    size: 18,
-                    weight: "bold",
-                },
-                color: "#0F172A"
-            },
-            tooltip: {
-                callbacks: {
-                    label: function (context) {
-                        const value = context.parsed.y;
-                        return units ? `${context.dataset.label}: ${formatValue(value)} ${units}` : `${context.dataset.label}: ${value}`;
-                    }
-                }
-            },
-        },
-        scales: {
-            x: {
-                ticks: { autoSkip: true, maxTicksLimit: 8, maxRotation: 45, minRotation: 0 },
-            },
-            y: {
-                beginAtZero: true,
-                ticks: {
-                    callback: function (value) {
-                        return units ? `${formatValue(value)} ${units}` : value;
-                    }
-                }
-            },
-        },
-    };
-    return (jsxRuntimeExports.jsx(Line, { data: data, options: options !== null && options !== void 0 ? options : defaultOptions }));
-}
-
 function DataGrid({ title, data, className }) {
     const { columns, rows } = data;
     if (!columns || columns.length === 0 || !rows || rows.length === 0) {
@@ -36099,24 +35963,12 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
     const baseState = { dataset_id: datasetId || "home_generation_dataset" };
     switch (agentType) {
         case "Pie Chart Agent":
-            return {
-                pie_chart_data: {
+            return Object.assign({ pie_chart_data: {
                     title: "",
                     type: "pie",
                     chart_type: "financial",
                     data: { labels: [], values: [], percentages: [], total: 0 },
                     metadata: { categories: 0, largest_category: "", largest_value: 0, largest_percentage: 0 }
-                }
-            };
-        case "Line Chart Agent":
-        case "Bar Chart Agent":
-            return Object.assign({ bar_chart_data: {
-                    title: "",
-                    type: agentType === "Line Chart Agent" ? "line" : "bar",
-                    chart_type: "financial",
-                    orientation: agentType === "Line Chart Agent" ? "horizontal" : "vertical",
-                    data: { labels: [], values: [], total: 0, average: 0 },
-                    metadata: { categories: 0, highest_category: "", highest_value: 0, lowest_category: "", lowest_value: 0, range: 0 }
                 } }, baseState);
         case "Series Bar Chart Agent":
         case "Series Line Chart Agent":
@@ -36125,6 +35977,7 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
                     type: "series_bar",
                     chart_type: "financial",
                     orientation: "vertical",
+                    units: "",
                     data: { labels: [], series: [], total: 0, average: 0 },
                     metadata: {
                         categories: 0, series_count: 0, highest_category: "", highest_value: 0, highest_series: "",
@@ -36147,7 +36000,7 @@ const createInitialChartState = (agentType, widgetIds, datasetId) => {
                     metadata: { created_at: "", content_type: "text", is_multiline: false }
                 } }, baseState);
         default:
-            return { widget_ids: widgetIds };
+            return Object.assign({ widget_ids: widgetIds }, baseState);
     }
 };
 const loadAgentState = async (widgetBackendUrl, threadId, agentName) => {
@@ -36245,58 +36098,6 @@ const getStyleValues = (styles = {}) => {
         textAlign: alignment,
     };
 };
-function BarChartComponent({ orientation, barChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
-    const hasCalledRef = React.useRef(false);
-    const labels = ((_b = (_a = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.labels) || ((_e = (_d = (_c = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _c === void 0 ? void 0 : _c.bar_chart_data) === null || _d === void 0 ? void 0 : _d.data) === null || _e === void 0 ? void 0 : _e.labels) || [];
-    const values = ((_g = (_f = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.values) || ((_k = (_j = (_h = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _h === void 0 ? void 0 : _h.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) || [];
-    const chartTitle = ((_l = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _l === void 0 ? void 0 : _l.title) || ((_o = (_m = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _m === void 0 ? void 0 : _m.bar_chart_data) === null || _o === void 0 ? void 0 : _o.title) || "";
-    const units = ((_p = barChartState === null || barChartState === void 0 ? void 0 : barChartState.bar_chart_data) === null || _p === void 0 ? void 0 : _p.units) || ((_r = (_q = barChartState === null || barChartState === void 0 ? void 0 : barChartState.state) === null || _q === void 0 ? void 0 : _q.bar_chart_data) === null || _r === void 0 ? void 0 : _r.units) || "";
-    values.length > 0 ? Math.max(...values) : 0;
-    const isEmpty = labels.length === 0 || values.length === 0;
-    const handleRefresh = async () => {
-        if (query && widgetBackendUrl && widgetId) {
-            await clearChat(widgetBackendUrl, widgetId);
-            appendMessage(new TextMessage({
-                content: `${query} and render data on a bar chart`,
-                role: Role.User,
-            }));
-        }
-    };
-    React.useEffect(() => {
-        if (isEmpty && query && isFirstLoad && !(barChartState === null || barChartState === void 0 ? void 0 : barChartState.agent_message) && !hasCalledRef.current) {
-            // console.log('calling append message')
-            hasCalledRef.current = true;
-            handleRefresh();
-        }
-    }, [isEmpty, query, isFirstLoad, barChartState]);
-    // Start timeout when chart is empty and loading
-    React.useEffect(() => {
-        if (isEmpty && startLoadingTimeout && !(barChartState === null || barChartState === void 0 ? void 0 : barChartState.agent_message)) {
-            startLoadingTimeout();
-        }
-        else if (!isEmpty && clearLoadingTimeout) {
-            clearLoadingTimeout();
-        }
-    }, [isEmpty, startLoadingTimeout, clearLoadingTimeout]);
-    const finalColors = React.useMemo(() => {
-        return shuffleColors(DEFAULT_COLORS, labels === null || labels === void 0 ? void 0 : labels.length);
-    }, [labels.length]);
-    if (isEmpty) {
-        return createLoadingComponent();
-    }
-    const transformedData = {
-        labels: labels,
-        datasets: [
-            {
-                label: chartTitle || "Data",
-                data: values,
-                backgroundColor: finalColors,
-            },
-        ],
-    };
-    return (jsxRuntimeExports.jsx(BarChart, { orientation: orientation, title: chartTitle, data: transformedData, units: units }));
-}
 function SeriesBarChartComponent({ orientation, seriesBarChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
     const hasCalledRef = React.useRef(false);
@@ -36307,7 +36108,7 @@ function SeriesBarChartComponent({ orientation, seriesBarChartState, styles, app
     const isEmpty = labels.length === 0 || series.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36358,7 +36159,7 @@ function SeriesLineChartComponent({ orientation, seriesLineChartState, styles, a
     const isEmpty = labels.length === 0 || series.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36417,7 +36218,7 @@ function PieChartComponent({ pieChartState, styles, appendMessage, query, isFirs
     const isEmpty = labels.length === 0 || values.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36461,63 +36262,6 @@ function PieChartComponent({ pieChartState, styles, appendMessage, query, isFirs
     };
     return (jsxRuntimeExports.jsx(PieChart, { title: chartTitle, data: transformedData, units: units }));
 }
-function LineChartComponent({ lineChartState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r;
-    const hasCalledRef = React.useRef(false);
-    const labels = ((_b = (_a = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.labels) || ((_e = (_d = (_c = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _c === void 0 ? void 0 : _c.bar_chart_data) === null || _d === void 0 ? void 0 : _d.data) === null || _e === void 0 ? void 0 : _e.labels) || [];
-    const values = ((_g = (_f = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.values) || ((_k = (_j = (_h = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _h === void 0 ? void 0 : _h.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) || [];
-    const chartTitle = ((_l = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _l === void 0 ? void 0 : _l.title) || ((_o = (_m = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _m === void 0 ? void 0 : _m.bar_chart_data) === null || _o === void 0 ? void 0 : _o.title) || "";
-    const units = ((_p = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.bar_chart_data) === null || _p === void 0 ? void 0 : _p.units) || ((_r = (_q = lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.state) === null || _q === void 0 ? void 0 : _q.bar_chart_data) === null || _r === void 0 ? void 0 : _r.units) || "";
-    values.length > 0 ? Math.max(...values) : 0;
-    const isEmpty = labels.length === 0 || values.length === 0;
-    const handleRefresh = async () => {
-        if (query && widgetBackendUrl && widgetId) {
-            await clearChat(widgetBackendUrl, widgetId);
-            appendMessage(new TextMessage({
-                content: `${query} and render data on a bar chart`,
-                role: Role.User,
-            }));
-        }
-    };
-    React.useEffect(() => {
-        if (isEmpty && query && isFirstLoad && !(lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.agent_message) && !hasCalledRef.current) {
-            hasCalledRef.current = true;
-            handleRefresh();
-        }
-    }, [isEmpty, query, isFirstLoad, lineChartState]);
-    // Start timeout when chart is empty and loading
-    React.useEffect(() => {
-        if (isEmpty && startLoadingTimeout && !(lineChartState === null || lineChartState === void 0 ? void 0 : lineChartState.agent_message)) {
-            startLoadingTimeout();
-        }
-        else if (!isEmpty && clearLoadingTimeout) {
-            clearLoadingTimeout();
-        }
-    }, [isEmpty, startLoadingTimeout, clearLoadingTimeout]);
-    const lineColor = React.useMemo(() => {
-        return LINE_COLORS[Math.floor(Math.random() * LINE_COLORS.length)];
-    }, [labels.length]);
-    if (isEmpty) {
-        return createLoadingComponent();
-    }
-    const transformedData = {
-        labels: labels,
-        datasets: [
-            {
-                label: chartTitle || "Data",
-                data: values,
-                borderColor: lineColor,
-                backgroundColor: "rgba(96, 165, 250, 0.1)",
-                borderWidth: 2,
-                pointBackgroundColor: lineColor,
-                pointBorderColor: lineColor,
-                pointRadius: 4,
-                fill: true,
-            },
-        ],
-    };
-    return (jsxRuntimeExports.jsx(LineChart, { title: chartTitle, data: transformedData, units: units }));
-}
 function DataGridComponent({ dataGridState, styles, appendMessage, query, isFirstLoad, widgetBackendUrl, widgetId, startLoadingTimeout, clearLoadingTimeout }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o;
     const hasCalledRef = React.useRef(false);
@@ -36527,7 +36271,7 @@ function DataGridComponent({ dataGridState, styles, appendMessage, query, isFirs
     const isEmpty = columns.length === 0 || rows.length === 0;
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36571,7 +36315,7 @@ function SummaryComponent({ summaryState, styles, appendMessage, query, isFirstL
     const isEmpty = !content || content.trim() === "";
     const handleRefresh = async () => {
         if (query) {
-            await clearChat();
+            await clearChat(widgetBackendUrl, widgetId);
             // Send trigger event to clear chart state
             console.log('Dispatching clearChartState event for widgetId:', widgetId);
             window.dispatchEvent(new CustomEvent('clearChartState', { detail: { widgetId } }));
@@ -36605,7 +36349,7 @@ function SummaryComponent({ summaryState, styles, appendMessage, query, isFirstL
     return (jsxRuntimeExports.jsx(SummaryWidget, { title: title, data: data, metadata: metadata, className: "" }));
 }
 function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds, widgetBackendUrl, datasetId }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18, _19, _20, _21, _22, _23, _24, _25, _26, _27, _28, _29, _30, _31;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, _16, _17, _18;
     const agentType = (_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentType;
     const orientation = (_b = widget.config) === null || _b === void 0 ? void 0 : _b.orientation;
     const isFirstLoad = (_c = widget.config) === null || _c === void 0 ? void 0 : _c.isFirstLoad;
@@ -36614,6 +36358,8 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
     const [isTimeoutTriggered, setIsTimeoutTriggered] = React.useState(false);
     const [apiCallCount, setApiCallCount] = React.useState(0);
     const [hasTimeoutError, setHasTimeoutError] = React.useState(false);
+    const [isAgentTriggered, setIsAgentTriggered] = React.useState(false);
+    const pollingIntervalRef = React.useRef(null);
     React.useEffect(() => {
         setThreadId(widget.id);
     }, [widget.id, setThreadId]);
@@ -36660,38 +36406,75 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
             timeoutRef.current = null;
         }
     }, []);
-    // Function to clear chat and reset chart state
-    React.useCallback(async (widgetId) => {
-        if (!widgetBackendUrl)
+    // Function to start polling for data availability
+    const startPollingForData = React.useCallback(() => {
+        var _a;
+        if (!widgetBackendUrl || !((_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentName))
             return;
-        try {
-            await fetch(`${widgetBackendUrl}/api/clear-chat`, {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({
-                    session_id: widgetId,
-                    delete_state: true
-                }),
-            });
-            setChartState(createInitialChartState(agentType || '', widgetIds, datasetId));
-            setApiCallCount(0);
-            setHasTimeoutError(false);
-        }
-        catch (error) {
-            console.error('Failed to clear chat and state:', error);
-        }
-    }, [widgetBackendUrl, setChartState, agentType, widgetIds]);
-    // Clean up timeout on unmount
+        setIsAgentTriggered(true);
+        setApiCallCount(0);
+        setHasTimeoutError(false);
+        // Clear any existing polling interval
+        if (pollingIntervalRef.current) {
+            clearInterval(pollingIntervalRef.current);
+        }
+        // Start polling immediately, then every CHART_REFRESH_TIMEOUT
+        const pollForData = async () => {
+            var _a, _b, _c;
+            if (apiCallCount >= 10) {
+                setHasTimeoutError(true);
+                setIsAgentTriggered(false);
+                if (pollingIntervalRef.current) {
+                    clearInterval(pollingIntervalRef.current);
+                    pollingIntervalRef.current = null;
+                }
+                return;
+            }
+            setApiCallCount(prev => prev + 1);
+            const agentName = ((_a = widget.config) === null || _a === void 0 ? void 0 : _a.agentName) || "adk-construction-project-agent";
+            const apiResponse = await loadAgentState(widgetBackendUrl, widget.id, agentName);
+            if (apiResponse && ((_c = (_b = apiResponse.data) === null || _b === void 0 ? void 0 : _b.loadAgentState) === null || _c === void 0 ? void 0 : _c.state)) {
+                parseAndUpdateChartState(apiResponse, setChartState);
+                setIsAgentTriggered(false);
+                if (pollingIntervalRef.current) {
+                    clearInterval(pollingIntervalRef.current);
+                    pollingIntervalRef.current = null;
+                }
+            }
+        };
+        // Poll immediately
+        pollForData();
+        // Then poll every CHART_REFRESH_TIMEOUT
+        pollingIntervalRef.current = setInterval(pollForData, CHART_REFRESH_TIMEOUT);
+        // Set a maximum timeout to stop polling after 10 attempts
+        setTimeout(() => {
+            if (pollingIntervalRef.current) {
+                clearInterval(pollingIntervalRef.current);
+                pollingIntervalRef.current = null;
+                setIsAgentTriggered(false);
+                if (apiCallCount >= 10) {
+                    setHasTimeoutError(true);
+                }
+            }
+        }, CHART_REFRESH_TIMEOUT * 10);
+    }, [widgetBackendUrl, (_f = widget.config) === null || _f === void 0 ? void 0 : _f.agentName, widget.id, apiCallCount, setChartState]);
+    // Clean up timeout and polling on unmount
     React.useEffect(() => {
         return () => {
             if (timeoutRef.current) {
                 clearTimeout(timeoutRef.current);
             }
+            if (pollingIntervalRef.current) {
+                clearInterval(pollingIntervalRef.current);
+            }
         };
     }, []);
-    const { appendMessage, reset } = reactCore.useCopilotChat();
+    const { appendMessage: originalAppendMessage, reset } = reactCore.useCopilotChat();
+    // Wrapper function for appendMessage that triggers polling
+    const appendMessage = React.useCallback((message) => {
+        startPollingForData();
+        return originalAppendMessage(message);
+    }, [originalAppendMessage, startPollingForData]);
     // Register the reset function with the parent component
     React.useEffect(() => {
         if (onResetReady && reset && widget.id) {
@@ -36753,17 +36536,16 @@ function CopilotKitAgent({ widget, showHeader, styles, onResetReady, widgetIds,
     }, [widget.id, appendMessage, agentType, setChartState]);
     return (jsxRuntimeExports.jsxs("div", { className: cn("flex flex-col h-full"), children: [showHeader && (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-between pb-2 border-b", children: jsxRuntimeExports.jsxs("div", { className: "flex items-center gap-2", children: [jsxRuntimeExports.jsx(lucideReact.Bot, { className: "h-4 w-4" }), jsxRuntimeExports.jsx("h3", { className: "text-sm font-medium", children: widget.title })] }) })), jsxRuntimeExports.jsx("div", { className: "flex-1 h-full", children: (chartState === null || chartState === void 0 ? void 0 : chartState.agent_message) && agentType !== "chatbot" && (
                 // Check if data is empty based on agent type
-                (agentType === "Bar Chart Agent" && (!((_h = (_g = (_f = chartState.bar_chart_data) === null || _f === void 0 ? void 0 : _f.data) === null || _g === void 0 ? void 0 : _g.labels) === null || _h === void 0 ? void 0 : _h.length) || !((_l = (_k = (_j = chartState.bar_chart_data) === null || _j === void 0 ? void 0 : _j.data) === null || _k === void 0 ? void 0 : _k.values) === null || _l === void 0 ? void 0 : _l.length))) ||
-                    (agentType === "Series Bar Chart Agent" && (!((_p = (_o = (_m = chartState.series_bar_chart_data) === null || _m === void 0 ? void 0 : _m.data) === null || _o === void 0 ? void 0 : _o.labels) === null || _p === void 0 ? void 0 : _p.length) || !((_s = (_r = (_q = chartState.series_bar_chart_data) === null || _q === void 0 ? void 0 : _q.data) === null || _r === void 0 ? void 0 : _r.series) === null || _s === void 0 ? void 0 : _s.length))) ||
-                    (agentType === "Series Line Chart Agent" && (!((_v = (_u = (_t = chartState.series_bar_chart_data) === null || _t === void 0 ? void 0 : _t.data) === null || _u === void 0 ? void 0 : _u.labels) === null || _v === void 0 ? void 0 : _v.length) || !((_y = (_x = (_w = chartState.series_bar_chart_data) === null || _w === void 0 ? void 0 : _w.data) === null || _x === void 0 ? void 0 : _x.series) === null || _y === void 0 ? void 0 : _y.length))) ||
-                    (agentType === "Pie Chart Agent" && (!((_1 = (_0 = (_z = chartState.pie_chart_data) === null || _z === void 0 ? void 0 : _z.data) === null || _0 === void 0 ? void 0 : _0.labels) === null || _1 === void 0 ? void 0 : _1.length) || !((_4 = (_3 = (_2 = chartState.pie_chart_data) === null || _2 === void 0 ? void 0 : _2.data) === null || _3 === void 0 ? void 0 : _3.values) === null || _4 === void 0 ? void 0 : _4.length))) ||
-                    (agentType === "Line Chart Agent" && (!((_7 = (_6 = (_5 = chartState.bar_chart_data) === null || _5 === void 0 ? void 0 : _5.data) === null || _6 === void 0 ? void 0 : _6.labels) === null || _7 === void 0 ? void 0 : _7.length) || !((_10 = (_9 = (_8 = chartState.bar_chart_data) === null || _8 === void 0 ? void 0 : _8.data) === null || _9 === void 0 ? void 0 : _9.values) === null || _10 === void 0 ? void 0 : _10.length))) ||
-                    (agentType === "Data Grid Agent" && (!((_13 = (_12 = (_11 = chartState.matrix_grid_data) === null || _11 === void 0 ? void 0 : _11.data) === null || _12 === void 0 ? void 0 : _12.headers) === null || _13 === void 0 ? void 0 : _13.length) || !((_16 = (_15 = (_14 = chartState.matrix_grid_data) === null || _14 === void 0 ? void 0 : _14.data) === null || _15 === void 0 ? void 0 : _15.rows) === null || _16 === void 0 ? void 0 : _16.length))) ||
-                    (agentType === "Summary Agent" && (!((_18 = (_17 = chartState.summary_data) === null || _17 === void 0 ? void 0 : _17.data) === null || _18 === void 0 ? void 0 : _18.content) || ((_21 = (_20 = (_19 = chartState.summary_data) === null || _19 === void 0 ? void 0 : _19.data) === null || _20 === void 0 ? void 0 : _20.content) === null || _21 === void 0 ? void 0 : _21.trim()) === ""))) ? (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-center h-full p-4", children: jsxRuntimeExports.jsx("div", { className: "text-center max-w-md", children: jsxRuntimeExports.jsx("p", { className: "text-sm text-gray-700", children: chartState.agent_message }) }) })) : agentType === "Bar Chart Agent" ? (jsxRuntimeExports.jsx(BarChartComponent, { barChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_22 = widget.config) === null || _22 === void 0 ? void 0 : _22.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Bar Chart Agent" ? (jsxRuntimeExports.jsx(SeriesBarChartComponent, { seriesBarChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_23 = widget.config) === null || _23 === void 0 ? void 0 : _23.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Line Chart Agent" ? (jsxRuntimeExports.jsx(SeriesLineChartComponent, { seriesLineChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_24 = widget.config) === null || _24 === void 0 ? void 0 : _24.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Pie Chart Agent" ? (jsxRuntimeExports.jsx(PieChartComponent, { pieChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_25 = widget.config) === null || _25 === void 0 ? void 0 : _25.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Line Chart Agent" ? (jsxRuntimeExports.jsx(LineChartComponent, { lineChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_26 = widget.config) === null || _26 === void 0 ? void 0 : _26.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Data Grid Agent" ? (jsxRuntimeExports.jsx(DataGridComponent, { dataGridState: chartState, styles: styles, appendMessage: appendMessage, query: (_27 = widget.config) === null || _27 === void 0 ? void 0 : _27.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Summary Agent" ? (jsxRuntimeExports.jsx(SummaryComponent, { summaryState: chartState, styles: styles, appendMessage: appendMessage, query: (_28 = widget.config) === null || _28 === void 0 ? void 0 : _28.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, widget_ids: widgetIds, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout, setChartState: setChartState })) : (jsxRuntimeExports.jsx(reactUi.CopilotChat, { className: "h-full text-xs [&_.copilot-chat-message]:text-xs [&_.copilot-chat-input]:text-xs", labels: {
-                        title: ((_29 = widget.config) === null || _29 === void 0 ? void 0 : _29.copilotTitle) || widget.title,
-                        initial: ((_30 = widget.config) === null || _30 === void 0 ? void 0 : _30.copilotInitialMessage) || ((_31 = widget.config) === null || _31 === void 0 ? void 0 : _31.placeholder) || "How can I help you today?"
+                (agentType === "Series Bar Chart Agent" && (!((_j = (_h = (_g = chartState.series_bar_chart_data) === null || _g === void 0 ? void 0 : _g.data) === null || _h === void 0 ? void 0 : _h.labels) === null || _j === void 0 ? void 0 : _j.length) || !((_m = (_l = (_k = chartState.series_bar_chart_data) === null || _k === void 0 ? void 0 : _k.data) === null || _l === void 0 ? void 0 : _l.series) === null || _m === void 0 ? void 0 : _m.length))) ||
+                    (agentType === "Series Line Chart Agent" && (!((_q = (_p = (_o = chartState.series_bar_chart_data) === null || _o === void 0 ? void 0 : _o.data) === null || _p === void 0 ? void 0 : _p.labels) === null || _q === void 0 ? void 0 : _q.length) || !((_t = (_s = (_r = chartState.series_bar_chart_data) === null || _r === void 0 ? void 0 : _r.data) === null || _s === void 0 ? void 0 : _s.series) === null || _t === void 0 ? void 0 : _t.length))) ||
+                    (agentType === "Pie Chart Agent" && (!((_w = (_v = (_u = chartState.pie_chart_data) === null || _u === void 0 ? void 0 : _u.data) === null || _v === void 0 ? void 0 : _v.labels) === null || _w === void 0 ? void 0 : _w.length) || !((_z = (_y = (_x = chartState.pie_chart_data) === null || _x === void 0 ? void 0 : _x.data) === null || _y === void 0 ? void 0 : _y.values) === null || _z === void 0 ? void 0 : _z.length))) ||
+                    (agentType === "Data Grid Agent" && (!((_2 = (_1 = (_0 = chartState.matrix_grid_data) === null || _0 === void 0 ? void 0 : _0.data) === null || _1 === void 0 ? void 0 : _1.headers) === null || _2 === void 0 ? void 0 : _2.length) || !((_5 = (_4 = (_3 = chartState.matrix_grid_data) === null || _3 === void 0 ? void 0 : _3.data) === null || _4 === void 0 ? void 0 : _4.rows) === null || _5 === void 0 ? void 0 : _5.length))) ||
+                    (agentType === "Summary Agent" && (!((_7 = (_6 = chartState.summary_data) === null || _6 === void 0 ? void 0 : _6.data) === null || _7 === void 0 ? void 0 : _7.content) || ((_10 = (_9 = (_8 = chartState.summary_data) === null || _8 === void 0 ? void 0 : _8.data) === null || _9 === void 0 ? void 0 : _9.content) === null || _10 === void 0 ? void 0 : _10.trim()) === ""))) ? (jsxRuntimeExports.jsx("div", { className: "flex items-center justify-center h-full p-4", children: jsxRuntimeExports.jsx("div", { className: "text-center max-w-md", children: jsxRuntimeExports.jsx("p", { className: "text-sm text-gray-700", children: chartState.agent_message }) }) })) : isAgentTriggered ? (createLoadingComponent("Agent processing your request...")) : agentType === "Series Bar Chart Agent" ? (jsxRuntimeExports.jsx(SeriesBarChartComponent, { seriesBarChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_11 = widget.config) === null || _11 === void 0 ? void 0 : _11.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Series Line Chart Agent" ? (jsxRuntimeExports.jsx(SeriesLineChartComponent, { seriesLineChartState: chartState, styles: styles, orientation: orientation, appendMessage: appendMessage, query: (_12 = widget.config) === null || _12 === void 0 ? void 0 : _12.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Pie Chart Agent" ? (jsxRuntimeExports.jsx(PieChartComponent, { pieChartState: chartState, styles: styles, appendMessage: appendMessage, query: (_13 = widget.config) === null || _13 === void 0 ? void 0 : _13.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Data Grid Agent" ? (jsxRuntimeExports.jsx(DataGridComponent, { dataGridState: chartState, styles: styles, appendMessage: appendMessage, query: (_14 = widget.config) === null || _14 === void 0 ? void 0 : _14.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout })) : agentType === "Summary Agent" ? (jsxRuntimeExports.jsx(SummaryComponent, { summaryState: chartState, styles: styles, appendMessage: appendMessage, query: (_15 = widget.config) === null || _15 === void 0 ? void 0 : _15.query, isFirstLoad: isFirstLoad, widgetBackendUrl: widgetBackendUrl, widgetId: widget.id, widget_ids: widgetIds, startLoadingTimeout: startLoadingTimeout, clearLoadingTimeout: clearLoadingTimeout, setChartState: setChartState })) : (jsxRuntimeExports.jsx(reactUi.CopilotChat, { className: "h-full text-xs [&_.copilot-chat-message]:text-xs [&_.copilot-chat-input]:text-xs", labels: {
+                        title: ((_16 = widget.config) === null || _16 === void 0 ? void 0 : _16.copilotTitle) || widget.title,
+                        initial: ((_17 = widget.config) === null || _17 === void 0 ? void 0 : _17.copilotInitialMessage) || ((_18 = widget.config) === null || _18 === void 0 ? void 0 : _18.placeholder) || "How can I help you today?"
                     }, onSubmitMessage: () => {
                         setChartState(prevState => (Object.assign(Object.assign({}, prevState), { widget_ids: widgetIds, dataset_id: datasetId })));
+                        startPollingForData();
                     } })) })] }));
 }
 function AgentWidget({ widget, showHeader = true, widgetBackendUrl, onResetReady, widgetIds, datasetId, }) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dp-widgets-framework",
-  "version": "1.1.2",
+  "version": "1.1.4",
   "private": false,
   "publishConfig": {
     "registry": "https://registry.npmjs.org"