From 9b5efc428674b964eef32b4db3f5205bdebb60f3 Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Fri, 3 Jan 2025 13:47:19 +0000
Subject: [PATCH 1/5] Recursively flatten types for better inference

---
 src/types.ts | 191 +++++++++++++++++++++++++--------------------------
 1 file changed, 93 insertions(+), 98 deletions(-)

diff --git a/src/types.ts b/src/types.ts
index 59b055a87..e13899e09 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -106,7 +106,7 @@ export enum ErrorCode {
   // SDK error codes
   ConnectionClosed = -32000,
   RequestTimeout = -32001,
- 
+
   // Standard JSON-RPC error codes
   ParseError = -32700,
   InvalidRequest = -32600,
@@ -1110,131 +1110,126 @@ export class McpError extends Error {
   }
 }
 
+type Primitive = string | number | boolean | bigint | null | undefined;
+type Flatten<T> = T extends Primitive
+  ? T
+  : T extends Array<infer U>
+  ? Array<Flatten<U>>
+  : T extends Set<infer U>
+  ? Set<Flatten<U>>
+  : T extends Map<infer K, infer V>
+  ? Map<Flatten<K>, Flatten<V>>
+  : T extends object
+  ? { [K in keyof T]: Flatten<T[K]> }
+  : T;
+
+type Infer<Parse extends (...args: any) => any> = Flatten<ReturnType<Parse>>;
+
 /* JSON-RPC types */
-export type ProgressToken = z.infer<typeof ProgressTokenSchema>;
-export type Cursor = z.infer<typeof CursorSchema>;
-export type Request = z.infer<typeof RequestSchema>;
-export type Notification = z.infer<typeof NotificationSchema>;
-export type Result = z.infer<typeof ResultSchema>;
-export type RequestId = z.infer<typeof RequestIdSchema>;
-export type JSONRPCRequest = z.infer<typeof JSONRPCRequestSchema>;
-export type JSONRPCNotification = z.infer<typeof JSONRPCNotificationSchema>;
-export type JSONRPCResponse = z.infer<typeof JSONRPCResponseSchema>;
-export type JSONRPCError = z.infer<typeof JSONRPCErrorSchema>;
-export type JSONRPCMessage = z.infer<typeof JSONRPCMessageSchema>;
+export type ProgressToken = Infer<typeof ProgressTokenSchema.parse>;
+export type Cursor = Infer<typeof CursorSchema.parse>;
+export type Request = Infer<typeof RequestSchema.parse>;
+export type Notification = Infer<typeof NotificationSchema.parse>;
+export type Result = Infer<typeof ResultSchema.parse>;
+export type RequestId = Infer<typeof RequestIdSchema.parse>;
+export type JSONRPCRequest = Infer<typeof JSONRPCRequestSchema.parse>;
+export type JSONRPCNotification = Infer<typeof JSONRPCNotificationSchema.parse>;
+export type JSONRPCResponse = Infer<typeof JSONRPCResponseSchema.parse>;
+export type JSONRPCError = Infer<typeof JSONRPCErrorSchema.parse>;
+export type JSONRPCMessage = Infer<typeof JSONRPCMessageSchema.parse>;
 
 /* Empty result */
-export type EmptyResult = z.infer<typeof EmptyResultSchema>;
+export type EmptyResult = Infer<typeof EmptyResultSchema.parse>;
 
 /* Cancellation */
-export type CancelledNotification = z.infer<typeof CancelledNotificationSchema>;
+export type CancelledNotification = Infer<typeof CancelledNotificationSchema.parse>;
 
 /* Initialization */
-export type Implementation = z.infer<typeof ImplementationSchema>;
-export type ClientCapabilities = z.infer<typeof ClientCapabilitiesSchema>;
-export type InitializeRequest = z.infer<typeof InitializeRequestSchema>;
-export type ServerCapabilities = z.infer<typeof ServerCapabilitiesSchema>;
-export type InitializeResult = z.infer<typeof InitializeResultSchema>;
-export type InitializedNotification = z.infer<
-  typeof InitializedNotificationSchema
->;
+export type Implementation = Infer<typeof ImplementationSchema.parse>;
+export type ClientCapabilities = Infer<typeof ClientCapabilitiesSchema.parse>;
+export type InitializeRequest = Infer<typeof InitializeRequestSchema.parse>;
+export type ServerCapabilities = Infer<typeof ServerCapabilitiesSchema.parse>;
+export type InitializeResult = Infer<typeof InitializeResultSchema.parse>;
+export type InitializedNotification = Infer<typeof InitializedNotificationSchema.parse>;
 
 /* Ping */
-export type PingRequest = z.infer<typeof PingRequestSchema>;
+export type PingRequest = Infer<typeof PingRequestSchema.parse>;
 
 /* Progress notifications */
-export type Progress = z.infer<typeof ProgressSchema>;
-export type ProgressNotification = z.infer<typeof ProgressNotificationSchema>;
+export type Progress = Infer<typeof ProgressSchema.parse>;
+export type ProgressNotification = Infer<typeof ProgressNotificationSchema.parse>;
 
 /* Pagination */
-export type PaginatedRequest = z.infer<typeof PaginatedRequestSchema>;
-export type PaginatedResult = z.infer<typeof PaginatedResultSchema>;
+export type PaginatedRequest = Infer<typeof PaginatedRequestSchema.parse>;
+export type PaginatedResult = Infer<typeof PaginatedResultSchema.parse>;
 
 /* Resources */
-export type ResourceContents = z.infer<typeof ResourceContentsSchema>;
-export type TextResourceContents = z.infer<typeof TextResourceContentsSchema>;
-export type BlobResourceContents = z.infer<typeof BlobResourceContentsSchema>;
-export type Resource = z.infer<typeof ResourceSchema>;
-export type ResourceTemplate = z.infer<typeof ResourceTemplateSchema>;
-export type ListResourcesRequest = z.infer<typeof ListResourcesRequestSchema>;
-export type ListResourcesResult = z.infer<typeof ListResourcesResultSchema>;
-export type ListResourceTemplatesRequest = z.infer<
-  typeof ListResourceTemplatesRequestSchema
->;
-export type ListResourceTemplatesResult = z.infer<
-  typeof ListResourceTemplatesResultSchema
->;
-export type ReadResourceRequest = z.infer<typeof ReadResourceRequestSchema>;
-export type ReadResourceResult = z.infer<typeof ReadResourceResultSchema>;
-export type ResourceListChangedNotification = z.infer<
-  typeof ResourceListChangedNotificationSchema
->;
-export type SubscribeRequest = z.infer<typeof SubscribeRequestSchema>;
-export type UnsubscribeRequest = z.infer<typeof UnsubscribeRequestSchema>;
-export type ResourceUpdatedNotification = z.infer<
-  typeof ResourceUpdatedNotificationSchema
->;
+export type ResourceContents = Infer<typeof ResourceContentsSchema.parse>;
+export type TextResourceContents = Infer<typeof TextResourceContentsSchema.parse>;
+export type BlobResourceContents = Infer<typeof BlobResourceContentsSchema.parse>;
+export type Resource = Infer<typeof ResourceSchema.parse>;
+export type ResourceTemplate = Infer<typeof ResourceTemplateSchema.parse>;
+export type ListResourcesRequest = Infer<typeof ListResourcesRequestSchema.parse>;
+export type ListResourcesResult = Infer<typeof ListResourcesResultSchema.parse>;
+export type ListResourceTemplatesRequest = Infer<typeof ListResourceTemplatesRequestSchema.parse>;
+export type ListResourceTemplatesResult = Infer<typeof ListResourceTemplatesResultSchema.parse>;
+export type ReadResourceRequest = Infer<typeof ReadResourceRequestSchema.parse>;
+export type ReadResourceResult = Infer<typeof ReadResourceResultSchema.parse>;
+export type ResourceListChangedNotification = Infer<typeof ResourceListChangedNotificationSchema.parse>;
+export type SubscribeRequest = Infer<typeof SubscribeRequestSchema.parse>;
+export type UnsubscribeRequest = Infer<typeof UnsubscribeRequestSchema.parse>;
+export type ResourceUpdatedNotification = Infer<typeof ResourceUpdatedNotificationSchema.parse>;
 
 /* Prompts */
-export type PromptArgument = z.infer<typeof PromptArgumentSchema>;
-export type Prompt = z.infer<typeof PromptSchema>;
-export type ListPromptsRequest = z.infer<typeof ListPromptsRequestSchema>;
-export type ListPromptsResult = z.infer<typeof ListPromptsResultSchema>;
-export type GetPromptRequest = z.infer<typeof GetPromptRequestSchema>;
-export type TextContent = z.infer<typeof TextContentSchema>;
-export type ImageContent = z.infer<typeof ImageContentSchema>;
-export type EmbeddedResource = z.infer<typeof EmbeddedResourceSchema>;
-export type PromptMessage = z.infer<typeof PromptMessageSchema>;
-export type GetPromptResult = z.infer<typeof GetPromptResultSchema>;
-export type PromptListChangedNotification = z.infer<
-  typeof PromptListChangedNotificationSchema
->;
+export type PromptArgument = Infer<typeof PromptArgumentSchema.parse>;
+export type Prompt = Infer<typeof PromptSchema.parse>;
+export type ListPromptsRequest = Infer<typeof ListPromptsRequestSchema.parse>;
+export type ListPromptsResult = Infer<typeof ListPromptsResultSchema.parse>;
+export type GetPromptRequest = Infer<typeof GetPromptRequestSchema.parse>;
+export type TextContent = Infer<typeof TextContentSchema.parse>;
+export type ImageContent = Infer<typeof ImageContentSchema.parse>;
+export type EmbeddedResource = Infer<typeof EmbeddedResourceSchema.parse>;
+export type PromptMessage = Infer<typeof PromptMessageSchema.parse>;
+export type GetPromptResult = Infer<typeof GetPromptResultSchema.parse>;
+export type PromptListChangedNotification = Infer<typeof PromptListChangedNotificationSchema.parse>;
 
 /* Tools */
-export type Tool = z.infer<typeof ToolSchema>;
-export type ListToolsRequest = z.infer<typeof ListToolsRequestSchema>;
-export type ListToolsResult = z.infer<typeof ListToolsResultSchema>;
-export type CallToolResult = z.infer<typeof CallToolResultSchema>;
-export type CompatibilityCallToolResult = z.infer<
-  typeof CompatibilityCallToolResultSchema
->;
-export type CallToolRequest = z.infer<typeof CallToolRequestSchema>;
-export type ToolListChangedNotification = z.infer<
-  typeof ToolListChangedNotificationSchema
->;
+export type Tool = Infer<typeof ToolSchema.parse>;
+export type ListToolsRequest = Infer<typeof ListToolsRequestSchema.parse>;
+export type ListToolsResult = Infer<typeof ListToolsResultSchema.parse>;
+export type CallToolResult = Infer<typeof CallToolResultSchema.parse>;
+export type CompatibilityCallToolResult = Infer<typeof CompatibilityCallToolResultSchema.parse>;
+export type CallToolRequest = Infer<typeof CallToolRequestSchema.parse>;
+export type ToolListChangedNotification = Infer<typeof ToolListChangedNotificationSchema.parse>;
 
 /* Logging */
-export type LoggingLevel = z.infer<typeof LoggingLevelSchema>;
-export type SetLevelRequest = z.infer<typeof SetLevelRequestSchema>;
-export type LoggingMessageNotification = z.infer<
-  typeof LoggingMessageNotificationSchema
->;
+export type LoggingLevel = Infer<typeof LoggingLevelSchema.parse>;
+export type SetLevelRequest = Infer<typeof SetLevelRequestSchema.parse>;
+export type LoggingMessageNotification = Infer<typeof LoggingMessageNotificationSchema.parse>;
 
 /* Sampling */
-export type SamplingMessage = z.infer<typeof SamplingMessageSchema>;
-export type CreateMessageRequest = z.infer<typeof CreateMessageRequestSchema>;
-export type CreateMessageResult = z.infer<typeof CreateMessageResultSchema>;
+export type SamplingMessage = Infer<typeof SamplingMessageSchema.parse>;
+export type CreateMessageRequest = Infer<typeof CreateMessageRequestSchema.parse>;
+export type CreateMessageResult = Infer<typeof CreateMessageResultSchema.parse>;
 
 /* Autocomplete */
-export type ResourceReference = z.infer<typeof ResourceReferenceSchema>;
-export type PromptReference = z.infer<typeof PromptReferenceSchema>;
-export type CompleteRequest = z.infer<typeof CompleteRequestSchema>;
-export type CompleteResult = z.infer<typeof CompleteResultSchema>;
+export type ResourceReference = Infer<typeof ResourceReferenceSchema.parse>;
+export type PromptReference = Infer<typeof PromptReferenceSchema.parse>;
+export type CompleteRequest = Infer<typeof CompleteRequestSchema.parse>;
+export type CompleteResult = Infer<typeof CompleteResultSchema.parse>;
 
 /* Roots */
-export type Root = z.infer<typeof RootSchema>;
-export type ListRootsRequest = z.infer<typeof ListRootsRequestSchema>;
-export type ListRootsResult = z.infer<typeof ListRootsResultSchema>;
-export type RootsListChangedNotification = z.infer<
-  typeof RootsListChangedNotificationSchema
->;
+export type Root = Infer<typeof RootSchema.parse>;
+export type ListRootsRequest = Infer<typeof ListRootsRequestSchema.parse>;
+export type ListRootsResult = Infer<typeof ListRootsResultSchema.parse>;
+export type RootsListChangedNotification = Infer<typeof RootsListChangedNotificationSchema.parse>;
 
 /* Client messages */
-export type ClientRequest = z.infer<typeof ClientRequestSchema>;
-export type ClientNotification = z.infer<typeof ClientNotificationSchema>;
-export type ClientResult = z.infer<typeof ClientResultSchema>;
+export type ClientRequest = Infer<typeof ClientRequestSchema.parse>;
+export type ClientNotification = Infer<typeof ClientNotificationSchema.parse>;
+export type ClientResult = Infer<typeof ClientResultSchema.parse>;
 
 /* Server messages */
-export type ServerRequest = z.infer<typeof ServerRequestSchema>;
-export type ServerNotification = z.infer<typeof ServerNotificationSchema>;
-export type ServerResult = z.infer<typeof ServerResultSchema>;
\ No newline at end of file
+export type ServerRequest = Infer<typeof ServerRequestSchema.parse>;
+export type ServerNotification = Infer<typeof ServerNotificationSchema.parse>;
+export type ServerResult = Infer<typeof ServerResultSchema.parse>;

From 160337cb3eb305a316111ca2bfa01cd27180fc23 Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Fri, 3 Jan 2025 13:48:31 +0000
Subject: [PATCH 2/5] Update package-lock.json

---
 package-lock.json | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 71b3ec927..5e68c8cd5 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@modelcontextprotocol/sdk",
-  "version": "1.0.3",
+  "version": "1.1.0",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "@modelcontextprotocol/sdk",
-      "version": "1.0.3",
+      "version": "1.1.0",
       "license": "MIT",
       "dependencies": {
         "content-type": "^1.0.5",

From 73ea68cf13ac0c90fe34b744de133fccf6d3adc5 Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Fri, 3 Jan 2025 13:49:07 +0000
Subject: [PATCH 3/5] Disable ESLint warning

---
 src/types.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/types.ts b/src/types.ts
index e13899e09..60b13f49b 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1123,6 +1123,7 @@ type Flatten<T> = T extends Primitive
   ? { [K in keyof T]: Flatten<T[K]> }
   : T;
 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
 type Infer<Parse extends (...args: any) => any> = Flatten<ReturnType<Parse>>;
 
 /* JSON-RPC types */

From b5711be1ad2ef861c3eeb52502b9d3370006b47c Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Fri, 3 Jan 2025 14:01:17 +0000
Subject: [PATCH 4/5] Can just reuse `z.infer` instead of using `ReturnType`

---
 src/types.ts | 159 +++++++++++++++++++++++++--------------------------
 1 file changed, 79 insertions(+), 80 deletions(-)

diff --git a/src/types.ts b/src/types.ts
index 60b13f49b..39177a6dc 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1,4 +1,4 @@
-import { z } from "zod";
+import { z, ZodTypeAny } from "zod";
 
 export const LATEST_PROTOCOL_VERSION = "2024-11-05";
 export const SUPPORTED_PROTOCOL_VERSIONS = [
@@ -1123,114 +1123,113 @@ type Flatten<T> = T extends Primitive
   ? { [K in keyof T]: Flatten<T[K]> }
   : T;
 
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-type Infer<Parse extends (...args: any) => any> = Flatten<ReturnType<Parse>>;
+type Infer<Schema extends ZodTypeAny> = Flatten<z.infer<Schema>>;
 
 /* JSON-RPC types */
-export type ProgressToken = Infer<typeof ProgressTokenSchema.parse>;
-export type Cursor = Infer<typeof CursorSchema.parse>;
-export type Request = Infer<typeof RequestSchema.parse>;
-export type Notification = Infer<typeof NotificationSchema.parse>;
-export type Result = Infer<typeof ResultSchema.parse>;
-export type RequestId = Infer<typeof RequestIdSchema.parse>;
-export type JSONRPCRequest = Infer<typeof JSONRPCRequestSchema.parse>;
-export type JSONRPCNotification = Infer<typeof JSONRPCNotificationSchema.parse>;
-export type JSONRPCResponse = Infer<typeof JSONRPCResponseSchema.parse>;
-export type JSONRPCError = Infer<typeof JSONRPCErrorSchema.parse>;
-export type JSONRPCMessage = Infer<typeof JSONRPCMessageSchema.parse>;
+export type ProgressToken = Infer<typeof ProgressTokenSchema>;
+export type Cursor = Infer<typeof CursorSchema>;
+export type Request = Infer<typeof RequestSchema>;
+export type Notification = Infer<typeof NotificationSchema>;
+export type Result = Infer<typeof ResultSchema>;
+export type RequestId = Infer<typeof RequestIdSchema>;
+export type JSONRPCRequest = Infer<typeof JSONRPCRequestSchema>;
+export type JSONRPCNotification = Infer<typeof JSONRPCNotificationSchema>;
+export type JSONRPCResponse = Infer<typeof JSONRPCResponseSchema>;
+export type JSONRPCError = Infer<typeof JSONRPCErrorSchema>;
+export type JSONRPCMessage = Infer<typeof JSONRPCMessageSchema>;
 
 /* Empty result */
-export type EmptyResult = Infer<typeof EmptyResultSchema.parse>;
+export type EmptyResult = Infer<typeof EmptyResultSchema>;
 
 /* Cancellation */
-export type CancelledNotification = Infer<typeof CancelledNotificationSchema.parse>;
+export type CancelledNotification = Infer<typeof CancelledNotificationSchema>;
 
 /* Initialization */
-export type Implementation = Infer<typeof ImplementationSchema.parse>;
-export type ClientCapabilities = Infer<typeof ClientCapabilitiesSchema.parse>;
-export type InitializeRequest = Infer<typeof InitializeRequestSchema.parse>;
-export type ServerCapabilities = Infer<typeof ServerCapabilitiesSchema.parse>;
-export type InitializeResult = Infer<typeof InitializeResultSchema.parse>;
-export type InitializedNotification = Infer<typeof InitializedNotificationSchema.parse>;
+export type Implementation = Infer<typeof ImplementationSchema>;
+export type ClientCapabilities = Infer<typeof ClientCapabilitiesSchema>;
+export type InitializeRequest = Infer<typeof InitializeRequestSchema>;
+export type ServerCapabilities = Infer<typeof ServerCapabilitiesSchema>;
+export type InitializeResult = Infer<typeof InitializeResultSchema>;
+export type InitializedNotification = Infer<typeof InitializedNotificationSchema>;
 
 /* Ping */
-export type PingRequest = Infer<typeof PingRequestSchema.parse>;
+export type PingRequest = Infer<typeof PingRequestSchema>;
 
 /* Progress notifications */
-export type Progress = Infer<typeof ProgressSchema.parse>;
-export type ProgressNotification = Infer<typeof ProgressNotificationSchema.parse>;
+export type Progress = Infer<typeof ProgressSchema>;
+export type ProgressNotification = Infer<typeof ProgressNotificationSchema>;
 
 /* Pagination */
-export type PaginatedRequest = Infer<typeof PaginatedRequestSchema.parse>;
-export type PaginatedResult = Infer<typeof PaginatedResultSchema.parse>;
+export type PaginatedRequest = Infer<typeof PaginatedRequestSchema>;
+export type PaginatedResult = Infer<typeof PaginatedResultSchema>;
 
 /* Resources */
-export type ResourceContents = Infer<typeof ResourceContentsSchema.parse>;
-export type TextResourceContents = Infer<typeof TextResourceContentsSchema.parse>;
-export type BlobResourceContents = Infer<typeof BlobResourceContentsSchema.parse>;
-export type Resource = Infer<typeof ResourceSchema.parse>;
-export type ResourceTemplate = Infer<typeof ResourceTemplateSchema.parse>;
-export type ListResourcesRequest = Infer<typeof ListResourcesRequestSchema.parse>;
-export type ListResourcesResult = Infer<typeof ListResourcesResultSchema.parse>;
-export type ListResourceTemplatesRequest = Infer<typeof ListResourceTemplatesRequestSchema.parse>;
-export type ListResourceTemplatesResult = Infer<typeof ListResourceTemplatesResultSchema.parse>;
-export type ReadResourceRequest = Infer<typeof ReadResourceRequestSchema.parse>;
-export type ReadResourceResult = Infer<typeof ReadResourceResultSchema.parse>;
-export type ResourceListChangedNotification = Infer<typeof ResourceListChangedNotificationSchema.parse>;
-export type SubscribeRequest = Infer<typeof SubscribeRequestSchema.parse>;
-export type UnsubscribeRequest = Infer<typeof UnsubscribeRequestSchema.parse>;
-export type ResourceUpdatedNotification = Infer<typeof ResourceUpdatedNotificationSchema.parse>;
+export type ResourceContents = Flatten<z.infer<typeof ResourceContentsSchema>>;
+export type TextResourceContents = Infer<typeof TextResourceContentsSchema>;
+export type BlobResourceContents = Infer<typeof BlobResourceContentsSchema>;
+export type Resource = Infer<typeof ResourceSchema>;
+export type ResourceTemplate = Infer<typeof ResourceTemplateSchema>;
+export type ListResourcesRequest = Infer<typeof ListResourcesRequestSchema>;
+export type ListResourcesResult = Infer<typeof ListResourcesResultSchema>;
+export type ListResourceTemplatesRequest = Infer<typeof ListResourceTemplatesRequestSchema>;
+export type ListResourceTemplatesResult = Infer<typeof ListResourceTemplatesResultSchema>;
+export type ReadResourceRequest = Infer<typeof ReadResourceRequestSchema>;
+export type ReadResourceResult = Infer<typeof ReadResourceResultSchema>;
+export type ResourceListChangedNotification = Infer<typeof ResourceListChangedNotificationSchema>;
+export type SubscribeRequest = Infer<typeof SubscribeRequestSchema>;
+export type UnsubscribeRequest = Infer<typeof UnsubscribeRequestSchema>;
+export type ResourceUpdatedNotification = Infer<typeof ResourceUpdatedNotificationSchema>;
 
 /* Prompts */
-export type PromptArgument = Infer<typeof PromptArgumentSchema.parse>;
-export type Prompt = Infer<typeof PromptSchema.parse>;
-export type ListPromptsRequest = Infer<typeof ListPromptsRequestSchema.parse>;
-export type ListPromptsResult = Infer<typeof ListPromptsResultSchema.parse>;
-export type GetPromptRequest = Infer<typeof GetPromptRequestSchema.parse>;
-export type TextContent = Infer<typeof TextContentSchema.parse>;
-export type ImageContent = Infer<typeof ImageContentSchema.parse>;
-export type EmbeddedResource = Infer<typeof EmbeddedResourceSchema.parse>;
-export type PromptMessage = Infer<typeof PromptMessageSchema.parse>;
-export type GetPromptResult = Infer<typeof GetPromptResultSchema.parse>;
-export type PromptListChangedNotification = Infer<typeof PromptListChangedNotificationSchema.parse>;
+export type PromptArgument = Infer<typeof PromptArgumentSchema>;
+export type Prompt = Infer<typeof PromptSchema>;
+export type ListPromptsRequest = Infer<typeof ListPromptsRequestSchema>;
+export type ListPromptsResult = Infer<typeof ListPromptsResultSchema>;
+export type GetPromptRequest = Infer<typeof GetPromptRequestSchema>;
+export type TextContent = Infer<typeof TextContentSchema>;
+export type ImageContent = Infer<typeof ImageContentSchema>;
+export type EmbeddedResource = Infer<typeof EmbeddedResourceSchema>;
+export type PromptMessage = Infer<typeof PromptMessageSchema>;
+export type GetPromptResult = Infer<typeof GetPromptResultSchema>;
+export type PromptListChangedNotification = Infer<typeof PromptListChangedNotificationSchema>;
 
 /* Tools */
-export type Tool = Infer<typeof ToolSchema.parse>;
-export type ListToolsRequest = Infer<typeof ListToolsRequestSchema.parse>;
-export type ListToolsResult = Infer<typeof ListToolsResultSchema.parse>;
-export type CallToolResult = Infer<typeof CallToolResultSchema.parse>;
-export type CompatibilityCallToolResult = Infer<typeof CompatibilityCallToolResultSchema.parse>;
-export type CallToolRequest = Infer<typeof CallToolRequestSchema.parse>;
-export type ToolListChangedNotification = Infer<typeof ToolListChangedNotificationSchema.parse>;
+export type Tool = Infer<typeof ToolSchema>;
+export type ListToolsRequest = Infer<typeof ListToolsRequestSchema>;
+export type ListToolsResult = Infer<typeof ListToolsResultSchema>;
+export type CallToolResult = Infer<typeof CallToolResultSchema>;
+export type CompatibilityCallToolResult = Infer<typeof CompatibilityCallToolResultSchema>;
+export type CallToolRequest = Infer<typeof CallToolRequestSchema>;
+export type ToolListChangedNotification = Infer<typeof ToolListChangedNotificationSchema>;
 
 /* Logging */
-export type LoggingLevel = Infer<typeof LoggingLevelSchema.parse>;
-export type SetLevelRequest = Infer<typeof SetLevelRequestSchema.parse>;
-export type LoggingMessageNotification = Infer<typeof LoggingMessageNotificationSchema.parse>;
+export type LoggingLevel = Infer<typeof LoggingLevelSchema>;
+export type SetLevelRequest = Infer<typeof SetLevelRequestSchema>;
+export type LoggingMessageNotification = Infer<typeof LoggingMessageNotificationSchema>;
 
 /* Sampling */
-export type SamplingMessage = Infer<typeof SamplingMessageSchema.parse>;
-export type CreateMessageRequest = Infer<typeof CreateMessageRequestSchema.parse>;
-export type CreateMessageResult = Infer<typeof CreateMessageResultSchema.parse>;
+export type SamplingMessage = Infer<typeof SamplingMessageSchema>;
+export type CreateMessageRequest = Infer<typeof CreateMessageRequestSchema>;
+export type CreateMessageResult = Infer<typeof CreateMessageResultSchema>;
 
 /* Autocomplete */
-export type ResourceReference = Infer<typeof ResourceReferenceSchema.parse>;
-export type PromptReference = Infer<typeof PromptReferenceSchema.parse>;
-export type CompleteRequest = Infer<typeof CompleteRequestSchema.parse>;
-export type CompleteResult = Infer<typeof CompleteResultSchema.parse>;
+export type ResourceReference = Infer<typeof ResourceReferenceSchema>;
+export type PromptReference = Infer<typeof PromptReferenceSchema>;
+export type CompleteRequest = Infer<typeof CompleteRequestSchema>;
+export type CompleteResult = Infer<typeof CompleteResultSchema>;
 
 /* Roots */
-export type Root = Infer<typeof RootSchema.parse>;
-export type ListRootsRequest = Infer<typeof ListRootsRequestSchema.parse>;
-export type ListRootsResult = Infer<typeof ListRootsResultSchema.parse>;
-export type RootsListChangedNotification = Infer<typeof RootsListChangedNotificationSchema.parse>;
+export type Root = Infer<typeof RootSchema>;
+export type ListRootsRequest = Infer<typeof ListRootsRequestSchema>;
+export type ListRootsResult = Infer<typeof ListRootsResultSchema>;
+export type RootsListChangedNotification = Infer<typeof RootsListChangedNotificationSchema>;
 
 /* Client messages */
-export type ClientRequest = Infer<typeof ClientRequestSchema.parse>;
-export type ClientNotification = Infer<typeof ClientNotificationSchema.parse>;
-export type ClientResult = Infer<typeof ClientResultSchema.parse>;
+export type ClientRequest = Infer<typeof ClientRequestSchema>;
+export type ClientNotification = Infer<typeof ClientNotificationSchema>;
+export type ClientResult = Infer<typeof ClientResultSchema>;
 
 /* Server messages */
-export type ServerRequest = Infer<typeof ServerRequestSchema.parse>;
-export type ServerNotification = Infer<typeof ServerNotificationSchema.parse>;
-export type ServerResult = Infer<typeof ServerResultSchema.parse>;
+export type ServerRequest = Infer<typeof ServerRequestSchema>;
+export type ServerNotification = Infer<typeof ServerNotificationSchema>;
+export type ServerResult = Infer<typeof ServerResultSchema>;

From 0c5d6c4f18bd2ce3d67da2df738f971bd28e14b2 Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Mon, 6 Jan 2025 21:37:30 +0000
Subject: [PATCH 5/5] Missed one

---
 src/types.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/types.ts b/src/types.ts
index 39177a6dc..15acd1491 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1164,7 +1164,7 @@ export type PaginatedRequest = Infer<typeof PaginatedRequestSchema>;
 export type PaginatedResult = Infer<typeof PaginatedResultSchema>;
 
 /* Resources */
-export type ResourceContents = Flatten<z.infer<typeof ResourceContentsSchema>>;
+export type ResourceContents = Infer<typeof ResourceContentsSchema>;
 export type TextResourceContents = Infer<typeof TextResourceContentsSchema>;
 export type BlobResourceContents = Infer<typeof BlobResourceContentsSchema>;
 export type Resource = Infer<typeof ResourceSchema>;