fastmcp 1.2.0 → 1.4.0

package/README.md

@@ -9,7 +9,8 @@ A TypeScript framework for building [MCP](https://modelcontextprotocol.io/) serv
 
 - Simple Tool, Resource, Prompt definition
 - Full TypeScript support
-- Built-in error handling
+- Built-in [logging](#logging)
+- Built-in [error handling](#errors)
 - Built-in CLI for [testing](#test-with-mcp-cli) and [debugging](#inspect-with-mcp-inspector)
 - Built-in support for [SSE](#sse)
 - Built-in [progress notifications](#progress)
@@ -39,7 +40,7 @@ server.addTool({
     b: z.number(),
   }),
   execute: async (args) => {
-    return args.a + args.b;
+    return String(args.a + args.b);
   },
 });
 
@@ -104,12 +105,104 @@ server.addTool({
     url: z.string(),
   }),
   execute: async (args) => {
-    const content = await fetchWebpageContent(args.url);
-    return content;
+    return await fetchWebpageContent(args.url);
   },
 });
 ```
 
+#### Returning a string
+
+`execute` can return a string:
+
+```js
+server.addTool({
+  name: "download",
+  description: "Download a file",
+  parameters: z.object({
+    url: z.string(),
+  }),
+  execute: async (args) => {
+    return "Hello, world!";
+  },
+});
+```
+
+The latter is equivalent to:
+
+```js
+server.addTool({
+  name: "download",
+  description: "Download a file",
+  parameters: z.object({
+    url: z.string(),
+  }),
+  execute: async (args) => {
+    return {
+      content: [
+        {
+          type: "text",
+          text: "Hello, world!",
+        },
+      ],
+    };
+  },
+});
+```
+
+#### Returning a list
+
+If you want to return a list of messages, you can return an object with a `content` property:
+
+```js
+server.addTool({
+  name: "download",
+  description: "Download a file",
+  parameters: z.object({
+    url: z.string(),
+  }),
+  execute: async (args) => {
+    return {
+      content: [
+        { type: "text", text: "First message" },
+        { type: "text", text: "Second message" },
+      ],
+    };
+  },
+});
+```
+
+#### Logging
+
+Tools can log messages to the client using the `log` object in the context object:
+
+```js
+server.addTool({
+  name: "download",
+  description: "Download a file",
+  parameters: z.object({
+    url: z.string(),
+  }),
+  execute: async (args, { log }) => {
+    log.info("Downloading file...", {
+      url,
+    });
+
+    // ...
+
+    log.info("Downloaded file");
+
+    return "done";
+  },
+});
+```
+
+The `log` object has the following methods:
+
+- `debug(message: string, data?: SerializableValue)`
+- `error(message: string, data?: SerializableValue)`
+- `info(message: string, data?: SerializableValue)`
+- `warn(message: string, data?: SerializableValue)`
+
 #### Errors
 
 The errors that are meant to be shown to the user should be thrown as `UserError` instances:

package/dist/FastMCP.d.ts

@@ -12,6 +12,10 @@ declare class UnexpectedStateError extends FastMCPError {
 declare class UserError extends UnexpectedStateError {
 }
 type ToolParameters = z.ZodTypeAny;
+type Literal = boolean | null | number | string | undefined;
+type SerializableValue = Literal | SerializableValue[] | {
+    [key: string]: SerializableValue;
+};
 type Progress = {
     /**
      * The progress thus far. This should increase every time progress is made, even if the total is unknown.
@@ -24,12 +28,73 @@ type Progress = {
 };
 type Context = {
     reportProgress: (progress: Progress) => Promise<void>;
+    log: {
+        debug: (message: string, data?: SerializableValue) => void;
+        error: (message: string, data?: SerializableValue) => void;
+        info: (message: string, data?: SerializableValue) => void;
+        warn: (message: string, data?: SerializableValue) => void;
+    };
 };
+declare const ContentResultZodSchema: z.ZodObject<{
+    content: z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+        type: z.ZodLiteral<"text">;
+        /**
+         * The text content of the message.
+         */
+        text: z.ZodString;
+    }, "strict", z.ZodTypeAny, {
+        type: "text";
+        text: string;
+    }, {
+        type: "text";
+        text: string;
+    }>, z.ZodObject<{
+        type: z.ZodLiteral<"image">;
+        /**
+         * The base64-encoded image data.
+         */
+        data: z.ZodString;
+        /**
+         * The MIME type of the image. Different providers may support different image types.
+         */
+        mimeType: z.ZodString;
+    }, "strict", z.ZodTypeAny, {
+        type: "image";
+        data: string;
+        mimeType: string;
+    }, {
+        type: "image";
+        data: string;
+        mimeType: string;
+    }>]>, "many">;
+    isError: z.ZodOptional<z.ZodBoolean>;
+}, "strict", z.ZodTypeAny, {
+    content: ({
+        type: "text";
+        text: string;
+    } | {
+        type: "image";
+        data: string;
+        mimeType: string;
+    })[];
+    isError?: boolean | undefined;
+}, {
+    content: ({
+        type: "text";
+        text: string;
+    } | {
+        type: "image";
+        data: string;
+        mimeType: string;
+    })[];
+    isError?: boolean | undefined;
+}>;
+type ContentResult = z.infer<typeof ContentResultZodSchema>;
 type Tool<Params extends ToolParameters = ToolParameters> = {
     name: string;
     description?: string;
     parameters?: Params;
-    execute: (args: z.infer<Params>, context: Context) => Promise<unknown>;
+    execute: (args: z.infer<Params>, context: Context) => Promise<string | ContentResult>;
 };
 type Resource = {
     uri: string;
@@ -68,6 +133,7 @@ declare class FastMCP {
     constructor(options: ServerOptions);
     private setupHandlers;
     private setupErrorHandling;
+    get loggingLevel(): "debug" | "info" | "notice" | "warning" | "error" | "critical" | "alert" | "emergency";
     private setupToolHandlers;
     private setupResourceHandlers;
     private setupPromptHandlers;

package/dist/FastMCP.js

@@ -10,9 +10,11 @@ import {
   ListResourcesRequestSchema,
   ListToolsRequestSchema,
   McpError,
-  ReadResourceRequestSchema
+  ReadResourceRequestSchema,
+  SetLevelRequestSchema
 } from "@modelcontextprotocol/sdk/types.js";
 import { zodToJsonSchema } from "zod-to-json-schema";
+import { z } from "zod";
 import http from "http";
 var FastMCPError = class extends Error {
   constructor(message) {
@@ -30,6 +32,32 @@ var UnexpectedStateError = class extends FastMCPError {
 };
 var UserError = class extends UnexpectedStateError {
 };
+var TextContentZodSchema = z.object({
+  type: z.literal("text"),
+  /**
+   * The text content of the message.
+   */
+  text: z.string()
+}).strict();
+var ImageContentZodSchema = z.object({
+  type: z.literal("image"),
+  /**
+   * The base64-encoded image data.
+   */
+  data: z.string().base64(),
+  /**
+   * The MIME type of the image. Different providers may support different image types.
+   */
+  mimeType: z.string()
+}).strict();
+var ContentZodSchema = z.discriminatedUnion("type", [
+  TextContentZodSchema,
+  ImageContentZodSchema
+]);
+var ContentResultZodSchema = z.object({
+  content: ContentZodSchema.array(),
+  isError: z.boolean().optional()
+}).strict();
 var FastMCP = class {
   constructor(options) {
     this.options = options;
@@ -43,6 +71,7 @@ var FastMCP = class {
   #prompts;
   #server = null;
   #options;
+  #loggingLevel = "info";
   setupHandlers(server) {
     this.setupErrorHandling(server);
     if (this.#tools.length) {
@@ -54,6 +83,10 @@ var FastMCP = class {
     if (this.#prompts.length) {
       this.setupPromptHandlers(server);
     }
+    server.setRequestHandler(SetLevelRequestSchema, (request) => {
+      this.#loggingLevel = request.params.level;
+      return {};
+    });
   }
   setupErrorHandling(server) {
     server.onerror = (error) => {
@@ -64,6 +97,9 @@ var FastMCP = class {
       process.exit(0);
     });
   }
+  get loggingLevel() {
+    return this.#loggingLevel;
+  }
   setupToolHandlers(server) {
     server.setRequestHandler(ListToolsRequestSchema, async () => {
       return {
@@ -111,9 +147,55 @@ var FastMCP = class {
               }
             });
           };
-          result = await tool.execute(args, {
-            reportProgress
+          const log = {
+            debug: (message, context) => {
+              server.sendLoggingMessage({
+                level: "debug",
+                data: {
+                  message,
+                  context
+                }
+              });
+            },
+            error: (message, context) => {
+              server.sendLoggingMessage({
+                level: "error",
+                data: {
+                  message,
+                  context
+                }
+              });
+            },
+            info: (message, context) => {
+              server.sendLoggingMessage({
+                level: "info",
+                data: {
+                  message,
+                  context
+                }
+              });
+            },
+            warn: (message, context) => {
+              server.sendLoggingMessage({
+                level: "warning",
+                data: {
+                  message,
+                  context
+                }
+              });
+            }
+          };
+          const maybeStringResult = await tool.execute(args, {
+            reportProgress,
+            log
           });
+          if (typeof maybeStringResult === "string") {
+            result = ContentResultZodSchema.parse({
+              content: [{ type: "text", text: maybeStringResult }]
+            });
+          } else {
+            result = ContentResultZodSchema.parse(maybeStringResult);
+          }
         } catch (error) {
           if (error instanceof UserError) {
             return {
@@ -126,14 +208,7 @@ var FastMCP = class {
             isError: true
           };
         }
-        if (typeof result === "string") {
-          return {
-            content: [{ type: "text", text: result }]
-          };
-        }
-        return {
-          content: [{ type: "text", text: JSON.stringify(result, null, 2) }]
-        };
+        return result;
       }
     );
   }
@@ -258,6 +333,7 @@ var FastMCP = class {
     if (this.#prompts.length) {
       capabilities.prompts = {};
     }
+    capabilities.logging = {};
     this.#server = new Server(
       { name: this.#options.name, version: this.#options.version },
       { capabilities }

package/dist/FastMCP.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/FastMCP.ts"],"sourcesContent":["import { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport { SSEServerTransport } from \"@modelcontextprotocol/sdk/server/sse.js\";\nimport {\n  CallToolRequestSchema,\n  ErrorCode,\n  GetPromptRequestSchema,\n  ListPromptsRequestSchema,\n  ListResourcesRequestSchema,\n  ListToolsRequestSchema,\n  McpError,\n  ReadResourceRequestSchema,\n  ServerCapabilities,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { zodToJsonSchema } from \"zod-to-json-schema\";\nimport type { z } from \"zod\";\nimport http from \"http\";\n\nabstract class FastMCPError extends Error {\n  public constructor(message?: string) {\n    super(message);\n    this.name = new.target.name;\n  }\n}\n\ntype Extra = unknown;\n\ntype Extras = Record<string, Extra>;\n\nclass UnexpectedStateError extends FastMCPError {\n  public extras?: Extras;\n\n  public constructor(message: string, extras?: Extras) {\n    super(message);\n    this.name = new.target.name;\n    this.extras = extras;\n  }\n}\n\nexport class UserError extends UnexpectedStateError {}\n\ntype ToolParameters = z.ZodTypeAny;\n\ntype Progress = {\n  /**\n   * The progress thus far. This should increase every time progress is made, even if the total is unknown.\n   */\n  progress: number;\n  /**\n   * Total number of items to process (or total progress required), if known.\n   */\n  total?: number;\n};\n\ntype Context = {\n  reportProgress: (progress: Progress) => Promise<void>;\n};\n\ntype Tool<Params extends ToolParameters = ToolParameters> = {\n  name: string;\n  description?: string;\n  parameters?: Params;\n  execute: (args: z.infer<Params>, context: Context) => Promise<unknown>;\n};\n\ntype Resource = {\n  uri: string;\n  name: string;\n  description?: string;\n  mimeType?: string;\n  load: () => Promise<{ text: string } | { blob: string }>;\n};\n\ntype PromptArgument = Readonly<{\n  name: string;\n  description?: string;\n  required?: boolean;\n}>;\n\ntype ArgumentsToObject<T extends PromptArgument[]> = {\n  [K in T[number][\"name\"]]: Extract<\n    T[number],\n    { name: K }\n  >[\"required\"] extends true\n    ? string\n    : string | undefined;\n};\n\ntype Prompt<\n  Arguments extends PromptArgument[] = PromptArgument[],\n  Args = ArgumentsToObject<Arguments>,\n> = {\n  name: string;\n  description?: string;\n  arguments?: Arguments;\n  load: (args: Args) => Promise<string>;\n};\n\ntype ServerOptions = {\n  name: string;\n  version: `${number}.${number}.${number}`;\n};\n\nexport class FastMCP {\n  #tools: Tool[];\n  #resources: Resource[];\n  #prompts: Prompt[];\n  #server: Server | null = null;\n  #options: ServerOptions;\n\n  constructor(public options: ServerOptions) {\n    this.#options = options;\n    this.#tools = [];\n    this.#resources = [];\n    this.#prompts = [];\n  }\n\n  private setupHandlers(server: Server) {\n    this.setupErrorHandling(server);\n\n    if (this.#tools.length) {\n      this.setupToolHandlers(server);\n    }\n\n    if (this.#resources.length) {\n      this.setupResourceHandlers(server);\n    }\n\n    if (this.#prompts.length) {\n      this.setupPromptHandlers(server);\n    }\n  }\n\n  private setupErrorHandling(server: Server) {\n    server.onerror = (error) => {\n      console.error(\"[MCP Error]\", error);\n    };\n    process.on(\"SIGINT\", async () => {\n      await server.close();\n      process.exit(0);\n    });\n  }\n\n  private setupToolHandlers(server: Server) {\n    server.setRequestHandler(ListToolsRequestSchema, async () => {\n      return {\n        tools: this.#tools.map((tool) => {\n          return {\n            name: tool.name,\n            description: tool.description,\n            inputSchema: tool.parameters\n              ? zodToJsonSchema(tool.parameters)\n              : undefined,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(\n      CallToolRequestSchema,\n      async (request, ...extra) => {\n        const tool = this.#tools.find(\n          (tool) => tool.name === request.params.name,\n        );\n\n        if (!tool) {\n          throw new McpError(\n            ErrorCode.MethodNotFound,\n            `Unknown tool: ${request.params.name}`,\n          );\n        }\n\n        let args: any = undefined;\n\n        if (tool.parameters) {\n          const parsed = tool.parameters.safeParse(request.params.arguments);\n\n          if (!parsed.success) {\n            throw new McpError(\n              ErrorCode.InvalidRequest,\n              `Invalid ${request.params.name} arguments`,\n            );\n          }\n\n          args = parsed.data;\n        }\n\n        const progressToken = request.params?._meta?.progressToken;\n\n        let result: any;\n\n        try {\n          const reportProgress = async (progress: Progress) => {\n            await server.notification({\n              method: \"notifications/progress\",\n              params: {\n                ...progress,\n                progressToken,\n              },\n            });\n          };\n\n          result = await tool.execute(args, {\n            reportProgress,\n          });\n        } catch (error) {\n          if (error instanceof UserError) {\n            return {\n              content: [{ type: \"text\", text: error.message }],\n              isError: true,\n            };\n          }\n\n          return {\n            content: [{ type: \"text\", text: `Error: ${error}` }],\n            isError: true,\n          };\n        }\n\n        if (typeof result === \"string\") {\n          return {\n            content: [{ type: \"text\", text: result }],\n          };\n        }\n\n        return {\n          content: [{ type: \"text\", text: JSON.stringify(result, null, 2) }],\n        };\n      },\n    );\n  }\n\n  private setupResourceHandlers(server: Server) {\n    server.setRequestHandler(ListResourcesRequestSchema, async () => {\n      return {\n        resources: this.#resources.map((resource) => {\n          return {\n            uri: resource.uri,\n            name: resource.name,\n            mimeType: resource.mimeType,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(ReadResourceRequestSchema, async (request) => {\n      const resource = this.#resources.find(\n        (resource) => resource.uri === request.params.uri,\n      );\n\n      if (!resource) {\n        throw new McpError(\n          ErrorCode.MethodNotFound,\n          `Unknown resource: ${request.params.uri}`,\n        );\n      }\n\n      let result: Awaited<ReturnType<Resource[\"load\"]>>;\n\n      try {\n        result = await resource.load();\n      } catch (error) {\n        throw new McpError(\n          ErrorCode.InternalError,\n          `Error reading resource: ${error}`,\n          {\n            uri: resource.uri,\n          },\n        );\n      }\n\n      return {\n        contents: [\n          {\n            uri: resource.uri,\n            mimeType: resource.mimeType,\n            ...result,\n          },\n        ],\n      };\n    });\n  }\n\n  private setupPromptHandlers(server: Server) {\n    server.setRequestHandler(ListPromptsRequestSchema, async () => {\n      return {\n        prompts: this.#prompts.map((prompt) => {\n          return {\n            name: prompt.name,\n            description: prompt.description,\n            arguments: prompt.arguments,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(GetPromptRequestSchema, async (request) => {\n      const prompt = this.#prompts.find(\n        (prompt) => prompt.name === request.params.name,\n      );\n\n      if (!prompt) {\n        throw new McpError(\n          ErrorCode.MethodNotFound,\n          `Unknown prompt: ${request.params.name}`,\n        );\n      }\n\n      const args = request.params.arguments;\n\n      if (prompt.arguments) {\n        for (const arg of prompt.arguments) {\n          if (arg.required && !(args && arg.name in args)) {\n            throw new McpError(\n              ErrorCode.InvalidRequest,\n              `Missing required argument: ${arg.name}`,\n            );\n          }\n        }\n      }\n\n      let result: Awaited<ReturnType<Prompt[\"load\"]>>;\n\n      try {\n        result = await prompt.load(args as Record<string, string | undefined>);\n      } catch (error) {\n        throw new McpError(\n          ErrorCode.InternalError,\n          `Error loading prompt: ${error}`,\n        );\n      }\n\n      return {\n        description: prompt.description,\n        messages: [\n          {\n            role: \"user\",\n            content: { type: \"text\", text: result },\n          },\n        ],\n      };\n    });\n  }\n\n  public addTool<Params extends ToolParameters>(tool: Tool<Params>) {\n    this.#tools.push(tool as unknown as Tool);\n  }\n\n  public addResource(resource: Resource) {\n    this.#resources.push(resource);\n  }\n\n  public addPrompt<const Args extends PromptArgument[]>(prompt: Prompt<Args>) {\n    this.#prompts.push(prompt);\n  }\n\n  #httpServer: http.Server | null = null;\n\n  public async start(\n    options:\n      | { transportType: \"stdio\" }\n      | {\n          transportType: \"sse\";\n          sse: { endpoint: `/${string}`; port: number };\n        } = {\n      transportType: \"stdio\",\n    },\n  ) {\n    const capabilities: ServerCapabilities = {};\n\n    if (this.#tools.length) {\n      capabilities.tools = {};\n    }\n\n    if (this.#resources.length) {\n      capabilities.resources = {};\n    }\n\n    if (this.#prompts.length) {\n      capabilities.prompts = {};\n    }\n\n    this.#server = new Server(\n      { name: this.#options.name, version: this.#options.version },\n      { capabilities },\n    );\n\n    this.setupHandlers(this.#server);\n\n    if (options.transportType === \"stdio\") {\n      const transport = new StdioServerTransport();\n\n      await this.#server.connect(transport);\n\n      console.error(`server is running on stdio`);\n    } else if (options.transportType === \"sse\") {\n      let activeTransport: SSEServerTransport | null = null;\n\n      /**\n       * Adopted from https://dev.classmethod.jp/articles/mcp-sse/\n       */\n      this.#httpServer = http.createServer(async (req, res) => {\n        if (req.method === \"GET\" && req.url === options.sse.endpoint) {\n          const transport = new SSEServerTransport(\"/messages\", res);\n\n          activeTransport = transport;\n\n          if (!this.#server) {\n            throw new Error(\"Server not initialized\");\n          }\n\n          await this.#server.connect(transport);\n\n          res.on(\"close\", () => {\n            console.log(\"SSE connection closed\");\n            if (activeTransport === transport) {\n              activeTransport = null;\n            }\n          });\n\n          this.startSending(transport);\n          return;\n        }\n\n        if (req.method === \"POST\" && req.url?.startsWith(\"/messages\")) {\n          if (!activeTransport) {\n            res.writeHead(400).end(\"No active transport\");\n            return;\n          }\n          await activeTransport.handlePostMessage(req, res);\n          return;\n        }\n\n        res.writeHead(404).end();\n      });\n\n      this.#httpServer.listen(options.sse.port, \"0.0.0.0\");\n\n      console.error(\n        `server is running on SSE at http://localhost:${options.sse.port}${options.sse.endpoint}`,\n      );\n    } else {\n      throw new Error(\"Invalid transport type\");\n    }\n  }\n\n  /**\n   * @see https://dev.classmethod.jp/articles/mcp-sse/\n   */\n  private async startSending(transport: SSEServerTransport) {\n    try {\n      await transport.send({\n        jsonrpc: \"2.0\",\n        method: \"sse/connection\",\n        params: { message: \"SSE Connection established\" },\n      });\n\n      let messageCount = 0;\n      const interval = setInterval(async () => {\n        messageCount++;\n\n        const message = `Message ${messageCount} at ${new Date().toISOString()}`;\n\n        try {\n          await transport.send({\n            jsonrpc: \"2.0\",\n            method: \"sse/message\",\n            params: { data: message },\n          });\n\n          console.log(`Sent: ${message}`);\n\n          if (messageCount === 10) {\n            clearInterval(interval);\n\n            await transport.send({\n              jsonrpc: \"2.0\",\n              method: \"sse/complete\",\n              params: { message: \"Stream completed\" },\n            });\n            console.log(\"Stream completed\");\n          }\n        } catch (error) {\n          console.error(\"Error sending message:\", error);\n          clearInterval(interval);\n        }\n      }, 1000);\n    } catch (error) {\n      console.error(\"Error in startSending:\", error);\n    }\n  }\n\n  public async stop() {\n    if (this.#httpServer) {\n      this.#httpServer.close();\n    }\n  }\n}\n"],"mappings":";AAAA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AACrC,SAAS,0BAA0B;AACnC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,uBAAuB;AAEhC,OAAO,UAAU;AAEjB,IAAe,eAAf,cAAoC,MAAM;AAAA,EACjC,YAAY,SAAkB;AACnC,UAAM,OAAO;AACb,SAAK,OAAO,WAAW;AAAA,EACzB;AACF;AAMA,IAAM,uBAAN,cAAmC,aAAa;AAAA,EACvC;AAAA,EAEA,YAAY,SAAiB,QAAiB;AACnD,UAAM,OAAO;AACb,SAAK,OAAO,WAAW;AACvB,SAAK,SAAS;AAAA,EAChB;AACF;AAEO,IAAM,YAAN,cAAwB,qBAAqB;AAAC;AAgE9C,IAAM,UAAN,MAAc;AAAA,EAOnB,YAAmB,SAAwB;AAAxB;AACjB,SAAK,WAAW;AAChB,SAAK,SAAS,CAAC;AACf,SAAK,aAAa,CAAC;AACnB,SAAK,WAAW,CAAC;AAAA,EACnB;AAAA,EAXA;AAAA,EACA;AAAA,EACA;AAAA,EACA,UAAyB;AAAA,EACzB;AAAA,EASQ,cAAc,QAAgB;AACpC,SAAK,mBAAmB,MAAM;AAE9B,QAAI,KAAK,OAAO,QAAQ;AACtB,WAAK,kBAAkB,MAAM;AAAA,IAC/B;AAEA,QAAI,KAAK,WAAW,QAAQ;AAC1B,WAAK,sBAAsB,MAAM;AAAA,IACnC;AAEA,QAAI,KAAK,SAAS,QAAQ;AACxB,WAAK,oBAAoB,MAAM;AAAA,IACjC;AAAA,EACF;AAAA,EAEQ,mBAAmB,QAAgB;AACzC,WAAO,UAAU,CAAC,UAAU;AAC1B,cAAQ,MAAM,eAAe,KAAK;AAAA,IACpC;AACA,YAAQ,GAAG,UAAU,YAAY;AAC/B,YAAM,OAAO,MAAM;AACnB,cAAQ,KAAK,CAAC;AAAA,IAChB,CAAC;AAAA,EACH;AAAA,EAEQ,kBAAkB,QAAgB;AACxC,WAAO,kBAAkB,wBAAwB,YAAY;AAC3D,aAAO;AAAA,QACL,OAAO,KAAK,OAAO,IAAI,CAAC,SAAS;AAC/B,iBAAO;AAAA,YACL,MAAM,KAAK;AAAA,YACX,aAAa,KAAK;AAAA,YAClB,aAAa,KAAK,aACd,gBAAgB,KAAK,UAAU,IAC/B;AAAA,UACN;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA,OAAO,YAAY,UAAU;AAC3B,cAAM,OAAO,KAAK,OAAO;AAAA,UACvB,CAACA,UAASA,MAAK,SAAS,QAAQ,OAAO;AAAA,QACzC;AAEA,YAAI,CAAC,MAAM;AACT,gBAAM,IAAI;AAAA,YACR,UAAU;AAAA,YACV,iBAAiB,QAAQ,OAAO,IAAI;AAAA,UACtC;AAAA,QACF;AAEA,YAAI,OAAY;AAEhB,YAAI,KAAK,YAAY;AACnB,gBAAM,SAAS,KAAK,WAAW,UAAU,QAAQ,OAAO,SAAS;AAEjE,cAAI,CAAC,OAAO,SAAS;AACnB,kBAAM,IAAI;AAAA,cACR,UAAU;AAAA,cACV,WAAW,QAAQ,OAAO,IAAI;AAAA,YAChC;AAAA,UACF;AAEA,iBAAO,OAAO;AAAA,QAChB;AAEA,cAAM,gBAAgB,QAAQ,QAAQ,OAAO;AAE7C,YAAI;AAEJ,YAAI;AACF,gBAAM,iBAAiB,OAAO,aAAuB;AACnD,kBAAM,OAAO,aAAa;AAAA,cACxB,QAAQ;AAAA,cACR,QAAQ;AAAA,gBACN,GAAG;AAAA,gBACH;AAAA,cACF;AAAA,YACF,CAAC;AAAA,UACH;AAEA,mBAAS,MAAM,KAAK,QAAQ,MAAM;AAAA,YAChC;AAAA,UACF,CAAC;AAAA,QACH,SAAS,OAAO;AACd,cAAI,iBAAiB,WAAW;AAC9B,mBAAO;AAAA,cACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,QAAQ,CAAC;AAAA,cAC/C,SAAS;AAAA,YACX;AAAA,UACF;AAEA,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,UAAU,KAAK,GAAG,CAAC;AAAA,YACnD,SAAS;AAAA,UACX;AAAA,QACF;AAEA,YAAI,OAAO,WAAW,UAAU;AAC9B,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,OAAO,CAAC;AAAA,UAC1C;AAAA,QACF;AAEA,eAAO;AAAA,UACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,KAAK,UAAU,QAAQ,MAAM,CAAC,EAAE,CAAC;AAAA,QACnE;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBAAsB,QAAgB;AAC5C,WAAO,kBAAkB,4BAA4B,YAAY;AAC/D,aAAO;AAAA,QACL,WAAW,KAAK,WAAW,IAAI,CAAC,aAAa;AAC3C,iBAAO;AAAA,YACL,KAAK,SAAS;AAAA,YACd,MAAM,SAAS;AAAA,YACf,UAAU,SAAS;AAAA,UACrB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO,kBAAkB,2BAA2B,OAAO,YAAY;AACrE,YAAM,WAAW,KAAK,WAAW;AAAA,QAC/B,CAACC,cAAaA,UAAS,QAAQ,QAAQ,OAAO;AAAA,MAChD;AAEA,UAAI,CAAC,UAAU;AACb,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,qBAAqB,QAAQ,OAAO,GAAG;AAAA,QACzC;AAAA,MACF;AAEA,UAAI;AAEJ,UAAI;AACF,iBAAS,MAAM,SAAS,KAAK;AAAA,MAC/B,SAAS,OAAO;AACd,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,2BAA2B,KAAK;AAAA,UAChC;AAAA,YACE,KAAK,SAAS;AAAA,UAChB;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,UAAU;AAAA,UACR;AAAA,YACE,KAAK,SAAS;AAAA,YACd,UAAU,SAAS;AAAA,YACnB,GAAG;AAAA,UACL;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,oBAAoB,QAAgB;AAC1C,WAAO,kBAAkB,0BAA0B,YAAY;AAC7D,aAAO;AAAA,QACL,SAAS,KAAK,SAAS,IAAI,CAAC,WAAW;AACrC,iBAAO;AAAA,YACL,MAAM,OAAO;AAAA,YACb,aAAa,OAAO;AAAA,YACpB,WAAW,OAAO;AAAA,UACpB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO,kBAAkB,wBAAwB,OAAO,YAAY;AAClE,YAAM,SAAS,KAAK,SAAS;AAAA,QAC3B,CAACC,YAAWA,QAAO,SAAS,QAAQ,OAAO;AAAA,MAC7C;AAEA,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,mBAAmB,QAAQ,OAAO,IAAI;AAAA,QACxC;AAAA,MACF;AAEA,YAAM,OAAO,QAAQ,OAAO;AAE5B,UAAI,OAAO,WAAW;AACpB,mBAAW,OAAO,OAAO,WAAW;AAClC,cAAI,IAAI,YAAY,EAAE,QAAQ,IAAI,QAAQ,OAAO;AAC/C,kBAAM,IAAI;AAAA,cACR,UAAU;AAAA,cACV,8BAA8B,IAAI,IAAI;AAAA,YACxC;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,UAAI;AAEJ,UAAI;AACF,iBAAS,MAAM,OAAO,KAAK,IAA0C;AAAA,MACvE,SAAS,OAAO;AACd,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,yBAAyB,KAAK;AAAA,QAChC;AAAA,MACF;AAEA,aAAO;AAAA,QACL,aAAa,OAAO;AAAA,QACpB,UAAU;AAAA,UACR;AAAA,YACE,MAAM;AAAA,YACN,SAAS,EAAE,MAAM,QAAQ,MAAM,OAAO;AAAA,UACxC;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEO,QAAuC,MAAoB;AAChE,SAAK,OAAO,KAAK,IAAuB;AAAA,EAC1C;AAAA,EAEO,YAAY,UAAoB;AACrC,SAAK,WAAW,KAAK,QAAQ;AAAA,EAC/B;AAAA,EAEO,UAA+C,QAAsB;AAC1E,SAAK,SAAS,KAAK,MAAM;AAAA,EAC3B;AAAA,EAEA,cAAkC;AAAA,EAElC,MAAa,MACX,UAKQ;AAAA,IACN,eAAe;AAAA,EACjB,GACA;AACA,UAAM,eAAmC,CAAC;AAE1C,QAAI,KAAK,OAAO,QAAQ;AACtB,mBAAa,QAAQ,CAAC;AAAA,IACxB;AAEA,QAAI,KAAK,WAAW,QAAQ;AAC1B,mBAAa,YAAY,CAAC;AAAA,IAC5B;AAEA,QAAI,KAAK,SAAS,QAAQ;AACxB,mBAAa,UAAU,CAAC;AAAA,IAC1B;AAEA,SAAK,UAAU,IAAI;AAAA,MACjB,EAAE,MAAM,KAAK,SAAS,MAAM,SAAS,KAAK,SAAS,QAAQ;AAAA,MAC3D,EAAE,aAAa;AAAA,IACjB;AAEA,SAAK,cAAc,KAAK,OAAO;AAE/B,QAAI,QAAQ,kBAAkB,SAAS;AACrC,YAAM,YAAY,IAAI,qBAAqB;AAE3C,YAAM,KAAK,QAAQ,QAAQ,SAAS;AAEpC,cAAQ,MAAM,4BAA4B;AAAA,IAC5C,WAAW,QAAQ,kBAAkB,OAAO;AAC1C,UAAI,kBAA6C;AAKjD,WAAK,cAAc,KAAK,aAAa,OAAO,KAAK,QAAQ;AACvD,YAAI,IAAI,WAAW,SAAS,IAAI,QAAQ,QAAQ,IAAI,UAAU;AAC5D,gBAAM,YAAY,IAAI,mBAAmB,aAAa,GAAG;AAEzD,4BAAkB;AAElB,cAAI,CAAC,KAAK,SAAS;AACjB,kBAAM,IAAI,MAAM,wBAAwB;AAAA,UAC1C;AAEA,gBAAM,KAAK,QAAQ,QAAQ,SAAS;AAEpC,cAAI,GAAG,SAAS,MAAM;AACpB,oBAAQ,IAAI,uBAAuB;AACnC,gBAAI,oBAAoB,WAAW;AACjC,gCAAkB;AAAA,YACpB;AAAA,UACF,CAAC;AAED,eAAK,aAAa,SAAS;AAC3B;AAAA,QACF;AAEA,YAAI,IAAI,WAAW,UAAU,IAAI,KAAK,WAAW,WAAW,GAAG;AAC7D,cAAI,CAAC,iBAAiB;AACpB,gBAAI,UAAU,GAAG,EAAE,IAAI,qBAAqB;AAC5C;AAAA,UACF;AACA,gBAAM,gBAAgB,kBAAkB,KAAK,GAAG;AAChD;AAAA,QACF;AAEA,YAAI,UAAU,GAAG,EAAE,IAAI;AAAA,MACzB,CAAC;AAED,WAAK,YAAY,OAAO,QAAQ,IAAI,MAAM,SAAS;AAEnD,cAAQ;AAAA,QACN,gDAAgD,QAAQ,IAAI,IAAI,GAAG,QAAQ,IAAI,QAAQ;AAAA,MACzF;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,wBAAwB;AAAA,IAC1C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,aAAa,WAA+B;AACxD,QAAI;AACF,YAAM,UAAU,KAAK;AAAA,QACnB,SAAS;AAAA,QACT,QAAQ;AAAA,QACR,QAAQ,EAAE,SAAS,6BAA6B;AAAA,MAClD,CAAC;AAED,UAAI,eAAe;AACnB,YAAM,WAAW,YAAY,YAAY;AACvC;AAEA,cAAM,UAAU,WAAW,YAAY,QAAO,oBAAI,KAAK,GAAE,YAAY,CAAC;AAEtE,YAAI;AACF,gBAAM,UAAU,KAAK;AAAA,YACnB,SAAS;AAAA,YACT,QAAQ;AAAA,YACR,QAAQ,EAAE,MAAM,QAAQ;AAAA,UAC1B,CAAC;AAED,kBAAQ,IAAI,SAAS,OAAO,EAAE;AAE9B,cAAI,iBAAiB,IAAI;AACvB,0BAAc,QAAQ;AAEtB,kBAAM,UAAU,KAAK;AAAA,cACnB,SAAS;AAAA,cACT,QAAQ;AAAA,cACR,QAAQ,EAAE,SAAS,mBAAmB;AAAA,YACxC,CAAC;AACD,oBAAQ,IAAI,kBAAkB;AAAA,UAChC;AAAA,QACF,SAAS,OAAO;AACd,kBAAQ,MAAM,0BAA0B,KAAK;AAC7C,wBAAc,QAAQ;AAAA,QACxB;AAAA,MACF,GAAG,GAAI;AAAA,IACT,SAAS,OAAO;AACd,cAAQ,MAAM,0BAA0B,KAAK;AAAA,IAC/C;AAAA,EACF;AAAA,EAEA,MAAa,OAAO;AAClB,QAAI,KAAK,aAAa;AACpB,WAAK,YAAY,MAAM;AAAA,IACzB;AAAA,EACF;AACF;","names":["tool","resource","prompt"]}
+{"version":3,"sources":["../src/FastMCP.ts"],"sourcesContent":["import { Server } from \"@modelcontextprotocol/sdk/server/index.js\";\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport { SSEServerTransport } from \"@modelcontextprotocol/sdk/server/sse.js\";\nimport {\n  CallToolRequestSchema,\n  ErrorCode,\n  GetPromptRequestSchema,\n  ListPromptsRequestSchema,\n  ListResourcesRequestSchema,\n  ListToolsRequestSchema,\n  LoggingLevel,\n  McpError,\n  ReadResourceRequestSchema,\n  ServerCapabilities,\n  SetLevelRequestSchema,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { zodToJsonSchema } from \"zod-to-json-schema\";\nimport { z } from \"zod\";\nimport http from \"http\";\n\nabstract class FastMCPError extends Error {\n  public constructor(message?: string) {\n    super(message);\n    this.name = new.target.name;\n  }\n}\n\ntype Extra = unknown;\n\ntype Extras = Record<string, Extra>;\n\nclass UnexpectedStateError extends FastMCPError {\n  public extras?: Extras;\n\n  public constructor(message: string, extras?: Extras) {\n    super(message);\n    this.name = new.target.name;\n    this.extras = extras;\n  }\n}\n\nexport class UserError extends UnexpectedStateError {}\n\ntype ToolParameters = z.ZodTypeAny;\n\ntype Literal = boolean | null | number | string | undefined;\n\ntype SerializableValue =\n  | Literal\n  | SerializableValue[]\n  | { [key: string]: SerializableValue };\n\ntype Progress = {\n  /**\n   * The progress thus far. This should increase every time progress is made, even if the total is unknown.\n   */\n  progress: number;\n  /**\n   * Total number of items to process (or total progress required), if known.\n   */\n  total?: number;\n};\n\ntype Context = {\n  reportProgress: (progress: Progress) => Promise<void>;\n  log: {\n    debug: (message: string, data?: SerializableValue) => void;\n    error: (message: string, data?: SerializableValue) => void;\n    info: (message: string, data?: SerializableValue) => void;\n    warn: (message: string, data?: SerializableValue) => void;\n  };\n};\n\nconst TextContentZodSchema = z\n  .object({\n    type: z.literal(\"text\"),\n    /**\n     * The text content of the message.\n     */\n    text: z.string(),\n  })\n  .strict();\n\ntype TextContent = z.infer<typeof TextContentZodSchema>;\n\nconst ImageContentZodSchema = z\n  .object({\n    type: z.literal(\"image\"),\n    /**\n     * The base64-encoded image data.\n     */\n    data: z.string().base64(),\n    /**\n     * The MIME type of the image. Different providers may support different image types.\n     */\n    mimeType: z.string(),\n  })\n  .strict();\n\ntype ImageContent = z.infer<typeof ImageContentZodSchema>;\n\nconst ContentZodSchema = z.discriminatedUnion(\"type\", [\n  TextContentZodSchema,\n  ImageContentZodSchema,\n]);\n\nconst ContentResultZodSchema = z\n  .object({\n    content: ContentZodSchema.array(),\n    isError: z.boolean().optional(),\n  })\n  .strict();\n\ntype ContentResult = z.infer<typeof ContentResultZodSchema>;\n\ntype Tool<Params extends ToolParameters = ToolParameters> = {\n  name: string;\n  description?: string;\n  parameters?: Params;\n  execute: (\n    args: z.infer<Params>,\n    context: Context,\n  ) => Promise<string | ContentResult>;\n};\n\ntype Resource = {\n  uri: string;\n  name: string;\n  description?: string;\n  mimeType?: string;\n  load: () => Promise<{ text: string } | { blob: string }>;\n};\n\ntype PromptArgument = Readonly<{\n  name: string;\n  description?: string;\n  required?: boolean;\n}>;\n\ntype ArgumentsToObject<T extends PromptArgument[]> = {\n  [K in T[number][\"name\"]]: Extract<\n    T[number],\n    { name: K }\n  >[\"required\"] extends true\n    ? string\n    : string | undefined;\n};\n\ntype Prompt<\n  Arguments extends PromptArgument[] = PromptArgument[],\n  Args = ArgumentsToObject<Arguments>,\n> = {\n  name: string;\n  description?: string;\n  arguments?: Arguments;\n  load: (args: Args) => Promise<string>;\n};\n\ntype ServerOptions = {\n  name: string;\n  version: `${number}.${number}.${number}`;\n};\n\nexport class FastMCP {\n  #tools: Tool[];\n  #resources: Resource[];\n  #prompts: Prompt[];\n  #server: Server | null = null;\n  #options: ServerOptions;\n  #loggingLevel: LoggingLevel = \"info\";\n\n  constructor(public options: ServerOptions) {\n    this.#options = options;\n    this.#tools = [];\n    this.#resources = [];\n    this.#prompts = [];\n  }\n\n  private setupHandlers(server: Server) {\n    this.setupErrorHandling(server);\n\n    if (this.#tools.length) {\n      this.setupToolHandlers(server);\n    }\n\n    if (this.#resources.length) {\n      this.setupResourceHandlers(server);\n    }\n\n    if (this.#prompts.length) {\n      this.setupPromptHandlers(server);\n    }\n\n    server.setRequestHandler(SetLevelRequestSchema, (request) => {\n      this.#loggingLevel = request.params.level;\n\n      return {};\n    });\n  }\n\n  private setupErrorHandling(server: Server) {\n    server.onerror = (error) => {\n      console.error(\"[MCP Error]\", error);\n    };\n    process.on(\"SIGINT\", async () => {\n      await server.close();\n      process.exit(0);\n    });\n  }\n\n  public get loggingLevel() {\n    return this.#loggingLevel;\n  }\n\n  private setupToolHandlers(server: Server) {\n    server.setRequestHandler(ListToolsRequestSchema, async () => {\n      return {\n        tools: this.#tools.map((tool) => {\n          return {\n            name: tool.name,\n            description: tool.description,\n            inputSchema: tool.parameters\n              ? zodToJsonSchema(tool.parameters)\n              : undefined,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(\n      CallToolRequestSchema,\n      async (request, ...extra) => {\n        const tool = this.#tools.find(\n          (tool) => tool.name === request.params.name,\n        );\n\n        if (!tool) {\n          throw new McpError(\n            ErrorCode.MethodNotFound,\n            `Unknown tool: ${request.params.name}`,\n          );\n        }\n\n        let args: any = undefined;\n\n        if (tool.parameters) {\n          const parsed = tool.parameters.safeParse(request.params.arguments);\n\n          if (!parsed.success) {\n            throw new McpError(\n              ErrorCode.InvalidRequest,\n              `Invalid ${request.params.name} arguments`,\n            );\n          }\n\n          args = parsed.data;\n        }\n\n        const progressToken = request.params?._meta?.progressToken;\n\n        let result: ContentResult;\n\n        try {\n          const reportProgress = async (progress: Progress) => {\n            await server.notification({\n              method: \"notifications/progress\",\n              params: {\n                ...progress,\n                progressToken,\n              },\n            });\n          };\n\n          const log = {\n            debug: (message: string, context?: SerializableValue) => {\n              server.sendLoggingMessage({\n                level: \"debug\",\n                data: {\n                  message,\n                  context,\n                },\n              });\n            },\n            error: (message: string, context?: SerializableValue) => {\n              server.sendLoggingMessage({\n                level: \"error\",\n                data: {\n                  message,\n                  context,\n                },\n              });\n            },\n            info: (message: string, context?: SerializableValue) => {\n              server.sendLoggingMessage({\n                level: \"info\",\n                data: {\n                  message,\n                  context,\n                },\n              });\n            },\n            warn: (message: string, context?: SerializableValue) => {\n              server.sendLoggingMessage({\n                level: \"warning\",\n                data: {\n                  message,\n                  context,\n                },\n              });\n            },\n          };\n\n          const maybeStringResult = await tool.execute(args, {\n            reportProgress,\n            log,\n          });\n\n          if (typeof maybeStringResult === \"string\") {\n            result = ContentResultZodSchema.parse({\n              content: [{ type: \"text\", text: maybeStringResult }],\n            });\n          } else {\n            result = ContentResultZodSchema.parse(maybeStringResult);\n          }\n        } catch (error) {\n          if (error instanceof UserError) {\n            return {\n              content: [{ type: \"text\", text: error.message }],\n              isError: true,\n            };\n          }\n\n          return {\n            content: [{ type: \"text\", text: `Error: ${error}` }],\n            isError: true,\n          };\n        }\n\n        return result;\n      },\n    );\n  }\n\n  private setupResourceHandlers(server: Server) {\n    server.setRequestHandler(ListResourcesRequestSchema, async () => {\n      return {\n        resources: this.#resources.map((resource) => {\n          return {\n            uri: resource.uri,\n            name: resource.name,\n            mimeType: resource.mimeType,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(ReadResourceRequestSchema, async (request) => {\n      const resource = this.#resources.find(\n        (resource) => resource.uri === request.params.uri,\n      );\n\n      if (!resource) {\n        throw new McpError(\n          ErrorCode.MethodNotFound,\n          `Unknown resource: ${request.params.uri}`,\n        );\n      }\n\n      let result: Awaited<ReturnType<Resource[\"load\"]>>;\n\n      try {\n        result = await resource.load();\n      } catch (error) {\n        throw new McpError(\n          ErrorCode.InternalError,\n          `Error reading resource: ${error}`,\n          {\n            uri: resource.uri,\n          },\n        );\n      }\n\n      return {\n        contents: [\n          {\n            uri: resource.uri,\n            mimeType: resource.mimeType,\n            ...result,\n          },\n        ],\n      };\n    });\n  }\n\n  private setupPromptHandlers(server: Server) {\n    server.setRequestHandler(ListPromptsRequestSchema, async () => {\n      return {\n        prompts: this.#prompts.map((prompt) => {\n          return {\n            name: prompt.name,\n            description: prompt.description,\n            arguments: prompt.arguments,\n          };\n        }),\n      };\n    });\n\n    server.setRequestHandler(GetPromptRequestSchema, async (request) => {\n      const prompt = this.#prompts.find(\n        (prompt) => prompt.name === request.params.name,\n      );\n\n      if (!prompt) {\n        throw new McpError(\n          ErrorCode.MethodNotFound,\n          `Unknown prompt: ${request.params.name}`,\n        );\n      }\n\n      const args = request.params.arguments;\n\n      if (prompt.arguments) {\n        for (const arg of prompt.arguments) {\n          if (arg.required && !(args && arg.name in args)) {\n            throw new McpError(\n              ErrorCode.InvalidRequest,\n              `Missing required argument: ${arg.name}`,\n            );\n          }\n        }\n      }\n\n      let result: Awaited<ReturnType<Prompt[\"load\"]>>;\n\n      try {\n        result = await prompt.load(args as Record<string, string | undefined>);\n      } catch (error) {\n        throw new McpError(\n          ErrorCode.InternalError,\n          `Error loading prompt: ${error}`,\n        );\n      }\n\n      return {\n        description: prompt.description,\n        messages: [\n          {\n            role: \"user\",\n            content: { type: \"text\", text: result },\n          },\n        ],\n      };\n    });\n  }\n\n  public addTool<Params extends ToolParameters>(tool: Tool<Params>) {\n    this.#tools.push(tool as unknown as Tool);\n  }\n\n  public addResource(resource: Resource) {\n    this.#resources.push(resource);\n  }\n\n  public addPrompt<const Args extends PromptArgument[]>(prompt: Prompt<Args>) {\n    this.#prompts.push(prompt);\n  }\n\n  #httpServer: http.Server | null = null;\n\n  public async start(\n    options:\n      | { transportType: \"stdio\" }\n      | {\n          transportType: \"sse\";\n          sse: { endpoint: `/${string}`; port: number };\n        } = {\n      transportType: \"stdio\",\n    },\n  ) {\n    const capabilities: ServerCapabilities = {};\n\n    if (this.#tools.length) {\n      capabilities.tools = {};\n    }\n\n    if (this.#resources.length) {\n      capabilities.resources = {};\n    }\n\n    if (this.#prompts.length) {\n      capabilities.prompts = {};\n    }\n\n    capabilities.logging = {};\n\n    this.#server = new Server(\n      { name: this.#options.name, version: this.#options.version },\n      { capabilities },\n    );\n\n    this.setupHandlers(this.#server);\n\n    if (options.transportType === \"stdio\") {\n      const transport = new StdioServerTransport();\n\n      await this.#server.connect(transport);\n\n      console.error(`server is running on stdio`);\n    } else if (options.transportType === \"sse\") {\n      let activeTransport: SSEServerTransport | null = null;\n\n      /**\n       * Adopted from https://dev.classmethod.jp/articles/mcp-sse/\n       */\n      this.#httpServer = http.createServer(async (req, res) => {\n        if (req.method === \"GET\" && req.url === options.sse.endpoint) {\n          const transport = new SSEServerTransport(\"/messages\", res);\n\n          activeTransport = transport;\n\n          if (!this.#server) {\n            throw new Error(\"Server not initialized\");\n          }\n\n          await this.#server.connect(transport);\n\n          res.on(\"close\", () => {\n            console.log(\"SSE connection closed\");\n            if (activeTransport === transport) {\n              activeTransport = null;\n            }\n          });\n\n          this.startSending(transport);\n          return;\n        }\n\n        if (req.method === \"POST\" && req.url?.startsWith(\"/messages\")) {\n          if (!activeTransport) {\n            res.writeHead(400).end(\"No active transport\");\n            return;\n          }\n          await activeTransport.handlePostMessage(req, res);\n          return;\n        }\n\n        res.writeHead(404).end();\n      });\n\n      this.#httpServer.listen(options.sse.port, \"0.0.0.0\");\n\n      console.error(\n        `server is running on SSE at http://localhost:${options.sse.port}${options.sse.endpoint}`,\n      );\n    } else {\n      throw new Error(\"Invalid transport type\");\n    }\n  }\n\n  /**\n   * @see https://dev.classmethod.jp/articles/mcp-sse/\n   */\n  private async startSending(transport: SSEServerTransport) {\n    try {\n      await transport.send({\n        jsonrpc: \"2.0\",\n        method: \"sse/connection\",\n        params: { message: \"SSE Connection established\" },\n      });\n\n      let messageCount = 0;\n      const interval = setInterval(async () => {\n        messageCount++;\n\n        const message = `Message ${messageCount} at ${new Date().toISOString()}`;\n\n        try {\n          await transport.send({\n            jsonrpc: \"2.0\",\n            method: \"sse/message\",\n            params: { data: message },\n          });\n\n          console.log(`Sent: ${message}`);\n\n          if (messageCount === 10) {\n            clearInterval(interval);\n\n            await transport.send({\n              jsonrpc: \"2.0\",\n              method: \"sse/complete\",\n              params: { message: \"Stream completed\" },\n            });\n            console.log(\"Stream completed\");\n          }\n        } catch (error) {\n          console.error(\"Error sending message:\", error);\n          clearInterval(interval);\n        }\n      }, 1000);\n    } catch (error) {\n      console.error(\"Error in startSending:\", error);\n    }\n  }\n\n  public async stop() {\n    if (this.#httpServer) {\n      this.#httpServer.close();\n    }\n  }\n}\n"],"mappings":";AAAA,SAAS,cAAc;AACvB,SAAS,4BAA4B;AACrC,SAAS,0BAA0B;AACnC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA;AAAA,EACA;AAAA,EAEA;AAAA,OACK;AACP,SAAS,uBAAuB;AAChC,SAAS,SAAS;AAClB,OAAO,UAAU;AAEjB,IAAe,eAAf,cAAoC,MAAM;AAAA,EACjC,YAAY,SAAkB;AACnC,UAAM,OAAO;AACb,SAAK,OAAO,WAAW;AAAA,EACzB;AACF;AAMA,IAAM,uBAAN,cAAmC,aAAa;AAAA,EACvC;AAAA,EAEA,YAAY,SAAiB,QAAiB;AACnD,UAAM,OAAO;AACb,SAAK,OAAO,WAAW;AACvB,SAAK,SAAS;AAAA,EAChB;AACF;AAEO,IAAM,YAAN,cAAwB,qBAAqB;AAAC;AAgCrD,IAAM,uBAAuB,EAC1B,OAAO;AAAA,EACN,MAAM,EAAE,QAAQ,MAAM;AAAA;AAAA;AAAA;AAAA,EAItB,MAAM,EAAE,OAAO;AACjB,CAAC,EACA,OAAO;AAIV,IAAM,wBAAwB,EAC3B,OAAO;AAAA,EACN,MAAM,EAAE,QAAQ,OAAO;AAAA;AAAA;AAAA;AAAA,EAIvB,MAAM,EAAE,OAAO,EAAE,OAAO;AAAA;AAAA;AAAA;AAAA,EAIxB,UAAU,EAAE,OAAO;AACrB,CAAC,EACA,OAAO;AAIV,IAAM,mBAAmB,EAAE,mBAAmB,QAAQ;AAAA,EACpD;AAAA,EACA;AACF,CAAC;AAED,IAAM,yBAAyB,EAC5B,OAAO;AAAA,EACN,SAAS,iBAAiB,MAAM;AAAA,EAChC,SAAS,EAAE,QAAQ,EAAE,SAAS;AAChC,CAAC,EACA,OAAO;AAoDH,IAAM,UAAN,MAAc;AAAA,EAQnB,YAAmB,SAAwB;AAAxB;AACjB,SAAK,WAAW;AAChB,SAAK,SAAS,CAAC;AACf,SAAK,aAAa,CAAC;AACnB,SAAK,WAAW,CAAC;AAAA,EACnB;AAAA,EAZA;AAAA,EACA;AAAA,EACA;AAAA,EACA,UAAyB;AAAA,EACzB;AAAA,EACA,gBAA8B;AAAA,EAStB,cAAc,QAAgB;AACpC,SAAK,mBAAmB,MAAM;AAE9B,QAAI,KAAK,OAAO,QAAQ;AACtB,WAAK,kBAAkB,MAAM;AAAA,IAC/B;AAEA,QAAI,KAAK,WAAW,QAAQ;AAC1B,WAAK,sBAAsB,MAAM;AAAA,IACnC;AAEA,QAAI,KAAK,SAAS,QAAQ;AACxB,WAAK,oBAAoB,MAAM;AAAA,IACjC;AAEA,WAAO,kBAAkB,uBAAuB,CAAC,YAAY;AAC3D,WAAK,gBAAgB,QAAQ,OAAO;AAEpC,aAAO,CAAC;AAAA,IACV,CAAC;AAAA,EACH;AAAA,EAEQ,mBAAmB,QAAgB;AACzC,WAAO,UAAU,CAAC,UAAU;AAC1B,cAAQ,MAAM,eAAe,KAAK;AAAA,IACpC;AACA,YAAQ,GAAG,UAAU,YAAY;AAC/B,YAAM,OAAO,MAAM;AACnB,cAAQ,KAAK,CAAC;AAAA,IAChB,CAAC;AAAA,EACH;AAAA,EAEA,IAAW,eAAe;AACxB,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ,kBAAkB,QAAgB;AACxC,WAAO,kBAAkB,wBAAwB,YAAY;AAC3D,aAAO;AAAA,QACL,OAAO,KAAK,OAAO,IAAI,CAAC,SAAS;AAC/B,iBAAO;AAAA,YACL,MAAM,KAAK;AAAA,YACX,aAAa,KAAK;AAAA,YAClB,aAAa,KAAK,aACd,gBAAgB,KAAK,UAAU,IAC/B;AAAA,UACN;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA,OAAO,YAAY,UAAU;AAC3B,cAAM,OAAO,KAAK,OAAO;AAAA,UACvB,CAACA,UAASA,MAAK,SAAS,QAAQ,OAAO;AAAA,QACzC;AAEA,YAAI,CAAC,MAAM;AACT,gBAAM,IAAI;AAAA,YACR,UAAU;AAAA,YACV,iBAAiB,QAAQ,OAAO,IAAI;AAAA,UACtC;AAAA,QACF;AAEA,YAAI,OAAY;AAEhB,YAAI,KAAK,YAAY;AACnB,gBAAM,SAAS,KAAK,WAAW,UAAU,QAAQ,OAAO,SAAS;AAEjE,cAAI,CAAC,OAAO,SAAS;AACnB,kBAAM,IAAI;AAAA,cACR,UAAU;AAAA,cACV,WAAW,QAAQ,OAAO,IAAI;AAAA,YAChC;AAAA,UACF;AAEA,iBAAO,OAAO;AAAA,QAChB;AAEA,cAAM,gBAAgB,QAAQ,QAAQ,OAAO;AAE7C,YAAI;AAEJ,YAAI;AACF,gBAAM,iBAAiB,OAAO,aAAuB;AACnD,kBAAM,OAAO,aAAa;AAAA,cACxB,QAAQ;AAAA,cACR,QAAQ;AAAA,gBACN,GAAG;AAAA,gBACH;AAAA,cACF;AAAA,YACF,CAAC;AAAA,UACH;AAEA,gBAAM,MAAM;AAAA,YACV,OAAO,CAAC,SAAiB,YAAgC;AACvD,qBAAO,mBAAmB;AAAA,gBACxB,OAAO;AAAA,gBACP,MAAM;AAAA,kBACJ;AAAA,kBACA;AAAA,gBACF;AAAA,cACF,CAAC;AAAA,YACH;AAAA,YACA,OAAO,CAAC,SAAiB,YAAgC;AACvD,qBAAO,mBAAmB;AAAA,gBACxB,OAAO;AAAA,gBACP,MAAM;AAAA,kBACJ;AAAA,kBACA;AAAA,gBACF;AAAA,cACF,CAAC;AAAA,YACH;AAAA,YACA,MAAM,CAAC,SAAiB,YAAgC;AACtD,qBAAO,mBAAmB;AAAA,gBACxB,OAAO;AAAA,gBACP,MAAM;AAAA,kBACJ;AAAA,kBACA;AAAA,gBACF;AAAA,cACF,CAAC;AAAA,YACH;AAAA,YACA,MAAM,CAAC,SAAiB,YAAgC;AACtD,qBAAO,mBAAmB;AAAA,gBACxB,OAAO;AAAA,gBACP,MAAM;AAAA,kBACJ;AAAA,kBACA;AAAA,gBACF;AAAA,cACF,CAAC;AAAA,YACH;AAAA,UACF;AAEA,gBAAM,oBAAoB,MAAM,KAAK,QAAQ,MAAM;AAAA,YACjD;AAAA,YACA;AAAA,UACF,CAAC;AAED,cAAI,OAAO,sBAAsB,UAAU;AACzC,qBAAS,uBAAuB,MAAM;AAAA,cACpC,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,kBAAkB,CAAC;AAAA,YACrD,CAAC;AAAA,UACH,OAAO;AACL,qBAAS,uBAAuB,MAAM,iBAAiB;AAAA,UACzD;AAAA,QACF,SAAS,OAAO;AACd,cAAI,iBAAiB,WAAW;AAC9B,mBAAO;AAAA,cACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,MAAM,QAAQ,CAAC;AAAA,cAC/C,SAAS;AAAA,YACX;AAAA,UACF;AAEA,iBAAO;AAAA,YACL,SAAS,CAAC,EAAE,MAAM,QAAQ,MAAM,UAAU,KAAK,GAAG,CAAC;AAAA,YACnD,SAAS;AAAA,UACX;AAAA,QACF;AAEA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBAAsB,QAAgB;AAC5C,WAAO,kBAAkB,4BAA4B,YAAY;AAC/D,aAAO;AAAA,QACL,WAAW,KAAK,WAAW,IAAI,CAAC,aAAa;AAC3C,iBAAO;AAAA,YACL,KAAK,SAAS;AAAA,YACd,MAAM,SAAS;AAAA,YACf,UAAU,SAAS;AAAA,UACrB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO,kBAAkB,2BAA2B,OAAO,YAAY;AACrE,YAAM,WAAW,KAAK,WAAW;AAAA,QAC/B,CAACC,cAAaA,UAAS,QAAQ,QAAQ,OAAO;AAAA,MAChD;AAEA,UAAI,CAAC,UAAU;AACb,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,qBAAqB,QAAQ,OAAO,GAAG;AAAA,QACzC;AAAA,MACF;AAEA,UAAI;AAEJ,UAAI;AACF,iBAAS,MAAM,SAAS,KAAK;AAAA,MAC/B,SAAS,OAAO;AACd,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,2BAA2B,KAAK;AAAA,UAChC;AAAA,YACE,KAAK,SAAS;AAAA,UAChB;AAAA,QACF;AAAA,MACF;AAEA,aAAO;AAAA,QACL,UAAU;AAAA,UACR;AAAA,YACE,KAAK,SAAS;AAAA,YACd,UAAU,SAAS;AAAA,YACnB,GAAG;AAAA,UACL;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,oBAAoB,QAAgB;AAC1C,WAAO,kBAAkB,0BAA0B,YAAY;AAC7D,aAAO;AAAA,QACL,SAAS,KAAK,SAAS,IAAI,CAAC,WAAW;AACrC,iBAAO;AAAA,YACL,MAAM,OAAO;AAAA,YACb,aAAa,OAAO;AAAA,YACpB,WAAW,OAAO;AAAA,UACpB;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAED,WAAO,kBAAkB,wBAAwB,OAAO,YAAY;AAClE,YAAM,SAAS,KAAK,SAAS;AAAA,QAC3B,CAACC,YAAWA,QAAO,SAAS,QAAQ,OAAO;AAAA,MAC7C;AAEA,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,mBAAmB,QAAQ,OAAO,IAAI;AAAA,QACxC;AAAA,MACF;AAEA,YAAM,OAAO,QAAQ,OAAO;AAE5B,UAAI,OAAO,WAAW;AACpB,mBAAW,OAAO,OAAO,WAAW;AAClC,cAAI,IAAI,YAAY,EAAE,QAAQ,IAAI,QAAQ,OAAO;AAC/C,kBAAM,IAAI;AAAA,cACR,UAAU;AAAA,cACV,8BAA8B,IAAI,IAAI;AAAA,YACxC;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAEA,UAAI;AAEJ,UAAI;AACF,iBAAS,MAAM,OAAO,KAAK,IAA0C;AAAA,MACvE,SAAS,OAAO;AACd,cAAM,IAAI;AAAA,UACR,UAAU;AAAA,UACV,yBAAyB,KAAK;AAAA,QAChC;AAAA,MACF;AAEA,aAAO;AAAA,QACL,aAAa,OAAO;AAAA,QACpB,UAAU;AAAA,UACR;AAAA,YACE,MAAM;AAAA,YACN,SAAS,EAAE,MAAM,QAAQ,MAAM,OAAO;AAAA,UACxC;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEO,QAAuC,MAAoB;AAChE,SAAK,OAAO,KAAK,IAAuB;AAAA,EAC1C;AAAA,EAEO,YAAY,UAAoB;AACrC,SAAK,WAAW,KAAK,QAAQ;AAAA,EAC/B;AAAA,EAEO,UAA+C,QAAsB;AAC1E,SAAK,SAAS,KAAK,MAAM;AAAA,EAC3B;AAAA,EAEA,cAAkC;AAAA,EAElC,MAAa,MACX,UAKQ;AAAA,IACN,eAAe;AAAA,EACjB,GACA;AACA,UAAM,eAAmC,CAAC;AAE1C,QAAI,KAAK,OAAO,QAAQ;AACtB,mBAAa,QAAQ,CAAC;AAAA,IACxB;AAEA,QAAI,KAAK,WAAW,QAAQ;AAC1B,mBAAa,YAAY,CAAC;AAAA,IAC5B;AAEA,QAAI,KAAK,SAAS,QAAQ;AACxB,mBAAa,UAAU,CAAC;AAAA,IAC1B;AAEA,iBAAa,UAAU,CAAC;AAExB,SAAK,UAAU,IAAI;AAAA,MACjB,EAAE,MAAM,KAAK,SAAS,MAAM,SAAS,KAAK,SAAS,QAAQ;AAAA,MAC3D,EAAE,aAAa;AAAA,IACjB;AAEA,SAAK,cAAc,KAAK,OAAO;AAE/B,QAAI,QAAQ,kBAAkB,SAAS;AACrC,YAAM,YAAY,IAAI,qBAAqB;AAE3C,YAAM,KAAK,QAAQ,QAAQ,SAAS;AAEpC,cAAQ,MAAM,4BAA4B;AAAA,IAC5C,WAAW,QAAQ,kBAAkB,OAAO;AAC1C,UAAI,kBAA6C;AAKjD,WAAK,cAAc,KAAK,aAAa,OAAO,KAAK,QAAQ;AACvD,YAAI,IAAI,WAAW,SAAS,IAAI,QAAQ,QAAQ,IAAI,UAAU;AAC5D,gBAAM,YAAY,IAAI,mBAAmB,aAAa,GAAG;AAEzD,4BAAkB;AAElB,cAAI,CAAC,KAAK,SAAS;AACjB,kBAAM,IAAI,MAAM,wBAAwB;AAAA,UAC1C;AAEA,gBAAM,KAAK,QAAQ,QAAQ,SAAS;AAEpC,cAAI,GAAG,SAAS,MAAM;AACpB,oBAAQ,IAAI,uBAAuB;AACnC,gBAAI,oBAAoB,WAAW;AACjC,gCAAkB;AAAA,YACpB;AAAA,UACF,CAAC;AAED,eAAK,aAAa,SAAS;AAC3B;AAAA,QACF;AAEA,YAAI,IAAI,WAAW,UAAU,IAAI,KAAK,WAAW,WAAW,GAAG;AAC7D,cAAI,CAAC,iBAAiB;AACpB,gBAAI,UAAU,GAAG,EAAE,IAAI,qBAAqB;AAC5C;AAAA,UACF;AACA,gBAAM,gBAAgB,kBAAkB,KAAK,GAAG;AAChD;AAAA,QACF;AAEA,YAAI,UAAU,GAAG,EAAE,IAAI;AAAA,MACzB,CAAC;AAED,WAAK,YAAY,OAAO,QAAQ,IAAI,MAAM,SAAS;AAEnD,cAAQ;AAAA,QACN,gDAAgD,QAAQ,IAAI,IAAI,GAAG,QAAQ,IAAI,QAAQ;AAAA,MACzF;AAAA,IACF,OAAO;AACL,YAAM,IAAI,MAAM,wBAAwB;AAAA,IAC1C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,aAAa,WAA+B;AACxD,QAAI;AACF,YAAM,UAAU,KAAK;AAAA,QACnB,SAAS;AAAA,QACT,QAAQ;AAAA,QACR,QAAQ,EAAE,SAAS,6BAA6B;AAAA,MAClD,CAAC;AAED,UAAI,eAAe;AACnB,YAAM,WAAW,YAAY,YAAY;AACvC;AAEA,cAAM,UAAU,WAAW,YAAY,QAAO,oBAAI,KAAK,GAAE,YAAY,CAAC;AAEtE,YAAI;AACF,gBAAM,UAAU,KAAK;AAAA,YACnB,SAAS;AAAA,YACT,QAAQ;AAAA,YACR,QAAQ,EAAE,MAAM,QAAQ;AAAA,UAC1B,CAAC;AAED,kBAAQ,IAAI,SAAS,OAAO,EAAE;AAE9B,cAAI,iBAAiB,IAAI;AACvB,0BAAc,QAAQ;AAEtB,kBAAM,UAAU,KAAK;AAAA,cACnB,SAAS;AAAA,cACT,QAAQ;AAAA,cACR,QAAQ,EAAE,SAAS,mBAAmB;AAAA,YACxC,CAAC;AACD,oBAAQ,IAAI,kBAAkB;AAAA,UAChC;AAAA,QACF,SAAS,OAAO;AACd,kBAAQ,MAAM,0BAA0B,KAAK;AAC7C,wBAAc,QAAQ;AAAA,QACxB;AAAA,MACF,GAAG,GAAI;AAAA,IACT,SAAS,OAAO;AACd,cAAQ,MAAM,0BAA0B,KAAK;AAAA,IAC/C;AAAA,EACF;AAAA,EAEA,MAAa,OAAO;AAClB,QAAI,KAAK,aAAa;AACpB,WAAK,YAAY,MAAM;AAAA,IACzB;AAAA,EACF;AACF;","names":["tool","resource","prompt"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fastmcp",
-  "version": "1.2.0",
+  "version": "1.4.0",
   "main": "dist/fastmcp.js",
   "scripts": {
     "build": "tsup",

package/src/FastMCP.test.ts

@@ -6,7 +6,11 @@ import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
 import { getRandomPort } from "get-port-please";
 import { EventSource } from "eventsource";
 import { setTimeout as delay } from "timers/promises";
-import { ErrorCode, McpError } from "@modelcontextprotocol/sdk/types.js";
+import {
+  ErrorCode,
+  LoggingMessageNotificationSchema,
+  McpError,
+} from "@modelcontextprotocol/sdk/types.js";
 
 // @ts-expect-error - figure out how to use --experimental-eventsource with vitest
 global.EventSource = EventSource;
@@ -16,7 +20,13 @@ const runWithTestServer = async ({
   start,
 }: {
   start: () => Promise<FastMCP>;
-  run: ({ client }: { client: Client }) => Promise<void>;
+  run: ({
+    client,
+    server,
+  }: {
+    client: Client;
+    server: FastMCP;
+  }) => Promise<void>;
 }) => {
   const port = await getRandomPort();
 
@@ -47,7 +57,7 @@ const runWithTestServer = async ({
 
     await client.connect(transport);
 
-    await run({ client });
+    await run({ client, server });
   } finally {
     await server.stop();
   }
@@ -71,7 +81,7 @@ test("adds tools", async () => {
           b: z.number(),
         }),
         execute: async (args) => {
-          return args.a + args.b;
+          return String(args.a + args.b);
         },
       });
 
@@ -116,7 +126,7 @@ test("calls a tool", async () => {
           b: z.number(),
         }),
         execute: async (args) => {
-          return args.a + args.b;
+          return String(args.a + args.b);
         },
       });
 
@@ -138,7 +148,7 @@ test("calls a tool", async () => {
   });
 });
 
-test("handles UserError errors", async () => {
+test("returns a list", async () => {
   await runWithTestServer({
     start: async () => {
       const server = new FastMCP({
@@ -154,6 +164,52 @@ test("handles UserError errors", async () => {
           b: z.number(),
         }),
         execute: async (args) => {
+          return {
+            content: [
+              { type: "text", text: "a" },
+              { type: "text", text: "b" },
+            ],
+          };
+        },
+      });
+
+      return server;
+    },
+    run: async ({ client }) => {
+      expect(
+        await client.callTool({
+          name: "add",
+          arguments: {
+            a: 1,
+            b: 2,
+          },
+        }),
+      ).toEqual({
+        content: [
+          { type: "text", text: "a" },
+          { type: "text", text: "b" },
+        ],
+      });
+    },
+  });
+});
+
+test("handles UserError errors", async () => {
+  await runWithTestServer({
+    start: async () => {
+      const server = new FastMCP({
+        name: "Test",
+        version: "1.0.0",
+      });
+
+      server.addTool({
+        name: "add",
+        description: "Add two numbers",
+        parameters: z.object({
+          a: z.number(),
+          b: z.number(),
+        }),
+        execute: async () => {
           throw new UserError("Something went wrong");
         },
       });
@@ -229,7 +285,7 @@ test("tracks tool progress", async () => {
 
           await delay(100);
 
-          return args.a + args.b;
+          return String(args.a + args.b);
         },
       });
 
@@ -261,6 +317,104 @@ test("tracks tool progress", async () => {
   });
 });
 
+test("sets logging levels", async () => {
+  await runWithTestServer({
+    start: async () => {
+      const server = new FastMCP({
+        name: "Test",
+        version: "1.0.0",
+      });
+
+      return server;
+    },
+    run: async ({ client, server }) => {
+      await client.setLoggingLevel("debug");
+
+      expect(server.loggingLevel).toBe("debug");
+
+      await client.setLoggingLevel("info");
+
+      expect(server.loggingLevel).toBe("info");
+    },
+  });
+});
+
+test("sends logging messages to the client", async () => {
+  await runWithTestServer({
+    start: async () => {
+      const server = new FastMCP({
+        name: "Test",
+        version: "1.0.0",
+      });
+
+      server.addTool({
+        name: "add",
+        description: "Add two numbers",
+        parameters: z.object({
+          a: z.number(),
+          b: z.number(),
+        }),
+        execute: async (args, { log }) => {
+          log.debug("debug message", {
+            foo: "bar",
+          });
+          log.error("error message");
+          log.info("info message");
+          log.warn("warn message");
+
+          return String(args.a + args.b);
+        },
+      });
+
+      return server;
+    },
+    run: async ({ client, server }) => {
+      const onLog = vi.fn();
+
+      client.setNotificationHandler(
+        LoggingMessageNotificationSchema,
+        (message) => {
+          if (message.method === "notifications/message") {
+            onLog({
+              level: message.params.level,
+              ...(message.params.data ?? {}),
+            });
+          }
+        },
+      );
+
+      await client.callTool({
+        name: "add",
+        arguments: {
+          a: 1,
+          b: 2,
+        },
+      });
+
+      expect(onLog).toHaveBeenCalledTimes(4);
+      expect(onLog).toHaveBeenNthCalledWith(1, {
+        level: "debug",
+        message: "debug message",
+        context: {
+          foo: "bar",
+        },
+      });
+      expect(onLog).toHaveBeenNthCalledWith(2, {
+        level: "error",
+        message: "error message",
+      });
+      expect(onLog).toHaveBeenNthCalledWith(3, {
+        level: "info",
+        message: "info message",
+      });
+      expect(onLog).toHaveBeenNthCalledWith(4, {
+        level: "warning",
+        message: "warn message",
+      });
+    },
+  });
+});
+
 test("adds resources", async () => {
   await runWithTestServer({
     start: async () => {

package/src/FastMCP.ts

@@ -8,12 +8,14 @@ import {
   ListPromptsRequestSchema,
   ListResourcesRequestSchema,
   ListToolsRequestSchema,
+  LoggingLevel,
   McpError,
   ReadResourceRequestSchema,
   ServerCapabilities,
+  SetLevelRequestSchema,
 } from "@modelcontextprotocol/sdk/types.js";
 import { zodToJsonSchema } from "zod-to-json-schema";
-import type { z } from "zod";
+import { z } from "zod";
 import http from "http";
 
 abstract class FastMCPError extends Error {
@@ -41,6 +43,13 @@ export class UserError extends UnexpectedStateError {}
 
 type ToolParameters = z.ZodTypeAny;
 
+type Literal = boolean | null | number | string | undefined;
+
+type SerializableValue =
+  | Literal
+  | SerializableValue[]
+  | { [key: string]: SerializableValue };
+
 type Progress = {
   /**
    * The progress thus far. This should increase every time progress is made, even if the total is unknown.
@@ -54,13 +63,64 @@ type Progress = {
 
 type Context = {
   reportProgress: (progress: Progress) => Promise<void>;
+  log: {
+    debug: (message: string, data?: SerializableValue) => void;
+    error: (message: string, data?: SerializableValue) => void;
+    info: (message: string, data?: SerializableValue) => void;
+    warn: (message: string, data?: SerializableValue) => void;
+  };
 };
 
+const TextContentZodSchema = z
+  .object({
+    type: z.literal("text"),
+    /**
+     * The text content of the message.
+     */
+    text: z.string(),
+  })
+  .strict();
+
+type TextContent = z.infer<typeof TextContentZodSchema>;
+
+const ImageContentZodSchema = z
+  .object({
+    type: z.literal("image"),
+    /**
+     * The base64-encoded image data.
+     */
+    data: z.string().base64(),
+    /**
+     * The MIME type of the image. Different providers may support different image types.
+     */
+    mimeType: z.string(),
+  })
+  .strict();
+
+type ImageContent = z.infer<typeof ImageContentZodSchema>;
+
+const ContentZodSchema = z.discriminatedUnion("type", [
+  TextContentZodSchema,
+  ImageContentZodSchema,
+]);
+
+const ContentResultZodSchema = z
+  .object({
+    content: ContentZodSchema.array(),
+    isError: z.boolean().optional(),
+  })
+  .strict();
+
+type ContentResult = z.infer<typeof ContentResultZodSchema>;
+
 type Tool<Params extends ToolParameters = ToolParameters> = {
   name: string;
   description?: string;
   parameters?: Params;
-  execute: (args: z.infer<Params>, context: Context) => Promise<unknown>;
+  execute: (
+    args: z.infer<Params>,
+    context: Context,
+  ) => Promise<string | ContentResult>;
 };
 
 type Resource = {
@@ -107,6 +167,7 @@ export class FastMCP {
   #prompts: Prompt[];
   #server: Server | null = null;
   #options: ServerOptions;
+  #loggingLevel: LoggingLevel = "info";
 
   constructor(public options: ServerOptions) {
     this.#options = options;
@@ -129,6 +190,12 @@ export class FastMCP {
     if (this.#prompts.length) {
       this.setupPromptHandlers(server);
     }
+
+    server.setRequestHandler(SetLevelRequestSchema, (request) => {
+      this.#loggingLevel = request.params.level;
+
+      return {};
+    });
   }
 
   private setupErrorHandling(server: Server) {
@@ -141,6 +208,10 @@ export class FastMCP {
     });
   }
 
+  public get loggingLevel() {
+    return this.#loggingLevel;
+  }
+
   private setupToolHandlers(server: Server) {
     server.setRequestHandler(ListToolsRequestSchema, async () => {
       return {
@@ -187,7 +258,7 @@ export class FastMCP {
 
         const progressToken = request.params?._meta?.progressToken;
 
-        let result: any;
+        let result: ContentResult;
 
         try {
           const reportProgress = async (progress: Progress) => {
@@ -200,9 +271,57 @@ export class FastMCP {
             });
           };
 
-          result = await tool.execute(args, {
+          const log = {
+            debug: (message: string, context?: SerializableValue) => {
+              server.sendLoggingMessage({
+                level: "debug",
+                data: {
+                  message,
+                  context,
+                },
+              });
+            },
+            error: (message: string, context?: SerializableValue) => {
+              server.sendLoggingMessage({
+                level: "error",
+                data: {
+                  message,
+                  context,
+                },
+              });
+            },
+            info: (message: string, context?: SerializableValue) => {
+              server.sendLoggingMessage({
+                level: "info",
+                data: {
+                  message,
+                  context,
+                },
+              });
+            },
+            warn: (message: string, context?: SerializableValue) => {
+              server.sendLoggingMessage({
+                level: "warning",
+                data: {
+                  message,
+                  context,
+                },
+              });
+            },
+          };
+
+          const maybeStringResult = await tool.execute(args, {
             reportProgress,
+            log,
           });
+
+          if (typeof maybeStringResult === "string") {
+            result = ContentResultZodSchema.parse({
+              content: [{ type: "text", text: maybeStringResult }],
+            });
+          } else {
+            result = ContentResultZodSchema.parse(maybeStringResult);
+          }
         } catch (error) {
           if (error instanceof UserError) {
             return {
@@ -217,15 +336,7 @@ export class FastMCP {
           };
         }
 
-        if (typeof result === "string") {
-          return {
-            content: [{ type: "text", text: result }],
-          };
-        }
-
-        return {
-          content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
-        };
+        return result;
       },
     );
   }
@@ -380,6 +491,8 @@ export class FastMCP {
       capabilities.prompts = {};
     }
 
+    capabilities.logging = {};
+
     this.#server = new Server(
       { name: this.#options.name, version: this.#options.version },
       { capabilities },