ai-plugin: Allow multiple tool calls to happen

Signed-off-by: ashu8912 <[email protected]>

Author: ashu8912

ai-assistant/src/ContentRenderer.tsx

@@ -100,12 +100,12 @@ const TableWrapper: React.FC<{ children: React.ReactNode }> = React.memo(({ chil
   // Extract table rows from children
   const tableElement = React.Children.only(children) as React.ReactElement;
   const tableChildren = tableElement.props.children;
-  
+
   if (!tableChildren) {
     // No children found, return table as is
     return <Box sx={{ overflowX: 'auto', width: '100%', mb: 2 }}>{children}</Box>;
   }
-  
+
   const tbody = React.Children.toArray(tableChildren).find(
     (child: any) => child?.type === 'tbody' || child?.props?.component === 'tbody'
   );

ai-assistant/src/components/common/InlineToolConfirmation.tsx

@@ -134,9 +134,12 @@ const InlineToolConfirmation: React.FC<InlineToolConfirmationProps> = ({
         }
       });
 
-      onApprove(selectedToolIds);
+      // Call onApprove and wait for it to complete
+      await onApprove(selectedToolIds);
     } catch (error) {
       console.error('Error during tool approval:', error);
+    } finally {
+      // Always reset the approving state, whether success or error
       setIsApproving(false);
     }
   };
@@ -147,9 +150,12 @@ const InlineToolConfirmation: React.FC<InlineToolConfirmationProps> = ({
     setIsDenying(true);
 
     try {
-      onDeny();
+      // Call onDeny and wait for it to complete
+      await onDeny();
     } catch (error) {
       console.error('Error during tool denial:', error);
+    } finally {
+      // Always reset the denying state
       setIsDenying(false);
     }
   };

ai-assistant/src/components/mcpOutput/MCPOutputDisplay.tsx

@@ -1276,7 +1276,7 @@ const ErrorDisplay: React.FC<{ data: any; onRetry?: () => void; width: string }>
             size="small"
             onClick={onRetry}
             startIcon={<Icon icon="mdi:refresh" />}
-            sx={{mr: 0.5}}
+            sx={{ mr: 0.5 }}
           >
             Retry Tool
           </Button>