fix: widget bugs and ai agent

Author: mohamadch91

Diff for: services/webui/src/components/widgets/KeyValue/index.tsx

@@ -104,6 +104,7 @@ export const getTable = (
 
 export default function KeyValueWidget({ kpis }: KPIProps) {
     const [items, setItems] = useState<any[]>([])
+    const [showError, setShowError] = useState(false)
 
     const RunQuery = (query_id: string) => {
         let url = ''
@@ -162,6 +163,7 @@ export default function KeyValueWidget({ kpis }: KPIProps) {
                 })
             }).catch((err)=>{
                 console.log("err",err)
+                setShowError(true)
             })
         })
 
@@ -173,12 +175,16 @@ export default function KeyValueWidget({ kpis }: KPIProps) {
 
     useEffect(() => {
         if (kpis.length > 0) {
+            setShowError(false)
             handleKPIs()
         }
     }, [kpis])
+    const GetItems = () => {
+        return items
+    }
     return (
         <>
-            {items.length == 0 || items.length != kpis.length ? (
+            {(items.length == 0 || items.length != kpis.length || showError) ? (
                 <>
                     <Alert header="Error" type="error">
                         Error fetching fata
@@ -187,9 +193,11 @@ export default function KeyValueWidget({ kpis }: KPIProps) {
             ) : (
                 <>
                     <KeyValuePairs
-                        columns={kpis.length >4 ? 4 : kpis.length}
+                        columns={kpis.length > 4 ? 4 : kpis.length}
                         minColumnWidth={250}
-                        items={items}
+                        // @ts-ignore
+
+                        items={GetItems()}
                     />
                 </>
             )}

Diff for: services/webui/src/components/widgets/WidgetLayout/index.tsx

@@ -235,19 +235,11 @@ export default function WidgetLayout({
                                                         ?.info
                                                 }
                                                 onChange={(e: any) => {
-                                                    setWidgetProps({
-                                                        ...widgetProps,
-                                                        kpis: [
-                                                            ...(widgetProps?.kpis ||
-                                                                []),
-                                                            {
-                                                                info: e.detail
-                                                                    .value,
-                                                                count_kpi: '',
-                                                                list_kpi: '',
-                                                            },
-                                                        ],
-                                                    })
+                                                    HandleKPIPropChange(
+                                                        index,
+                                                        e.detail.value,
+                                                        'info'
+                                                    )
                                                 }}
                                             />
                                             <Input
@@ -257,23 +249,11 @@ export default function WidgetLayout({
                                                         ?.count_kpi
                                                 }
                                                 onChange={(e: any) => {
-                                                    setWidgetProps({
-                                                        ...widgetProps,
-                                                        kpis: [
-                                                            ...(widgetProps?.kpis ||
-                                                                []),
-                                                            {
-                                                                info: widgetProps
-                                                                    ?.kpis?.[
-                                                                    index
-                                                                ]?.info,
-                                                                count_kpi:
-                                                                    e.detail
-                                                                        .value,
-                                                                list_kpi: '',
-                                                            },
-                                                        ],
-                                                    })
+                                                    HandleKPIPropChange(
+                                                        index,
+                                                        e.detail.value,
+                                                        'count_kpi'
+                                                    )
                                                 }}
                                             />
                                             <Input
@@ -283,28 +263,11 @@ export default function WidgetLayout({
                                                         ?.list_kpi
                                                 }
                                                 onChange={(e: any) => {
-                                                    setWidgetProps({
-                                                        ...widgetProps,
-                                                        kpis: [
-                                                            ...(widgetProps?.kpis ||
-                                                                []),
-                                                            {
-                                                                info: widgetProps
-                                                                    ?.kpis?.[
-                                                                    index
-                                                                ]?.info,
-                                                                count_kpi:
-                                                                    widgetProps
-                                                                        ?.kpis?.[
-                                                                        index
-                                                                    ]
-                                                                        ?.count_kpi,
-                                                                list_kpi:
-                                                                    e.detail
-                                                                        .value,
-                                                            },
-                                                        ],
-                                                    })
+                                                    HandleKPIPropChange(
+                                                        index,
+                                                        e.detail.value,
+                                                        'list_kpi'
+                                                    )
                                                 }}
                                             />
                                     </>

Diff for: services/webui/src/pages/AI/chat/AIChat.tsx

@@ -344,9 +344,17 @@ function AIChat({ setOpen }: any) {
             user_clarification_response:answer
 
         }
+        // @ts-ignore
+         const token = JSON.parse(localStorage.getItem('openg_auth')).token
+
+         const config = {
+             headers: {
+                 Authorization: `Bearer ${token}`,
+             },
+         }
 
         axios
-            .post(`/main/core/api/v4/chatbot/generate-query`, body)
+            .post(`/main/core/api/v4/chatbot/generate-query`,body,config)
             .then((res) => {
                 if (res?.data) {
                     const output = res?.data