refactor: pass specTypeSchema() directly to ExtensionHandle (typescript-sdk#1846 StandardSchemaV1 widening)

setCustomRequestHandler/ExtensionHandle now accept StandardSchemaV1, so
specTypeSchema('Name') can be passed directly without z.custom wrapping.
Drops zod import from app.ts/app-bridge.ts where no longer needed.

Author: felixweinberger

package-lock.json

@@ -193,16 +193,16 @@ function postProcess(content: string): string {
   content = content.replace(
     'import { z } from "zod";',
     `import { z } from "zod/v4";
-import { isSpecType } from "@modelcontextprotocol/client";
+import { specTypeSchema } from "@modelcontextprotocol/client";
 import type { ${typeImports} } from "@modelcontextprotocol/client";`,
   );
 
-  // 2. Replace z.any() placeholders for external SDK types with isSpecType-backed z.custom
+  // 2. Replace z.any() placeholders for external SDK types with specTypeSchema()
   for (const schema of EXTERNAL_TYPE_SCHEMAS) {
     const typeName = schema.replace(/Schema$/, "");
     content = content.replace(
       new RegExp(`((?:export )?const ${schema}) = z\\.any\\(\\);`, "g"),
-      `$1 = z.custom<${typeName}>((v) => isSpecType("${typeName}", v));`,
+      `$1 = specTypeSchema("${typeName}");`,
     );
   }
 

scripts/generate-schemas.ts

@@ -26,7 +26,7 @@ import {
 } from "@modelcontextprotocol/server";
 
 import { EventDispatcher } from "./events";
-import { isSpecType } from "@modelcontextprotocol/server";
+import { specTypeSchema } from "@modelcontextprotocol/server";
 import {
   LATEST_PROTOCOL_VERSION,
   McpUiAppCapabilities,
@@ -93,9 +93,7 @@ function toExtra(ctx: ServerContext): RequestHandlerExtra {
   return { signal: ctx.mcpReq.signal };
 }
 
-const LogParamsSchema = z.custom<LoggingMessageNotification["params"]>(
-  (v) => v != null && typeof v === "object",
-);
+const LogParamsSchema = specTypeSchema("LoggingMessageNotificationParams");
 
 /** Options for constructing an {@link AppBridge `AppBridge`}. */
 export interface HostOptions extends ProtocolOptions {
@@ -634,7 +632,7 @@ export class AppBridge extends EventDispatcher<AppBridgeEventMap> {
     return this.ui.sendRequest(
       "ui/call-view-tool",
       params,
-      z.custom<CallToolResult>((v) => isSpecType("CallToolResult", v)),
+      specTypeSchema("CallToolResult"),
       options,
     );
   }
@@ -648,7 +646,7 @@ export class AppBridge extends EventDispatcher<AppBridgeEventMap> {
     return this.ui.sendRequest(
       "ui/list-view-tools",
       params,
-      z.custom<ListToolsResult>((v) => isSpecType("ListToolsResult", v)),
+      specTypeSchema("ListToolsResult"),
       options,
     );
   }

src/app-bridge.ts

@@ -1,4 +1,4 @@
-import {
+specTypeSchema("PaginatedRequestParams"),specTypeSchema("PaginatedRequestParams"),import {
   Client,
   type CallToolRequest,
   type ClientContext,
@@ -20,7 +20,7 @@ import {
 import { EventDispatcher } from "./events";
 export { EventDispatcher, ProtocolWithEvents } from "./events";
 import { PostMessageTransport } from "./message-transport";
-import { isSpecType } from "@modelcontextprotocol/client";
+import { specTypeSchema } from "@modelcontextprotocol/client";
 import { z } from "zod/v4";
 import {
   LATEST_PROTOCOL_VERSION,
@@ -226,9 +226,7 @@ export class App extends EventDispatcher<AppEventMap> {
     // Non-spec host→iframe tool surface (renamed from tools/call & tools/list).
     this.ui.setRequestHandler(
       "ui/call-view-tool",
-      z.custom<CallToolRequest["params"]>((v) =>
-        isSpecType("CallToolRequestParams", v),
-      ),
+      specTypeSchema("CallToolRequestParams"),
       async (params, ctx) => {
         if (!this._oncalltool) throw new Error("No oncalltool handler set");
         return this._oncalltool(params, toExtra(ctx));
@@ -430,7 +428,7 @@ export class App extends EventDispatcher<AppEventMap> {
     return this.ui.sendRequest(
       "ui/update-model-context",
       params,
-      z.custom<Record<string, never>>((v) => isSpecType("EmptyResult", v)),
+      specTypeSchema("EmptyResult"),
       options,
     );
   }