ai-plugin: Add utility enhancements and helper functions

- Add isElectron utility function for environment detection
- Enhance ToolConfigManager with configuration validation
- Update AIInputSection with improved tool integration
- Add electron client improvements for MCP communication
- Include helper functions for better component integration

Signed-off-by: ashu8912 <aghildiyal@microsoft.com>

Author: ashu8912

ai-assistant/src/ai/mcp/electron-client.ts

@@ -90,6 +90,7 @@ class ElectronMCPClient {
     }
 
     try {
+      console.log("args for tool executed is ", args)
       const response = await window.desktopApi!.mcp.executeTool(toolName, args, toolCallId);
 
       if (response.success) {

ai-assistant/src/components/assistant/AIInputSection.tsx

@@ -15,6 +15,7 @@ import { getProviderById } from '../../config/modelConfig';
 import { getModelDisplayName, getProviderModelsForChat } from '../../utils/modalUtils';
 import { StoredProviderConfig } from '../../utils/ProviderConfigManager';
 import TestModeInput from './TestModeInput';
+import { ToolsDialog } from './ToolsDialog';
 
 interface AIInputSectionProps {
   promptVal: string;
@@ -24,11 +25,13 @@ interface AIInputSectionProps {
   activeConfig: StoredProviderConfig | null;
   availableConfigs: StoredProviderConfig[];
   selectedModel: string;
+  enabledTools: string[];
   onSend: (prompt: string) => void;
   onStop: () => void;
   onClearHistory: () => void;
   onConfigChange: (config: StoredProviderConfig, model: string) => void;
   onTestModeResponse: (content: string, type: 'assistant' | 'user', hasError?: boolean) => void;
+  onToolsChange: (enabledTools: string[]) => void;
 }
 
 export const AIInputSection: React.FC<AIInputSectionProps> = ({
@@ -39,12 +42,15 @@ export const AIInputSection: React.FC<AIInputSectionProps> = ({
   activeConfig,
   availableConfigs,
   selectedModel,
+  enabledTools,
   onSend,
   onStop,
   onClearHistory,
   onConfigChange,
   onTestModeResponse,
+  onToolsChange,
 }) => {
+  const [showToolsDialog, setShowToolsDialog] = React.useState(false);
   const handleKeyDown = (e: React.KeyboardEvent) => {
     if (e.key === 'Enter' && !e.shiftKey) {
       e.preventDefault();
@@ -188,6 +194,20 @@ export const AIInputSection: React.FC<AIInputSectionProps> = ({
               </Select>
             </Box>
           )}
+
+          {/* Tools Button */}
+          {!isTestMode && (
+            <Box ml={1}>
+              <ActionButton
+                description="Manage Tools"
+                onClick={() => setShowToolsDialog(true)}
+                icon="mdi:tools"
+                iconButtonProps={{
+                  size: 'small',
+                }}
+              />
+            </Box>
+          )}
         </Grid>
 
         <Grid item>
@@ -214,6 +234,14 @@ export const AIInputSection: React.FC<AIInputSectionProps> = ({
           )}
         </Grid>
       </Grid>
+
+      {/* Tools Dialog */}
+      <ToolsDialog
+        open={showToolsDialog}
+        onClose={() => setShowToolsDialog(false)}
+        enabledTools={enabledTools}
+        onToolsChange={onToolsChange}
+      />
     </Box>
   );
 };

ai-assistant/src/utils.tsx

@@ -21,6 +21,9 @@ function usePluginSettings() {
   // Add state to control UI panel visibility - initialize from stored settings
   const [isUIPanelOpen, setIsUIPanelOpenState] = React.useState(conf?.isUIPanelOpen ?? false);
 
+  // Add state for enabled tools - initialize from stored settings
+  const [enabledTools, setEnabledToolsState] = React.useState<string[]>(conf?.enabledTools ?? []);
+
   // Wrap setIsUIPanelOpen to also update the stored configuration
   const setIsUIPanelOpen = (isOpen: boolean) => {
     setIsUIPanelOpenState(isOpen);
@@ -32,6 +35,17 @@ function usePluginSettings() {
     });
   };
 
+  // Wrap setEnabledTools to also update the stored configuration
+  const setEnabledTools = (tools: string[]) => {
+    setEnabledToolsState(tools);
+    // Save the tools configuration
+    const currentConf = pluginStore.get() || {};
+    pluginStore.update({
+      ...currentConf,
+      enabledTools: tools,
+    });
+  };
+
   return {
     event,
     setEvent,
@@ -41,6 +55,8 @@ function usePluginSettings() {
     setActiveProvider,
     isUIPanelOpen,
     setIsUIPanelOpen,
+    enabledTools,
+    setEnabledTools,
   };
 }
 

ai-assistant/src/utils/ToolConfigManager.ts

@@ -52,3 +52,19 @@ export function getEnabledToolIds(pluginSettings: any): string[] {
   const allTools = getAllAvailableTools();
   return allTools.map(tool => tool.id).filter(toolId => isToolEnabled(pluginSettings, toolId));
 }
+
+// Sets the enabled tools list in plugin settings
+export function setEnabledTools(pluginSettings: any, enabledToolIds: string[]): any {
+  const enabledTools: Record<string, boolean> = {};
+  
+  // Get all available tools and set their enabled state
+  const allTools = getAllAvailableTools();
+  allTools.forEach(tool => {
+    enabledTools[tool.id] = enabledToolIds.includes(tool.id);
+  });
+
+  return {
+    ...pluginSettings,
+    enabledTools,
+  };
+}