tools
This commit is contained in:
@@ -1,5 +1,12 @@
|
|||||||
import { useState, useEffect } from "react";
|
import { useState, useEffect } from "react";
|
||||||
import { Send, Bell, Terminal, Files, MessageSquare } from "lucide-react";
|
import {
|
||||||
|
Send,
|
||||||
|
Bell,
|
||||||
|
Terminal,
|
||||||
|
Files,
|
||||||
|
MessageSquare,
|
||||||
|
Hammer,
|
||||||
|
} from "lucide-react";
|
||||||
import { Tabs, TabsList, TabsTrigger } from "@/components/ui/tabs";
|
import { Tabs, TabsList, TabsTrigger } from "@/components/ui/tabs";
|
||||||
|
|
||||||
import ConsoleTab from "./components/ConsoleTab";
|
import ConsoleTab from "./components/ConsoleTab";
|
||||||
@@ -8,6 +15,7 @@ import RequestsTab from "./components/RequestsTabs";
|
|||||||
import ResourcesTab, { Resource } from "./components/ResourcesTab";
|
import ResourcesTab, { Resource } from "./components/ResourcesTab";
|
||||||
import NotificationsTab from "./components/NotificationsTab";
|
import NotificationsTab from "./components/NotificationsTab";
|
||||||
import PromptsTab, { Prompt } from "./components/PromptsTab";
|
import PromptsTab, { Prompt } from "./components/PromptsTab";
|
||||||
|
import ToolsTab, { Tool as ToolType } from "./components/ToolsTab";
|
||||||
|
|
||||||
const App = () => {
|
const App = () => {
|
||||||
const [socket, setSocket] = useState<WebSocket | null>(null);
|
const [socket, setSocket] = useState<WebSocket | null>(null);
|
||||||
@@ -18,6 +26,8 @@ const App = () => {
|
|||||||
const [resourceContent, setResourceContent] = useState<string>("");
|
const [resourceContent, setResourceContent] = useState<string>("");
|
||||||
const [prompts, setPrompts] = useState<Prompt[]>([]);
|
const [prompts, setPrompts] = useState<Prompt[]>([]);
|
||||||
const [promptContent, setPromptContent] = useState<string>("");
|
const [promptContent, setPromptContent] = useState<string>("");
|
||||||
|
const [tools, setTools] = useState<ToolType[]>([]);
|
||||||
|
const [toolResult, setToolResult] = useState<string>("");
|
||||||
const [error, setError] = useState<string | null>(null);
|
const [error, setError] = useState<string | null>(null);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -44,6 +54,12 @@ const App = () => {
|
|||||||
} else if (message.type === "prompt") {
|
} else if (message.type === "prompt") {
|
||||||
setPromptContent(JSON.stringify(message.data, null, 2));
|
setPromptContent(JSON.stringify(message.data, null, 2));
|
||||||
setError(null);
|
setError(null);
|
||||||
|
} else if (message.type === "tools") {
|
||||||
|
setTools(message.data.tools);
|
||||||
|
setError(null);
|
||||||
|
} else if (message.type === "toolResult") {
|
||||||
|
setToolResult(JSON.stringify(message.data, null, 2));
|
||||||
|
setError(null);
|
||||||
} else if (message.type === "error") {
|
} else if (message.type === "error") {
|
||||||
setError(message.message);
|
setError(message.message);
|
||||||
}
|
}
|
||||||
@@ -71,6 +87,7 @@ const App = () => {
|
|||||||
null,
|
null,
|
||||||
);
|
);
|
||||||
const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null);
|
const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null);
|
||||||
|
const [selectedTool, setSelectedTool] = useState<ToolType | null>(null);
|
||||||
|
|
||||||
const listResources = () => {
|
const listResources = () => {
|
||||||
sendWebSocketMessage({ type: "listResources" });
|
sendWebSocketMessage({ type: "listResources" });
|
||||||
@@ -88,6 +105,14 @@ const App = () => {
|
|||||||
sendWebSocketMessage({ type: "getPrompt", name });
|
sendWebSocketMessage({ type: "getPrompt", name });
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const listTools = () => {
|
||||||
|
sendWebSocketMessage({ type: "listTools" });
|
||||||
|
};
|
||||||
|
|
||||||
|
const callTool = (name: string, params: Record<string, unknown>) => {
|
||||||
|
sendWebSocketMessage({ type: "callTool", name, params });
|
||||||
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="flex h-screen bg-gray-100">
|
<div className="flex h-screen bg-gray-100">
|
||||||
<Sidebar connectionStatus={connectionStatus} />
|
<Sidebar connectionStatus={connectionStatus} />
|
||||||
@@ -96,14 +121,6 @@ const App = () => {
|
|||||||
<div className="flex-1 overflow-auto">
|
<div className="flex-1 overflow-auto">
|
||||||
<Tabs defaultValue="requests" className="w-full p-4">
|
<Tabs defaultValue="requests" className="w-full p-4">
|
||||||
<TabsList className="mb-4">
|
<TabsList className="mb-4">
|
||||||
<TabsTrigger value="requests">
|
|
||||||
<Send className="w-4 h-4 mr-2" />
|
|
||||||
Requests
|
|
||||||
</TabsTrigger>
|
|
||||||
<TabsTrigger value="notifications">
|
|
||||||
<Bell className="w-4 h-4 mr-2" />
|
|
||||||
Notifications
|
|
||||||
</TabsTrigger>
|
|
||||||
<TabsTrigger value="resources">
|
<TabsTrigger value="resources">
|
||||||
<Files className="w-4 h-4 mr-2" />
|
<Files className="w-4 h-4 mr-2" />
|
||||||
Resources
|
Resources
|
||||||
@@ -112,7 +129,19 @@ const App = () => {
|
|||||||
<MessageSquare className="w-4 h-4 mr-2" />
|
<MessageSquare className="w-4 h-4 mr-2" />
|
||||||
Prompts
|
Prompts
|
||||||
</TabsTrigger>
|
</TabsTrigger>
|
||||||
<TabsTrigger value="console">
|
<TabsTrigger value="requests" disabled>
|
||||||
|
<Send className="w-4 h-4 mr-2" />
|
||||||
|
Requests
|
||||||
|
</TabsTrigger>
|
||||||
|
<TabsTrigger value="notifications" disabled>
|
||||||
|
<Bell className="w-4 h-4 mr-2" />
|
||||||
|
Notifications
|
||||||
|
</TabsTrigger>
|
||||||
|
<TabsTrigger value="tools" disabled>
|
||||||
|
<Hammer className="w-4 h-4 mr-2" />
|
||||||
|
Tools
|
||||||
|
</TabsTrigger>
|
||||||
|
<TabsTrigger value="console" disabled>
|
||||||
<Terminal className="w-4 h-4 mr-2" />
|
<Terminal className="w-4 h-4 mr-2" />
|
||||||
Console
|
Console
|
||||||
</TabsTrigger>
|
</TabsTrigger>
|
||||||
@@ -139,6 +168,15 @@ const App = () => {
|
|||||||
promptContent={promptContent}
|
promptContent={promptContent}
|
||||||
error={error}
|
error={error}
|
||||||
/>
|
/>
|
||||||
|
<ToolsTab
|
||||||
|
tools={tools}
|
||||||
|
listTools={listTools}
|
||||||
|
callTool={callTool}
|
||||||
|
selectedTool={selectedTool}
|
||||||
|
setSelectedTool={setSelectedTool}
|
||||||
|
toolResult={toolResult}
|
||||||
|
error={error}
|
||||||
|
/>
|
||||||
<ConsoleTab />
|
<ConsoleTab />
|
||||||
</div>
|
</div>
|
||||||
</Tabs>
|
</Tabs>
|
||||||
|
|||||||
101
client/src/components/ToolsTab.tsx
Normal file
101
client/src/components/ToolsTab.tsx
Normal file
@@ -0,0 +1,101 @@
|
|||||||
|
import { TabsContent } from "@/components/ui/tabs";
|
||||||
|
import { Button } from "@/components/ui/button";
|
||||||
|
import { Textarea } from "@/components/ui/textarea";
|
||||||
|
import { Send, AlertCircle } from "lucide-react";
|
||||||
|
import { Alert, AlertDescription, AlertTitle } from "@/components/ui/alert";
|
||||||
|
import { useState } from "react";
|
||||||
|
|
||||||
|
export type Tool = {
|
||||||
|
name: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
const ToolsTab = ({
|
||||||
|
tools,
|
||||||
|
listTools,
|
||||||
|
callTool,
|
||||||
|
selectedTool,
|
||||||
|
setSelectedTool,
|
||||||
|
toolResult,
|
||||||
|
error,
|
||||||
|
}: {
|
||||||
|
tools: Tool[];
|
||||||
|
listTools: () => void;
|
||||||
|
callTool: (name: string, params: Record<string, unknown>) => void;
|
||||||
|
selectedTool: Tool | null;
|
||||||
|
setSelectedTool: (tool: Tool) => void;
|
||||||
|
toolResult: string;
|
||||||
|
error: string | null;
|
||||||
|
}) => {
|
||||||
|
const [params, setParams] = useState("");
|
||||||
|
|
||||||
|
return (
|
||||||
|
<TabsContent value="tools" className="grid grid-cols-2 gap-4">
|
||||||
|
<div className="bg-white rounded-lg shadow">
|
||||||
|
<div className="p-4 border-b border-gray-200">
|
||||||
|
<h3 className="font-semibold">Tools</h3>
|
||||||
|
</div>
|
||||||
|
<div className="p-4">
|
||||||
|
<Button variant="outline" className="w-full mb-4" onClick={listTools}>
|
||||||
|
List Tools
|
||||||
|
</Button>
|
||||||
|
<div className="space-y-2">
|
||||||
|
{tools.map((tool, index) => (
|
||||||
|
<div
|
||||||
|
key={index}
|
||||||
|
className="flex items-center p-2 rounded hover:bg-gray-50 cursor-pointer"
|
||||||
|
onClick={() => setSelectedTool(tool)}
|
||||||
|
>
|
||||||
|
<span className="flex-1">{tool.name}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="bg-white rounded-lg shadow">
|
||||||
|
<div className="p-4 border-b border-gray-200">
|
||||||
|
<h3 className="font-semibold">
|
||||||
|
{selectedTool ? selectedTool.name : "Select a tool"}
|
||||||
|
</h3>
|
||||||
|
</div>
|
||||||
|
<div className="p-4">
|
||||||
|
{error ? (
|
||||||
|
<Alert variant="destructive">
|
||||||
|
<AlertCircle className="h-4 w-4" />
|
||||||
|
<AlertTitle>Error</AlertTitle>
|
||||||
|
<AlertDescription>{error}</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
) : selectedTool ? (
|
||||||
|
<div className="space-y-4">
|
||||||
|
<Textarea
|
||||||
|
placeholder="Tool parameters (JSON)"
|
||||||
|
className="h-32 font-mono"
|
||||||
|
value={params}
|
||||||
|
onChange={(e) => setParams(e.target.value)}
|
||||||
|
/>
|
||||||
|
<Button
|
||||||
|
onClick={() => callTool(selectedTool.name, JSON.parse(params))}
|
||||||
|
>
|
||||||
|
<Send className="w-4 h-4 mr-2" />
|
||||||
|
Run Tool
|
||||||
|
</Button>
|
||||||
|
{toolResult && (
|
||||||
|
<pre className="bg-gray-50 p-4 rounded text-sm overflow-auto max-h-64">
|
||||||
|
{JSON.stringify(toolResult, null, 2)}
|
||||||
|
</pre>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<Alert>
|
||||||
|
<AlertDescription>
|
||||||
|
Select a tool from the list to view its details and run it
|
||||||
|
</AlertDescription>
|
||||||
|
</Alert>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</TabsContent>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default ToolsTab;
|
||||||
@@ -10,6 +10,10 @@ import {
|
|||||||
ListPromptsResultSchema,
|
ListPromptsResultSchema,
|
||||||
GetPromptResult,
|
GetPromptResult,
|
||||||
GetPromptResultSchema,
|
GetPromptResultSchema,
|
||||||
|
ListToolsResult,
|
||||||
|
ListToolsResultSchema,
|
||||||
|
CallToolResult,
|
||||||
|
CallToolResultSchema,
|
||||||
} from "mcp-typescript/types.js";
|
} from "mcp-typescript/types.js";
|
||||||
|
|
||||||
export class McpClient {
|
export class McpClient {
|
||||||
@@ -86,6 +90,29 @@ export class McpClient {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Tool Operations
|
||||||
|
async listTools(): Promise<ListToolsResult> {
|
||||||
|
return await this.client.request(
|
||||||
|
{
|
||||||
|
method: "tools/list",
|
||||||
|
},
|
||||||
|
ListToolsResultSchema,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
async callTool(
|
||||||
|
name: string,
|
||||||
|
params: Record<string, unknown>,
|
||||||
|
): Promise<CallToolResult> {
|
||||||
|
return await this.client.request(
|
||||||
|
{
|
||||||
|
method: "tools/call",
|
||||||
|
params: { name, ...params },
|
||||||
|
},
|
||||||
|
CallToolResultSchema,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
getServerCapabilities() {
|
getServerCapabilities() {
|
||||||
return this.client.getServerCapabilities();
|
return this.client.getServerCapabilities();
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -30,6 +30,19 @@ wss.on("connection", (ws: WebSocket) => {
|
|||||||
} else if (command.type === "getPrompt" && command.name) {
|
} else if (command.type === "getPrompt" && command.name) {
|
||||||
const prompt = await mcpClient.getPrompt(command.name);
|
const prompt = await mcpClient.getPrompt(command.name);
|
||||||
ws.send(JSON.stringify({ type: "prompt", data: prompt }));
|
ws.send(JSON.stringify({ type: "prompt", data: prompt }));
|
||||||
|
} else if (command.type === "listTools") {
|
||||||
|
const tools = await mcpClient.listTools();
|
||||||
|
ws.send(JSON.stringify({ type: "tools", data: tools }));
|
||||||
|
} else if (
|
||||||
|
command.type === "callTool" &&
|
||||||
|
command.name &&
|
||||||
|
command.params
|
||||||
|
) {
|
||||||
|
const result = await mcpClient.callTool(
|
||||||
|
command.name + "asdf",
|
||||||
|
command.params,
|
||||||
|
);
|
||||||
|
ws.send(JSON.stringify({ type: "toolResult", data: result }));
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
console.error("Error:", error);
|
console.error("Error:", error);
|
||||||
|
|||||||
Reference in New Issue
Block a user