@revojs/ai 0.2.20 → 0.2.21

package/dist/index.d.ts

@@ -50,6 +50,9 @@ export interface ToolDefinition {
 export interface ToolExecutionOptions {
     signal?: AbortSignal;
 }
+export interface ModelContextProtocolServerOptions {
+    filter?: (tool: ToolDescription) => boolean;
+}
 export interface ToolDescription extends LanguageModelFunctionTool {
     inputSchema: JSONObject;
     annotations?: ToolAnnotations;
@@ -683,7 +686,7 @@ export declare function useTools(scope: Scope): {
     callTool: (name: string, input?: JSONObject, options?: ToolExecutionOptions) => Promise<CallToolResult>;
     executeTool(call: LanguageModelToolCall, options?: ToolExecutionOptions): Promise<LanguageModelToolResultPart>;
 };
-export declare function handleModelContextProtocol(scope: Scope): Promise<Response>;
+export declare function handleModelContextProtocol(scope: Scope, options?: ModelContextProtocolServerOptions): Promise<Response>;
 export declare function useModelContextProtocol(options: ModelContextProtocolConnectOptions): Promise<ModelContextProtocolConnection>;
 export declare function defineProvider<TProvider extends Provider>(provider: TProvider): TProvider;
 export declare function useAi<TSource extends Source<Provider> = typeof import("#providers").default>(scope: Scope): AiConnections<TSource>;

package/dist/index.js

@@ -206,7 +206,7 @@ function useTools(scope) {
 		}
 	};
 }
-async function handleModelContextProtocol(scope) {
+async function handleModelContextProtocol(scope, options = {}) {
 	requireSameOrigin(scope);
 	const { request } = useServer(scope);
 	if (request.method !== "POST") return new Response(null, {
@@ -223,13 +223,18 @@ async function handleModelContextProtocol(scope) {
 		jsonSchemaValidator: new CfWorkerJsonSchemaValidator()
 	});
 	const transport = new WebStandardStreamableHTTPServerTransport({ enableJsonResponse: true });
-	server.server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: (await collection.listTools()).map((tool) => ({
+	const listTools = async () => {
+		const tools = await collection.listTools();
+		return options.filter ? tools.filter(options.filter) : tools;
+	};
+	server.server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: (await listTools()).map((tool) => ({
 		name: tool.name,
 		description: tool.description,
 		inputSchema: tool.inputSchema,
 		annotations: tool.annotations
 	})) }));
 	server.server.setRequestHandler(CallToolRequestSchema, async ({ params }, extra) => {
+		if (options.filter && !(await listTools()).some((tool) => tool.name === params.name)) throw new McpError(ErrorCode.InvalidParams, `Unknown tool "${params.name}".`);
 		try {
 			return await collection.callTool(params.name, params.arguments ?? {}, { signal: AbortSignal.any([request.signal, extra.signal]) });
 		} catch (error) {

package/dist/routes/v1/model-context-protocol/delete.d.ts

@@ -1,5 +1,4 @@
-import { handleModelContextProtocol } from "../../..";
 declare const _default: {
-    fetch: typeof handleModelContextProtocol;
+    fetch: (scope: import("revojs").Scope) => Promise<Response>;
 };
 export default _default;

package/dist/routes/v1/model-context-protocol/delete.js

@@ -1,6 +1,6 @@
 import { handleModelContextProtocol } from "../../../index.js";
 import { defineRoute } from "@revojs/server";
 //#region src/routes/v1/model-context-protocol/delete.ts
-var delete_default = defineRoute({ fetch: handleModelContextProtocol });
+var delete_default = defineRoute({ fetch: (scope) => handleModelContextProtocol(scope) });
 //#endregion
 export { delete_default as default };

package/dist/routes/v1/model-context-protocol/get.d.ts

@@ -1,5 +1,4 @@
-import { handleModelContextProtocol } from "../../..";
 declare const _default: {
-    fetch: typeof handleModelContextProtocol;
+    fetch: (scope: import("revojs").Scope) => Promise<Response>;
 };
 export default _default;

package/dist/routes/v1/model-context-protocol/get.js

@@ -1,6 +1,6 @@
 import { handleModelContextProtocol } from "../../../index.js";
 import { defineRoute } from "@revojs/server";
 //#region src/routes/v1/model-context-protocol/get.ts
-var get_default = defineRoute({ fetch: handleModelContextProtocol });
+var get_default = defineRoute({ fetch: (scope) => handleModelContextProtocol(scope) });
 //#endregion
 export { get_default as default };

package/dist/routes/v1/model-context-protocol/post.d.ts

@@ -1,5 +1,4 @@
-import { handleModelContextProtocol } from "../../..";
 declare const _default: {
-    fetch: typeof handleModelContextProtocol;
+    fetch: (scope: import("revojs").Scope) => Promise<Response>;
 };
 export default _default;

package/dist/routes/v1/model-context-protocol/post.js

@@ -1,6 +1,6 @@
 import { handleModelContextProtocol } from "../../../index.js";
 import { defineRoute } from "@revojs/server";
 //#region src/routes/v1/model-context-protocol/post.ts
-var post_default = defineRoute({ fetch: handleModelContextProtocol });
+var post_default = defineRoute({ fetch: (scope) => handleModelContextProtocol(scope) });
 //#endregion
 export { post_default as default };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@revojs/ai",
-    "version": "0.2.20",
+    "version": "0.2.21",
     "license": "MIT",
     "repository": {
         "type": "git",