From b400ce5782916211744fa3792dbb4f864549a50b Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Wed, 2 Oct 2024 14:22:42 +0100
Subject: [PATCH] Mark numbers as integers where possible

---
 src/types.ts | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/src/types.ts b/src/types.ts
index 39d5756f..5f0a62ec 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -13,7 +13,7 @@ export const JSONRPC_VERSION = "2.0";
 /**
  * A progress token, used to associate progress notifications with the original request.
  */
-export const ProgressTokenSchema = z.union([z.string(), z.number()]);
+export const ProgressTokenSchema = z.union([z.string(), z.number().int()]);
 
 export const RequestSchema = z.object({
   method: z.string(),
@@ -47,7 +47,7 @@ export const ResultSchema = z.object({
 /**
  * A uniquely identifying ID for a request in JSON-RPC.
  */
-export const RequestIdSchema = z.union([z.string(), z.number()]);
+export const RequestIdSchema = z.union([z.string(), z.number().int()]);
 
 /**
  * A request that expects a response.
@@ -90,7 +90,7 @@ export const JSONRPCErrorSchema = z.object({
     /**
      * The error type that occurred.
      */
-    code: z.number(),
+    code: z.number().int(),
     /**
      * A short description of the error. The message SHOULD be limited to a concise single sentence.
      */
@@ -218,14 +218,10 @@ export const ProgressNotificationSchema = NotificationSchema.extend({
     progressToken: ProgressTokenSchema,
     /**
      * The progress thus far. This should increase every time progress is made, even if the total is unknown.
-     *
-     * @TJS-type number
      */
     progress: z.number(),
     /**
      * Total number of items to process (or total progress required), if known.
-     *
-     * @TJS-type number
      */
     total: z.optional(z.number())
   })
@@ -599,14 +595,11 @@ export const CreateMessageRequestSchema = RequestSchema.extend({
      * A request to include context from one or more MCP servers (including the caller), to be attached to the prompt. The client MAY ignore this request.
      */
     includeContext: z.optional(z.enum(["none", "thisServer", "allServers"])),
-    /**
-     * @TJS-type number
-     */
     temperature: z.optional(z.number()),
     /**
      * The maximum number of tokens to sample, as requested by the server. The client MAY choose to sample fewer tokens than requested.
      */
-    maxTokens: z.number(),
+    maxTokens: z.number().int(),
     stopSequences: z.optional(z.array(z.string())),
     /**
      * Optional metadata to pass through to the LLM provider. The format of this metadata is provider-specific.
@@ -703,7 +696,7 @@ export const CompleteResultSchema = ResultSchema.extend({
     /**
      * The total number of completion options available. This can exceed the number of values actually sent in the response.
      */
-    total: z.optional(z.number()),
+    total: z.optional(z.number().int()),
     /**
      * Indicates whether there are additional completion options beyond those provided in the current response, even if the exact total is unknown.
      */