fastmcp 4.13.1 → 4.14.1

package/dist/{chunk-3BMGHXXY.js → chunk-SGCRCQNU.js}

@@ -355,6 +355,8 @@ function assertToolSchemas(tool) {
     assertStandardSchema(tool.name, "outputSchema", tool.outputSchema);
   }
 }
+var STREAM_KEEPALIVE_LOGGER = "fastmcp-keepalive";
+var STREAM_KEEPALIVE_DEFAULT_INTERVAL_MS = 2e4;
 var TextContentZodSchema = z.object({
   /**
    * The text content of the message.
@@ -421,10 +423,23 @@ var CompletionZodSchema = z.object({
    */
   total: z.optional(z.number().int()),
   /**
-   * An array of completion values. Must not exceed 100 items.
+   * An array of completion values. The MCP spec caps this at 100 items; values
+   * beyond the cap are trimmed by `capCompletionValues` (which sets `hasMore`)
+   * rather than rejected, so the schema itself does not enforce the limit.
    */
-  values: z.array(z.string()).max(100)
+  values: z.array(z.string())
 });
+var COMPLETION_VALUES_LIMIT = 100;
+var capCompletionValues = (completion) => {
+  if (completion.values.length <= COMPLETION_VALUES_LIMIT) {
+    return completion;
+  }
+  return {
+    ...completion,
+    hasMore: true,
+    values: completion.values.slice(0, COMPLETION_VALUES_LIMIT)
+  };
+};
 var FastMCPSessionEventEmitterBase = EventEmitter;
 var ServerState = /* @__PURE__ */ ((ServerState2) => {
   ServerState2["Error"] = "error";
@@ -483,6 +498,7 @@ var FastMCPSession = class extends FastMCPSessionEventEmitter {
    * instance — so capabilities can never be inferred here.
    */
   #stateless;
+  #streamKeepaliveConfig;
   /**
    * Resource URIs the connected client has subscribed to via
    * `resources/subscribe`. Used to scope `notifications/resources/updated`
@@ -503,6 +519,7 @@ var FastMCPSession = class extends FastMCPSessionEventEmitter {
     roots,
     sessionId,
     stateless = false,
+    streamKeepalive,
     tools,
     transportType,
     utils,
@@ -516,6 +533,7 @@ var FastMCPSession = class extends FastMCPSessionEventEmitter {
     this.#rootsConfig = roots;
     this.#sessionId = sessionId;
     this.#stateless = stateless;
+    this.#streamKeepaliveConfig = streamKeepalive;
     this.#needsEventLoopFlush = transportType === "httpStream";
     if (tools.length) {
       this.#capabilities.tools = {};
@@ -843,6 +861,44 @@ ${error instanceof Error ? error.stack : JSON.stringify(error)}`
       logLevel: pingConfig.logLevel || "debug"
     };
   }
+  /**
+   * Periodically writes to the response stream of an in-flight tool call, so an
+   * idle-connection timeout (proxy, load balancer) does not close it while a
+   * long-running tool produces no output of its own.
+   *
+   * The notification is related to the tool call, so it travels on that
+   * request's own stream, which is the only server-to-client route that exists
+   * when running stateless.
+   *
+   * @returns a function that stops the keepalive.
+   */
+  #startStreamKeepalive(extra, toolName) {
+    const config = this.#streamKeepaliveConfig;
+    if (!config?.enabled) {
+      return () => {
+      };
+    }
+    const intervalMs = config.intervalMs && config.intervalMs > 0 ? config.intervalMs : STREAM_KEEPALIVE_DEFAULT_INTERVAL_MS;
+    const timer = setInterval(() => {
+      extra.sendNotification({
+        method: "notifications/message",
+        params: {
+          data: { message: `keepalive while '${toolName}' is running` },
+          level: config.logLevel ?? "debug",
+          logger: STREAM_KEEPALIVE_LOGGER
+        }
+      }).catch((error) => {
+        this.#logger.debug(
+          `[FastMCP debug] stream keepalive for '${toolName}' failed:`,
+          error instanceof Error ? error.message : String(error)
+        );
+      });
+    }, intervalMs);
+    timer.unref?.();
+    const stop = () => clearInterval(timer);
+    extra.signal?.addEventListener("abort", stop, { once: true });
+    return stop;
+  }
   async #validateStructuredContent(tool, value, toolName) {
     if (!tool.outputSchema) {
       return value;
@@ -936,11 +992,13 @@ ${error instanceof Error ? error.stack : JSON.stringify(error)}`
             request
           });
         }
-        const completion = CompletionZodSchema.parse(
-          await prompt.complete(
-            request.params.argument.name,
-            request.params.argument.value,
-            this.#auth
+        const completion = capCompletionValues(
+          CompletionZodSchema.parse(
+            await prompt.complete(
+              request.params.argument.name,
+              request.params.argument.value,
+              this.#auth
+            )
           )
         );
         return {
@@ -968,11 +1026,13 @@ ${error instanceof Error ? error.stack : JSON.stringify(error)}`
             }
           );
         }
-        const completion = CompletionZodSchema.parse(
-          await resource.complete(
-            request.params.argument.name,
-            request.params.argument.value,
-            this.#auth
+        const completion = capCompletionValues(
+          CompletionZodSchema.parse(
+            await resource.complete(
+              request.params.argument.name,
+              request.params.argument.value,
+              this.#auth
+            )
           )
         );
         return {
@@ -1260,169 +1320,178 @@ ${error instanceof Error ? error.stack : JSON.stringify(error)}`
         tools: cachedToolsList
       };
     });
-    this.#server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const tool = toolsMap.get(request.params.name);
-      if (!tool) {
-        throw new McpError(
-          ErrorCode.MethodNotFound,
-          `Unknown tool: ${request.params.name}`
-        );
-      }
-      let args = void 0;
-      if (tool.parameters) {
-        const parsed = await tool.parameters["~standard"].validate(
-          request.params.arguments
-        );
-        if (parsed.issues) {
-          const friendlyErrors = this.#formatSchemaIssues(parsed.issues);
+    this.#server.setRequestHandler(
+      CallToolRequestSchema,
+      async (request, extra) => {
+        const tool = toolsMap.get(request.params.name);
+        if (!tool) {
           throw new McpError(
-            ErrorCode.InvalidParams,
-            `Tool '${request.params.name}' parameter validation failed: ${friendlyErrors}. Please check the parameter types and values according to the tool's schema.`
+            ErrorCode.MethodNotFound,
+            `Unknown tool: ${request.params.name}`
           );
         }
-        args = parsed.value;
-      }
-      const progressToken = request.params?._meta?.progressToken;
-      let result;
-      try {
-        const reportProgress = async (progress) => {
-          if (progressToken === void 0) {
-            return;
+        let args = void 0;
+        if (tool.parameters) {
+          const parsed = await tool.parameters["~standard"].validate(
+            request.params.arguments
+          );
+          if (parsed.issues) {
+            const friendlyErrors = this.#formatSchemaIssues(parsed.issues);
+            throw new McpError(
+              ErrorCode.InvalidParams,
+              `Tool '${request.params.name}' parameter validation failed: ${friendlyErrors}. Please check the parameter types and values according to the tool's schema.`
+            );
           }
-          try {
-            await this.#server.notification({
-              method: "notifications/progress",
-              params: {
-                ...progress,
-                progressToken
+          args = parsed.value;
+        }
+        const progressToken = request.params?._meta?.progressToken;
+        let result;
+        try {
+          const reportProgress = async (progress) => {
+            if (progressToken === void 0) {
+              return;
+            }
+            try {
+              await this.#server.notification({
+                method: "notifications/progress",
+                params: {
+                  ...progress,
+                  progressToken
+                }
+              });
+              if (this.#needsEventLoopFlush) {
+                await new Promise((resolve) => setImmediate(resolve));
               }
-            });
-            if (this.#needsEventLoopFlush) {
-              await new Promise((resolve) => setImmediate(resolve));
+            } catch (progressError) {
+              this.#logger.warn(
+                `[FastMCP warning] Failed to report progress for tool '${request.params.name}':`,
+                progressError instanceof Error ? progressError.message : String(progressError)
+              );
             }
-          } catch (progressError) {
-            this.#logger.warn(
-              `[FastMCP warning] Failed to report progress for tool '${request.params.name}':`,
-              progressError instanceof Error ? progressError.message : String(progressError)
-            );
-          }
-        };
-        const log = this.#createLog();
-        const streamContent = async (content) => {
-          const contentArray = Array.isArray(content) ? content : [content];
-          try {
-            await this.#server.notification({
-              method: "notifications/tool/streamContent",
-              params: {
-                content: contentArray,
-                toolName: request.params.name
+          };
+          const log = this.#createLog();
+          const streamContent = async (content) => {
+            const contentArray = Array.isArray(content) ? content : [content];
+            try {
+              await this.#server.notification({
+                method: "notifications/tool/streamContent",
+                params: {
+                  content: contentArray,
+                  toolName: request.params.name
+                }
+              });
+              if (this.#needsEventLoopFlush) {
+                await new Promise((resolve) => setImmediate(resolve));
               }
-            });
-            if (this.#needsEventLoopFlush) {
-              await new Promise((resolve) => setImmediate(resolve));
+            } catch (streamError) {
+              this.#logger.warn(
+                `[FastMCP warning] Failed to stream content for tool '${request.params.name}':`,
+                streamError instanceof Error ? streamError.message : String(streamError)
+              );
             }
-          } catch (streamError) {
-            this.#logger.warn(
-              `[FastMCP warning] Failed to stream content for tool '${request.params.name}':`,
-              streamError instanceof Error ? streamError.message : String(streamError)
-            );
+          };
+          if (this.#onToolCall) {
+            await this.#onToolCall({
+              arguments: args ?? {},
+              toolName: request.params.name
+            });
           }
-        };
-        if (this.#onToolCall) {
-          await this.#onToolCall({
-            arguments: args ?? {},
-            toolName: request.params.name
-          });
-        }
-        const executeToolPromise = tool.execute(args, {
-          client: {
-            version: this.#server.getClientVersion()
-          },
-          elicit: (params, options) => this.#server.elicitInput(params, options),
-          log,
-          reportProgress,
-          requestId: typeof request.params?._meta?.requestId === "string" ? request.params._meta.requestId : void 0,
-          session: this.#auth,
-          sessionId: this.#sessionId,
-          streamContent
-        });
-        const maybeStringResult = await (tool.timeoutMs ? Promise.race([
-          executeToolPromise,
-          new Promise((_, reject) => {
-            const timeoutId = setTimeout(() => {
-              reject(
-                new UserError(
-                  `Tool '${request.params.name}' timed out after ${tool.timeoutMs}ms. Consider increasing timeoutMs or optimizing the tool implementation.`
-                )
+          const executeToolPromise = Promise.resolve(
+            tool.execute(args, {
+              client: {
+                version: this.#server.getClientVersion()
+              },
+              elicit: (params, options) => this.#server.elicitInput(params, options),
+              log,
+              reportProgress,
+              requestId: typeof request.params?._meta?.requestId === "string" ? request.params._meta.requestId : void 0,
+              session: this.#auth,
+              sessionId: this.#sessionId,
+              streamContent
+            })
+          );
+          const stopStreamKeepalive = this.#startStreamKeepalive(
+            extra,
+            request.params.name
+          );
+          const maybeStringResult = await (tool.timeoutMs ? Promise.race([
+            executeToolPromise,
+            new Promise((_, reject) => {
+              const timeoutId = setTimeout(() => {
+                reject(
+                  new UserError(
+                    `Tool '${request.params.name}' timed out after ${tool.timeoutMs}ms. Consider increasing timeoutMs or optimizing the tool implementation.`
+                  )
+                );
+              }, tool.timeoutMs);
+              executeToolPromise.then(
+                () => clearTimeout(timeoutId),
+                () => clearTimeout(timeoutId)
               );
-            }, tool.timeoutMs);
-            executeToolPromise.then(
-              () => clearTimeout(timeoutId),
-              () => clearTimeout(timeoutId)
-            );
-          })
-        ]) : executeToolPromise);
-        await delay(1);
-        if (maybeStringResult === void 0 || maybeStringResult === null) {
-          result = ContentResultZodSchema.parse({
-            content: []
-          });
-        } else if (typeof maybeStringResult === "string") {
-          result = ContentResultZodSchema.parse({
-            content: [{ text: maybeStringResult, type: "text" }]
-          });
-        } else if ("type" in maybeStringResult) {
-          result = ContentResultZodSchema.parse({
-            content: [maybeStringResult]
-          });
-        } else if ("content" in maybeStringResult) {
-          result = ContentResultZodSchema.parse(maybeStringResult);
-          if (result.structuredContent !== void 0 && tool.outputSchema) {
-            result.structuredContent = await this.#validateStructuredContent(
+            })
+          ]) : executeToolPromise).finally(stopStreamKeepalive);
+          await delay(1);
+          if (maybeStringResult === void 0 || maybeStringResult === null) {
+            result = ContentResultZodSchema.parse({
+              content: []
+            });
+          } else if (typeof maybeStringResult === "string") {
+            result = ContentResultZodSchema.parse({
+              content: [{ text: maybeStringResult, type: "text" }]
+            });
+          } else if ("type" in maybeStringResult) {
+            result = ContentResultZodSchema.parse({
+              content: [maybeStringResult]
+            });
+          } else if ("content" in maybeStringResult) {
+            result = ContentResultZodSchema.parse(maybeStringResult);
+            if (result.structuredContent !== void 0 && tool.outputSchema) {
+              result.structuredContent = await this.#validateStructuredContent(
+                tool,
+                result.structuredContent,
+                request.params.name
+              );
+            }
+          } else if (tool.outputSchema) {
+            const structuredContent = await this.#validateStructuredContent(
               tool,
-              result.structuredContent,
+              maybeStringResult,
               request.params.name
             );
+            result = ContentResultZodSchema.parse({
+              content: [
+                {
+                  text: JSON.stringify(structuredContent),
+                  type: "text"
+                }
+              ],
+              structuredContent
+            });
+          } else {
+            result = ContentResultZodSchema.parse(maybeStringResult);
           }
-        } else if (tool.outputSchema) {
-          const structuredContent = await this.#validateStructuredContent(
-            tool,
-            maybeStringResult,
-            request.params.name
-          );
-          result = ContentResultZodSchema.parse({
+        } catch (error) {
+          if (error instanceof UserError) {
+            return {
+              content: [{ text: error.message, type: "text" }],
+              isError: true,
+              ...error.extras ? { structuredContent: error.extras } : {}
+            };
+          }
+          const errorMessage = error instanceof Error ? error.message : String(error);
+          return {
             content: [
               {
-                text: JSON.stringify(structuredContent),
+                text: `Tool '${request.params.name}' execution failed: ${errorMessage}`,
                 type: "text"
               }
             ],
-            structuredContent
-          });
-        } else {
-          result = ContentResultZodSchema.parse(maybeStringResult);
-        }
-      } catch (error) {
-        if (error instanceof UserError) {
-          return {
-            content: [{ text: error.message, type: "text" }],
-            isError: true,
-            ...error.extras ? { structuredContent: error.extras } : {}
+            isError: true
           };
         }
-        const errorMessage = error instanceof Error ? error.message : String(error);
-        return {
-          content: [
-            {
-              text: `Tool '${request.params.name}' execution failed: ${errorMessage}`,
-              type: "text"
-            }
-          ],
-          isError: true
-        };
+        return result;
       }
-      return result;
-    });
+    );
   }
 };
 function camelToSnakeCase(str) {
@@ -1863,6 +1932,7 @@ var FastMCP = class extends FastMCPEventEmitter {
         resources: this.#resources,
         resourcesTemplates: this.#resourcesTemplates,
         roots: this.#options.roots,
+        streamKeepalive: this.#options.streamKeepalive,
         tools: this.#tools,
         transportType: "stdio",
         utils: this.#options.utils,
@@ -2062,6 +2132,7 @@ var FastMCP = class extends FastMCPEventEmitter {
       roots: this.#options.roots,
       sessionId,
       stateless,
+      streamKeepalive: this.#options.streamKeepalive,
       tools: allowedTools,
       transportType: "httpStream",
       utils: this.#options.utils,
@@ -2676,4 +2747,4 @@ export {
   FastMCPSession,
   FastMCP
 };
-//# sourceMappingURL=chunk-3BMGHXXY.js.map
+//# sourceMappingURL=chunk-SGCRCQNU.js.map