diff --git a/ui/litellm-dashboard/src/components/mcp_tools/mcp_tools.tsx b/ui/litellm-dashboard/src/components/mcp_tools/mcp_tools.tsx index 7c24b1b377..5a971e8b47 100644 --- a/ui/litellm-dashboard/src/components/mcp_tools/mcp_tools.tsx +++ b/ui/litellm-dashboard/src/components/mcp_tools/mcp_tools.tsx @@ -1,7 +1,7 @@ import React, { useState } from "react"; import { useQuery, useMutation } from "@tanstack/react-query"; import { ToolTestPanel } from "./ToolTestPanel"; -import { MCPTool, MCPToolsViewerProps, CallMCPToolResponse } from "./types"; +import { MCPTool, MCPToolsViewerProps, MCPContent, CallMCPToolResponse } from "./types"; import { listMCPTools, callMCPTool } from "../networking"; import { Card, Title, Text } from "@tremor/react"; @@ -16,7 +16,7 @@ const MCPToolsViewer = ({ serverAlias, // Add serverAlias prop }: MCPToolsViewerProps) => { const [selectedTool, setSelectedTool] = useState(null); - const [toolResult, setToolResult] = useState(null); + const [toolResult, setToolResult] = useState(null); const [toolError, setToolError] = useState(null); // Query to fetch MCP tools @@ -40,14 +40,14 @@ const MCPToolsViewer = ({ if (!accessToken) throw new Error("Access Token required"); try { - const result = await callMCPTool(accessToken, args.tool.name, args.arguments); + const result: CallMCPToolResponse = await callMCPTool(accessToken, args.tool.name, args.arguments); return result; } catch (error) { throw error; } }, onSuccess: (data) => { - setToolResult(data); + setToolResult(data.content); setToolError(null); }, onError: (error: Error) => { diff --git a/ui/litellm-dashboard/src/components/mcp_tools/types.tsx b/ui/litellm-dashboard/src/components/mcp_tools/types.tsx index 6c01ca2e61..fd19a23a8b 100644 --- a/ui/litellm-dashboard/src/components/mcp_tools/types.tsx +++ b/ui/litellm-dashboard/src/components/mcp_tools/types.tsx @@ -110,7 +110,12 @@ export interface MCPEmbeddedResource { export type MCPContent = MCPTextContent | MCPImageContent | MCPEmbeddedResource; // Define the response structure for the callMCPTool endpoint -export type CallMCPToolResponse = MCPContent[]; +export type CallMCPToolResponse = { + content: MCPContent[]; + _meta: any; + isError: boolean; + structuredContent: any; +}; // Props for the main component export interface MCPToolsViewerProps {