tmcp 1.9.0 → 1.10.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tmcp",
-  "version": "1.9.0",
+  "version": "1.10.0",
   "description": "The main tmcp library",
   "main": "src/index.js",
   "type": "module",

package/src/index.js

@@ -4,7 +4,7 @@
  * @import { JSONRPCRequest, JSONRPCParams } from "json-rpc-2.0";
  * @import { ExtractURITemplateVariables } from "./internal/uri-template.js";
  * @import { CallToolResult, ReadResourceResult, GetPromptResult, ClientCapabilities as ClientCapabilitiesType, JSONRPCRequest as JSONRPCRequestType, JSONRPCResponse, CreateMessageRequestParams, CreateMessageResult, Resource, LoggingLevel, ToolAnnotations } from "./validation/index.js";
- * @import { Tool, Completion, Prompt, StoredResource, ServerOptions, ServerInfo, SubscriptionsKeys, McpEvents, Context as ContextType } from "./internal/internal.js";
+ * @import { Tool, Completion, Prompt, StoredResource, ServerOptions, ServerInfo, SubscriptionsKeys, McpEvents } from "./internal/internal.js";
  */
 import { JSONRPCClient, JSONRPCServer } from 'json-rpc-2.0';
 import { AsyncLocalStorage } from 'node:async_hooks';
@@ -92,7 +92,7 @@ export class McpServer {
 	#client;
 	#options;
 	/**
-	 * @type {Map<string, Tool<any>>}
+	 * @type {Map<string, Tool<any, any>>}
 	 */
 	#tools = new Map();
 	/**
@@ -328,12 +328,25 @@ export class McpServer {
 						name,
 						title: tool.title || tool.description,
 						description: tool.description,
-						annotations: tool.annotations,
+
 						inputSchema: tool.schema
 							? await this.#options.adapter.toJsonSchema(
 									tool.schema,
 								)
 							: { type: 'object', properties: {} },
+						...(tool.outputSchema
+							? {
+									outputSchema:
+										await this.#options.adapter.toJsonSchema(
+											tool.outputSchema,
+										),
+								}
+							: {}),
+						...(tool.annotations
+							? {
+									annotations: tool.annotations,
+								}
+							: {}),
 					};
 				}),
 			);
@@ -348,22 +361,57 @@ export class McpServer {
 				if (!tool) {
 					throw new McpError(-32601, `Tool ${name} not found`);
 				}
-				if (!tool.schema) {
-					return v.parse(CallToolResultSchema, await tool.execute());
-				}
-				let validated_args = tool.schema['~standard'].validate(args);
-				if (validated_args instanceof Promise)
-					validated_args = await validated_args;
-				if (validated_args.issues) {
-					throw new McpError(
-						-32602,
-						`Invalid arguments for tool ${name}: ${JSON.stringify(validated_args.issues)}`,
-					);
+
+				// Validate input arguments if schema is provided
+				let validated_args = args;
+				if (tool.schema) {
+					let validation_result =
+						tool.schema['~standard'].validate(args);
+					if (validation_result instanceof Promise)
+						validation_result = await validation_result;
+					if (validation_result.issues) {
+						throw new McpError(
+							-32602,
+							`Invalid arguments for tool ${name}: ${JSON.stringify(validation_result.issues)}`,
+						);
+					}
+					validated_args = validation_result.value;
 				}
-				return v.parse(
+
+				// Execute the tool
+				const tool_result = tool.schema
+					? await tool.execute(validated_args)
+					: await tool.execute();
+
+				console.log({ tool_result });
+
+				// Parse the basic result structure
+				const parsed_result = v.parse(
 					CallToolResultSchema,
-					await tool.execute(validated_args.value),
+					tool_result,
 				);
+
+				// If tool has outputSchema, validate and populate structuredContent
+				if (
+					tool.outputSchema &&
+					parsed_result.structuredContent !== undefined
+				) {
+					let output_validation = tool.outputSchema[
+						'~standard'
+					].validate(parsed_result.structuredContent);
+					if (output_validation instanceof Promise)
+						output_validation = await output_validation;
+					if (output_validation.issues) {
+						throw new McpError(
+							-32603,
+							`Tool ${name} returned invalid structured content: ${JSON.stringify(output_validation.issues)}`,
+						);
+					}
+					// Update with validated structured content
+					parsed_result.structuredContent = output_validation.value;
+				}
+
+				return parsed_result;
 			},
 		);
 	}
@@ -600,7 +648,7 @@ export class McpServer {
 	#init_completion() {
 		this.#server.addMethod(
 			'completion/complete',
-			({ argument, ref, context }) => {
+			async ({ argument, ref, context }) => {
 				const completions = this.#completions[ref.type];
 				if (!completions) return null;
 				const complete = completions.get(ref.uri ?? ref.name);
@@ -609,7 +657,7 @@ export class McpServer {
 				if (!actual_complete) return null;
 				return v.parse(
 					CompleteResultSchema,
-					actual_complete(argument.value, context),
+					await actual_complete(argument.value, context),
 				);
 			},
 		);
@@ -627,10 +675,14 @@ export class McpServer {
 	}
 	/**
 	 * @template {StandardSchema | undefined} [TSchema=undefined]
-	 * @param {{ name: string; description: string; title?: string; schema?: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema> extends Record<string, unknown> ? TSchema : never; annotations?: ToolAnnotations }} options
-	 * @param {TSchema extends undefined ? (()=>Promise<CallToolResult> | CallToolResult) : ((input: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema>) => Promise<CallToolResult> | CallToolResult)} execute
+	 * @template {StandardSchema | undefined} [TOutputSchema=undefined]
+	 * @param {{ name: string; description: string; title?: string; schema?: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema> extends Record<string, unknown> ? TSchema : never; outputSchema?: StandardSchemaV1.InferOutput<TOutputSchema extends undefined ? never : TOutputSchema> extends Record<string, unknown> ? TOutputSchema : never; annotations?: ToolAnnotations }} options
+	 * @param {TSchema extends undefined ? (()=>Promise<CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>> | CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>) : ((input: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema>) => Promise<CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>> | CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>)} execute
 	 */
-	tool({ name, description, title, schema, annotations }, execute) {
+	tool(
+		{ name, description, title, schema, outputSchema, annotations },
+		execute,
+	) {
 		if (this.#options.capabilities?.tools?.listChanged) {
 			this.#notify('notifications/tools/list_changed', {});
 		}
@@ -638,6 +690,7 @@ export class McpServer {
 			description,
 			title,
 			schema,
+			outputSchema,
 			execute,
 			annotations,
 		});

package/src/internal/internal.d.ts

@@ -13,9 +13,10 @@ import {
 	ToolAnnotations
 } from '../validation/index.js';
 
-export type Tool<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
+export type Tool<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>, TOutputSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
 	description: string;
 	schema: TSchema;
+	outputSchema?: TOutputSchema;
 	title?: string;
 	annotations?: ToolAnnotations;
 	execute: (
@@ -26,7 +27,7 @@ export type Tool<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
 export type Completion = (
 	query: string,
 	context: { arguments: Record<string, string> },
-) => CompleteResult;
+) => CompleteResult | Promise<CompleteResult>;
 
 export type Prompt<TSchema extends StandardSchemaV1 = StandardSchemaV1<any>> = {
 	description: string;

package/src/types/index.d.ts

@@ -23,13 +23,14 @@ declare module 'tmcp' {
 		
 		on<TEvent extends keyof McpEvents>(event: TEvent, callback: McpEvents[TEvent], options?: AddEventListenerOptions): () => void;
 		
-		tool<TSchema extends StandardSchema | undefined = undefined>({ name, description, title, schema, annotations }: {
+		tool<TSchema extends StandardSchema | undefined = undefined, TOutputSchema extends StandardSchema | undefined = undefined>({ name, description, title, schema, outputSchema, annotations }: {
 			name: string;
 			description: string;
 			title?: string;
 			schema?: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema> extends Record<string, unknown> ? TSchema : never;
+			outputSchema?: StandardSchemaV1.InferOutput<TOutputSchema extends undefined ? never : TOutputSchema> extends Record<string, unknown> ? TOutputSchema : never;
 			annotations?: ToolAnnotations;
-		}, execute: TSchema extends undefined ? (() => Promise<CallToolResult> | CallToolResult) : ((input: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema>) => Promise<CallToolResult> | CallToolResult)): void;
+		}, execute: TSchema extends undefined ? (() => Promise<CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>> | CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>) : ((input: StandardSchemaV1.InferInput<TSchema extends undefined ? never : TSchema>) => Promise<CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>> | CallToolResult<TOutputSchema extends undefined ? undefined : StandardSchemaV1.InferInput<TOutputSchema extends undefined ? never : TOutputSchema>>)): void;
 		
 		prompt<TSchema extends StandardSchema | undefined = undefined>({ name, description, title, schema, complete }: {
 			name: string;
@@ -116,7 +117,7 @@ declare module 'tmcp' {
 	type Completion = (
 		query: string,
 		context: { arguments: Record<string, string> },
-	) => CompleteResult;
+	) => CompleteResult | Promise<CompleteResult>;
 
 	type ServerOptions<TSchema extends StandardSchemaV1> = {
 		capabilities?: ServerCapabilities;
@@ -895,7 +896,11 @@ declare module 'tmcp' {
 	type ClientCapabilities_1 = v.InferInput<typeof ClientCapabilitiesSchema>;
 	type ServerCapabilities = v.InferInput<typeof ServerCapabilitiesSchema>;
 	type InitializeRequestParams = v.InferInput<typeof InitializeRequestParamsSchema>;
-	type CallToolResult = v.InferInput<typeof CallToolResultSchema>;
+	type CallToolResult<TStructuredContent extends Record<string, unknown> | undefined> = Omit<v.InferInput<typeof CallToolResultSchema>, "structuredContent"> & (undefined extends TStructuredContent ? {
+		structuredContent?: undefined;
+	} : {
+		structuredContent: TStructuredContent;
+	});
 	type ReadResourceResult = v.InferInput<typeof ReadResourceResultSchema>;
 	type GetPromptResult = v.InferInput<typeof GetPromptResultSchema>;
 	type CompleteResult = v.InferInput<typeof CompleteResultSchema>;

package/src/types/index.d.ts.map

@@ -24,7 +24,6 @@
 		"CompleteResultSchema",
 		"ServerCapabilities",
 		"InitializeRequestParams",
-		"CallToolResult",
 		"ReadResourceResult",
 		"GetPromptResult",
 		"CompleteResult",
@@ -56,6 +55,6 @@
 		null,
 		null
 	],
-	"mappings": ";;;;;cAsFaA,SAASA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aCi3CsCC,kBAAkBA;MC96ClEC,UAAUA;;;;;MAsCVC,aAAaA;;;;;;;;;;;;;MAabC,UAAUA;;;;;;MAMVC,iBAAiBA;;MAEjBC,SAASA;;;;;;;;;;;;ODgBRC,qBAAqBA;;;;;;;;;;;;;;OAuGrBC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;OA6BxBC,6BAA6BA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAuB7BC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmMxBC,cAAcA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4GdC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8PxBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BrBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2GrBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoEpBC,kBAAkBA;OAuGlBC,gCAAgCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8ChCC,yBAAyBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+LzBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;MA8H2BC,kBAAkBA;MASbC,uBAAuBA;MAGhCC,cAAcA;MAGVC,kBAAkBA;MAGrBC,eAAeA;MAGhBC,cAAcA;MAGFC,0BAA0BA;MAGjCC,mBAAmBA;MAY9BC,QAAQA;MAMDC,eAAeA;MAGlBC,YAAYA;MAGTC,eAAeA;;ME5/CnEC,IAAIA;;;;;;;MAOJC,cAAcA;;;;;;;MAOdC,cAAcA;;;;;MAKdC,qBAAqBA;;;;;;;;;;;;;;;;;MAiBrBC,4BAA4BA;;;;;;MAMrBC,2BAA2BA;;;;;;;OCjC1BC,iBAAiBA;;;;;;;;;;;;;;;;cAAjBA,iBAAiBA",
+	"mappings": ";;;;;cAsFaA,SAASA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;aCi3CsCC,kBAAkBA;MC76ClEC,UAAUA;;;;;MAsCVC,aAAaA;;;;;;;;;;;;;MAabC,UAAUA;;;;;;MAMVC,iBAAiBA;;MAEjBC,SAASA;;;;;;;;;;;;ODeRC,qBAAqBA;;;;;;;;;;;;;;OAuGrBC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;OA6BxBC,6BAA6BA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAuB7BC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAmMxBC,cAAcA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4GdC,wBAAwBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8PxBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA6BrBC,qBAAqBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA2GrBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAoEpBC,kBAAkBA;OAuGlBC,gCAAgCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA8ChCC,yBAAyBA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA+LzBC,oBAAoBA;;;;;;;;;;;;;;;;;;;;;;MA8H2BC,kBAAkBA;MASbC,uBAAuBA;;;;;;MAO5BC,kBAAkBA;MAGrBC,eAAeA;MAGhBC,cAAcA;MAGFC,0BAA0BA;MAGjCC,mBAAmBA;MAY9BC,QAAQA;MAMDC,eAAeA;MAGlBC,YAAYA;MAGTC,eAAeA;;ME7/CnEC,IAAIA;;;;;;;MAOJC,cAAcA;;;;;;;MAOdC,cAAcA;;;;;MAKdC,qBAAqBA;;;;;;;;;;;;;;;;;MAiBrBC,4BAA4BA;;;;;;MAMrBC,2BAA2BA;;;;;;;OCjC1BC,iBAAiBA;;;;;;;;;;;;;;;;cAAjBA,iBAAiBA",
 	"ignoreList": []
 }

package/src/validation/index.js

@@ -1492,7 +1492,8 @@ export const ServerResultSchema = v.union([
  * @typedef {v.InferInput<typeof InitializeRequestParamsSchema>} InitializeRequestParams
  */
 /**
- * @typedef {v.InferInput<typeof CallToolResultSchema>} CallToolResult
+ * @template {Record<string, unknown> | undefined} TStructuredContent
+ * @typedef {Omit<v.InferInput<typeof CallToolResultSchema>, "structuredContent"> & (undefined extends TStructuredContent ? { structuredContent?: undefined } : { structuredContent: TStructuredContent })} CallToolResult
  */
 /**
  * @typedef {v.InferInput<typeof ReadResourceResultSchema>} ReadResourceResult