Record 'roots list changed' notifications in history sidebar

This commit is contained in:
Justin Spahr-Summers
2024-11-08 12:04:45 +00:00
parent 57f0c49154
commit 2867173e7b
2 changed files with 23 additions and 11 deletions

View File

@@ -20,6 +20,7 @@ import {
ServerNotification,
Tool,
CompatibilityCallToolResult,
ClientNotification,
} from "@modelcontextprotocol/sdk/types.js";
import { useEffect, useRef, useState } from "react";
@@ -90,7 +91,7 @@ const App = () => {
const [url, setUrl] = useState<string>("http://localhost:3001/sse");
const [transportType, setTransportType] = useState<"stdio" | "sse">("stdio");
const [requestHistory, setRequestHistory] = useState<
{ request: string; response: string }[]
{ request: string; response?: string }[]
>([]);
const [mcpClient, setMcpClient] = useState<Client | null>(null);
const [notifications, setNotifications] = useState<ServerNotification[]>([]);
@@ -158,10 +159,13 @@ const App = () => {
);
}, []);
const pushHistory = (request: object, response: object) => {
const pushHistory = (request: object, response?: object) => {
setRequestHistory((prev) => [
...prev,
{ request: JSON.stringify(request), response: JSON.stringify(response) },
{
request: JSON.stringify(request),
response: response !== undefined ? JSON.stringify(response) : undefined,
},
]);
};
@@ -183,6 +187,20 @@ const App = () => {
}
};
const sendNotification = async (notification: ClientNotification) => {
if (!mcpClient) {
throw new Error("MCP client not connected");
}
try {
await mcpClient.notification(notification);
pushHistory(notification);
} catch (e: unknown) {
setError((e as Error).message);
throw e;
}
};
const listResources = async () => {
const response = await makeRequest(
{
@@ -275,13 +293,7 @@ const App = () => {
};
const handleRootsChange = async () => {
if (mcpClient) {
try {
await mcpClient.sendRootsListChanged();
} catch (e) {
console.error("Failed to send roots list changed notification:", e);
}
}
sendNotification({ method: "notifications/roots/list_changed" });
};
const connectMcpServer = async () => {

View File

@@ -6,7 +6,7 @@ const HistoryAndNotifications = ({
requestHistory,
serverNotifications,
}: {
requestHistory: Array<{ request: string; response: string | null }>;
requestHistory: Array<{ request: string; response?: string }>;
serverNotifications: ServerNotification[];
}) => {
const [expandedRequests, setExpandedRequests] = useState<{