@mcp-ts/sdk 2.4.4 → 2.5.0

package/dist/client/react.mjs

@@ -37,6 +37,12 @@ var SSEClient = class {
   async disconnectFromServer(sessionId) {
     return this.sendRequest("disconnect", { sessionId });
   }
+  async setToolPolicy(sessionId, toolPolicy) {
+    return this.sendRequest("setToolPolicy", { sessionId, toolPolicy });
+  }
+  async getToolPolicy(sessionId) {
+    return this.sendRequest("getToolPolicy", { sessionId });
+  }
   async listTools(sessionId) {
     return this.sendRequest("listTools", { sessionId });
   }
@@ -423,6 +429,7 @@ function useMcp(options) {
             state: getInitialConnectionState(s.status),
             createdAt: new Date(s.createdAt),
             updatedAt: new Date(s.updatedAt ?? s.createdAt),
+            toolPolicy: s.toolPolicy,
             tools: []
           }))
         );
@@ -530,6 +537,29 @@ function useMcp(options) {
     }
     return await clientRef.current.listTools(sessionId);
   }, []);
+  const updateToolPolicy = useCallback(async (sessionId, toolPolicy) => {
+    if (!clientRef.current) {
+      throw new Error("SSE client not initialized");
+    }
+    const result = await clientRef.current.setToolPolicy(sessionId, toolPolicy);
+    if (isMountedRef.current) {
+      setConnections((prev) => prev.map(
+        (connection) => connection.sessionId === sessionId ? {
+          ...connection,
+          toolPolicy: result.toolPolicy,
+          tools: result.tools,
+          updatedAt: /* @__PURE__ */ new Date()
+        } : connection
+      ));
+    }
+    return result;
+  }, []);
+  const getToolAccess = useCallback(async (sessionId) => {
+    if (!clientRef.current) {
+      throw new Error("SSE client not initialized");
+    }
+    return await clientRef.current.getToolPolicy(sessionId);
+  }, []);
   const listPrompts = useCallback(async (sessionId) => {
     if (!clientRef.current) {
       throw new Error("SSE client not initialized");
@@ -598,6 +628,8 @@ function useMcp(options) {
       resumeAuth,
       callTool,
       listTools,
+      updateToolPolicy,
+      getToolAccess,
       listPrompts,
       getPrompt,
       listResources,
@@ -622,6 +654,8 @@ function useMcp(options) {
       resumeAuth,
       callTool,
       listTools,
+      updateToolPolicy,
+      getToolAccess,
       listPrompts,
       getPrompt,
       listResources,
@@ -632,12 +666,8 @@ function useMcp(options) {
 }
 var OAUTH_STATE_SEPARATOR = ".";
 customAlphabet(
-  "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  1
-);
-customAlphabet(
-  "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789",
-  11
+  "abcdefghijklmnopqrstuvwxyz0123456789",
+  12
 );
 function parseOAuthState(state) {
   const separatorIndex = state.indexOf(OAUTH_STATE_SEPARATOR);