Fix TypeScript build errors

- Remove duplicate 'sampling' property in ClientCapabilitiesSchema (types.ts:299)
- Add type guards for union types in backfillSampling.ts:
  - Check 'text' in c.resource before accessing text property
  - Check 'blob' in c.resource before accessing blob property
  - Fix c.resource.data to c.resource.blob for PDF (blob is the correct property)
  - Fix c.url to c.uri for resource_link (uri is the correct property)
- Fix toolLoop.ts:
  - Change tool_choice to toolChoice (camelCase convention)
  - Add type guard for text blocks when joining content
- Fix toolRegistry.ts:
  - Remove .shape access on Zod schemas (not available on ZodType)
  - Add type assertion for callback to handle optional inputSchema

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

Co-Authored-By: Claude <[email protected]>

Author: ochafik

src/examples/backfill/backfillSampling.ts

@@ -256,16 +256,16 @@ function contentBlockFromMcp(content: AssistantMessageContent | UserMessageConte
                     } else if (c.type === 'image') {
                         return makeImageBlock(c.data, c.mimeType);
                     } else if (c.type === 'resource') {
-                        if (c.resource.mimeType === 'text/plain') {
+                        if (c.resource.mimeType === 'text/plain' && 'text' in c.resource) {
                             return <TextBlockParam>{type: 'text', text: c.resource.text};
-                        } else if (c.resource.mimeType?.startsWith('image/') && typeof c.resource.blob === 'string') {
+                        } else if (c.resource.mimeType?.startsWith('image/') && 'blob' in c.resource && typeof c.resource.blob === 'string') {
                             return makeImageBlock(c.resource.blob, c.resource.mimeType);
-                        } else if (c.resource.mimeType === 'application/pdf') {
+                        } else if (c.resource.mimeType === 'application/pdf' && 'blob' in c.resource) {
                             return <DocumentBlockParam>{
                                 type: 'document',
                                 source: {
                                     type: 'base64',
-                                    data: c.resource.data,
+                                    data: c.resource.blob,
                                     media_type: 'application/pdf',
                                 },
                             };
@@ -281,7 +281,7 @@ function contentBlockFromMcp(content: AssistantMessageContent | UserMessageConte
                             type: 'document',
                             source: {
                                 type: 'url',
-                                url: c.url,
+                                url: c.uri,
                             },
                         }
                     } else {

src/examples/server/toolLoop.ts

@@ -80,7 +80,7 @@ export async function runToolLoop(
       maxTokens: 4000,
       tools: iteration < maxIterations ? options.registry.tools : undefined,
       // Don't allow tool calls at the last iteration: finish with an answer no matter what!
-      tool_choice: iteration < maxIterations ? defaultToolChoice : { mode: "none" },
+      toolChoice: iteration < maxIterations ? defaultToolChoice : { mode: "none" },
     });
 
     // Aggregate usage statistics from the response
@@ -138,14 +138,14 @@ export async function runToolLoop(
       if (unexpectedBlocks.length > 0) {
         throw new Error(`Expected text content in final answer, but got: ${unexpectedBlocks.map(b => b.type).join(", ")}`);
       }
-      
+
       await options.server.sendLoggingMessage({
         level: "info",
         data: `Tool loop completed after ${iteration} iteration(s)`,
       });
 
       return {
-        answer: contentArray.map(block => block.text).join("\n\n"),
+        answer: contentArray.map(block => block.type === 'text' ? block.text : '').join("\n\n"),
         transcript: messages,
         usage
       };

src/examples/server/toolRegistry.ts

@@ -32,11 +32,11 @@ export class ToolRegistry {
       server.registerTool(name, {
         title: tool.title,
         description: tool.description,
-        inputSchema: tool.inputSchema?.shape,
-        outputSchema: tool.outputSchema?.shape,
+        inputSchema: tool.inputSchema,
+        outputSchema: tool.outputSchema,
         annotations: tool.annotations,
         _meta: tool._meta,
-      }, tool.callback);
+      }, tool.callback as any);
     }
   }
 

src/types.ts

@@ -296,7 +296,6 @@ export const ClientCapabilitiesSchema = z.object({
             tools: AssertObjectSchema.optional(),
         })
         .optional(),
-    sampling: AssertObjectSchema.optional(),
     /**
      * Present if the client supports eliciting user input.
      */