feat: add .strict() transform and re-export JSON-RPC schemas

Add post-processing transform to append .strict() to JSON-RPC schemas:
- JSONRPCRequestSchema
- JSONRPCNotificationSchema
- JSONRPCResultResponseSchema
- JSONRPCErrorResponseSchema

This makes them compatible with types.ts's stricter validation, allowing re-export.

Total schemas re-exported: 49
Lines saved: ~452 (2600 → 2148)

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: ochafik

scripts/generate-schemas.ts

@@ -89,6 +89,16 @@ const FIELD_OVERRIDES: Record<string, Record<string, string>> = {
  */
 const INTEGER_SCHEMAS = ['ProgressTokenSchema', 'RequestIdSchema'];
 
+/**
+ * Schemas that need .strict() added for stricter validation.
+ */
+const STRICT_SCHEMAS = [
+    'JSONRPCRequestSchema',
+    'JSONRPCNotificationSchema',
+    'JSONRPCResultResponseSchema',
+    'JSONRPCErrorResponseSchema',
+];
+
 // =============================================================================
 // Pre-processing: Transform spec types to SDK-compatible hierarchy
 // =============================================================================
@@ -261,6 +271,7 @@ const AST_TRANSFORMS: Transform[] = [
     transformIntegerRefinements,
     transformUnionToEnum,
     applyFieldOverrides,
+    addStrictToSchemas,
 ];
 
 /**
@@ -472,6 +483,25 @@ function applyFieldOverrides(sourceFile: SourceFile): void {
     }
 }
 
+/**
+ * Add .strict() to specified schemas for stricter validation.
+ * This matches the SDK's behavior of rejecting unknown properties.
+ */
+function addStrictToSchemas(sourceFile: SourceFile): void {
+    for (const schemaName of STRICT_SCHEMAS) {
+        const varDecl = sourceFile.getVariableDeclaration(schemaName);
+        if (!varDecl) continue;
+
+        const initializer = varDecl.getInitializer();
+        if (!initializer) continue;
+
+        // Append .strict() to the schema
+        const currentText = initializer.getText();
+        varDecl.setInitializer(`${currentText}.strict()`);
+        console.log(`    ✓ Added .strict() to ${schemaName}`);
+    }
+}
+
 // =============================================================================
 // Main
 // =============================================================================

src/generated/sdk.schemas.ts

@@ -132,29 +132,33 @@ export const RequestSchema = z.object({
  */
 export const JSONRPCNotificationSchema = NotificationSchema.extend({
     jsonrpc: z.literal('2.0')
-});
+}).strict();
 
 /**
  * A successful (non-error) response to a request.
  *
  * @category JSON-RPC
  */
-export const JSONRPCResultResponseSchema = z.object({
-    jsonrpc: z.literal('2.0'),
-    id: RequestIdSchema,
-    result: ResultSchema
-});
+export const JSONRPCResultResponseSchema = z
+    .object({
+        jsonrpc: z.literal('2.0'),
+        id: RequestIdSchema,
+        result: ResultSchema
+    })
+    .strict();
 
 /**
  * A response to a request that indicates an error occurred.
  *
  * @category JSON-RPC
  */
-export const JSONRPCErrorResponseSchema = z.object({
-    jsonrpc: z.literal('2.0'),
-    id: RequestIdSchema.optional(),
-    error: ErrorSchema
-});
+export const JSONRPCErrorResponseSchema = z
+    .object({
+        jsonrpc: z.literal('2.0'),
+        id: RequestIdSchema.optional(),
+        error: ErrorSchema
+    })
+    .strict();
 
 /**
  * A response to a request, containing either the result or error.
@@ -1964,7 +1968,7 @@ export const ListToolsResultSchema = PaginatedResultSchema.extend({
 export const JSONRPCRequestSchema = RequestSchema.extend({
     jsonrpc: z.literal('2.0'),
     id: RequestIdSchema
-});
+}).strict();
 
 /**
  * An error response that indicates that the server requires the client to provide additional information via an elicitation request.

src/types.ts

@@ -70,6 +70,11 @@ import {
     PromptListChangedNotificationSchema,
     ToolListChangedNotificationSchema,
     RootsListChangedNotificationSchema,
+    // JSON-RPC schemas (with .strict() for validation)
+    JSONRPCRequestSchema,
+    JSONRPCNotificationSchema,
+    JSONRPCResultResponseSchema,
+    JSONRPCErrorResponseSchema,
 } from './generated/sdk.schemas.js';
 
 // Alias RequestParamsSchema to BaseRequestParamsSchema for internal use
@@ -122,6 +127,10 @@ export {
     PromptListChangedNotificationSchema,
     ToolListChangedNotificationSchema,
     RootsListChangedNotificationSchema,
+    JSONRPCRequestSchema,
+    JSONRPCNotificationSchema,
+    JSONRPCResultResponseSchema,
+    JSONRPCErrorResponseSchema,
 };
 
 export const LATEST_PROTOCOL_VERSION = '2025-11-25';
@@ -182,47 +191,15 @@ const RequestMetaSchema = z.looseObject({
 export const isTaskAugmentedRequestParams = (value: unknown): value is TaskAugmentedRequestParams =>
     TaskAugmentedRequestParamsSchema.safeParse(value).success;
 
-// Note: RequestSchema, NotificationSchema, ResultSchema are re-exported from generated.
-// They include proper _meta typing with RELATED_TASK_META_KEY.
-
-/**
- * A request that expects a response.
- */
-export const JSONRPCRequestSchema = z
-    .object({
-        jsonrpc: z.literal(JSONRPC_VERSION),
-        id: RequestIdSchema,
-        ...RequestSchema.shape
-    })
-    .strict();
+// Note: RequestSchema, NotificationSchema, ResultSchema, and JSON-RPC schemas
+// are re-exported from generated. They include proper _meta typing and .strict().
 
 export const isJSONRPCRequest = (value: unknown): value is JSONRPCRequest => JSONRPCRequestSchema.safeParse(value).success;
-
-/**
- * A notification which does not expect a response.
- */
-export const JSONRPCNotificationSchema = z
-    .object({
-        jsonrpc: z.literal(JSONRPC_VERSION),
-        ...NotificationSchema.shape
-    })
-    .strict();
-
 export const isJSONRPCNotification = (value: unknown): value is JSONRPCNotification => JSONRPCNotificationSchema.safeParse(value).success;
-
-/**
- * A successful (non-error) response to a request.
- */
-export const JSONRPCResultResponseSchema = z
-    .object({
-        jsonrpc: z.literal(JSONRPC_VERSION),
-        id: RequestIdSchema,
-        result: ResultSchema
-    })
-    .strict();
-
 export const isJSONRPCResultResponse = (value: unknown): value is JSONRPCResultResponse =>
     JSONRPCResultResponseSchema.safeParse(value).success;
+export const isJSONRPCErrorResponse = (value: unknown): value is JSONRPCErrorResponse =>
+    JSONRPCErrorResponseSchema.safeParse(value).success;
 
 /**
  * Error codes defined by the JSON-RPC specification.
@@ -243,33 +220,6 @@ export enum ErrorCode {
     UrlElicitationRequired = -32042
 }
 
-/**
- * A response to a request that indicates an error occurred.
- */
-export const JSONRPCErrorResponseSchema = z
-    .object({
-        jsonrpc: z.literal(JSONRPC_VERSION),
-        id: RequestIdSchema.optional(),
-        error: z.object({
-            /**
-             * The error type that occurred.
-             */
-            code: z.number().int(),
-            /**
-             * A short description of the error. The message SHOULD be limited to a concise single sentence.
-             */
-            message: z.string(),
-            /**
-             * Additional information about the error. The value of this member is defined by the sender (e.g. detailed error information, nested errors etc.).
-             */
-            data: z.unknown().optional()
-        })
-    })
-    .strict();
-
-export const isJSONRPCErrorResponse = (value: unknown): value is JSONRPCErrorResponse =>
-    JSONRPCErrorResponseSchema.safeParse(value).success;
-
 export const JSONRPCMessageSchema = z.union([
     JSONRPCRequestSchema,
     JSONRPCNotificationSchema,