@cyanheads/mcp-ts-core 0.1.0-beta.25 → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CLAUDE.md +1 -4
- package/README.md +5 -3
- package/dist/core/app.d.ts.map +1 -1
- package/dist/core/app.js +58 -5
- package/dist/core/app.js.map +1 -1
- package/dist/mcp-server/prompts/prompt-registration.d.ts.map +1 -1
- package/dist/mcp-server/prompts/prompt-registration.js +2 -1
- package/dist/mcp-server/prompts/prompt-registration.js.map +1 -1
- package/dist/mcp-server/resources/utils/resourceHandlerFactory.d.ts.map +1 -1
- package/dist/mcp-server/resources/utils/resourceHandlerFactory.js +4 -1
- package/dist/mcp-server/resources/utils/resourceHandlerFactory.js.map +1 -1
- package/dist/mcp-server/tasks/core/taskManager.d.ts +0 -12
- package/dist/mcp-server/tasks/core/taskManager.d.ts.map +1 -1
- package/dist/mcp-server/tasks/core/taskManager.js +64 -2
- package/dist/mcp-server/tasks/core/taskManager.js.map +1 -1
- package/dist/mcp-server/transports/auth/authMiddleware.d.ts.map +1 -1
- package/dist/mcp-server/transports/auth/authMiddleware.js +47 -12
- package/dist/mcp-server/transports/auth/authMiddleware.js.map +1 -1
- package/dist/mcp-server/transports/http/sessionStore.d.ts.map +1 -1
- package/dist/mcp-server/transports/http/sessionStore.js +16 -0
- package/dist/mcp-server/transports/http/sessionStore.js.map +1 -1
- package/dist/services/graph/core/GraphService.d.ts +5 -0
- package/dist/services/graph/core/GraphService.d.ts.map +1 -1
- package/dist/services/graph/core/GraphService.js +56 -9
- package/dist/services/graph/core/GraphService.js.map +1 -1
- package/dist/services/llm/providers/openrouter.provider.d.ts.map +1 -1
- package/dist/services/llm/providers/openrouter.provider.js +77 -2
- package/dist/services/llm/providers/openrouter.provider.js.map +1 -1
- package/dist/services/speech/core/speechMetrics.d.ts +17 -0
- package/dist/services/speech/core/speechMetrics.d.ts.map +1 -0
- package/dist/services/speech/core/speechMetrics.js +36 -0
- package/dist/services/speech/core/speechMetrics.js.map +1 -0
- package/dist/services/speech/providers/elevenlabs.provider.d.ts.map +1 -1
- package/dist/services/speech/providers/elevenlabs.provider.js +59 -41
- package/dist/services/speech/providers/elevenlabs.provider.js.map +1 -1
- package/dist/services/speech/providers/whisper.provider.d.ts.map +1 -1
- package/dist/services/speech/providers/whisper.provider.js +77 -59
- package/dist/services/speech/providers/whisper.provider.js.map +1 -1
- package/dist/storage/core/StorageService.d.ts +5 -0
- package/dist/storage/core/StorageService.d.ts.map +1 -1
- package/dist/storage/core/StorageService.js +64 -19
- package/dist/storage/core/StorageService.js.map +1 -1
- package/dist/utils/internal/error-handler/errorHandler.d.ts.map +1 -1
- package/dist/utils/internal/error-handler/errorHandler.js +13 -0
- package/dist/utils/internal/error-handler/errorHandler.js.map +1 -1
- package/dist/utils/internal/performance.d.ts +29 -0
- package/dist/utils/internal/performance.d.ts.map +1 -1
- package/dist/utils/internal/performance.js +184 -9
- package/dist/utils/internal/performance.js.map +1 -1
- package/dist/utils/telemetry/semconv.d.ts +82 -0
- package/dist/utils/telemetry/semconv.d.ts.map +1 -1
- package/dist/utils/telemetry/semconv.js +113 -0
- package/dist/utils/telemetry/semconv.js.map +1 -1
- package/package.json +1 -1
- package/skills/add-export/SKILL.md +15 -8
- package/skills/add-provider/SKILL.md +67 -18
- package/skills/add-resource/SKILL.md +16 -6
- package/skills/add-tool/SKILL.md +4 -4
- package/skills/api-auth/SKILL.md +12 -12
- package/skills/api-config/SKILL.md +158 -15
- package/skills/api-context/SKILL.md +18 -13
- package/skills/api-errors/SKILL.md +77 -2
- package/skills/api-services/references/graph.md +1 -0
- package/skills/api-services/references/llm.md +1 -1
- package/skills/api-services/references/speech.md +5 -5
- package/skills/api-testing/SKILL.md +10 -39
- package/skills/api-utils/SKILL.md +77 -12
- package/skills/api-workers/SKILL.md +14 -9
- package/skills/devcheck/SKILL.md +19 -10
- package/skills/maintenance/SKILL.md +7 -4
- package/skills/migrate-mcp-ts-template/SKILL.md +5 -5
- package/skills/release/SKILL.md +5 -5
- package/skills/setup/SKILL.md +22 -13
- package/skills/walkthrough-init/SKILL.md +1 -1
- package/templates/AGENTS.md +130 -4
- package/templates/CLAUDE.md +128 -2
- package/templates/package.json +1 -1
package/CLAUDE.md
CHANGED
|
@@ -1,12 +1,10 @@
|
|
|
1
1
|
# Agent Protocol
|
|
2
2
|
|
|
3
|
-
**Package:** `@cyanheads/mcp-ts-core` · **Version:** 0.1.0
|
|
3
|
+
**Package:** `@cyanheads/mcp-ts-core` · **Version:** 0.1.0
|
|
4
4
|
**npm:** [@cyanheads/mcp-ts-core](https://www.npmjs.com/package/@cyanheads/mcp-ts-core) · **Docker:** [ghcr.io/cyanheads/mcp-ts-core](https://ghcr.io/cyanheads/mcp-ts-core)
|
|
5
5
|
|
|
6
6
|
> **Developer note:** Never assume. Read related files and docs before making changes. Read full file content for context. Never edit a file before reading it.
|
|
7
7
|
|
|
8
|
-
> **Extraction status:** Phases 1–4 complete (foundation, builders, skills, templates, examples, tests). Phase 5 (publish `@cyanheads/mcp-ts-core@0.1.0`) not started. See `core-extraction/` for details.
|
|
9
|
-
|
|
10
8
|
---
|
|
11
9
|
|
|
12
10
|
## Core Rules
|
|
@@ -61,7 +59,6 @@ Build configs exported for consumer extension: `tsconfig.json` extends `@cyanhea
|
|
|
61
59
|
### Node.js — `createApp(options)`
|
|
62
60
|
|
|
63
61
|
```ts
|
|
64
|
-
#!/usr/bin/env node
|
|
65
62
|
import { createApp } from '@cyanheads/mcp-ts-core';
|
|
66
63
|
import { allToolDefinitions } from './mcp-server/tools/index.js';
|
|
67
64
|
import { allResourceDefinitions } from './mcp-server/resources/index.js';
|
package/README.md
CHANGED
|
@@ -5,7 +5,7 @@
|
|
|
5
5
|
|
|
6
6
|
<div align="center">
|
|
7
7
|
|
|
8
|
-
[](./CHANGELOG.md) [](https://github.com/modelcontextprotocol/modelcontextprotocol/blob/main/docs/specification/2025-11-25/changelog.mdx) [](https://modelcontextprotocol.io/) [](./LICENSE)
|
|
9
9
|
|
|
10
10
|
[](https://www.typescriptlang.org/) [](https://bun.sh/)
|
|
11
11
|
|
|
@@ -21,8 +21,10 @@
|
|
|
21
21
|
import { createApp, tool, z } from '@cyanheads/mcp-ts-core';
|
|
22
22
|
|
|
23
23
|
const greet = tool('greet', {
|
|
24
|
-
description: 'Greet someone by name.',
|
|
25
|
-
|
|
24
|
+
description: 'Greet someone by name and return a personalized message.',
|
|
25
|
+
annotations: { readOnlyHint: true },
|
|
26
|
+
input: z.object({ name: z.string().describe('Name of the person to greet') }),
|
|
27
|
+
output: z.object({ message: z.string().describe('The greeting message') }),
|
|
26
28
|
handler: async (input) => ({ message: `Hello, ${input.name}!` }),
|
|
27
29
|
});
|
|
28
30
|
|
package/dist/core/app.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"app.d.ts","sourceRoot":"","sources":["../../src/core/app.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,OAAO,EAAE,MAAM,EAAe,MAAM,mBAAmB,CAAC;AAExD,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gDAAgD,CAAC;AAE1F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oDAAoD,CAAC;AAIhG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yCAAyC,CAAC;AAE1E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2CAA2C,CAAC;AAElF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qCAAqC,CAAC;AAExE,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAElE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gDAAgD,CAAC;AAE/E,OAAO,EAAE,MAAM,EAAoB,MAAM,4BAA4B,CAAC;AAGtE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;
|
|
1
|
+
{"version":3,"file":"app.d.ts","sourceRoot":"","sources":["../../src/core/app.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,OAAO,EAAE,MAAM,EAAe,MAAM,mBAAmB,CAAC;AAExD,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gDAAgD,CAAC;AAE1F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oDAAoD,CAAC;AAIhG,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,yCAAyC,CAAC;AAE1E,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2CAA2C,CAAC;AAElF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qCAAqC,CAAC;AAExE,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAElE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gDAAgD,CAAC;AAE/E,OAAO,EAAE,MAAM,EAAoB,MAAM,4BAA4B,CAAC;AAGtE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAQ9D,gGAAgG;AAChG,MAAM,WAAW,gBAAgB;IAC/B,wEAAwE;IACxE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,0BAA0B;IAC1B,OAAO,CAAC,EAAE,mBAAmB,EAAE,CAAC;IAChC,4BAA4B;IAC5B,SAAS,CAAC,EAAE,qBAAqB,EAAE,CAAC;IACpC,yEAAyE;IACzE,KAAK,CAAC,EAAE,CAAC,IAAI,EAAE,YAAY,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACrD,qDAAqD;IACrD,KAAK,CAAC,EAAE,UAAU,EAAE,CAAC;IACrB,8EAA8E;IAC9E,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wEAAwE;AACxE,MAAM,WAAW,YAAY;IAC3B,MAAM,EAAE,OAAO,MAAM,CAAC;IACtB,WAAW,CAAC,EAAE,YAAY,CAAC;IAC3B,MAAM,EAAE,OAAO,MAAM,CAAC;IACtB,WAAW,EAAE,WAAW,CAAC;IACzB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,OAAO,EAAE,cAAc,CAAC;IACxB,QAAQ,CAAC,EAAE,cAAc,CAAC,QAAQ,CAAC,CAAC;CACrC;AAED,iFAAiF;AACjF,MAAM,WAAW,YAAY;IAC3B,8EAA8E;IAC9E,QAAQ,CAAC,QAAQ,EAAE,YAAY,CAAC;IAChC,kFAAkF;IAClF,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1C;AAMD,uFAAuF;AACvF,MAAM,WAAW,WAAW;IAC1B,YAAY,EAAE,YAAY,CAAC;IAC3B,YAAY,EAAE,MAAM,OAAO,CAAC,SAAS,CAAC,CAAC;IACvC,gBAAgB,EAAE,gBAAgB,CAAC;CACpC;AAED;;;;;GAKG;AACH,wBAAsB,eAAe,CAAC,OAAO,GAAE,gBAAqB,GAAG,OAAO,CAAC,WAAW,CAAC,CAsG1F;AAoBD;;;;;;;;;GASG;AACH,wBAAsB,SAAS,CAAC,OAAO,GAAE,gBAAqB,GAAG,OAAO,CAAC,YAAY,CAAC,CAyLrF;AAMD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,YAAY,EAAE,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAC/F,YAAY,EAAE,gBAAgB,EAAE,MAAM,gDAAgD,CAAC;AACvF,OAAO,EAAE,MAAM,EAAE,MAAM,gDAAgD,CAAC;AACxE,YAAY,EACV,qBAAqB,EACrB,kBAAkB,GACnB,MAAM,oDAAoD,CAAC;AAC5D,OAAO,EAAE,QAAQ,EAAE,MAAM,oDAAoD,CAAC;AAC9E,YAAY,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,GACf,MAAM,4CAA4C,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,4CAA4C,CAAC"}
|
package/dist/core/app.js
CHANGED
|
@@ -23,6 +23,8 @@ import { initHighResTimer } from '../utils/internal/performance.js';
|
|
|
23
23
|
import { requestContextService } from '../utils/internal/requestContext.js';
|
|
24
24
|
import { RateLimiter } from '../utils/security/rateLimiter.js';
|
|
25
25
|
import { initializeOpenTelemetry, shutdownOpenTelemetry, } from '../utils/telemetry/instrumentation.js';
|
|
26
|
+
import { createObservableGauge } from '../utils/telemetry/metrics.js';
|
|
27
|
+
import { withSpan } from '../utils/telemetry/trace.js';
|
|
26
28
|
/**
|
|
27
29
|
* Constructs core services, registries, and server factory.
|
|
28
30
|
* Shared by createApp() (Node) and createWorkerHandler() (Workers).
|
|
@@ -152,6 +154,40 @@ export async function createApp(options = {}) {
|
|
|
152
154
|
}),
|
|
153
155
|
initHighResTimer(),
|
|
154
156
|
]);
|
|
157
|
+
// --- Process-level observable gauges (registered once after OTEL init) ---
|
|
158
|
+
if (typeof process !== 'undefined' && typeof process.memoryUsage === 'function') {
|
|
159
|
+
// Share a single memoryUsage() syscall across the three gauge callbacks per
|
|
160
|
+
// collection cycle. The snapshot is refreshed at most once per 100 ms.
|
|
161
|
+
let memSnapshot;
|
|
162
|
+
let memSnapshotTs = 0;
|
|
163
|
+
const getMemSnapshot = () => {
|
|
164
|
+
const now = Date.now();
|
|
165
|
+
if (!memSnapshot || now - memSnapshotTs > 100) {
|
|
166
|
+
memSnapshot = process.memoryUsage();
|
|
167
|
+
memSnapshotTs = now;
|
|
168
|
+
}
|
|
169
|
+
return memSnapshot;
|
|
170
|
+
};
|
|
171
|
+
createObservableGauge('process.memory.rss', 'Process resident set size', () => getMemSnapshot().rss, 'bytes');
|
|
172
|
+
createObservableGauge('process.memory.heap_used', 'V8 heap memory used', () => getMemSnapshot().heapUsed, 'bytes');
|
|
173
|
+
createObservableGauge('process.memory.heap_total', 'V8 total heap size', () => getMemSnapshot().heapTotal, 'bytes');
|
|
174
|
+
}
|
|
175
|
+
if (typeof process !== 'undefined' && typeof process.uptime === 'function') {
|
|
176
|
+
createObservableGauge('process.uptime', 'Process uptime in seconds', () => process.uptime(), 's');
|
|
177
|
+
}
|
|
178
|
+
// --- Event loop delay gauge (best single signal for Node.js health) ---
|
|
179
|
+
if (typeof process !== 'undefined') {
|
|
180
|
+
try {
|
|
181
|
+
const { monitorEventLoopDelay } = await import('node:perf_hooks');
|
|
182
|
+
const eld = monitorEventLoopDelay({ resolution: 20 });
|
|
183
|
+
eld.enable();
|
|
184
|
+
createObservableGauge('process.event_loop.delay', 'Event loop delay p99 in milliseconds', () => eld.percentile(99) / 1e6, // ns → ms
|
|
185
|
+
'ms');
|
|
186
|
+
}
|
|
187
|
+
catch {
|
|
188
|
+
// perf_hooks unavailable in this runtime — skip silently
|
|
189
|
+
}
|
|
190
|
+
}
|
|
155
191
|
// --- Compose services (handles name/version overrides + resetConfig) ---
|
|
156
192
|
const { coreServices, createServer, definitionCounts } = await composeServices(options);
|
|
157
193
|
// --- Initialize logger (after composeServices so config reflects overrides) ---
|
|
@@ -198,8 +234,14 @@ export async function createApp(options = {}) {
|
|
|
198
234
|
});
|
|
199
235
|
logger.info(`Received ${signal}. Initiating graceful shutdown...`, shutdownContext);
|
|
200
236
|
try {
|
|
201
|
-
await
|
|
202
|
-
|
|
237
|
+
await withSpan('mcp.server.shutdown', async (span) => {
|
|
238
|
+
span.setAttribute('mcp.server.shutdown.signal', signal);
|
|
239
|
+
await withSpan('mcp.server.shutdown.transport', async () => {
|
|
240
|
+
await transportManager.stop(signal);
|
|
241
|
+
});
|
|
242
|
+
logger.info('Graceful shutdown completed successfully.', shutdownContext);
|
|
243
|
+
});
|
|
244
|
+
// Flush telemetry after shutdown span is recorded
|
|
203
245
|
await shutdownOpenTelemetry();
|
|
204
246
|
await logger.close();
|
|
205
247
|
}
|
|
@@ -207,18 +249,29 @@ export async function createApp(options = {}) {
|
|
|
207
249
|
const err = error instanceof Error ? error : new Error(String(error));
|
|
208
250
|
logger.error('Critical error during shutdown process.', err, shutdownContext);
|
|
209
251
|
try {
|
|
252
|
+
await shutdownOpenTelemetry();
|
|
210
253
|
await logger.close();
|
|
211
254
|
}
|
|
212
255
|
catch {
|
|
213
|
-
// Ignore errors during final
|
|
256
|
+
// Ignore errors during final cleanup
|
|
214
257
|
}
|
|
215
258
|
}
|
|
216
259
|
};
|
|
217
260
|
// --- Register error/signal handlers ---
|
|
218
261
|
process.on('uncaughtException', onUncaughtException);
|
|
219
262
|
process.on('unhandledRejection', onUnhandledRejection);
|
|
220
|
-
// --- Start transport ---
|
|
221
|
-
await
|
|
263
|
+
// --- Start transport (wrapped in startup span with phase breakdown) ---
|
|
264
|
+
await withSpan('mcp.server.startup', async (span) => {
|
|
265
|
+
span.setAttribute('mcp.server.name', config.mcpServerName);
|
|
266
|
+
span.setAttribute('mcp.server.version', config.mcpServerVersion);
|
|
267
|
+
span.setAttribute('mcp.server.transport', config.mcpTransportType);
|
|
268
|
+
span.setAttribute('mcp.server.tools_count', definitionCounts.tools);
|
|
269
|
+
span.setAttribute('mcp.server.resources_count', definitionCounts.resources);
|
|
270
|
+
span.setAttribute('mcp.server.prompts_count', definitionCounts.prompts);
|
|
271
|
+
await withSpan('mcp.server.startup.transport', async () => {
|
|
272
|
+
await transportManager.start();
|
|
273
|
+
});
|
|
274
|
+
});
|
|
222
275
|
logger.info(`${config.mcpServerName} is now running and ready.`, startupContext);
|
|
223
276
|
process.on('SIGTERM', onSigterm);
|
|
224
277
|
process.on('SIGINT', onSigint);
|
package/dist/core/app.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"app.js","sourceRoot":"","sources":["../../src/core/app.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAKH,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,6CAA6C,CAAC;AAE7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,iDAAiD,CAAC;AAEnF,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AACzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,wCAAwC,CAAC;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAEvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AAEtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iDAAiD,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAEzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAoB,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EACL,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,sCAAsC,CAAC;
|
|
1
|
+
{"version":3,"file":"app.js","sourceRoot":"","sources":["../../src/core/app.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAKH,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,6CAA6C,CAAC;AAE7E,OAAO,EAAE,gBAAgB,EAAE,MAAM,iDAAiD,CAAC;AAEnF,OAAO,EAAE,aAAa,EAAE,MAAM,0CAA0C,CAAC;AACzE,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,wCAAwC,CAAC;AAErE,OAAO,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAEvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AAEtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,iDAAiD,CAAC;AACrF,OAAO,EAAE,aAAa,EAAE,MAAM,yCAAyC,CAAC;AAExE,OAAO,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAClE,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAEzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAoB,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EACL,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AAgDtD;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,UAA4B,EAAE;IAClE,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,SAAS,GAAG,EAAE,EAAE,OAAO,GAAG,EAAE,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAEpE,+DAA+D;IAC/D,uEAAuE;IACvE,4EAA4E;IAC5E,4DAA4D;IAC5D,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,OAAO,CAAC,IAAI;YAAE,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,OAAO,CAAC,IAAI,CAAC;QAC7D,IAAI,OAAO,CAAC,OAAO;YAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC;QACtE,WAAW,EAAE,CAAC;IAChB,CAAC;IAED,wBAAwB;IAExB,IAAI,cAAoD,CAAC;IACzD,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,cAAc,EAAE,CAAC;YAC9D,MAAM,kBAAkB,CAAC,+DAA+D,CAAC,CAAC;QAC5F,CAAC;QACD,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACxE,MAAM,kBAAkB,CACtB,wFAAwF,CACzF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,cAAc,GAAG,YAAY,CAAW,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,QAAQ,CAAC,cAAc,EAAE;YAC3F,IAAI,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE;SACzD,CAAC,CAAC;IACL,CAAC;IAED,MAAM,eAAe,GAAG,qBAAqB,CAAC,MAAM,EAAE;QACpD,GAAG,CAAC,cAAc,IAAI,EAAE,cAAc,EAAE,CAAC;KAC1C,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,eAAe,CAAC,CAAC;IAC3D,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAEpD,iEAAiE;IAEjE,IAAI,WAAqC,CAAC;IAC1C,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;QAC5B,WAAW,GAAG,IAAI,kBAAkB,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACpE,CAAC;IAED,IAAI,aAAwC,CAAC;IAC7C,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,MAAM,SAAS,GAAqC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO;YAC5E,CAAC,CAAE,MAAM,CAAC,MAAM,CAAC,GAA4B;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,SAAS,GAAqC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO;YAC5E,CAAC,CAAE,MAAM,CAAC,MAAM,CAAC,GAA4B;YAC7C,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,SAAS,IAAI,SAAS,EAAE,CAAC;YAC3B,aAAa,GAAG,IAAI,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAiB;QACjC,MAAM;QACN,MAAM;QACN,WAAW;QACX,OAAO,EAAE,cAAc;QACvB,GAAG,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,CAAC;QACnC,GAAG,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,CAAC;QACvC,GAAG,CAAC,cAAc,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;KACpD,CAAC;IAEF,gCAAgC;IAChC,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,KAAK,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;IAED,uBAAuB;IAEvB,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,KAAK,EAAE;QAC3C,MAAM;QACN,OAAO,EAAE,cAAc;KACxB,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC,SAAS,EAAE;QACvD,MAAM;QACN,OAAO,EAAE,cAAc;KACxB,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IAEhD,MAAM,YAAY,GAAG,GAAG,EAAE,CACxB,uBAAuB,CAAC;QACtB,MAAM;QACN,cAAc;QACd,gBAAgB;QAChB,aAAa;QACb,YAAY;KACb,CAAC,CAAC;IAEL,OAAO;QACL,YAAY;QACZ,YAAY;QACZ,gBAAgB,EAAE;YAChB,OAAO,EAAE,OAAO,CAAC,MAAM;YACvB,SAAS,EAAE,SAAS,CAAC,MAAM;YAC3B,KAAK,EAAE,KAAK,CAAC,MAAM;SACpB;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,sBAAsB;AACtB,8EAA8E;AAE9E;;GAEG;AACH,SAAS,cAAc;IACrB,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,WAAW,EAAE,CAAC;IACpE,MAAM,WAAW,GAAG,CAAC,aAAa,IAAI,aAAa,KAAK,OAAO,CAAC;IAChE,MAAM,oBAAoB,GAAG,aAAa,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;IAE/E,IAAI,WAAW,IAAI,oBAAoB,EAAE,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC;QAC3B,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,GAAG,CAAC;IAChC,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAAC,UAA4B,EAAE;IAC5D,cAAc,EAAE,CAAC;IAEjB,0EAA0E;IAC1E,MAAM,OAAO,CAAC,GAAG,CAAC;QAChB,uBAAuB,EAAE,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YAC/C,OAAO,CAAC,KAAK,CAAC,+CAA+C,EAAE,GAAG,CAAC,CAAC;QACtE,CAAC,CAAC;QACF,gBAAgB,EAAE;KACnB,CAAC,CAAC;IAEH,4EAA4E;IAC5E,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,OAAO,OAAO,CAAC,WAAW,KAAK,UAAU,EAAE,CAAC;QAChF,4EAA4E;QAC5E,uEAAuE;QACvE,IAAI,WAA2C,CAAC;QAChD,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,MAAM,cAAc,GAAG,GAAuB,EAAE;YAC9C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,IAAI,CAAC,WAAW,IAAI,GAAG,GAAG,aAAa,GAAG,GAAG,EAAE,CAAC;gBAC9C,WAAW,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;gBACpC,aAAa,GAAG,GAAG,CAAC;YACtB,CAAC;YACD,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC;QAEF,qBAAqB,CACnB,oBAAoB,EACpB,2BAA2B,EAC3B,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,GAAG,EAC1B,OAAO,CACR,CAAC;QACF,qBAAqB,CACnB,0BAA0B,EAC1B,qBAAqB,EACrB,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,QAAQ,EAC/B,OAAO,CACR,CAAC;QACF,qBAAqB,CACnB,2BAA2B,EAC3B,oBAAoB,EACpB,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,SAAS,EAChC,OAAO,CACR,CAAC;IACJ,CAAC;IACD,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,OAAO,OAAO,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;QAC3E,qBAAqB,CACnB,gBAAgB,EAChB,2BAA2B,EAC3B,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EACtB,GAAG,CACJ,CAAC;IACJ,CAAC;IAED,yEAAyE;IACzE,IAAI,OAAO,OAAO,KAAK,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,EAAE,qBAAqB,EAAE,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC;YAClE,MAAM,GAAG,GAAG,qBAAqB,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;YACtD,GAAG,CAAC,MAAM,EAAE,CAAC;YACb,qBAAqB,CACnB,0BAA0B,EAC1B,sCAAsC,EACtC,GAAG,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,UAAU;YAC1C,IAAI,CACL,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,yDAAyD;QAC3D,CAAC;IACH,CAAC;IAED,0EAA0E;IAC1E,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,MAAM,eAAe,CAAC,OAAO,CAAC,CAAC;IAExF,iFAAiF;IACjF,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,QAAuB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAEjF,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;IAC1C,MAAM,CAAC,IAAI,CACT,8CAA8C,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,EAC3E,qBAAqB,CAAC,oBAAoB,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CACzE,CAAC;IAEF,oBAAoB;IACpB,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,OAAO,CAAC,CAAC;IAClE,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,MAAM,EACN,MAAM,EACN,YAAY,EACZ,WAAW,EACX,gBAAgB,CACjB,CAAC;IAEF,0BAA0B;IAC1B,MAAM,cAAc,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;QAChE,SAAS,EAAE,eAAe;QAC1B,eAAe,EAAE,MAAM,CAAC,aAAa;QACrC,kBAAkB,EAAE,MAAM,CAAC,gBAAgB;QAC3C,eAAe,EAAE,MAAM,CAAC,WAAW;KACpC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,YAAY,MAAM,CAAC,aAAa,MAAM,MAAM,CAAC,gBAAgB,MAAM,EAAE,cAAc,CAAC,CAAC;IAEjG,0DAA0D;IAC1D,MAAM,mBAAmB,GAAG,CAAC,KAAY,EAAE,EAAE;QAC3C,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC;QAC3E,KAAK,QAAQ,CAAC,mBAAmB,CAAC,CAAC;IACrC,CAAC,CAAC;IACF,MAAM,oBAAoB,GAAG,CAAC,MAAe,EAAE,EAAE;QAC/C,MAAM,GAAG,GAAG,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QACzE,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE,GAAG,EAAE,cAAc,CAAC,CAAC;QAClF,KAAK,QAAQ,CAAC,oBAAoB,CAAC,CAAC;IACtC,CAAC,CAAC;IACF,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,KAAK,QAAQ,CAAC,SAAS,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,KAAK,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAE/C,mBAAmB;IACnB,IAAI,cAAc,GAAG,KAAK,CAAC;IAE3B,MAAM,QAAQ,GAAG,KAAK,EAAE,MAAM,GAAG,UAAU,EAAiB,EAAE;QAC5D,IAAI,cAAc;YAAE,OAAO;QAC3B,cAAc,GAAG,IAAI,CAAC;QAEtB,6DAA6D;QAC7D,OAAO,CAAC,cAAc,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;QACjE,OAAO,CAAC,cAAc,CAAC,oBAAoB,EAAE,oBAAoB,CAAC,CAAC;QACnE,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QAC7C,OAAO,CAAC,cAAc,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAE3C,MAAM,eAAe,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YACjE,SAAS,EAAE,gBAAgB;YAC3B,YAAY,EAAE,MAAM;SACrB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,YAAY,MAAM,mCAAmC,EAAE,eAAe,CAAC,CAAC;QAEpF,IAAI,CAAC;YACH,MAAM,QAAQ,CAAC,qBAAqB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBACnD,IAAI,CAAC,YAAY,CAAC,4BAA4B,EAAE,MAAM,CAAC,CAAC;gBAExD,MAAM,QAAQ,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;oBACzD,MAAM,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtC,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE,eAAe,CAAC,CAAC;YAC5E,CAAC,CAAC,CAAC;YACH,kDAAkD;YAClD,MAAM,qBAAqB,EAAE,CAAC;YAC9B,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACtE,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE,GAAG,EAAE,eAAe,CAAC,CAAC;YAC9E,IAAI,CAAC;gBACH,MAAM,qBAAqB,EAAE,CAAC;gBAC9B,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,qCAAqC;YACvC,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,yCAAyC;IACzC,OAAO,CAAC,EAAE,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;IACrD,OAAO,CAAC,EAAE,CAAC,oBAAoB,EAAE,oBAAoB,CAAC,CAAC;IAEvD,yEAAyE;IACzE,MAAM,QAAQ,CAAC,oBAAoB,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;QAClD,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QAC3D,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,sBAAsB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACnE,IAAI,CAAC,YAAY,CAAC,wBAAwB,EAAE,gBAAgB,CAAC,KAAK,CAAC,CAAC;QACpE,IAAI,CAAC,YAAY,CAAC,4BAA4B,EAAE,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAC5E,IAAI,CAAC,YAAY,CAAC,0BAA0B,EAAE,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAExE,MAAM,QAAQ,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,gBAAgB,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,aAAa,4BAA4B,EAAE,cAAc,CAAC,CAAC;IAEjF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IACjC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAE/B,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;AAC9C,CAAC;AAED,8EAA8E;AAC9E,mDAAmD;AACnD,8EAA8E;AAE9E,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAE,MAAM,gDAAgD,CAAC;AAKxE,OAAO,EAAE,QAAQ,EAAE,MAAM,oDAAoD,CAAC;AAM9E,OAAO,EAAE,IAAI,EAAE,MAAM,4CAA4C,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"prompt-registration.d.ts","sourceRoot":"","sources":["../../../src/mcp-server/prompts/prompt-registration.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gDAAgD,CAAC;AAG1F,OAAO,KAAK,EAAE,MAAM,IAAI,aAAa,EAAE,MAAM,4BAA4B,CAAC;
|
|
1
|
+
{"version":3,"file":"prompt-registration.d.ts","sourceRoot":"","sources":["../../../src/mcp-server/prompts/prompt-registration.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gDAAgD,CAAC;AAG1F,OAAO,KAAK,EAAE,MAAM,IAAI,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAI1E,qBAAa,cAAc;IAEvB,OAAO,CAAC,UAAU;IAClB,OAAO,CAAC,MAAM;gBADN,UAAU,EAAE,mBAAmB,EAAE,EACjC,MAAM,EAAE,OAAO,aAAa;IAGtC;;OAEG;IACG,WAAW,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;YAcrC,cAAc;CAkD7B"}
|
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
import { JsonRpcErrorCode, McpError } from '../../types-global/errors.js';
|
|
2
2
|
import { ErrorHandler } from '../../utils/internal/error-handler/errorHandler.js';
|
|
3
|
+
import { measurePromptGeneration } from '../../utils/internal/performance.js';
|
|
3
4
|
import { requestContextService } from '../../utils/internal/requestContext.js';
|
|
4
5
|
export class PromptRegistry {
|
|
5
6
|
promptDefs;
|
|
@@ -32,7 +33,7 @@ export class PromptRegistry {
|
|
|
32
33
|
}, async (args) => {
|
|
33
34
|
try {
|
|
34
35
|
const validatedArgs = promptDef.args ? promptDef.args.parse(args) : args;
|
|
35
|
-
const messages = await promptDef.generate(validatedArgs);
|
|
36
|
+
const messages = await measurePromptGeneration(() => Promise.resolve(promptDef.generate(validatedArgs)), { ...context, promptName: promptDef.name });
|
|
36
37
|
return { messages };
|
|
37
38
|
}
|
|
38
39
|
catch (error) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"prompt-registration.js","sourceRoot":"","sources":["../../../src/mcp-server/prompts/prompt-registration.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,gDAAgD,CAAC;AAE9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAE3E,MAAM,OAAO,cAAc;IAEf;IACA;IAFV,YACU,UAAiC,EACjC,MAA4B;QAD5B,eAAU,GAAV,UAAU,CAAuB;QACjC,WAAM,GAAN,MAAM,CAAsB;IACnC,CAAC;IAEJ;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,MAAiB;QACjC,MAAM,OAAO,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YACzD,SAAS,EAAE,4BAA4B;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,UAAU,CAAC,MAAM,eAAe,EAAE,OAAO,CAAC,CAAC;QAEjF,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,CAAC,UAAU,CAAC,MAAM,UAAU,EAAE,OAAO,CAAC,CAAC;IACzF,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,MAAiB,EACjB,SAA8B,EAC9B,OAAsE;QAEtE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAEpE,MAAM,YAAY,CAAC,QAAQ,CACzB,GAAG,EAAE;YACH,MAAM,CAAC,cAAc,CACnB,SAAS,CAAC,IAAI,EACd;gBACE,WAAW,EAAE,SAAS,CAAC,WAAW;gBAClC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI;oBACpB,UAAU,EAAE,SAAS,CAAC,IAAI,CAAC,KAAK;iBACjC,CAAC;aACH,EACD,KAAK,EAAE,IAA6B,EAAE,EAAE;gBACtC,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;oBACzE,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,QAAQ,
|
|
1
|
+
{"version":3,"file":"prompt-registration.js","sourceRoot":"","sources":["../../../src/mcp-server/prompts/prompt-registration.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,YAAY,EAAE,MAAM,gDAAgD,CAAC;AAE9E,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAE3E,MAAM,OAAO,cAAc;IAEf;IACA;IAFV,YACU,UAAiC,EACjC,MAA4B;QAD5B,eAAU,GAAV,UAAU,CAAuB;QACjC,WAAM,GAAN,MAAM,CAAsB;IACnC,CAAC;IAEJ;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,MAAiB;QACjC,MAAM,OAAO,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YACzD,SAAS,EAAE,4BAA4B;SACxC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,UAAU,CAAC,MAAM,eAAe,EAAE,OAAO,CAAC,CAAC;QAEjF,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,IAAI,CAAC,UAAU,CAAC,MAAM,UAAU,EAAE,OAAO,CAAC,CAAC;IACzF,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,MAAiB,EACjB,SAA8B,EAC9B,OAAsE;QAEtE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QAEpE,MAAM,YAAY,CAAC,QAAQ,CACzB,GAAG,EAAE;YACH,MAAM,CAAC,cAAc,CACnB,SAAS,CAAC,IAAI,EACd;gBACE,WAAW,EAAE,SAAS,CAAC,WAAW;gBAClC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI;oBACpB,UAAU,EAAE,SAAS,CAAC,IAAI,CAAC,KAAK;iBACjC,CAAC;aACH,EACD,KAAK,EAAE,IAA6B,EAAE,EAAE;gBACtC,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;oBACzE,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAC5C,GAAG,EAAE,CACH,OAAO,CAAC,OAAO,CACb,SAAS,CAAC,QAAQ,CAAC,aAAyD,CAAC,CAC9E,EACH,EAAE,GAAG,OAAO,EAAE,UAAU,EAAE,SAAS,CAAC,IAAI,EAAE,CAC3C,CAAC;oBACF,OAAO,EAAE,QAAQ,EAAE,CAAC;gBACtB,CAAC;gBAAC,OAAO,KAAc,EAAE,CAAC;oBACxB,MAAM,OAAO,GAAG,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE;wBAC9C,SAAS,EAAE,UAAU,SAAS,CAAC,IAAI,EAAE;wBACrC,OAAO;qBACR,CAAC,CAAC;oBACH,MAAM,OAAO,YAAY,QAAQ;wBAC/B,CAAC,CAAC,OAAO;wBACT,CAAC,CAAC,IAAI,QAAQ,CAAC,gBAAgB,CAAC,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;gBACpE,CAAC;YACH,CAAC,CACF,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sBAAsB,SAAS,CAAC,IAAI,EAAE,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,EACD;YACE,SAAS,EAAE,qBAAqB,SAAS,CAAC,IAAI,EAAE;YAChD,OAAO;YACP,SAAS,EAAE,gBAAgB,CAAC,oBAAoB;YAChD,QAAQ,EAAE,IAAI;SACf,CACF,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"resourceHandlerFactory.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/resources/utils/resourceHandlerFactory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AACxF,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iDAAiD,CAAC;AACjF,OAAO,KAAK,EACV,kBAAkB,EAClB,kBAAkB,EAClB,aAAa,EACd,MAAM,oCAAoC,CAAC;AAK5C,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oDAAoD,CAAC;AAEhG,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAEvE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;
|
|
1
|
+
{"version":3,"file":"resourceHandlerFactory.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/resources/utils/resourceHandlerFactory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,8CAA8C,CAAC;AACxF,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iDAAiD,CAAC;AACjF,OAAO,KAAK,EACV,kBAAkB,EAClB,kBAAkB,EAClB,aAAa,EACd,MAAM,oCAAoC,CAAC;AAK5C,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oDAAoD,CAAC;AAEhG,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAEvE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAQzD,KAAK,QAAQ,GAAG,mBAAmB,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC;AAOvE,qEAAqE;AACrE,MAAM,WAAW,8BAA8B;IAC7C,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,cAAc,CAAC;CACzB;AAgDD;;;;;;;;;;;GAWG;AACH,wBAAgB,qBAAqB,CACnC,GAAG,EAAE,qBAAqB,EAC1B,QAAQ,EAAE,8BAA8B,GACvC,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,QAAQ,KAAK,OAAO,CAAC,kBAAkB,CAAC,CA8DlF"}
|
|
@@ -6,6 +6,7 @@
|
|
|
6
6
|
import { createContext } from '../../../core/context.js';
|
|
7
7
|
import { withRequiredScopes } from '../../../mcp-server/transports/auth/lib/authUtils.js';
|
|
8
8
|
import { ErrorHandler } from '../../../utils/internal/error-handler/errorHandler.js';
|
|
9
|
+
import { measureResourceExecution } from '../../../utils/internal/performance.js';
|
|
9
10
|
import { requestContextService } from '../../../utils/internal/requestContext.js';
|
|
10
11
|
// ---------------------------------------------------------------------------
|
|
11
12
|
// Default formatter
|
|
@@ -86,7 +87,9 @@ export function createResourceHandler(def, services) {
|
|
|
86
87
|
sample: wrapSample(sdkCaps),
|
|
87
88
|
uri,
|
|
88
89
|
});
|
|
89
|
-
|
|
90
|
+
// Execute handler with performance measurement
|
|
91
|
+
const resourceName = def.name ?? def.uriTemplate;
|
|
92
|
+
const result = await measureResourceExecution(() => Promise.resolve(def.handler(validatedParams, ctx)), { ...appContext, resourceName }, { uri: uri.href, mimeType });
|
|
90
93
|
const contents = formatter(result, { uri, mimeType });
|
|
91
94
|
return { contents };
|
|
92
95
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"resourceHandlerFactory.js","sourceRoot":"","sources":["../../../../src/mcp-server/resources/utils/resourceHandlerFactory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAYH,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,OAAO,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AAEnF,OAAO,EAAE,YAAY,EAAE,MAAM,gDAAgD,CAAC;AAE9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAmB3E,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAE9E,SAAS,wBAAwB,CAC/B,MAAe,EACf,IAAoC;IAEpC,OAAO;QACL;YACE,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI;YAClB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,+BAA+B;AAC/B,8EAA8E;AAE9E,SAAS,UAAU,CAAC,UAAkC;IACpD,IAAI,OAAO,UAAU,CAAC,WAAW,KAAK,UAAU;QAAE,OAAO;IACzD,MAAM,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC;IAClC,OAAO,CAAC,GAAW,EAAE,MAA8B,EAAE,EAAE,CACrD,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,EAAE,CAE3C,CAAC;AACN,CAAC;AAED,SAAS,UAAU,CAAC,UAAkC;IACpD,IAAI,OAAO,UAAU,CAAC,aAAa,KAAK,UAAU;QAAE,OAAO;IAC3D,MAAM,EAAE,GAAG,UAAU,CAAC,aAAa,CAAC;IACpC,OAAO,CACL,IAA+E,EAC/E,IAAmB,EACnB,EAAE,CACF,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,CAE7B,CAAC;AACN,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,qBAAqB,CACnC,GAA0B,EAC1B,QAAwC;IAExC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,IAAI,kBAAkB,CAAC;IACpD,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,IAAI,wBAAwB,CAAC;IAEzD,OAAO,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,WAAW,EAA+B,EAAE;QACxE,MAAM,UAAU,GAAG,WAAkC,CAAC;QACtD,MAAM,OAAO,GAAG,WAAgD,CAAC;QAEjE,MAAM,SAAS,GAAG,OAAO,UAAU,EAAE,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAE/F,MAAM,UAAU,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YAC5D,aAAa,EAAE;gBACb,GAAG,CAAC,OAAO,UAAU,EAAE,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzF,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpC;YACD,SAAS,EAAE,oBAAoB;YAC/B,iBAAiB,EAAE;gBACjB,YAAY,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW;gBACzC,WAAW,EAAE,GAAG,CAAC,IAAI;gBACrB,SAAS;gBACT,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,2BAA2B;YAC3B,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YAC3C,CAAC;YAED,wCAAwC;YACxC,MAAM,eAAe,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAE7E,iCAAiC;YACjC,MAAM,GAAG,GAAG,aAAa,CAAC;gBACxB,UAAU;gBACV,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC;gBAC3B,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC;gBAC3B,GAAG;aACJ,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"resourceHandlerFactory.js","sourceRoot":"","sources":["../../../../src/mcp-server/resources/utils/resourceHandlerFactory.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAYH,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAElD,OAAO,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AAEnF,OAAO,EAAE,YAAY,EAAE,MAAM,gDAAgD,CAAC;AAE9E,OAAO,EAAE,wBAAwB,EAAE,MAAM,iCAAiC,CAAC;AAC3E,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAmB3E,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAE9E,SAAS,wBAAwB,CAC/B,MAAe,EACf,IAAoC;IAEpC,OAAO;QACL;YACE,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI;YAClB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB;KACF,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,+BAA+B;AAC/B,8EAA8E;AAE9E,SAAS,UAAU,CAAC,UAAkC;IACpD,IAAI,OAAO,UAAU,CAAC,WAAW,KAAK,UAAU;QAAE,OAAO;IACzD,MAAM,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC;IAClC,OAAO,CAAC,GAAW,EAAE,MAA8B,EAAE,EAAE,CACrD,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,eAAe,EAAE,MAAM,EAAE,CAE3C,CAAC;AACN,CAAC;AAED,SAAS,UAAU,CAAC,UAAkC;IACpD,IAAI,OAAO,UAAU,CAAC,aAAa,KAAK,UAAU;QAAE,OAAO;IAC3D,MAAM,EAAE,GAAG,UAAU,CAAC,aAAa,CAAC;IACpC,OAAO,CACL,IAA+E,EAC/E,IAAmB,EACnB,EAAE,CACF,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,IAAI,EAAE,CAE7B,CAAC;AACN,CAAC;AAED,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,qBAAqB,CACnC,GAA0B,EAC1B,QAAwC;IAExC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,IAAI,kBAAkB,CAAC;IACpD,MAAM,SAAS,GAAG,GAAG,CAAC,MAAM,IAAI,wBAAwB,CAAC;IAEzD,OAAO,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,WAAW,EAA+B,EAAE;QACxE,MAAM,UAAU,GAAG,WAAkC,CAAC;QACtD,MAAM,OAAO,GAAG,WAAgD,CAAC;QAEjE,MAAM,SAAS,GAAG,OAAO,UAAU,EAAE,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAE/F,MAAM,UAAU,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YAC5D,aAAa,EAAE;gBACb,GAAG,CAAC,OAAO,UAAU,EAAE,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACzF,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpC;YACD,SAAS,EAAE,oBAAoB;YAC/B,iBAAiB,EAAE;gBACjB,YAAY,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW;gBACzC,WAAW,EAAE,GAAG,CAAC,IAAI;gBACrB,SAAS;gBACT,WAAW,EAAE,SAAS;aACvB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,2BAA2B;YAC3B,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpC,kBAAkB,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YAC3C,CAAC;YAED,wCAAwC;YACxC,MAAM,eAAe,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAE7E,iCAAiC;YACjC,MAAM,GAAG,GAAG,aAAa,CAAC;gBACxB,UAAU;gBACV,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,MAAM,EAAE,UAAU,CAAC,MAAM;gBACzB,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC;gBAC3B,MAAM,EAAE,UAAU,CAAC,OAAO,CAAC;gBAC3B,GAAG;aACJ,CAAC,CAAC;YAEH,+CAA+C;YAC/C,MAAM,YAAY,GAAG,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAC3C,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,EACxD,EAAE,GAAG,UAAU,EAAE,YAAY,EAAE,EAC/B,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,EAAE,QAAQ,EAAE,CAC5B,CAAC;YAEF,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,CAAC;YACtD,OAAO,EAAE,QAAQ,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,YAAY,CAAC,WAAW,CAAC,KAAK,EAAE;gBACpC,SAAS,EAAE,YAAY,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,WAAW,cAAc;gBAChE,OAAO,EAAE,UAAU;gBACnB,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE;aACpC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -1,15 +1,3 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* @fileoverview Singleton service for managing MCP task state and message queues.
|
|
3
|
-
* Supports both in-memory and storage-backed task stores based on configuration.
|
|
4
|
-
*
|
|
5
|
-
* Configure via environment variables:
|
|
6
|
-
* - TASK_STORE_TYPE: 'in-memory' (default) or 'storage'
|
|
7
|
-
* - TASK_STORE_TENANT_ID: Tenant ID for storage isolation (default: 'system-tasks')
|
|
8
|
-
* - TASK_STORE_DEFAULT_TTL_MS: Default TTL in milliseconds (optional)
|
|
9
|
-
*
|
|
10
|
-
* @experimental These APIs are experimental and may change without notice.
|
|
11
|
-
* @module src/mcp-server/tasks/core/taskManager
|
|
12
|
-
*/
|
|
13
1
|
import type { config as configType } from '../../../config/index.js';
|
|
14
2
|
import type { StorageService } from '../../../storage/core/StorageService.js';
|
|
15
3
|
import type { RequestContext } from '../../../utils/internal/requestContext.js';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"taskManager.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/taskManager.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"taskManager.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/taskManager.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,MAAM,IAAI,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kCAAkC,CAAC;AAEvE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AAMzE,OAAO,EAKL,KAAK,gBAAgB,EACrB,KAAK,SAAS,EACf,MAAM,gBAAgB,CAAC;AAiFxB;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAY;IACtC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAkC;IACpE,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA2B;IACxD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA0B;IACpD,OAAO,CAAC,cAAc,CAAS;gBAEnB,MAAM,EAAE,OAAO,UAAU,EAAE,cAAc,EAAE,cAAc;IAwCrE;;;;;;;;;;OAUG;IACI,YAAY,IAAI,SAAS;IAIhC;;;;;;;;;OASG;IACI,eAAe,IAAI,gBAAgB;IAI1C;;;;OAIG;IACI,YAAY,IAAI,WAAW,GAAG,SAAS;IAI9C;;;;;;;;;;OAUG;IACI,OAAO,CAAC,OAAO,CAAC,EAAE,cAAc,GAAG,IAAI;IAuB9C;;;;;OAKG;IACI,YAAY,IAAI,MAAM,GAAG,IAAI;IAQpC;;;;OAIG;IACI,YAAY,IAAI,OAAO;CAG/B"}
|
|
@@ -1,8 +1,62 @@
|
|
|
1
1
|
import { logger } from '../../../utils/internal/logger.js';
|
|
2
2
|
import { idGenerator } from '../../../utils/security/idGenerator.js';
|
|
3
|
+
import { createCounter, createObservableGauge } from '../../../utils/telemetry/metrics.js';
|
|
4
|
+
import { ATTR_MCP_TASK_STATUS, ATTR_MCP_TASK_STORE_TYPE } from '../../../utils/telemetry/semconv.js';
|
|
3
5
|
import { SessionAwareTaskStore } from './sessionAwareTaskStore.js';
|
|
4
6
|
import { StorageBackedTaskStore } from './storageBackedTaskStore.js';
|
|
5
7
|
import { InMemoryTaskMessageQueue, InMemoryTaskStore, } from './taskTypes.js';
|
|
8
|
+
// ---------------------------------------------------------------------------
|
|
9
|
+
// Task lifecycle metrics
|
|
10
|
+
// ---------------------------------------------------------------------------
|
|
11
|
+
let taskCreatedCounter;
|
|
12
|
+
let taskStatusCounter;
|
|
13
|
+
function getTaskMetrics() {
|
|
14
|
+
taskCreatedCounter ??= createCounter('mcp.tasks.created', 'Total tasks created', '{tasks}');
|
|
15
|
+
taskStatusCounter ??= createCounter('mcp.tasks.status_changes', 'Total task status transitions', '{transitions}');
|
|
16
|
+
return { taskCreatedCounter, taskStatusCounter };
|
|
17
|
+
}
|
|
18
|
+
/**
|
|
19
|
+
* Wraps a TaskStore to record lifecycle metrics on create, status update,
|
|
20
|
+
* and result storage. Delegates all actual logic to the inner store.
|
|
21
|
+
*/
|
|
22
|
+
class InstrumentedTaskStore {
|
|
23
|
+
inner;
|
|
24
|
+
storeTypeLabel;
|
|
25
|
+
constructor(inner, storeTypeLabel) {
|
|
26
|
+
this.inner = inner;
|
|
27
|
+
this.storeTypeLabel = storeTypeLabel;
|
|
28
|
+
}
|
|
29
|
+
async createTask(taskParams, requestId, request, sessionId) {
|
|
30
|
+
const task = await this.inner.createTask(taskParams, requestId, request, sessionId);
|
|
31
|
+
getTaskMetrics().taskCreatedCounter.add(1, {
|
|
32
|
+
[ATTR_MCP_TASK_STORE_TYPE]: this.storeTypeLabel,
|
|
33
|
+
});
|
|
34
|
+
return task;
|
|
35
|
+
}
|
|
36
|
+
async storeTaskResult(taskId, status, result, sessionId) {
|
|
37
|
+
await this.inner.storeTaskResult(taskId, status, result, sessionId);
|
|
38
|
+
getTaskMetrics().taskStatusCounter.add(1, {
|
|
39
|
+
[ATTR_MCP_TASK_STATUS]: status,
|
|
40
|
+
[ATTR_MCP_TASK_STORE_TYPE]: this.storeTypeLabel,
|
|
41
|
+
});
|
|
42
|
+
}
|
|
43
|
+
async updateTaskStatus(taskId, status, statusMessage, sessionId) {
|
|
44
|
+
await this.inner.updateTaskStatus(taskId, status, statusMessage, sessionId);
|
|
45
|
+
getTaskMetrics().taskStatusCounter.add(1, {
|
|
46
|
+
[ATTR_MCP_TASK_STATUS]: status,
|
|
47
|
+
[ATTR_MCP_TASK_STORE_TYPE]: this.storeTypeLabel,
|
|
48
|
+
});
|
|
49
|
+
}
|
|
50
|
+
getTask(taskId, sessionId) {
|
|
51
|
+
return this.inner.getTask(taskId, sessionId);
|
|
52
|
+
}
|
|
53
|
+
getTaskResult(taskId, sessionId) {
|
|
54
|
+
return this.inner.getTaskResult(taskId, sessionId);
|
|
55
|
+
}
|
|
56
|
+
listTasks(cursor, sessionId) {
|
|
57
|
+
return this.inner.listTasks(cursor, sessionId);
|
|
58
|
+
}
|
|
59
|
+
}
|
|
6
60
|
/**
|
|
7
61
|
* Singleton service that manages task state and message queues for the MCP server.
|
|
8
62
|
*
|
|
@@ -36,8 +90,9 @@ export class TaskManager {
|
|
|
36
90
|
constructor(config, storageService) {
|
|
37
91
|
this.storeType = config.tasks.storeType;
|
|
38
92
|
this.messageQueue = new InMemoryTaskMessageQueue();
|
|
93
|
+
let baseStore;
|
|
39
94
|
if (this.storeType === 'storage') {
|
|
40
|
-
|
|
95
|
+
baseStore = new StorageBackedTaskStore(storageService, {
|
|
41
96
|
tenantId: config.tasks.tenantId,
|
|
42
97
|
defaultTtl: config.tasks.defaultTtlMs ?? null,
|
|
43
98
|
});
|
|
@@ -46,8 +101,10 @@ export class TaskManager {
|
|
|
46
101
|
this.inMemoryTaskStore = new InMemoryTaskStore();
|
|
47
102
|
// Wrap with session ownership enforcement — the SDK's InMemoryTaskStore
|
|
48
103
|
// ignores sessionId parameters, so SessionAwareTaskStore adds that layer.
|
|
49
|
-
|
|
104
|
+
baseStore = new SessionAwareTaskStore(this.inMemoryTaskStore);
|
|
50
105
|
}
|
|
106
|
+
// Wrap with lifecycle metrics (outermost layer)
|
|
107
|
+
this.taskStore = new InstrumentedTaskStore(baseStore, this.storeType);
|
|
51
108
|
logger.info(`TaskManager initialized with ${this.storeType} task store`, {
|
|
52
109
|
operation: 'TaskManager.constructor',
|
|
53
110
|
requestId: idGenerator.generate('req'),
|
|
@@ -55,6 +112,11 @@ export class TaskManager {
|
|
|
55
112
|
storeType: this.storeType,
|
|
56
113
|
...(this.storeType === 'storage' && { tenantId: config.tasks.tenantId }),
|
|
57
114
|
});
|
|
115
|
+
// Wire task count to OTel observable gauge
|
|
116
|
+
if (this.inMemoryTaskStore) {
|
|
117
|
+
const store = this.inMemoryTaskStore;
|
|
118
|
+
createObservableGauge('mcp.tasks.active', 'Number of active tasks in the in-memory store', () => store.getAllTasks().length, '{tasks}');
|
|
119
|
+
}
|
|
58
120
|
}
|
|
59
121
|
/**
|
|
60
122
|
* Returns the TaskStore instance for managing task lifecycle.
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"taskManager.js","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/taskManager.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"taskManager.js","sourceRoot":"","sources":["../../../../src/mcp-server/tasks/core/taskManager.ts"],"names":[],"mappings":"AAgBA,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAEpD,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,aAAa,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACpF,OAAO,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EAEL,wBAAwB,EACxB,iBAAiB,GAIlB,MAAM,gBAAgB,CAAC;AAExB,8EAA8E;AAC9E,yBAAyB;AACzB,8EAA8E;AAE9E,IAAI,kBAAgE,CAAC;AACrE,IAAI,iBAA+D,CAAC;AAEpE,SAAS,cAAc;IACrB,kBAAkB,KAAK,aAAa,CAAC,mBAAmB,EAAE,qBAAqB,EAAE,SAAS,CAAC,CAAC;IAC5F,iBAAiB,KAAK,aAAa,CACjC,0BAA0B,EAC1B,+BAA+B,EAC/B,eAAe,CAChB,CAAC;IACF,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;AACnD,CAAC;AAED;;;GAGG;AACH,MAAM,qBAAqB;IAEN;IACA;IAFnB,YACmB,KAAgB,EAChB,cAAsB;QADtB,UAAK,GAAL,KAAK,CAAW;QAChB,mBAAc,GAAd,cAAc,CAAQ;IACtC,CAAC;IAEJ,KAAK,CAAC,UAAU,CACd,UAA6B,EAC7B,SAAoB,EACpB,OAAgB,EAChB,SAAkB;QAElB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;QACpF,cAAc,EAAE,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE;YACzC,CAAC,wBAAwB,CAAC,EAAE,IAAI,CAAC,cAAc;SAChD,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAA8B,EAC9B,MAAc,EACd,SAAkB;QAElB,MAAM,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;QACpE,cAAc,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE;YACxC,CAAC,oBAAoB,CAAC,EAAE,MAAM;YAC9B,CAAC,wBAAwB,CAAC,EAAE,IAAI,CAAC,cAAc;SAChD,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,MAAc,EACd,MAAsB,EACtB,aAAsB,EACtB,SAAkB;QAElB,MAAM,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;QAC5E,cAAc,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE;YACxC,CAAC,oBAAoB,CAAC,EAAE,MAAM;YAC9B,CAAC,wBAAwB,CAAC,EAAE,IAAI,CAAC,cAAc;SAChD,CAAC,CAAC;IACL,CAAC;IAED,OAAO,CAAC,MAAc,EAAE,SAAkB;QACxC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,aAAa,CAAC,MAAc,EAAE,SAAkB;QAC9C,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAED,SAAS,CAAC,MAAe,EAAE,SAAkB;QAC3C,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;CACF;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,OAAO,WAAW;IACL,SAAS,CAAY;IACrB,iBAAiB,GAA6B,IAAI,CAAC;IACnD,YAAY,CAA2B;IACvC,SAAS,CAA0B;IAC5C,cAAc,GAAG,KAAK,CAAC;IAE/B,YAAY,MAAyB,EAAE,cAA8B;QACnE,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,IAAI,wBAAwB,EAAE,CAAC;QAEnD,IAAI,SAAoB,CAAC;QACzB,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACjC,SAAS,GAAG,IAAI,sBAAsB,CAAC,cAAc,EAAE;gBACrD,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ;gBAC/B,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY,IAAI,IAAI;aAC9C,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC;YACjD,wEAAwE;YACxE,0EAA0E;YAC1E,SAAS,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAChE,CAAC;QAED,gDAAgD;QAChD,IAAI,CAAC,SAAS,GAAG,IAAI,qBAAqB,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAEtE,MAAM,CAAC,IAAI,CAAC,gCAAgC,IAAI,CAAC,SAAS,aAAa,EAAE;YACvE,SAAS,EAAE,yBAAyB;YACpC,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC;YACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,CAAC,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,EAAE,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;SACzE,CAAC,CAAC;QAEH,2CAA2C;QAC3C,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC;YACrC,qBAAqB,CACnB,kBAAkB,EAClB,+CAA+C,EAC/C,GAAG,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,MAAM,EAChC,SAAS,CACV,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;;;;;;;OAUG;IACI,YAAY;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;;;;;;;;OASG;IACI,eAAe;QACpB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;;;OAIG;IACI,YAAY;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED;;;;;;;;;;OAUG;IACI,OAAO,CAAC,OAAwB;QACrC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAE3B,MAAM,UAAU,GAAG,OAAO,IAAI;YAC5B,SAAS,EAAE,qBAAqB;YAChC,SAAS,EAAE,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC;YACtC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,sCAAsC,EAAE,UAAU,CAAC,CAAC;QAEhE,4CAA4C;QAC5C,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC;QACnC,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE,UAAU,CAAC,CAAC;IAC1D,CAAC;IAED;;;;;OAKG;IACI,YAAY;QACjB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC,iBAAiB,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC;QACrD,CAAC;QACD,wEAAwE;QACxE,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACI,YAAY;QACjB,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"authMiddleware.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/transports/auth/authMiddleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEtD,OAAO,KAAK,EAAW,iBAAiB,EAAQ,MAAM,MAAM,CAAC;AAG7D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yDAAyD,CAAC;
|
|
1
|
+
{"version":3,"file":"authMiddleware.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/transports/auth/authMiddleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEtD,OAAO,KAAK,EAAW,iBAAiB,EAAQ,MAAM,MAAM,CAAC;AAG7D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yDAAyD,CAAC;AAiC5F;;;;;GAKG;AACH,wBAAgB,oBAAoB,CAClC,QAAQ,EAAE,YAAY,GACrB,iBAAiB,CAAC;IAAE,QAAQ,EAAE,YAAY,CAAA;CAAE,CAAC,CA6E/C"}
|
|
@@ -13,7 +13,17 @@ import { trace } from '@opentelemetry/api';
|
|
|
13
13
|
import { authContext } from '../../../mcp-server/transports/auth/lib/authContext.js';
|
|
14
14
|
import { unauthorized } from '../../../types-global/errors.js';
|
|
15
15
|
import { logger } from '../../../utils/internal/logger.js';
|
|
16
|
+
import { nowMs } from '../../../utils/internal/performance.js';
|
|
16
17
|
import { requestContextService } from '../../../utils/internal/requestContext.js';
|
|
18
|
+
import { createCounter, createHistogram } from '../../../utils/telemetry/metrics.js';
|
|
19
|
+
import { ATTR_MCP_AUTH_FAILURE_REASON, ATTR_MCP_AUTH_METHOD, ATTR_MCP_AUTH_OUTCOME, ATTR_MCP_AUTH_SCOPES, ATTR_MCP_AUTH_SUBJECT, ATTR_MCP_CLIENT_ID, ATTR_MCP_TENANT_ID, } from '../../../utils/telemetry/semconv.js';
|
|
20
|
+
let authAttemptCounter;
|
|
21
|
+
let authDuration;
|
|
22
|
+
function getAuthMetrics() {
|
|
23
|
+
authAttemptCounter ??= createCounter('mcp.auth.attempts', 'Total authentication attempts', '{attempts}');
|
|
24
|
+
authDuration ??= createHistogram('mcp.auth.duration', 'Authentication verification duration', 'ms');
|
|
25
|
+
return { authAttemptCounter, authDuration };
|
|
26
|
+
}
|
|
17
27
|
/**
|
|
18
28
|
* Creates a Hono middleware function that enforces authentication using a given strategy.
|
|
19
29
|
*
|
|
@@ -30,19 +40,48 @@ export function createAuthMiddleware(strategy) {
|
|
|
30
40
|
},
|
|
31
41
|
});
|
|
32
42
|
logger.debug('Initiating authentication check.', context);
|
|
43
|
+
const m = getAuthMetrics();
|
|
44
|
+
const activeSpan = trace.getActiveSpan();
|
|
45
|
+
const recordAuthEvent = (outcome, failureReason, durationMs) => {
|
|
46
|
+
const counterAttrs = { [ATTR_MCP_AUTH_OUTCOME]: outcome };
|
|
47
|
+
if (failureReason)
|
|
48
|
+
counterAttrs[ATTR_MCP_AUTH_FAILURE_REASON] = failureReason;
|
|
49
|
+
m.authAttemptCounter.add(1, counterAttrs);
|
|
50
|
+
if (durationMs !== undefined)
|
|
51
|
+
m.authDuration.record(durationMs, counterAttrs);
|
|
52
|
+
const spanAttrs = {
|
|
53
|
+
[ATTR_MCP_AUTH_METHOD]: 'bearer',
|
|
54
|
+
[ATTR_MCP_AUTH_OUTCOME]: outcome,
|
|
55
|
+
};
|
|
56
|
+
if (failureReason)
|
|
57
|
+
spanAttrs[ATTR_MCP_AUTH_FAILURE_REASON] = failureReason;
|
|
58
|
+
activeSpan?.setAttributes(spanAttrs);
|
|
59
|
+
};
|
|
33
60
|
const authHeader = c.req.header('Authorization');
|
|
34
61
|
if (!authHeader?.startsWith('Bearer ')) {
|
|
35
62
|
logger.warning('Authorization header missing or invalid.', context);
|
|
63
|
+
recordAuthEvent('missing', 'missing_header');
|
|
36
64
|
throw unauthorized('Missing or invalid Authorization header. Bearer scheme required.');
|
|
37
65
|
}
|
|
38
66
|
const token = authHeader.substring(7);
|
|
39
67
|
if (!token) {
|
|
40
68
|
logger.warning('Bearer token is missing from Authorization header.', context);
|
|
69
|
+
recordAuthEvent('missing', 'missing_token');
|
|
41
70
|
throw unauthorized('Authentication token is missing.');
|
|
42
71
|
}
|
|
43
72
|
logger.debug('Extracted Bearer token, proceeding to verification.', context);
|
|
44
73
|
// Strategy.verify() throws McpError on failure — errors propagate to httpErrorHandler.
|
|
45
|
-
const
|
|
74
|
+
const t0 = nowMs();
|
|
75
|
+
let authInfo;
|
|
76
|
+
try {
|
|
77
|
+
authInfo = await strategy.verify(token);
|
|
78
|
+
}
|
|
79
|
+
catch (err) {
|
|
80
|
+
recordAuthEvent('failure', undefined, Math.round((nowMs() - t0) * 100) / 100);
|
|
81
|
+
throw err;
|
|
82
|
+
}
|
|
83
|
+
const durationMs = Math.round((nowMs() - t0) * 100) / 100;
|
|
84
|
+
recordAuthEvent('success', undefined, durationMs);
|
|
46
85
|
const authLogContext = {
|
|
47
86
|
...context,
|
|
48
87
|
...(authInfo.tenantId ? { tenantId: authInfo.tenantId } : {}),
|
|
@@ -51,17 +90,13 @@ export function createAuthMiddleware(strategy) {
|
|
|
51
90
|
scopes: authInfo.scopes,
|
|
52
91
|
};
|
|
53
92
|
logger.info('Authentication successful. Auth context populated.', authLogContext);
|
|
54
|
-
// Add authentication
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
'auth.subject': authInfo.subject ?? 'unknown',
|
|
62
|
-
'auth.method': 'bearer',
|
|
63
|
-
});
|
|
64
|
-
}
|
|
93
|
+
// Add authentication identity to OpenTelemetry span for distributed tracing
|
|
94
|
+
activeSpan?.setAttributes({
|
|
95
|
+
[ATTR_MCP_CLIENT_ID]: authInfo.clientId,
|
|
96
|
+
[ATTR_MCP_TENANT_ID]: authInfo.tenantId ?? 'none',
|
|
97
|
+
[ATTR_MCP_AUTH_SCOPES]: authInfo.scopes.join(','),
|
|
98
|
+
[ATTR_MCP_AUTH_SUBJECT]: authInfo.subject ?? 'unknown',
|
|
99
|
+
});
|
|
65
100
|
// Run the next middleware in the chain within the populated auth context.
|
|
66
101
|
await authContext.run({ authInfo }, next);
|
|
67
102
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"authMiddleware.js","sourceRoot":"","sources":["../../../../src/mcp-server/transports/auth/authMiddleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAG3C,OAAO,EAAE,WAAW,EAAE,MAAM,iDAAiD,CAAC;AAE9E,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;
|
|
1
|
+
{"version":3,"file":"authMiddleware.js","sourceRoot":"","sources":["../../../../src/mcp-server/transports/auth/authMiddleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAGH,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAG3C,OAAO,EAAE,WAAW,EAAE,MAAM,iDAAiD,CAAC;AAE9E,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AACxD,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,KAAK,EAAE,MAAM,iCAAiC,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,oCAAoC,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC9E,OAAO,EACL,4BAA4B,EAC5B,oBAAoB,EACpB,qBAAqB,EACrB,oBAAoB,EACpB,qBAAqB,EACrB,kBAAkB,EAClB,kBAAkB,GACnB,MAAM,8BAA8B,CAAC;AAEtC,IAAI,kBAAgE,CAAC;AACrE,IAAI,YAA4D,CAAC;AAEjE,SAAS,cAAc;IACrB,kBAAkB,KAAK,aAAa,CAClC,mBAAmB,EACnB,+BAA+B,EAC/B,YAAY,CACb,CAAC;IACF,YAAY,KAAK,eAAe,CAC9B,mBAAmB,EACnB,sCAAsC,EACtC,IAAI,CACL,CAAC;IACF,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,CAAC;AAC9C,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,oBAAoB,CAClC,QAAsB;IAEtB,OAAO,KAAK,UAAU,cAAc,CAAC,CAAsC,EAAE,IAAU;QACrF,MAAM,OAAO,GAAG,qBAAqB,CAAC,oBAAoB,CAAC;YACzD,SAAS,EAAE,gBAAgB;YAC3B,iBAAiB,EAAE;gBACjB,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,MAAM;gBACpB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI;aACjB;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,OAAO,CAAC,CAAC;QAE1D,MAAM,CAAC,GAAG,cAAc,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC;QAEzC,MAAM,eAAe,GAAG,CAAC,OAAe,EAAE,aAAsB,EAAE,UAAmB,EAAE,EAAE;YACvF,MAAM,YAAY,GAA2B,EAAE,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,CAAC;YAClF,IAAI,aAAa;gBAAE,YAAY,CAAC,4BAA4B,CAAC,GAAG,aAAa,CAAC;YAC9E,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;YAC1C,IAAI,UAAU,KAAK,SAAS;gBAAE,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAE9E,MAAM,SAAS,GAA2B;gBACxC,CAAC,oBAAoB,CAAC,EAAE,QAAQ;gBAChC,CAAC,qBAAqB,CAAC,EAAE,OAAO;aACjC,CAAC;YACF,IAAI,aAAa;gBAAE,SAAS,CAAC,4BAA4B,CAAC,GAAG,aAAa,CAAC;YAC3E,UAAU,EAAE,aAAa,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QACjD,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YACvC,MAAM,CAAC,OAAO,CAAC,0CAA0C,EAAE,OAAO,CAAC,CAAC;YACpE,eAAe,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC;YAC7C,MAAM,YAAY,CAAC,kEAAkE,CAAC,CAAC;QACzF,CAAC;QAED,MAAM,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,CAAC,OAAO,CAAC,oDAAoD,EAAE,OAAO,CAAC,CAAC;YAC9E,eAAe,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;YAC5C,MAAM,YAAY,CAAC,kCAAkC,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,qDAAqD,EAAE,OAAO,CAAC,CAAC;QAE7E,uFAAuF;QACvF,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;QACnB,IAAI,QAAqD,CAAC;QAC1D,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAe,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;YAC9E,MAAM,GAAG,CAAC;QACZ,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC;QAC1D,eAAe,CAAC,SAAS,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QAElD,MAAM,cAAc,GAAG;YACrB,GAAG,OAAO;YACV,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;SACxB,CAAC;QACF,MAAM,CAAC,IAAI,CAAC,oDAAoD,EAAE,cAAc,CAAC,CAAC;QAElF,4EAA4E;QAC5E,UAAU,EAAE,aAAa,CAAC;YACxB,CAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,QAAQ;YACvC,CAAC,kBAAkB,CAAC,EAAE,QAAQ,CAAC,QAAQ,IAAI,MAAM;YACjD,CAAC,oBAAoB,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC;YACjD,CAAC,qBAAqB,CAAC,EAAE,QAAQ,CAAC,OAAO,IAAI,SAAS;SACvD,CAAC,CAAC;QAEH,0EAA0E;QAC1E,MAAM,WAAW,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sessionStore.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/transports/http/sessionStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;
|
|
1
|
+
{"version":3,"file":"sessionStore.d.ts","sourceRoot":"","sources":["../../../../src/mcp-server/transports/http/sessionStore.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAoBH;;;GAGG;AACH,MAAM,WAAW,eAAe;IAC9B,iDAAiD;IACjD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,mCAAmC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,qCAAqC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,UAAU,OAAO;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,EAAE,EAAE,MAAM,CAAC;IACX,cAAc,EAAE,IAAI,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IAGjB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,eAAe,CAAiC;gBAE5C,cAAc,EAAE,MAAM;IAOlC;;;OAGG;IACH,OAAO,IAAI,IAAI;IAKf;;;;;;;;OAQG;IACH,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,eAAe,GAAG,OAAO;IA2DnE;;;;;;;;;;;;;;OAcG;IACH,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,eAAe,GAAG,OAAO;IAkE1E;;;OAGG;IACH,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAYlC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAyB5B;;;OAGG;IACH,eAAe,IAAI,MAAM;CAG1B"}
|
|
@@ -8,6 +8,13 @@ import { validateSessionIdFormat } from '../../../mcp-server/transports/http/ses
|
|
|
8
8
|
import { invalidParams } from '../../../types-global/errors.js';
|
|
9
9
|
import { logger } from '../../../utils/internal/logger.js';
|
|
10
10
|
import { requestContextService } from '../../../utils/internal/requestContext.js';
|
|
11
|
+
import { createCounter } from '../../../utils/telemetry/metrics.js';
|
|
12
|
+
import { ATTR_MCP_SESSION_EVENT } from '../../../utils/telemetry/semconv.js';
|
|
13
|
+
let sessionEventCounter;
|
|
14
|
+
function getSessionMetrics() {
|
|
15
|
+
sessionEventCounter ??= createCounter('mcp.sessions.events', 'Session lifecycle events', '{events}');
|
|
16
|
+
return { sessionEventCounter };
|
|
17
|
+
}
|
|
11
18
|
/**
|
|
12
19
|
* Simple in-memory session store for stateful MCP sessions.
|
|
13
20
|
* In production, consider using Redis or another persistent store.
|
|
@@ -66,6 +73,7 @@ export class SessionStore {
|
|
|
66
73
|
newSession.subject = identity.subject;
|
|
67
74
|
session = newSession;
|
|
68
75
|
this.sessions.set(sessionId, session);
|
|
76
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'created' });
|
|
69
77
|
const context = requestContextService.createRequestContext({
|
|
70
78
|
operation: 'SessionStore.create',
|
|
71
79
|
sessionId,
|
|
@@ -134,6 +142,7 @@ export class SessionStore {
|
|
|
134
142
|
// If request has no identity but session does, reject (security: session was authenticated)
|
|
135
143
|
if (!identity) {
|
|
136
144
|
warn('Session requires authentication but request has no identity');
|
|
145
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'rejected' });
|
|
137
146
|
return false;
|
|
138
147
|
}
|
|
139
148
|
// Verify tenant ID match — reject if session is bound but request lacks or mismatches
|
|
@@ -142,6 +151,7 @@ export class SessionStore {
|
|
|
142
151
|
sessionTenant: session.tenantId,
|
|
143
152
|
requestTenant: identity.tenantId,
|
|
144
153
|
});
|
|
154
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'rejected' });
|
|
145
155
|
return false;
|
|
146
156
|
}
|
|
147
157
|
// Verify client ID match — reject if session is bound but request lacks or mismatches
|
|
@@ -150,6 +160,7 @@ export class SessionStore {
|
|
|
150
160
|
sessionClient: session.clientId,
|
|
151
161
|
requestClient: identity.clientId,
|
|
152
162
|
});
|
|
163
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'rejected' });
|
|
153
164
|
return false;
|
|
154
165
|
}
|
|
155
166
|
// Verify subject match — reject if session is bound but request lacks or mismatches
|
|
@@ -158,6 +169,7 @@ export class SessionStore {
|
|
|
158
169
|
sessionSubject: session.subject,
|
|
159
170
|
requestSubject: identity.subject,
|
|
160
171
|
});
|
|
172
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'rejected' });
|
|
161
173
|
return false;
|
|
162
174
|
}
|
|
163
175
|
return true;
|
|
@@ -169,6 +181,7 @@ export class SessionStore {
|
|
|
169
181
|
terminate(sessionId) {
|
|
170
182
|
const deleted = this.sessions.delete(sessionId);
|
|
171
183
|
if (deleted) {
|
|
184
|
+
getSessionMetrics().sessionEventCounter.add(1, { [ATTR_MCP_SESSION_EVENT]: 'terminated' });
|
|
172
185
|
const context = requestContextService.createRequestContext({
|
|
173
186
|
operation: 'SessionStore.terminate',
|
|
174
187
|
sessionId,
|
|
@@ -189,6 +202,9 @@ export class SessionStore {
|
|
|
189
202
|
}
|
|
190
203
|
}
|
|
191
204
|
if (cleanedCount > 0) {
|
|
205
|
+
getSessionMetrics().sessionEventCounter.add(cleanedCount, {
|
|
206
|
+
[ATTR_MCP_SESSION_EVENT]: 'stale_cleanup',
|
|
207
|
+
});
|
|
192
208
|
const context = requestContextService.createRequestContext({
|
|
193
209
|
operation: 'SessionStore.cleanup',
|
|
194
210
|
});
|