tmcp 1.9.0 → 1.9.1

package/package.json

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

package/src/index.js

@@ -600,7 +600,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 +609,7 @@ export class McpServer {
 				if (!actual_complete) return null;
 				return v.parse(
 					CompleteResultSchema,
-					actual_complete(argument.value, context),
+					await actual_complete(argument.value, context),
 				);
 			},
 		);

package/src/internal/internal.d.ts

@@ -26,7 +26,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

@@ -116,7 +116,7 @@ declare module 'tmcp' {
 	type Completion = (
 		query: string,
 		context: { arguments: Record<string, string> },
-	) => CompleteResult;
+	) => CompleteResult | Promise<CompleteResult>;
 
 	type ServerOptions<TSchema extends StandardSchemaV1> = {
 		capabilities?: ServerCapabilities;