Merge pull request #45 from modelcontextprotocol/davidsp/fix-capture

modelcontextprotocol · Nov 11, 2024 · 94e4e92 · 94e4e92
2 parents 148e843 + ea4484c
commit 94e4e92
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/client/src/App.tsx b/client/src/App.tsx
@@ -108,6 +108,7 @@ const App = () => {
     >
   >([]);
   const nextRequestId = useRef(0);
+  const rootsRef = useRef<Root[]>([]);
 
   const handleApproveSampling = (id: number, result: CreateMessageResult) => {
     setPendingSampleRequests((prev) => {
@@ -159,6 +160,10 @@ const App = () => {
       );
   }, []);
 
+  useEffect(() => {
+    rootsRef.current = roots;
+  }, [roots]);
+
   const pushHistory = (request: object, response?: object) => {
     setRequestHistory((prev) => [
       ...prev,
@@ -293,7 +298,7 @@ const App = () => {
   };
 
   const handleRootsChange = async () => {
-    sendNotification({ method: "notifications/roots/list_changed" });
+    await sendNotification({ method: "notifications/roots/list_changed" });
   };
 
   const connectMcpServer = async () => {
@@ -337,7 +342,7 @@ const App = () => {
       });
 
       client.setRequestHandler(ListRootsRequestSchema, async () => {
-        return { roots };
+        return { roots: rootsRef.current };
       });
 
       setMcpClient(client);