@frontmcp/sdk 0.2.5 → 0.3.1

package/src/tool/flows/call-tool.flow.js

@@ -0,0 +1,249 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+// tools/flows/call-tool.flow.ts
+const common_1 = require("../../common");
+const zod_1 = require("zod");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
+const inputSchema = zod_1.z.object({
+    request: types_js_1.CallToolRequestSchema,
+    ctx: zod_1.z.any(),
+});
+const outputSchema = types_js_1.CallToolResultSchema;
+const stateSchema = zod_1.z.object({
+    input: zod_1.z.object({
+        name: zod_1.z.string().min(1).max(64),
+        arguments: zod_1.z.object({}).passthrough().optional(),
+    }).passthrough(),
+    authInfo: zod_1.z.any().optional(),
+    tool: zod_1.z.instanceof(common_1.ToolEntry),
+    toolContext: zod_1.z.instanceof(common_1.ToolContext),
+});
+const plan = {
+    pre: [
+        'parseInput',
+        'findTool',
+        'createToolCallContext',
+        'acquireQuota',
+        'acquireSemaphore',
+    ],
+    execute: [
+        'validateInput',
+        'execute',
+        'validateOutput',
+    ],
+    finalize: [
+        'releaseSemaphore',
+        'releaseQuota',
+        'finalize',
+    ],
+};
+const name = 'tools:call-tool';
+const { Stage } = (0, common_1.FlowHooksOf)(name);
+let CallToolFlow = class CallToolFlow extends common_1.FlowBase {
+    logger = this.scopeLogger.child('CallToolFlow');
+    async parseInput() {
+        this.logger.verbose('parseInput:start');
+        const { request: { method, params }, ctx } = inputSchema.parse(this.rawInput);
+        if (method !== 'tools/call') {
+            this.logger.warn(`parseInput: invalid method "${method}"`);
+            throw new Error('Invalid method');
+        }
+        this.state.set({ input: params, authInfo: ctx.authInfo });
+        this.logger.verbose('parseInput:done');
+    }
+    async findTool() {
+        this.logger.verbose('findTool:start');
+        // TODO: add support for session based tools
+        const activeTools = this.scope.tools.getTools(true);
+        this.logger.info(`findTool: discovered ${activeTools.length} active tool(s) (including hidden)`);
+        const { name } = this.state.required.input;
+        const tool = activeTools.find(t => t.metadata.name === name);
+        if (!tool) {
+            const errorMessage = `Tool "${name}" not found`;
+            this.logger.warn(errorMessage);
+            this.fail(new Error(errorMessage));
+        }
+        this.logger = this.logger.child(`CallToolFlow(${name})`);
+        this.state.set('tool', tool);
+        this.logger.info(`findTool: tool "${name}" found`);
+        this.logger.verbose('findTool:done');
+    }
+    async createToolCallContext() {
+        this.logger.verbose('createToolCallContext:start');
+        const { ctx } = this.input;
+        const { tool, input } = this.state.required;
+        const context = tool.create(input, ctx);
+        const toolHooks = this.scope.hooks.getClsHooks(tool.record.provide)
+            .map(hook => {
+            hook.run = async () => {
+                return context[hook.metadata.method]();
+            };
+            return hook;
+        });
+        this.appendContextHooks(toolHooks);
+        context.input = input;
+        context.mark('createToolCallContext');
+        this.state.set('toolContext', context);
+        this.logger.verbose('createToolCallContext:done');
+    }
+    async acquireQuota() {
+        this.logger.verbose('acquireQuota:start');
+        // used for rate limiting
+        this.state.toolContext?.mark('acquireQuota');
+        this.logger.verbose('acquireQuota:done');
+    }
+    async acquireSemaphore() {
+        this.logger.verbose('acquireSemaphore:start');
+        // used for concurrency control
+        this.state.toolContext?.mark('acquireSemaphore');
+        this.logger.verbose('acquireSemaphore:done');
+    }
+    async validateInput() {
+        this.logger.verbose('validateInput:start');
+        const { tool, input } = this.state.required;
+        const { toolContext } = this.state;
+        if (!toolContext) {
+            return;
+        }
+        toolContext.mark('validateInput');
+        try {
+            toolContext.input = tool.inputSchema.parse(input.arguments ?? {});
+        }
+        catch (err) {
+            this.fail(new Error(`Invalid input: validation failed`));
+        }
+        this.logger.verbose('validateInput:done');
+    }
+    async execute() {
+        this.logger.verbose('execute:start');
+        const toolContext = this.state.toolContext;
+        if (!toolContext) {
+            return;
+        }
+        toolContext.mark('execute');
+        toolContext.output = await toolContext.execute(toolContext.input);
+        this.logger.verbose('execute:done');
+    }
+    async validateOutput() {
+        this.logger.verbose('validateOutput:start');
+        const { tool, toolContext } = this.state;
+        if (!toolContext || !tool) {
+            return;
+        }
+        toolContext.mark('validateOutput');
+        toolContext.output = tool.outputSchema.parse(toolContext.output);
+        this.logger.verbose('validateOutput:done');
+    }
+    async releaseSemaphore() {
+        this.logger.verbose('releaseSemaphore:start');
+        // release concurrency control
+        this.state.toolContext?.mark('releaseSemaphore');
+        this.logger.verbose('releaseSemaphore:done');
+    }
+    async releaseQuota() {
+        this.logger.verbose('releaseQuota:start');
+        // release rate limiting
+        this.state.toolContext?.mark('releaseQuota');
+        this.logger.verbose('releaseQuota:done');
+    }
+    async finalize() {
+        this.logger.verbose('finalize:start');
+        if (!this.state.toolContext) {
+            this.fail(new Error("error"));
+            return;
+        }
+        const { tool, toolContext } = this.state.required;
+        const success = tool.outputSchema.safeParse(toolContext.output).success;
+        if (success) {
+            const response = toolContext.output;
+            this.respond({
+                content: [{
+                        type: 'text',
+                        text: JSON.stringify(response)
+                    }],
+            });
+        }
+        else {
+            this.fail(new Error("invalid output schema"));
+        }
+        this.logger.verbose('finalize:done');
+    }
+};
+tslib_1.__decorate([
+    Stage('parseInput'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "parseInput", null);
+tslib_1.__decorate([
+    Stage('findTool'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "findTool", null);
+tslib_1.__decorate([
+    Stage('createToolCallContext'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "createToolCallContext", null);
+tslib_1.__decorate([
+    Stage('acquireQuota'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "acquireQuota", null);
+tslib_1.__decorate([
+    Stage('acquireSemaphore'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "acquireSemaphore", null);
+tslib_1.__decorate([
+    Stage('validateInput'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "validateInput", null);
+tslib_1.__decorate([
+    Stage('execute'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "execute", null);
+tslib_1.__decorate([
+    Stage('validateOutput'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "validateOutput", null);
+tslib_1.__decorate([
+    Stage('releaseSemaphore'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "releaseSemaphore", null);
+tslib_1.__decorate([
+    Stage('releaseQuota'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "releaseQuota", null);
+tslib_1.__decorate([
+    Stage('finalize'),
+    tslib_1.__metadata("design:type", Function),
+    tslib_1.__metadata("design:paramtypes", []),
+    tslib_1.__metadata("design:returntype", Promise)
+], CallToolFlow.prototype, "finalize", null);
+CallToolFlow = tslib_1.__decorate([
+    (0, common_1.Flow)({
+        name,
+        plan,
+        inputSchema,
+        outputSchema,
+        access: 'authorized',
+    })
+], CallToolFlow);
+exports.default = CallToolFlow;
+//# sourceMappingURL=call-tool.flow.js.map

package/src/tool/flows/call-tool.flow.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"call-tool.flow.js","sourceRoot":"","sources":["../../../../src/tool/flows/call-tool.flow.ts"],"names":[],"mappings":";;;AAAA,gCAAgC;AAChC,yCAA2G;AAC3G,6BAAsB;AACtB,iEAA+F;AAG/F,MAAM,WAAW,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3B,OAAO,EAAE,gCAAqB;IAC9B,GAAG,EAAE,OAAC,CAAC,GAAG,EAAE;CACb,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,+BAAoB,CAAC;AAE1C,MAAM,WAAW,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3B,KAAK,EAAE,OAAC,CAAC,MAAM,CAAC;QACd,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/B,SAAS,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE;KACjD,CAAC,CAAC,WAAW,EAAE;IAChB,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAyB;IACnD,IAAI,EAAE,OAAC,CAAC,UAAU,CAAC,kBAAS,CAAC;IAC7B,WAAW,EAAE,OAAC,CAAC,UAAU,CAAC,oBAAW,CAAC;CACvC,CAAC,CAAC;AAEH,MAAM,IAAI,GAAG;IACX,GAAG,EAAE;QACH,YAAY;QACZ,UAAU;QACV,uBAAuB;QACvB,cAAc;QACd,kBAAkB;KACnB;IACD,OAAO,EAAE;QACP,eAAe;QACf,SAAS;QACT,gBAAgB;KACjB;IACD,QAAQ,EAAE;QACR,kBAAkB;QAClB,cAAc;QACd,UAAU;KACX;CACkC,CAAC;AActC,MAAM,IAAI,GAAG,iBAA0B,CAAC;AACxC,MAAM,EAAC,KAAK,EAAC,GAAG,IAAA,oBAAW,EAAoB,IAAI,CAAC,CAAC;AAStC,IAAM,YAAY,GAAlB,MAAM,YAAa,SAAQ,iBAAqB;IAC7D,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IAG1C,AAAN,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACxC,MAAM,EAAC,OAAO,EAAE,EAAC,MAAM,EAAE,MAAM,EAAC,EAAE,GAAG,EAAC,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE1E,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;YAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,MAAM,GAAG,CAAC,CAAC;YAC3D,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAC,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IACzC,CAAC;IAIK,AAAN,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACtC,4CAA4C;QAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;QACnD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wBAAwB,WAAW,CAAC,MAAM,oCAAoC,CAAC,CAAC;QAEjG,MAAM,EAAC,IAAI,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;QACzC,MAAM,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC7D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,YAAY,GAAG,SAAS,IAAI,aAAa,CAAA;YAC/C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAA;QACpC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,IAAI,GAAG,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,IAAK,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mBAAmB,IAAI,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACvC,CAAC;IAGK,AAAN,KAAK,CAAC,qBAAqB;QACzB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,6BAA6B,CAAC,CAAC;QACnD,MAAM,EAAC,GAAG,EAAC,GAAG,IAAI,CAAC,KAAK,CAAA;QACxB,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;aAChE,GAAG,CAAC,IAAI,CAAC,EAAE;YACV,IAAI,CAAC,GAAG,GAAG,KAAK,IAAI,EAAE;gBACpB,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAA;YACxC,CAAC,CAAA;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAA;QAEJ,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAA;QAClC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC;QACtB,OAAO,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAA;QACrC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAA;QACtC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;IACpD,CAAC;IAGK,AAAN,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC1C,yBAAyB;QACzB,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAA;QAC5C,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC3C,CAAC;IAGK,AAAN,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;QAC9C,+BAA+B;QAC/B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAChD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAC/C,CAAC;IAIK,AAAN,KAAK,CAAC,aAAa;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAC3C,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAC1C,MAAM,EAAC,WAAW,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC;QACjC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,WAAW,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACjC,IAAI,CAAC;YACH,WAAW,CAAC,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;QACpE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAC5C,CAAC;IAGK,AAAN,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;QAC3C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC3B,WAAW,CAAC,MAAM,GAAG,MAAM,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;QACjE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC;IAGK,AAAN,KAAK,CAAC,cAAc;QAClB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAC5C,MAAM,EAAC,IAAI,EAAE,WAAW,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC;QACvC,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QACD,WAAW,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAClC,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;QAChE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC7C,CAAC;IAGK,AAAN,KAAK,CAAC,gBAAgB;QACpB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;QAC9C,8BAA8B;QAC9B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAA;QAChD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;IAC/C,CAAC;IAGK,AAAN,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QAC1C,wBAAwB;QACxB,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,CAAA;QAC5C,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC3C,CAAC;IAGK,AAAN,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAA;YAC7B,OAAO;QACT,CAAC;QACD,MAAM,EAAC,IAAI,EAAE,WAAW,EAAC,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;QACxE,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC;YACpC,IAAI,CAAC,OAAO,CAAC;gBACX,OAAO,EAAE,CAAC;wBACR,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC;qBAC/B,CAAC;aACH,CAAC,CAAA;QAEJ,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAA;QAC/C,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACvC,CAAC;CACF,CAAA;AAxJO;IADL,KAAK,CAAC,YAAY,CAAC;;;;8CAYnB;AAIK;IADL,KAAK,CAAC,UAAU,CAAC;;;;4CAkBjB;AAGK;IADL,KAAK,CAAC,uBAAuB,CAAC;;;;yDAmB9B;AAGK;IADL,KAAK,CAAC,cAAc,CAAC;;;;gDAMrB;AAGK;IADL,KAAK,CAAC,kBAAkB,CAAC;;;;oDAMzB;AAIK;IADL,KAAK,CAAC,eAAe,CAAC;;;;iDAetB;AAGK;IADL,KAAK,CAAC,SAAS,CAAC;;;;2CAUhB;AAGK;IADL,KAAK,CAAC,gBAAgB,CAAC;;;;kDAUvB;AAGK;IADL,KAAK,CAAC,kBAAkB,CAAC;;;;oDAMzB;AAGK;IADL,KAAK,CAAC,cAAc,CAAC;;;;gDAMrB;AAGK;IADL,KAAK,CAAC,UAAU,CAAC;;;;4CAsBjB;AA3JkB,YAAY;IAPhC,IAAA,aAAI,EAAC;QACJ,IAAI;QACJ,IAAI;QACJ,WAAW;QACX,YAAY;QACZ,MAAM,EAAE,YAAY;KACrB,CAAC;GACmB,YAAY,CA4JhC;kBA5JoB,YAAY","sourcesContent":["// tools/flows/call-tool.flow.ts\nimport {Flow, FlowBase, FlowHooksOf, FlowPlan, FlowRunOptions, ToolContext, ToolEntry} from '../../common';\nimport {z} from 'zod';\nimport {CallToolRequestSchema, CallToolResultSchema} from '@modelcontextprotocol/sdk/types.js';\nimport {AuthInfo} from \"@modelcontextprotocol/sdk/server/auth/types.js\";\n\nconst inputSchema = z.object({\n  request: CallToolRequestSchema,\n  ctx: z.any(),\n});\n\nconst outputSchema = CallToolResultSchema;\n\nconst stateSchema = z.object({\n  input: z.object({\n    name: z.string().min(1).max(64),\n    arguments: z.object({}).passthrough().optional(),\n  }).passthrough(),\n  authInfo: z.any().optional() as z.ZodType<AuthInfo>,\n  tool: z.instanceof(ToolEntry),\n  toolContext: z.instanceof(ToolContext),\n});\n\nconst plan = {\n  pre: [\n    'parseInput',\n    'findTool',\n    'createToolCallContext',\n    'acquireQuota',\n    'acquireSemaphore',\n  ],\n  execute: [\n    'validateInput',\n    'execute',\n    'validateOutput',\n  ],\n  finalize: [\n    'releaseSemaphore',\n    'releaseQuota',\n    'finalize',\n  ],\n} as const satisfies FlowPlan<string>;\n\ndeclare global {\n  interface ExtendFlows {\n    'tools:call-tool': FlowRunOptions<\n      CallToolFlow,\n      typeof plan,\n      typeof inputSchema,\n      typeof outputSchema,\n      typeof stateSchema\n    >;\n  }\n}\n\nconst name = 'tools:call-tool' as const;\nconst {Stage} = FlowHooksOf<'tools:call-tool'>(name);\n\n@Flow({\n  name,\n  plan,\n  inputSchema,\n  outputSchema,\n  access: 'authorized',\n})\nexport default class CallToolFlow extends FlowBase<typeof name> {\n  logger = this.scopeLogger.child('CallToolFlow');\n\n  @Stage('parseInput')\n  async parseInput() {\n    this.logger.verbose('parseInput:start');\n    const {request: {method, params}, ctx} = inputSchema.parse(this.rawInput);\n\n    if (method !== 'tools/call') {\n      this.logger.warn(`parseInput: invalid method \"${method}\"`);\n      throw new Error('Invalid method');\n    }\n\n    this.state.set({input: params, authInfo: ctx.authInfo});\n    this.logger.verbose('parseInput:done');\n  }\n\n\n  @Stage('findTool')\n  async findTool() {\n    this.logger.verbose('findTool:start');\n    // TODO: add support for session based tools\n    const activeTools = this.scope.tools.getTools(true)\n    this.logger.info(`findTool: discovered ${activeTools.length} active tool(s) (including hidden)`);\n\n    const {name} = this.state.required.input;\n    const tool = activeTools.find(t => t.metadata.name === name);\n    if (!tool) {\n      const errorMessage = `Tool \"${name}\" not found`\n      this.logger.warn(errorMessage);\n      this.fail(new Error(errorMessage))\n    }\n    this.logger = this.logger.child(`CallToolFlow(${name})`);\n    this.state.set('tool', tool!);\n    this.logger.info(`findTool: tool \"${name}\" found`);\n    this.logger.verbose('findTool:done');\n  }\n\n  @Stage('createToolCallContext')\n  async createToolCallContext() {\n    this.logger.verbose('createToolCallContext:start');\n    const {ctx} = this.input\n    const {tool, input} = this.state.required;\n    const context = tool.create(input, ctx)\n    const toolHooks = this.scope.hooks.getClsHooks(tool.record.provide)\n      .map(hook => {\n        hook.run = async () => {\n          return context[hook.metadata.method]()\n        }\n        return hook;\n      })\n\n    this.appendContextHooks(toolHooks)\n    context.input = input;\n    context.mark('createToolCallContext')\n    this.state.set('toolContext', context)\n    this.logger.verbose('createToolCallContext:done');\n  }\n\n  @Stage('acquireQuota')\n  async acquireQuota() {\n    this.logger.verbose('acquireQuota:start');\n    // used for rate limiting\n    this.state.toolContext?.mark('acquireQuota')\n    this.logger.verbose('acquireQuota:done');\n  }\n\n  @Stage('acquireSemaphore')\n  async acquireSemaphore() {\n    this.logger.verbose('acquireSemaphore:start');\n    // used for concurrency control\n    this.state.toolContext?.mark('acquireSemaphore')\n    this.logger.verbose('acquireSemaphore:done');\n  }\n\n\n  @Stage('validateInput')\n  async validateInput() {\n    this.logger.verbose('validateInput:start');\n    const {tool, input} = this.state.required;\n    const {toolContext} = this.state;\n    if (!toolContext) {\n      return;\n    }\n    toolContext.mark('validateInput')\n    try {\n      toolContext.input = tool.inputSchema.parse(input.arguments ?? {});\n    } catch (err) {\n      this.fail(new Error(`Invalid input: validation failed`));\n    }\n    this.logger.verbose('validateInput:done');\n  }\n\n  @Stage('execute')\n  async execute() {\n    this.logger.verbose('execute:start');\n    const toolContext = this.state.toolContext;\n    if (!toolContext) {\n      return;\n    }\n    toolContext.mark('execute')\n    toolContext.output = await toolContext.execute(toolContext.input)\n    this.logger.verbose('execute:done');\n  }\n\n  @Stage('validateOutput')\n  async validateOutput() {\n    this.logger.verbose('validateOutput:start');\n    const {tool, toolContext} = this.state;\n    if (!toolContext || !tool) {\n      return;\n    }\n    toolContext.mark('validateOutput')\n    toolContext.output = tool.outputSchema.parse(toolContext.output)\n    this.logger.verbose('validateOutput:done');\n  }\n\n  @Stage('releaseSemaphore')\n  async releaseSemaphore() {\n    this.logger.verbose('releaseSemaphore:start');\n    // release concurrency control\n    this.state.toolContext?.mark('releaseSemaphore')\n    this.logger.verbose('releaseSemaphore:done');\n  }\n\n  @Stage('releaseQuota')\n  async releaseQuota() {\n    this.logger.verbose('releaseQuota:start');\n    // release rate limiting\n    this.state.toolContext?.mark('releaseQuota')\n    this.logger.verbose('releaseQuota:done');\n  }\n\n  @Stage('finalize')\n  async finalize() {\n    this.logger.verbose('finalize:start');\n    if (!this.state.toolContext) {\n      this.fail(new Error(\"error\"))\n      return;\n    }\n    const {tool, toolContext} = this.state.required;\n    const success = tool.outputSchema.safeParse(toolContext.output).success;\n    if (success) {\n      const response = toolContext.output;\n      this.respond({\n        content: [{\n          type: 'text',\n          text: JSON.stringify(response)\n        }],\n      })\n\n    } else {\n      this.fail(new Error(\"invalid output schema\"))\n    }\n    this.logger.verbose('finalize:done');\n  }\n}\n"]}