@orkestrel/mcp 0.0.9 → 0.0.11

package/dist/src/core/index.cjs

@@ -1383,7 +1383,7 @@ var MCPServer = class {
 			keys: this.#limits.keys,
 			depth: this.#limits.depth
 		})) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: `_meta` exceeds the configured limit or contains an unsafe value");
-		return modern ? this.#modern(request, id, options) : this.#legacy(request, id);
+		return modern ? this.#modern(request, id, options) : this.#legacy(request, id, options);
 	}
 	async handle(message, options) {
 		if (!isBoundedString(message, this.#limits.message)) return JSON.stringify(buildJSONRPCError(null, JSONRPC_PARSE_ERROR, "Parse error"));
@@ -1399,7 +1399,7 @@ var MCPServer = class {
 		if (answer === void 0) return void 0;
 		return Symbol.asyncIterator in answer ? serializeStream(answer) : JSON.stringify(answer);
 	}
-	async #legacy(request, id) {
+	async #legacy(request, id, options) {
 		switch (request.method) {
 			case "initialize": {
 				const requested = request.params?.["protocolVersion"];
@@ -1408,7 +1408,7 @@ var MCPServer = class {
 			case "ping": return buildJSONRPCResult(id, {});
 			case "tools/list": return buildJSONRPCResult(id, { tools: buildToolDescriptors(this.#options.tools) });
 			case "tools/call": {
-				const result = await this.#runTool(request, id);
+				const result = await this.#runTool(request, id, options);
 				return "jsonrpc" in result ? result : buildJSONRPCResult(id, result);
 			}
 			default: return buildJSONRPCError(id, JSONRPC_METHOD_NOT_FOUND, `Method not found: ${request.method}`);
@@ -1441,7 +1441,7 @@ var MCPServer = class {
 		const id = request.id ?? null;
 		const input = await this.#input(request, options);
 		if (input !== void 0) return input;
-		const result = await this.#runTool(request, id);
+		const result = await this.#runTool(request, id, options);
 		return "jsonrpc" in result ? result : buildJSONRPCResult(id, buildModernResult(result, this.#options.identity));
 	}
 	async #input(request, options) {
@@ -1463,12 +1463,12 @@ var MCPServer = class {
 				arguments: args
 			}, options);
 			if (elicitation === void 0) return void 0;
-			const principal = await configured.principal(request);
+			const principal = await configured.principal(request, options);
 			return this.#required(request, name, elicitation, principal);
 		}
 		if (!isBoundedString(requestState, this.#limits.state) || !(0, _orkestrel_contract.isRecord)(inputResponses)) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: `inputResponses` and `requestState` are required together");
 		const state = parseMCPInputState(await (0, _orkestrel_server.verifyToken)(requestState, configured.secret));
-		const principal = await configured.principal(request);
+		const principal = await configured.principal(request, options);
 		if (state === void 0 || state.principal !== principal || state.ttl !== configured.ttl || state.origin === id || state.name !== name || !Object.hasOwn(inputResponses, state.key)) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: request state could not be verified for this retry");
 		const response = inputResponses[state.key];
 		if (!isElicitResult(response)) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: the elicitation response is missing or malformed");
@@ -1543,7 +1543,7 @@ var MCPServer = class {
 			this.#subscriptions -= 1;
 		}
 	}
-	async #runTool(request, id) {
+	async #runTool(request, id, options) {
 		const params = request.params;
 		const name = params?.["name"];
 		if (!(0, _orkestrel_contract.isString)(name)) return buildJSONRPCError(id, JSONRPC_INVALID_PARAMS, "Invalid params: a string `name` is required");
@@ -1553,7 +1553,8 @@ var MCPServer = class {
 		const result = await this.#options.tools.execute({
 			id: callId,
 			name,
-			arguments: args
+			arguments: args,
+			...options.caller === void 0 ? {} : { caller: options.caller }
 		});
 		if (!result.success && !isBoundedString(result.error, this.#limits.content)) return buildJSONRPCError(id, JSONRPC_SERVER_ERROR, "Server limit exceeded: tool content is too large");
 		if (result.success && result.value !== void 0 && !isBoundedJSON(result.value, {