@agentlensai/mcp 0.11.0 → 0.13.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/LICENSE +21 -0
- package/dist/__tests__/benchmark.test.d.ts +5 -0
- package/dist/__tests__/benchmark.test.js +1 -1
- package/dist/__tests__/capabilities.test.d.ts +5 -0
- package/dist/__tests__/capabilities.test.d.ts.map +1 -0
- package/dist/__tests__/capabilities.test.js +127 -0
- package/dist/__tests__/capabilities.test.js.map +1 -0
- package/dist/__tests__/context.test.d.ts +5 -0
- package/dist/__tests__/context.test.d.ts.map +1 -0
- package/dist/__tests__/context.test.js +203 -0
- package/dist/__tests__/context.test.js.map +1 -0
- package/dist/__tests__/guardrail-middleware.test.d.ts +2 -0
- package/dist/__tests__/guardrail-middleware.test.d.ts.map +1 -0
- package/dist/__tests__/guardrail-middleware.test.js +107 -0
- package/dist/__tests__/guardrail-middleware.test.js.map +1 -0
- package/dist/__tests__/health.test.d.ts +5 -0
- package/dist/__tests__/health.test.js +1 -1
- package/dist/__tests__/llm-call.test.d.ts +2 -0
- package/dist/__tests__/llm-call.test.js +1 -1
- package/dist/__tests__/optimize.test.d.ts +5 -0
- package/dist/__tests__/optimize.test.js +1 -1
- package/dist/__tests__/recall.test.d.ts +2 -0
- package/dist/__tests__/recall.test.js +4 -4
- package/dist/__tests__/recall.test.js.map +1 -1
- package/dist/__tests__/reflect.test.d.ts +5 -0
- package/dist/__tests__/reflect.test.d.ts.map +1 -0
- package/dist/__tests__/reflect.test.js +165 -0
- package/dist/__tests__/reflect.test.js.map +1 -0
- package/dist/__tests__/replay.test.d.ts +5 -0
- package/dist/__tests__/replay.test.d.ts.map +1 -0
- package/dist/__tests__/replay.test.js +245 -0
- package/dist/__tests__/replay.test.js.map +1 -0
- package/dist/capabilities.d.ts +28 -0
- package/dist/capabilities.d.ts.map +1 -0
- package/dist/capabilities.js +68 -0
- package/dist/capabilities.js.map +1 -0
- package/dist/guardrail-middleware.d.ts +36 -0
- package/dist/guardrail-middleware.d.ts.map +1 -0
- package/dist/guardrail-middleware.js +70 -0
- package/dist/guardrail-middleware.js.map +1 -0
- package/dist/index.d.ts +19 -5
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +42 -8
- package/dist/index.js.map +1 -1
- package/dist/server.test.d.ts +2 -0
- package/dist/server.test.js +12 -5
- package/dist/server.test.js.map +1 -1
- package/dist/tools/__tests__/agents.test.d.ts +5 -0
- package/dist/tools/__tests__/agents.test.d.ts.map +1 -0
- package/dist/tools/__tests__/agents.test.js +56 -0
- package/dist/tools/__tests__/agents.test.js.map +1 -0
- package/dist/tools/__tests__/alerts.test.d.ts +5 -0
- package/dist/tools/__tests__/alerts.test.d.ts.map +1 -0
- package/dist/tools/__tests__/alerts.test.js +83 -0
- package/dist/tools/__tests__/alerts.test.js.map +1 -0
- package/dist/tools/__tests__/analytics.test.d.ts +5 -0
- package/dist/tools/__tests__/analytics.test.d.ts.map +1 -0
- package/dist/tools/__tests__/analytics.test.js +53 -0
- package/dist/tools/__tests__/analytics.test.js.map +1 -0
- package/dist/tools/__tests__/cost-budgets.test.d.ts +5 -0
- package/dist/tools/__tests__/cost-budgets.test.d.ts.map +1 -0
- package/dist/tools/__tests__/cost-budgets.test.js +86 -0
- package/dist/tools/__tests__/cost-budgets.test.js.map +1 -0
- package/dist/tools/__tests__/error-helpers.test.d.ts +5 -0
- package/dist/tools/__tests__/error-helpers.test.d.ts.map +1 -0
- package/dist/tools/__tests__/error-helpers.test.js +38 -0
- package/dist/tools/__tests__/error-helpers.test.js.map +1 -0
- package/dist/tools/__tests__/lessons.test.d.ts +5 -0
- package/dist/tools/__tests__/lessons.test.d.ts.map +1 -0
- package/dist/tools/__tests__/lessons.test.js +64 -0
- package/dist/tools/__tests__/lessons.test.js.map +1 -0
- package/dist/tools/__tests__/prompts.test.d.ts +5 -0
- package/dist/tools/__tests__/prompts.test.d.ts.map +1 -0
- package/dist/tools/__tests__/prompts.test.js +112 -0
- package/dist/tools/__tests__/prompts.test.js.map +1 -0
- package/dist/tools/__tests__/sessions.test.d.ts +5 -0
- package/dist/tools/__tests__/sessions.test.d.ts.map +1 -0
- package/dist/tools/__tests__/sessions.test.js +62 -0
- package/dist/tools/__tests__/sessions.test.js.map +1 -0
- package/dist/tools/__tests__/stats.test.d.ts +5 -0
- package/dist/tools/__tests__/stats.test.d.ts.map +1 -0
- package/dist/tools/__tests__/stats.test.js +41 -0
- package/dist/tools/__tests__/stats.test.js.map +1 -0
- package/dist/tools/__tests__/trust.test.d.ts +5 -0
- package/dist/tools/__tests__/trust.test.d.ts.map +1 -0
- package/dist/tools/__tests__/trust.test.js +46 -0
- package/dist/tools/__tests__/trust.test.js.map +1 -0
- package/dist/tools/agents.d.ts +9 -0
- package/dist/tools/agents.d.ts.map +1 -0
- package/dist/tools/agents.js +67 -0
- package/dist/tools/agents.js.map +1 -0
- package/dist/tools/alerts.d.ts +9 -0
- package/dist/tools/alerts.d.ts.map +1 -0
- package/dist/tools/alerts.js +126 -0
- package/dist/tools/alerts.js.map +1 -0
- package/dist/tools/analytics.d.ts +9 -0
- package/dist/tools/analytics.d.ts.map +1 -0
- package/dist/tools/analytics.js +67 -0
- package/dist/tools/analytics.js.map +1 -0
- package/dist/tools/cost-budgets.d.ts +9 -0
- package/dist/tools/cost-budgets.d.ts.map +1 -0
- package/dist/tools/cost-budgets.js +144 -0
- package/dist/tools/cost-budgets.js.map +1 -0
- package/dist/tools/error-helpers.d.ts +14 -0
- package/dist/tools/error-helpers.d.ts.map +1 -0
- package/dist/tools/error-helpers.js +17 -0
- package/dist/tools/error-helpers.js.map +1 -0
- package/dist/tools/lessons.d.ts +9 -0
- package/dist/tools/lessons.d.ts.map +1 -0
- package/dist/tools/lessons.js +107 -0
- package/dist/tools/lessons.js.map +1 -0
- package/dist/tools/prompts.d.ts +9 -0
- package/dist/tools/prompts.d.ts.map +1 -0
- package/dist/tools/prompts.js +117 -0
- package/dist/tools/prompts.js.map +1 -0
- package/dist/tools/recall.js +3 -3
- package/dist/tools/recall.js.map +1 -1
- package/dist/tools/reflect.js +1 -1
- package/dist/tools/reflect.js.map +1 -1
- package/dist/tools/sessions.d.ts +9 -0
- package/dist/tools/sessions.d.ts.map +1 -0
- package/dist/tools/sessions.js +88 -0
- package/dist/tools/sessions.js.map +1 -0
- package/dist/tools/stats.d.ts +9 -0
- package/dist/tools/stats.d.ts.map +1 -0
- package/dist/tools/stats.js +43 -0
- package/dist/tools/stats.js.map +1 -0
- package/dist/tools/trust.d.ts +9 -0
- package/dist/tools/trust.d.ts.map +1 -0
- package/dist/tools/trust.js +42 -0
- package/dist/tools/trust.js.map +1 -0
- package/dist/tools.d.ts +11 -2
- package/dist/tools.d.ts.map +1 -1
- package/dist/tools.js +70 -47
- package/dist/tools.js.map +1 -1
- package/dist/tools.test.d.ts +2 -0
- package/dist/tools.test.js +10 -3
- package/dist/tools.test.js.map +1 -1
- package/dist/transport.d.ts +59 -0
- package/dist/transport.d.ts.map +1 -1
- package/dist/transport.js +245 -0
- package/dist/transport.js.map +1 -1
- package/dist/transport.test.d.ts +2 -0
- package/dist/transport.test.d.ts.map +1 -0
- package/dist/transport.test.js +170 -0
- package/dist/transport.test.js.map +1 -0
- package/package.json +10 -10
- package/dist/__tests__/learn.test.d.ts.map +0 -1
- package/dist/__tests__/learn.test.js +0 -381
- package/dist/__tests__/learn.test.js.map +0 -1
- package/dist/tools/__tests__/community.test.d.ts +0 -5
- package/dist/tools/__tests__/community.test.d.ts.map +0 -1
- package/dist/tools/__tests__/community.test.js +0 -172
- package/dist/tools/__tests__/community.test.js.map +0 -1
- package/dist/tools/__tests__/transport-community.test.d.ts +0 -5
- package/dist/tools/__tests__/transport-community.test.d.ts.map +0 -1
- package/dist/tools/__tests__/transport-community.test.js +0 -99
- package/dist/tools/__tests__/transport-community.test.js.map +0 -1
- package/dist/tools/community.d.ts +0 -10
- package/dist/tools/community.d.ts.map +0 -1
- package/dist/tools/community.js +0 -167
- package/dist/tools/community.js.map +0 -1
- package/dist/tools/learn.d.ts +0 -10
- package/dist/tools/learn.d.ts.map +0 -1
- package/dist/tools/learn.js +0 -277
- package/dist/tools/learn.js.map +0 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"guardrail-middleware.d.ts","sourceRoot":"","sources":["../src/guardrail-middleware.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAChE,YAAY,EAAE,sBAAsB,EAAE,CAAC;AAEvC,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,OAAO,GAAG,QAAQ,CAAC;CAC/B;AAED,MAAM,WAAW,yBAAyB;IACxC,eAAe,CACb,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,kBAAkB,EAC3B,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,sBAAsB,CAAC,CAAC;CACpC;AAED,MAAM,WAAW,0BAA0B;IACzC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;IACrC,WAAW,EAAE,MAAM,MAAM,CAAC;IAC1B,SAAS,EAAE,yBAAyB,CAAC;IACrC,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,UAAU,UAAU;IAClB,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/C,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,KAAK,WAAW,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;AAE1E;;GAEG;AACH,wBAAgB,aAAa,CAC3B,OAAO,EAAE,WAAW,EACpB,OAAO,EAAE,0BAA0B,GAClC,WAAW,CAgFb"}
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Wrap a tool handler with content guardrail enforcement.
|
|
3
|
+
*/
|
|
4
|
+
export function guardrailWrap(handler, options) {
|
|
5
|
+
return async (args) => {
|
|
6
|
+
const agentId = options.getAgentId();
|
|
7
|
+
if (!agentId) {
|
|
8
|
+
return handler(args);
|
|
9
|
+
}
|
|
10
|
+
const baseContext = {
|
|
11
|
+
tenantId: options.getTenantId(),
|
|
12
|
+
agentId,
|
|
13
|
+
toolName: options.toolName,
|
|
14
|
+
};
|
|
15
|
+
// Phase 1: Scan Input
|
|
16
|
+
const inputText = JSON.stringify(args);
|
|
17
|
+
if (inputText) {
|
|
18
|
+
try {
|
|
19
|
+
const inputResult = await options.evaluator.evaluateContent(inputText, { ...baseContext, direction: 'input' }, options.timeoutMs);
|
|
20
|
+
if (inputResult.decision === 'block') {
|
|
21
|
+
return {
|
|
22
|
+
content: [{ type: 'text', text: `Guardrail policy violation: request blocked.` }],
|
|
23
|
+
isError: true,
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
if (inputResult.decision === 'redact' && inputResult.redactedContent) {
|
|
27
|
+
try {
|
|
28
|
+
args = JSON.parse(inputResult.redactedContent);
|
|
29
|
+
}
|
|
30
|
+
catch {
|
|
31
|
+
// Redaction broke JSON — fall back to original args
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
catch {
|
|
36
|
+
// Fail-open on evaluator error
|
|
37
|
+
}
|
|
38
|
+
}
|
|
39
|
+
// Phase 2: Execute Handler
|
|
40
|
+
const result = await handler(args);
|
|
41
|
+
// Phase 3: Scan Output
|
|
42
|
+
if (!result.isError) {
|
|
43
|
+
const outputText = result.content
|
|
44
|
+
.filter((c) => c.type === 'text')
|
|
45
|
+
.map((c) => c.text)
|
|
46
|
+
.join('\n');
|
|
47
|
+
if (outputText) {
|
|
48
|
+
try {
|
|
49
|
+
const outputResult = await options.evaluator.evaluateContent(outputText, { ...baseContext, direction: 'output' }, options.timeoutMs);
|
|
50
|
+
if (outputResult.decision === 'block') {
|
|
51
|
+
return {
|
|
52
|
+
content: [{ type: 'text', text: 'Guardrail policy violation: response blocked due to sensitive content.' }],
|
|
53
|
+
isError: true,
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
if (outputResult.decision === 'redact' && outputResult.redactedContent) {
|
|
57
|
+
return {
|
|
58
|
+
content: [{ type: 'text', text: outputResult.redactedContent }],
|
|
59
|
+
};
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
catch {
|
|
63
|
+
// Fail-open
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
return result;
|
|
68
|
+
};
|
|
69
|
+
}
|
|
70
|
+
//# sourceMappingURL=guardrail-middleware.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"guardrail-middleware.js","sourceRoot":"","sources":["../src/guardrail-middleware.ts"],"names":[],"mappings":"AAsCA;;GAEG;AACH,MAAM,UAAU,aAAa,CAC3B,OAAoB,EACpB,OAAmC;IAEnC,OAAO,KAAK,EAAE,IAA6B,EAAuB,EAAE;QAClE,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;QACrC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG;YAClB,QAAQ,EAAE,OAAO,CAAC,WAAW,EAAE;YAC/B,OAAO;YACP,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC3B,CAAC;QAEF,sBAAsB;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACvC,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,eAAe,CACzD,SAAS,EACT,EAAE,GAAG,WAAW,EAAE,SAAS,EAAE,OAAgB,EAAE,EAC/C,OAAO,CAAC,SAAS,CAClB,CAAC;gBAEF,IAAI,WAAW,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;oBACrC,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8CAA8C,EAAE,CAAC;wBACjF,OAAO,EAAE,IAAI;qBACd,CAAC;gBACJ,CAAC;gBAED,IAAI,WAAW,CAAC,QAAQ,KAAK,QAAQ,IAAI,WAAW,CAAC,eAAe,EAAE,CAAC;oBACrE,IAAI,CAAC;wBACH,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;oBACjD,CAAC;oBAAC,MAAM,CAAC;wBACP,oDAAoD;oBACtD,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QAED,2BAA2B;QAC3B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;QAEnC,uBAAuB;QACvB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO;iBAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,IAAI,UAAU,EAAE,CAAC;gBACf,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,eAAe,CAC1D,UAAU,EACV,EAAE,GAAG,WAAW,EAAE,SAAS,EAAE,QAAiB,EAAE,EAChD,OAAO,CAAC,SAAS,CAClB,CAAC;oBAEF,IAAI,YAAY,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;wBACtC,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,wEAAwE,EAAE,CAAC;4BAC3G,OAAO,EAAE,IAAI;yBACd,CAAC;oBACJ,CAAC;oBAED,IAAI,YAAY,CAAC,QAAQ,KAAK,QAAQ,IAAI,YAAY,CAAC,eAAe,EAAE,CAAC;wBACvE,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,CAAC,eAAe,EAAE,CAAC;yBAChE,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,YAAY;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC"}
|
package/dist/index.d.ts
CHANGED
|
@@ -1,22 +1,36 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
/**
|
|
3
|
-
* @agentlensai/mcp — MCP Server Entrypoint (Story 5.1)
|
|
3
|
+
* @agentlensai/mcp — MCP Server Entrypoint (Story 5.1, Feature 10)
|
|
4
4
|
*
|
|
5
5
|
* Starts an MCP server over stdio transport that exposes AgentLens
|
|
6
6
|
* instrumentation tools to MCP-compatible AI agents.
|
|
7
7
|
*
|
|
8
8
|
* Environment variables:
|
|
9
|
-
* AGENTLENS_URL
|
|
10
|
-
* AGENTLENS_API_KEY
|
|
9
|
+
* AGENTLENS_URL — API server base URL (default: http://localhost:3400)
|
|
10
|
+
* AGENTLENS_API_KEY — API key for authentication (optional)
|
|
11
|
+
* AGENTLENS_MCP_TOOLS — Comma-separated allowlist of tool names to register
|
|
12
|
+
* AGENTLENS_MCP_TOOLS_EXCLUDE — Comma-separated denylist of tool names to exclude
|
|
11
13
|
*/
|
|
12
14
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
13
15
|
import { AgentLensTransport } from './transport.js';
|
|
14
16
|
export { AgentLensTransport } from './transport.js';
|
|
15
17
|
export { registerTools } from './tools.js';
|
|
18
|
+
export type { ToolRegistrationOptions } from './capabilities.js';
|
|
19
|
+
export { shouldRegisterTool, TOOL_FEATURE_MAP } from './capabilities.js';
|
|
20
|
+
export type { ServerInfo } from './capabilities.js';
|
|
16
21
|
/**
|
|
17
|
-
* Create and configure the MCP server
|
|
22
|
+
* Create and configure the MCP server with auto-discovery (async).
|
|
23
|
+
* Probes server capabilities and conditionally registers tools.
|
|
18
24
|
*/
|
|
19
|
-
export declare function createServer(): {
|
|
25
|
+
export declare function createServer(): Promise<{
|
|
26
|
+
mcpServer: McpServer;
|
|
27
|
+
transport: AgentLensTransport;
|
|
28
|
+
}>;
|
|
29
|
+
/**
|
|
30
|
+
* Create and configure the MCP server synchronously (no auto-discovery).
|
|
31
|
+
* Registers all tools unconditionally. Backward compatible with old createServer().
|
|
32
|
+
*/
|
|
33
|
+
export declare function createServerSync(): {
|
|
20
34
|
mcpServer: McpServer;
|
|
21
35
|
transport: AgentLensTransport;
|
|
22
36
|
};
|
package/dist/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAIpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,YAAY,EAAE,uBAAuB,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACzE,YAAY,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAQpD;;;GAGG;AACH,wBAAsB,YAAY,IAAI,OAAO,CAAC;IAAE,SAAS,EAAE,SAAS,CAAC;IAAC,SAAS,EAAE,kBAAkB,CAAA;CAAE,CAAC,CA6BrG;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,IAAI;IAAE,SAAS,EAAE,SAAS,CAAC;IAAC,SAAS,EAAE,kBAAkB,CAAA;CAAE,CAsB1F"}
|
package/dist/index.js
CHANGED
|
@@ -1,13 +1,15 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
/**
|
|
3
|
-
* @agentlensai/mcp — MCP Server Entrypoint (Story 5.1)
|
|
3
|
+
* @agentlensai/mcp — MCP Server Entrypoint (Story 5.1, Feature 10)
|
|
4
4
|
*
|
|
5
5
|
* Starts an MCP server over stdio transport that exposes AgentLens
|
|
6
6
|
* instrumentation tools to MCP-compatible AI agents.
|
|
7
7
|
*
|
|
8
8
|
* Environment variables:
|
|
9
|
-
* AGENTLENS_URL
|
|
10
|
-
* AGENTLENS_API_KEY
|
|
9
|
+
* AGENTLENS_URL — API server base URL (default: http://localhost:3400)
|
|
10
|
+
* AGENTLENS_API_KEY — API key for authentication (optional)
|
|
11
|
+
* AGENTLENS_MCP_TOOLS — Comma-separated allowlist of tool names to register
|
|
12
|
+
* AGENTLENS_MCP_TOOLS_EXCLUDE — Comma-separated denylist of tool names to exclude
|
|
11
13
|
*/
|
|
12
14
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
13
15
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
@@ -15,17 +17,50 @@ import { AgentLensTransport } from './transport.js';
|
|
|
15
17
|
import { registerTools } from './tools.js';
|
|
16
18
|
export { AgentLensTransport } from './transport.js';
|
|
17
19
|
export { registerTools } from './tools.js';
|
|
20
|
+
export { shouldRegisterTool, TOOL_FEATURE_MAP } from './capabilities.js';
|
|
21
|
+
function parseEnvList(value) {
|
|
22
|
+
if (!value)
|
|
23
|
+
return null;
|
|
24
|
+
const items = value.split(',').map(s => s.trim()).filter(Boolean);
|
|
25
|
+
return items.length > 0 ? items : null;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Create and configure the MCP server with auto-discovery (async).
|
|
29
|
+
* Probes server capabilities and conditionally registers tools.
|
|
30
|
+
*/
|
|
31
|
+
export async function createServer() {
|
|
32
|
+
const baseUrl = process.env['AGENTLENS_URL'] ?? 'http://localhost:3400';
|
|
33
|
+
const apiKey = process.env['AGENTLENS_API_KEY'];
|
|
34
|
+
const transport = new AgentLensTransport({ baseUrl, apiKey });
|
|
35
|
+
transport.installShutdownHandlers();
|
|
36
|
+
const mcpServer = new McpServer({
|
|
37
|
+
name: 'agentlens',
|
|
38
|
+
version: '0.12.1',
|
|
39
|
+
}, {
|
|
40
|
+
capabilities: {
|
|
41
|
+
tools: {},
|
|
42
|
+
},
|
|
43
|
+
});
|
|
44
|
+
// Auto-discovery: probe server capabilities
|
|
45
|
+
const serverInfo = await transport.probeCapabilities();
|
|
46
|
+
const allowlist = parseEnvList(process.env['AGENTLENS_MCP_TOOLS']);
|
|
47
|
+
const denylist = parseEnvList(process.env['AGENTLENS_MCP_TOOLS_EXCLUDE']);
|
|
48
|
+
const registrationOpts = { serverInfo, allowlist, denylist };
|
|
49
|
+
registerTools(mcpServer, transport, registrationOpts);
|
|
50
|
+
return { mcpServer, transport };
|
|
51
|
+
}
|
|
18
52
|
/**
|
|
19
|
-
* Create and configure the MCP server (
|
|
53
|
+
* Create and configure the MCP server synchronously (no auto-discovery).
|
|
54
|
+
* Registers all tools unconditionally. Backward compatible with old createServer().
|
|
20
55
|
*/
|
|
21
|
-
export function
|
|
56
|
+
export function createServerSync() {
|
|
22
57
|
const baseUrl = process.env['AGENTLENS_URL'] ?? 'http://localhost:3400';
|
|
23
58
|
const apiKey = process.env['AGENTLENS_API_KEY'];
|
|
24
59
|
const transport = new AgentLensTransport({ baseUrl, apiKey });
|
|
25
60
|
transport.installShutdownHandlers();
|
|
26
61
|
const mcpServer = new McpServer({
|
|
27
62
|
name: 'agentlens',
|
|
28
|
-
version: '0.
|
|
63
|
+
version: '0.12.1',
|
|
29
64
|
}, {
|
|
30
65
|
capabilities: {
|
|
31
66
|
tools: {},
|
|
@@ -38,10 +73,9 @@ export function createServer() {
|
|
|
38
73
|
* Main entry point — start the MCP server on stdio.
|
|
39
74
|
*/
|
|
40
75
|
async function main() {
|
|
41
|
-
const { mcpServer } = createServer();
|
|
76
|
+
const { mcpServer } = await createServer();
|
|
42
77
|
const stdioTransport = new StdioServerTransport();
|
|
43
78
|
await mcpServer.connect(stdioTransport);
|
|
44
|
-
// Log to stderr so it doesn't interfere with MCP stdio protocol
|
|
45
79
|
process.stderr.write('AgentLens MCP server running\n');
|
|
46
80
|
}
|
|
47
81
|
// Run only when executed directly (not when imported as a library)
|
package/dist/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;;;;;;;;;GAWG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAG3C,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE3C,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAGzE,SAAS,YAAY,CAAC,KAAyB;IAC7C,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAClE,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;AACzC,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY;IAChC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,uBAAuB,CAAC;IACxE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9D,SAAS,CAAC,uBAAuB,EAAE,CAAC;IAEpC,MAAM,SAAS,GAAG,IAAI,SAAS,CAC7B;QACE,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,QAAQ;KAClB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,4CAA4C;IAC5C,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,iBAAiB,EAAE,CAAC;IAEvD,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC;IACnE,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC,CAAC;IAE1E,MAAM,gBAAgB,GAA4B,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtF,aAAa,CAAC,SAAS,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;IAEtD,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAClC,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,uBAAuB,CAAC;IACxE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;IAEhD,MAAM,SAAS,GAAG,IAAI,kBAAkB,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;IAC9D,SAAS,CAAC,uBAAuB,EAAE,CAAC;IAEpC,MAAM,SAAS,GAAG,IAAI,SAAS,CAC7B;QACE,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,QAAQ;KAClB,EACD;QACE,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE;SACV;KACF,CACF,CAAC;IAEF,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAEpC,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;AAClC,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI;IACjB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,YAAY,EAAE,CAAC;IAE3C,MAAM,cAAc,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAClD,MAAM,SAAS,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;IAExC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;AACzD,CAAC;AAED,mEAAmE;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,iBAAiB,GAAG,CAAC,GAAG,EAAE;IAC9B,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAClE,OAAO,QAAQ,KAAK,SAAS,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC,CAAC,EAAE,CAAC;AAEL,IAAI,iBAAiB,EAAE,CAAC;IACtB,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;QAC9B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;AACL,CAAC"}
|
package/dist/server.test.js
CHANGED
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
5
5
|
import { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
6
6
|
import { InMemoryTransport } from '@modelcontextprotocol/sdk/inMemory.js';
|
|
7
|
-
import { createServer } from './index.js';
|
|
7
|
+
import { createServerSync as createServer } from './index.js';
|
|
8
8
|
// Mock global fetch for the transport layer
|
|
9
9
|
const mockFetch = vi.fn();
|
|
10
10
|
beforeEach(() => {
|
|
@@ -24,33 +24,40 @@ describe('MCP Server Entrypoint (Story 5.1)', () => {
|
|
|
24
24
|
const { transport } = createServer();
|
|
25
25
|
expect(transport).toBeDefined();
|
|
26
26
|
});
|
|
27
|
-
it('lists all
|
|
27
|
+
it('lists all 24 tools via MCP protocol', async () => {
|
|
28
28
|
const { mcpServer } = createServer();
|
|
29
29
|
const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair();
|
|
30
30
|
const client = new Client({ name: 'test-client', version: '1.0.0' });
|
|
31
31
|
await mcpServer.connect(serverTransport);
|
|
32
32
|
await client.connect(clientTransport);
|
|
33
33
|
const result = await client.listTools();
|
|
34
|
-
expect(result.tools).toHaveLength(
|
|
34
|
+
expect(result.tools).toHaveLength(24);
|
|
35
35
|
const names = result.tools.map((t) => t.name).sort();
|
|
36
36
|
expect(names).toEqual([
|
|
37
|
+
'agentlens_agents',
|
|
38
|
+
'agentlens_alerts',
|
|
39
|
+
'agentlens_analytics',
|
|
37
40
|
'agentlens_benchmark',
|
|
38
|
-
'agentlens_community',
|
|
39
41
|
'agentlens_context',
|
|
42
|
+
'agentlens_cost_budgets',
|
|
40
43
|
'agentlens_delegate',
|
|
41
44
|
'agentlens_discover',
|
|
42
45
|
'agentlens_guardrails',
|
|
43
46
|
'agentlens_health',
|
|
44
|
-
'
|
|
47
|
+
'agentlens_lessons',
|
|
45
48
|
'agentlens_log_event',
|
|
46
49
|
'agentlens_log_llm_call',
|
|
47
50
|
'agentlens_optimize',
|
|
51
|
+
'agentlens_prompts',
|
|
48
52
|
'agentlens_query_events',
|
|
49
53
|
'agentlens_recall',
|
|
50
54
|
'agentlens_reflect',
|
|
51
55
|
'agentlens_replay',
|
|
52
56
|
'agentlens_session_end',
|
|
53
57
|
'agentlens_session_start',
|
|
58
|
+
'agentlens_sessions',
|
|
59
|
+
'agentlens_stats',
|
|
60
|
+
'agentlens_trust',
|
|
54
61
|
]);
|
|
55
62
|
});
|
|
56
63
|
it('all tools have descriptions and input schemas', async () => {
|
package/dist/server.test.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../src/server.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;
|
|
1
|
+
{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../src/server.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC1E,OAAO,EAAE,gBAAgB,IAAI,YAAY,EAAE,MAAM,YAAY,CAAC;AAE9D,4CAA4C;AAC5C,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE1B,UAAU,CAAC,GAAG,EAAE;IACd,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAClC,SAAS,CAAC,SAAS,EAAE,CAAC;AACxB,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACb,EAAE,CAAC,gBAAgB,EAAE,CAAC;AACxB,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACpC,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QAErC,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,SAAS,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACzC,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEtC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAEtC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YACpB,kBAAkB;YAClB,kBAAkB;YAClB,qBAAqB;YACrB,qBAAqB;YACrB,mBAAmB;YACnB,wBAAwB;YACxB,oBAAoB;YACpB,oBAAoB;YACpB,sBAAsB;YACtB,kBAAkB;YAClB,mBAAmB;YACnB,qBAAqB;YACrB,wBAAwB;YACxB,oBAAoB;YACpB,mBAAmB;YACnB,wBAAwB;YACxB,kBAAkB;YAClB,mBAAmB;YACnB,kBAAkB;YAClB,uBAAuB;YACvB,yBAAyB;YACzB,oBAAoB;YACpB,iBAAiB;YACjB,iBAAiB;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QAErC,MAAM,CAAC,eAAe,EAAE,eAAe,CAAC,GAAG,iBAAiB,CAAC,gBAAgB,EAAE,CAAC;QAChF,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAErE,MAAM,SAAS,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QACzC,MAAM,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;QAEtC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QACxC,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,oBAAoB,CAAC;QACpD,MAAM,EAAE,SAAS,EAAE,GAAG,YAAY,EAAE,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;QAChC,6BAA6B;QAC7B,SAAS,CAAC,iBAAiB,CACzB,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CAAC,CACrF,CAAC;QACF,KAAK,SAAS,CAAC,kBAAkB,CAAC;YAChC,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,GAAG;YACZ,SAAS,EAAE,QAAQ;YACnB,OAAO,EAAE,EAAE;SACZ,CAAC,CAAC;QACH,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,+BAA+B,EAC/B,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACF,OAAO,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agents.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/agents.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tests for agentlens_agents MCP Tool (Feature 10, Story 10.7)
|
|
3
|
+
*/
|
|
4
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
5
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
6
|
+
import { AgentLensTransport } from '../../transport.js';
|
|
7
|
+
import { registerAgentsTool } from '../agents.js';
|
|
8
|
+
function mockResponse(data, ok = true, status = 200) {
|
|
9
|
+
return { ok, status, json: () => Promise.resolve(data), text: () => Promise.resolve(JSON.stringify(data)) };
|
|
10
|
+
}
|
|
11
|
+
describe('agentlens_agents', () => {
|
|
12
|
+
let transport;
|
|
13
|
+
let toolHandler;
|
|
14
|
+
beforeEach(() => {
|
|
15
|
+
const server = new McpServer({ name: 'test', version: '0.0.0' }, { capabilities: { tools: {} } });
|
|
16
|
+
transport = new AgentLensTransport({ baseUrl: 'http://localhost:3400' });
|
|
17
|
+
const origTool = server.tool.bind(server);
|
|
18
|
+
vi.spyOn(server, 'tool').mockImplementation((...args) => {
|
|
19
|
+
toolHandler = args[args.length - 1];
|
|
20
|
+
return origTool(...args);
|
|
21
|
+
});
|
|
22
|
+
registerAgentsTool(server, transport);
|
|
23
|
+
});
|
|
24
|
+
it('lists agents', async () => {
|
|
25
|
+
vi.spyOn(transport, 'listAgents').mockResolvedValue(mockResponse({ agents: [] }));
|
|
26
|
+
const result = await toolHandler({ action: 'list' });
|
|
27
|
+
expect(result.isError).toBeUndefined();
|
|
28
|
+
});
|
|
29
|
+
it('gets agent detail', async () => {
|
|
30
|
+
vi.spyOn(transport, 'getAgent').mockResolvedValue(mockResponse({ id: 'a1' }));
|
|
31
|
+
const result = await toolHandler({ action: 'detail', agentId: 'a1' });
|
|
32
|
+
expect(result.isError).toBeUndefined();
|
|
33
|
+
});
|
|
34
|
+
it('requires agentId for detail', async () => {
|
|
35
|
+
const result = await toolHandler({ action: 'detail' });
|
|
36
|
+
expect(result.isError).toBe(true);
|
|
37
|
+
expect(result.content[0].text).toContain('agentId');
|
|
38
|
+
});
|
|
39
|
+
it('unpauses agent', async () => {
|
|
40
|
+
vi.spyOn(transport, 'unpauseAgent').mockResolvedValue(mockResponse({ ok: true }));
|
|
41
|
+
const result = await toolHandler({ action: 'unpause', agentId: 'a1', clearModelOverride: true });
|
|
42
|
+
expect(result.isError).toBeUndefined();
|
|
43
|
+
expect(transport.unpauseAgent).toHaveBeenCalledWith('a1', true);
|
|
44
|
+
});
|
|
45
|
+
it('requires agentId for unpause', async () => {
|
|
46
|
+
const result = await toolHandler({ action: 'unpause' });
|
|
47
|
+
expect(result.isError).toBe(true);
|
|
48
|
+
});
|
|
49
|
+
it('handles 404', async () => {
|
|
50
|
+
vi.spyOn(transport, 'getAgent').mockResolvedValue(mockResponse('Not found', false, 404));
|
|
51
|
+
const result = await toolHandler({ action: 'detail', agentId: 'x' });
|
|
52
|
+
expect(result.isError).toBe(true);
|
|
53
|
+
expect(result.content[0].text).toContain('newer AgentLens server version');
|
|
54
|
+
});
|
|
55
|
+
});
|
|
56
|
+
//# sourceMappingURL=agents.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agents.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/agents.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,SAAS,YAAY,CAAC,IAAa,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,GAAG,GAAG;IAC1D,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAc,CAAC;AAC1H,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,SAA6B,CAAC;IAClC,IAAI,WAA8H,CAAC;IAEnI,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAClG,SAAS,GAAG,IAAI,kBAAkB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACzE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;YACjE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAuB,CAAC;YAC1D,OAAO,QAAQ,CAAC,GAAI,IAAoC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QACH,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;QAC5B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC9E,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAClF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QACjG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC3B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QACzF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"alerts.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/alerts.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tests for agentlens_alerts MCP Tool (Feature 10, Story 10.8)
|
|
3
|
+
*/
|
|
4
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
5
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
6
|
+
import { AgentLensTransport } from '../../transport.js';
|
|
7
|
+
import { registerAlertsTool } from '../alerts.js';
|
|
8
|
+
function mockResponse(data, ok = true, status = 200) {
|
|
9
|
+
return { ok, status, json: () => Promise.resolve(data), text: () => Promise.resolve(JSON.stringify(data)) };
|
|
10
|
+
}
|
|
11
|
+
describe('agentlens_alerts', () => {
|
|
12
|
+
let transport;
|
|
13
|
+
let toolHandler;
|
|
14
|
+
beforeEach(() => {
|
|
15
|
+
const server = new McpServer({ name: 'test', version: '0.0.0' }, { capabilities: { tools: {} } });
|
|
16
|
+
transport = new AgentLensTransport({ baseUrl: 'http://localhost:3400' });
|
|
17
|
+
const origTool = server.tool.bind(server);
|
|
18
|
+
vi.spyOn(server, 'tool').mockImplementation((...args) => {
|
|
19
|
+
toolHandler = args[args.length - 1];
|
|
20
|
+
return origTool(...args);
|
|
21
|
+
});
|
|
22
|
+
registerAlertsTool(server, transport);
|
|
23
|
+
});
|
|
24
|
+
it('lists alert rules', async () => {
|
|
25
|
+
vi.spyOn(transport, 'listAlertRules').mockResolvedValue(mockResponse({ rules: [] }));
|
|
26
|
+
const result = await toolHandler({ action: 'list' });
|
|
27
|
+
expect(result.isError).toBeUndefined();
|
|
28
|
+
});
|
|
29
|
+
it('creates alert rule', async () => {
|
|
30
|
+
vi.spyOn(transport, 'createAlertRule').mockResolvedValue(mockResponse({ id: 'r1' }));
|
|
31
|
+
const result = await toolHandler({
|
|
32
|
+
action: 'create', name: 'High errors', condition: 'error_rate_above', threshold: 0.1, windowMinutes: 60,
|
|
33
|
+
});
|
|
34
|
+
expect(result.isError).toBeUndefined();
|
|
35
|
+
expect(transport.createAlertRule).toHaveBeenCalledWith(expect.objectContaining({ name: 'High errors', condition: 'error_rate_above', threshold: 0.1 }));
|
|
36
|
+
});
|
|
37
|
+
it('validates create requires name', async () => {
|
|
38
|
+
const result = await toolHandler({ action: 'create', condition: 'error_rate_above', threshold: 0.1 });
|
|
39
|
+
expect(result.isError).toBe(true);
|
|
40
|
+
expect(result.content[0].text).toContain('name');
|
|
41
|
+
});
|
|
42
|
+
it('validates create requires condition', async () => {
|
|
43
|
+
const result = await toolHandler({ action: 'create', name: 'Test', threshold: 0.1 });
|
|
44
|
+
expect(result.isError).toBe(true);
|
|
45
|
+
expect(result.content[0].text).toContain('condition');
|
|
46
|
+
});
|
|
47
|
+
it('validates create requires threshold', async () => {
|
|
48
|
+
const result = await toolHandler({ action: 'create', name: 'Test', condition: 'error_rate_above' });
|
|
49
|
+
expect(result.isError).toBe(true);
|
|
50
|
+
expect(result.content[0].text).toContain('threshold');
|
|
51
|
+
});
|
|
52
|
+
it('updates alert rule', async () => {
|
|
53
|
+
vi.spyOn(transport, 'updateAlertRule').mockResolvedValue(mockResponse({ id: 'r1' }));
|
|
54
|
+
const result = await toolHandler({ action: 'update', ruleId: 'r1', enabled: false });
|
|
55
|
+
expect(result.isError).toBeUndefined();
|
|
56
|
+
});
|
|
57
|
+
it('validates update requires ruleId', async () => {
|
|
58
|
+
const result = await toolHandler({ action: 'update', enabled: false });
|
|
59
|
+
expect(result.isError).toBe(true);
|
|
60
|
+
});
|
|
61
|
+
it('deletes alert rule', async () => {
|
|
62
|
+
vi.spyOn(transport, 'deleteAlertRule').mockResolvedValue(mockResponse(null));
|
|
63
|
+
const result = await toolHandler({ action: 'delete', ruleId: 'r1' });
|
|
64
|
+
expect(result.isError).toBeUndefined();
|
|
65
|
+
expect(result.content[0].text).toContain('r1');
|
|
66
|
+
});
|
|
67
|
+
it('validates delete requires ruleId', async () => {
|
|
68
|
+
const result = await toolHandler({ action: 'delete' });
|
|
69
|
+
expect(result.isError).toBe(true);
|
|
70
|
+
});
|
|
71
|
+
it('gets alert history', async () => {
|
|
72
|
+
vi.spyOn(transport, 'getAlertHistory').mockResolvedValue(mockResponse({ history: [] }));
|
|
73
|
+
const result = await toolHandler({ action: 'history', limit: 10 });
|
|
74
|
+
expect(result.isError).toBeUndefined();
|
|
75
|
+
});
|
|
76
|
+
it('handles 404', async () => {
|
|
77
|
+
vi.spyOn(transport, 'listAlertRules').mockResolvedValue(mockResponse('Not found', false, 404));
|
|
78
|
+
const result = await toolHandler({ action: 'list' });
|
|
79
|
+
expect(result.isError).toBe(true);
|
|
80
|
+
expect(result.content[0].text).toContain('newer AgentLens server version');
|
|
81
|
+
});
|
|
82
|
+
});
|
|
83
|
+
//# sourceMappingURL=alerts.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"alerts.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/alerts.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,SAAS,YAAY,CAAC,IAAa,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,GAAG,GAAG;IAC1D,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAc,CAAC;AAC1H,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,SAA6B,CAAC;IAClC,IAAI,WAA8H,CAAC;IAEnI,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAClG,SAAS,GAAG,IAAI,kBAAkB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACzE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;YACjE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAuB,CAAC;YAC1D,OAAO,QAAQ,CAAC,GAAI,IAAoC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QACH,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC;YAC/B,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE;SACxG,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;IAC1J,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,kBAAkB,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QACtG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QACrF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACpG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACrF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7E,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACxF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC3B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QAC/F,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"analytics.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/analytics.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Tests for agentlens_analytics MCP Tool (Feature 10, Story 10.9)
|
|
3
|
+
*/
|
|
4
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
5
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
6
|
+
import { AgentLensTransport } from '../../transport.js';
|
|
7
|
+
import { registerAnalyticsTool } from '../analytics.js';
|
|
8
|
+
function mockResponse(data, ok = true, status = 200) {
|
|
9
|
+
return { ok, status, json: () => Promise.resolve(data), text: () => Promise.resolve(JSON.stringify(data)) };
|
|
10
|
+
}
|
|
11
|
+
describe('agentlens_analytics', () => {
|
|
12
|
+
let transport;
|
|
13
|
+
let toolHandler;
|
|
14
|
+
beforeEach(() => {
|
|
15
|
+
const server = new McpServer({ name: 'test', version: '0.0.0' }, { capabilities: { tools: {} } });
|
|
16
|
+
transport = new AgentLensTransport({ baseUrl: 'http://localhost:3400' });
|
|
17
|
+
const origTool = server.tool.bind(server);
|
|
18
|
+
vi.spyOn(server, 'tool').mockImplementation((...args) => {
|
|
19
|
+
toolHandler = args[args.length - 1];
|
|
20
|
+
return origTool(...args);
|
|
21
|
+
});
|
|
22
|
+
registerAnalyticsTool(server, transport);
|
|
23
|
+
});
|
|
24
|
+
it('gets metrics', async () => {
|
|
25
|
+
vi.spyOn(transport, 'getAnalytics').mockResolvedValue(mockResponse({ buckets: [] }));
|
|
26
|
+
const result = await toolHandler({ action: 'metrics', range: '24h' });
|
|
27
|
+
expect(result.isError).toBeUndefined();
|
|
28
|
+
expect(transport.getAnalytics).toHaveBeenCalledWith(expect.objectContaining({ range: '24h' }));
|
|
29
|
+
});
|
|
30
|
+
it('gets costs', async () => {
|
|
31
|
+
vi.spyOn(transport, 'getAnalyticsCosts').mockResolvedValue(mockResponse({ costs: {} }));
|
|
32
|
+
const result = await toolHandler({ action: 'costs' });
|
|
33
|
+
expect(result.isError).toBeUndefined();
|
|
34
|
+
});
|
|
35
|
+
it('gets agent metrics with date range', async () => {
|
|
36
|
+
vi.spyOn(transport, 'getAnalyticsAgents').mockResolvedValue(mockResponse({ agents: [] }));
|
|
37
|
+
const result = await toolHandler({ action: 'agents', from: '2026-01-01', to: '2026-02-01' });
|
|
38
|
+
expect(result.isError).toBeUndefined();
|
|
39
|
+
expect(transport.getAnalyticsAgents).toHaveBeenCalledWith(expect.objectContaining({ from: '2026-01-01', to: '2026-02-01' }));
|
|
40
|
+
});
|
|
41
|
+
it('gets tool usage', async () => {
|
|
42
|
+
vi.spyOn(transport, 'getAnalyticsTools').mockResolvedValue(mockResponse({ tools: [] }));
|
|
43
|
+
const result = await toolHandler({ action: 'tools' });
|
|
44
|
+
expect(result.isError).toBeUndefined();
|
|
45
|
+
});
|
|
46
|
+
it('handles 404', async () => {
|
|
47
|
+
vi.spyOn(transport, 'getAnalytics').mockResolvedValue(mockResponse('Not found', false, 404));
|
|
48
|
+
const result = await toolHandler({ action: 'metrics' });
|
|
49
|
+
expect(result.isError).toBe(true);
|
|
50
|
+
expect(result.content[0].text).toContain('newer AgentLens server version');
|
|
51
|
+
});
|
|
52
|
+
});
|
|
53
|
+
//# sourceMappingURL=analytics.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"analytics.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/analytics.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AAExD,SAAS,YAAY,CAAC,IAAa,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,GAAG,GAAG;IAC1D,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAc,CAAC;AAC1H,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,SAA6B,CAAC;IAClC,IAAI,WAA8H,CAAC;IAEnI,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAClG,SAAS,GAAG,IAAI,kBAAkB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE,CAAC,CAAC;QACzE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;YACjE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAuB,CAAC;YAC1D,OAAO,QAAQ,CAAC,GAAI,IAAoC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QACH,qBAAqB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;QAC5B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IACjG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;QAC1B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACxF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1F,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;QAC7F,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;IAC/H,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;QAC/B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QACxF,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;QAC3B,EAAE,CAAC,KAAK,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC,iBAAiB,CAAC,YAAY,CAAC,WAAW,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;QAC7F,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gCAAgC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-budgets.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/cost-budgets.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}
|