Add ability to run mcp server from ai-assistant plugin

ai-assistant/package.json

@@ -38,9 +38,12 @@
     "@langchain/community": "^0.3.42",
     "@langchain/core": "^0.3.51",
     "@langchain/google-genai": "^0.2.5",
+    "@langchain/langgraph": "^0.4.9",
+    "@langchain/mcp-adapters": "^0.6.0",
     "@langchain/mistralai": "^0.2.0",
     "@langchain/ollama": "^0.2.0",
     "@langchain/openai": "^0.5.5",
+    "@modelcontextprotocol/sdk": "^1.17.5",
     "@monaco-editor/react": "^4.5.2",
     "@types/prismjs": "^1.26.5",
     "@types/react-syntax-highlighter": "^15.5.13",

ai-assistant/src/ContentRenderer.tsx

@@ -6,6 +6,7 @@ import { Link as RouterLink, useHistory } from 'react-router-dom';
 import remarkGfm from 'remark-gfm';
 import YAML from 'yaml';
 import { LogsButton, YamlDisplay } from './components';
+import MCPFormattedMessage from './components/chat/MCPFormattedMessage';
 import { getHeadlampLink } from './utils/promptLinkHelper';
 import { parseKubernetesYAML } from './utils/SampleYamlLibrary';
 
@@ -87,6 +88,8 @@ const parseLogsButtonData = (content: string, logsButtonIndex: number): ParseRes
 interface ContentRendererProps {
   content: string;
   onYamlDetected?: (yaml: string, resourceType: string) => void;
+  promptWidth?: string; // Add width prop
+  onRetryTool?: (toolName: string, args: Record<string, any>) => void;
 }
 
 // Table wrapper component with show more functionality - moved outside to preserve state
@@ -276,7 +279,7 @@ markdownComponents.li.displayName = 'MarkdownLi';
 markdownComponents.blockquote.displayName = 'MarkdownBlockquote';
 
 const ContentRenderer: React.FC<ContentRendererProps> = React.memo(
-  ({ content, onYamlDetected }) => {
+  ({ content, onYamlDetected, onRetryTool }) => {
     const history = useHistory();
     // Create code component that has access to onYamlDetected
     const CodeComponent = React.useMemo(() => {
@@ -524,7 +527,18 @@ const ContentRenderer: React.FC<ContentRendererProps> = React.memo(
     const processedContent = useMemo(() => {
       if (!content) return null;
 
-      // First, check if content is a JSON response with error or success keys
+      // First, check if content is a formatted MCP output (pure JSON)
+      try {
+        const parsed = JSON.parse(content.trim());
+        if (parsed.formatted && parsed.mcpOutput) {
+          // This is a formatted MCP output, use our specialized component
+          return <MCPFormattedMessage content={content} isAssistant onRetryTool={onRetryTool} />;
+        }
+      } catch (error) {
+        // Not JSON or not formatted MCP output, continue with normal processing
+      }
+
+      // Second, check if content is a JSON response with error or success keys
       const jsonParseResult = parseJsonContent(content.trim());
       if (jsonParseResult.success) {
         const parsedContent = jsonParseResult.data;
@@ -556,8 +570,8 @@ const ContentRenderer: React.FC<ContentRendererProps> = React.memo(
             <Box
               component="pre"
               sx={{
-                backgroundColor: theme => theme.palette.grey[100],
-                color: theme => theme.palette.grey[900],
+                backgroundColor: (theme: any) => theme.palette.grey[100],
+                color: (theme: any) => theme.palette.grey[900],
                 padding: 2,
                 borderRadius: 1,
                 overflowX: 'auto',
@@ -633,7 +647,7 @@ const ContentRenderer: React.FC<ContentRendererProps> = React.memo(
           {content}
         </ReactMarkdown>
       );
-    }, [content, onYamlDetected, processUnformattedYaml]);
+    }, [content, onYamlDetected, onRetryTool, processUnformattedYaml]);
 
     return (
       <Box sx={{ width: '100%', overflowWrap: 'break-word', wordWrap: 'break-word' }}>
@@ -645,7 +659,8 @@ const ContentRenderer: React.FC<ContentRendererProps> = React.memo(
     // Only re-render if content or onYamlDetected actually changed
     return (
       prevProps.content === nextProps.content &&
-      prevProps.onYamlDetected === nextProps.onYamlDetected
+      prevProps.onYamlDetected === nextProps.onYamlDetected &&
+      prevProps.onRetryTool === nextProps.onRetryTool
     );
   }
 );

ai-assistant/src/ai/manager.ts

@@ -1,3 +1,11 @@
+export type ToolCall = {
+  id: string;
+  name: string;
+  description?: string;
+  arguments: Record<string, any>;
+  type: 'mcp' | 'regular';
+};
+
 export type Prompt = {
   role: string;
   content: string;
@@ -9,6 +17,14 @@ export type Prompt = {
   contentFilterError?: boolean;
   alreadyDisplayed?: boolean;
   isDisplayOnly?: boolean; // Mark messages that shouldn't be sent to LLM
+  requestId?: string; // For tracking tool confirmation messages
+  // Add support for inline tool confirmations
+  toolConfirmation?: {
+    tools: ToolCall[];
+    onApprove: (approvedToolIds: string[]) => void;
+    onDeny: () => void;
+    loading?: boolean;
+  };
 };
 
 export default abstract class AIManager {