@a5c-ai/tools-adapter 5.1.1-staging.00ceebd28cf2
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/README.md +57 -0
- package/dist/__tests__/dispatch.test.d.ts +2 -0
- package/dist/__tests__/dispatch.test.d.ts.map +1 -0
- package/dist/__tests__/dispatch.test.js +251 -0
- package/dist/__tests__/dispatch.test.js.map +1 -0
- package/dist/__tests__/hooks.test.d.ts +2 -0
- package/dist/__tests__/hooks.test.d.ts.map +1 -0
- package/dist/__tests__/hooks.test.js +172 -0
- package/dist/__tests__/hooks.test.js.map +1 -0
- package/dist/__tests__/mcp-bridge.test.d.ts +2 -0
- package/dist/__tests__/mcp-bridge.test.d.ts.map +1 -0
- package/dist/__tests__/mcp-bridge.test.js +129 -0
- package/dist/__tests__/mcp-bridge.test.js.map +1 -0
- package/dist/__tests__/registry.test.d.ts +2 -0
- package/dist/__tests__/registry.test.d.ts.map +1 -0
- package/dist/__tests__/registry.test.js +196 -0
- package/dist/__tests__/registry.test.js.map +1 -0
- package/dist/__tests__/schema-translation.test.d.ts +2 -0
- package/dist/__tests__/schema-translation.test.d.ts.map +1 -0
- package/dist/__tests__/schema-translation.test.js +126 -0
- package/dist/__tests__/schema-translation.test.js.map +1 -0
- package/dist/__tests__/unified-contract.test.d.ts +2 -0
- package/dist/__tests__/unified-contract.test.d.ts.map +1 -0
- package/dist/__tests__/unified-contract.test.js +98 -0
- package/dist/__tests__/unified-contract.test.js.map +1 -0
- package/dist/dispatch.d.ts +43 -0
- package/dist/dispatch.d.ts.map +1 -0
- package/dist/dispatch.js +157 -0
- package/dist/dispatch.js.map +1 -0
- package/dist/hooks.d.ts +99 -0
- package/dist/hooks.d.ts.map +1 -0
- package/dist/hooks.js +110 -0
- package/dist/hooks.js.map +1 -0
- package/dist/index.d.ts +13 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +25 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp-bridge.d.ts +54 -0
- package/dist/mcp-bridge.d.ts.map +1 -0
- package/dist/mcp-bridge.js +78 -0
- package/dist/mcp-bridge.js.map +1 -0
- package/dist/registry.d.ts +70 -0
- package/dist/registry.d.ts.map +1 -0
- package/dist/registry.js +221 -0
- package/dist/registry.js.map +1 -0
- package/dist/schema-translation.d.ts +38 -0
- package/dist/schema-translation.d.ts.map +1 -0
- package/dist/schema-translation.js +61 -0
- package/dist/schema-translation.js.map +1 -0
- package/dist/types.d.ts +159 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +39 -0
- package/dist/types.js.map +1 -0
- package/package.json +61 -0
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
import { describe, expect, it, beforeEach } from 'vitest';
|
|
2
|
+
import { McpBridge } from '../mcp-bridge.js';
|
|
3
|
+
import { ToolRegistry } from '../registry.js';
|
|
4
|
+
/* -------------------------------------------------------------------------- */
|
|
5
|
+
/* Helpers */
|
|
6
|
+
/* -------------------------------------------------------------------------- */
|
|
7
|
+
function makeServerConfig(overrides = {}) {
|
|
8
|
+
return {
|
|
9
|
+
id: 'mcp-srv-1',
|
|
10
|
+
name: 'Test MCP Server',
|
|
11
|
+
transport: 'stdio',
|
|
12
|
+
command: 'node',
|
|
13
|
+
args: ['server.js'],
|
|
14
|
+
...overrides,
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
function makeMcpTool(overrides = {}) {
|
|
18
|
+
return {
|
|
19
|
+
name: 'mcp_tool',
|
|
20
|
+
description: 'An MCP tool',
|
|
21
|
+
inputSchema: {
|
|
22
|
+
type: 'object',
|
|
23
|
+
properties: {
|
|
24
|
+
query: { type: 'string' },
|
|
25
|
+
},
|
|
26
|
+
},
|
|
27
|
+
...overrides,
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
/* ========================================================================== */
|
|
31
|
+
/* McpBridge */
|
|
32
|
+
/* ========================================================================== */
|
|
33
|
+
describe('McpBridge', () => {
|
|
34
|
+
let registry;
|
|
35
|
+
let bridge;
|
|
36
|
+
beforeEach(() => {
|
|
37
|
+
registry = new ToolRegistry();
|
|
38
|
+
bridge = new McpBridge(registry);
|
|
39
|
+
});
|
|
40
|
+
it('registerServer adds tools to the underlying registry', () => {
|
|
41
|
+
const config = makeServerConfig({ id: 'srv-a', name: 'Server A' });
|
|
42
|
+
const tools = [
|
|
43
|
+
makeMcpTool({ name: 'search' }),
|
|
44
|
+
makeMcpTool({ name: 'fetch' }),
|
|
45
|
+
];
|
|
46
|
+
bridge.registerServer(config, tools);
|
|
47
|
+
expect(registry.has('search')).toBe(true);
|
|
48
|
+
expect(registry.has('fetch')).toBe(true);
|
|
49
|
+
expect(registry.size).toBe(2);
|
|
50
|
+
});
|
|
51
|
+
it('unregisterServer removes tools from the registry', () => {
|
|
52
|
+
const config = makeServerConfig({ id: 'srv-b' });
|
|
53
|
+
bridge.registerServer(config, [makeMcpTool({ name: 'tool_to_remove' })]);
|
|
54
|
+
expect(registry.has('tool_to_remove')).toBe(true);
|
|
55
|
+
bridge.unregisterServer('srv-b');
|
|
56
|
+
expect(registry.has('tool_to_remove')).toBe(false);
|
|
57
|
+
expect(registry.getServer('srv-b')).toBeUndefined();
|
|
58
|
+
});
|
|
59
|
+
it('listServers returns all registered MCP server configs', () => {
|
|
60
|
+
bridge.registerServer(makeServerConfig({ id: 's1', name: 'S1' }), []);
|
|
61
|
+
bridge.registerServer(makeServerConfig({ id: 's2', name: 'S2' }), []);
|
|
62
|
+
const servers = bridge.listServers();
|
|
63
|
+
expect(servers).toHaveLength(2);
|
|
64
|
+
expect(servers.map((s) => s.id).sort()).toEqual(['s1', 's2']);
|
|
65
|
+
});
|
|
66
|
+
it('listServers returns empty array when nothing is registered', () => {
|
|
67
|
+
expect(bridge.listServers()).toEqual([]);
|
|
68
|
+
});
|
|
69
|
+
it('getServerTools returns the correct tools for a server', () => {
|
|
70
|
+
const config = makeServerConfig({ id: 'query-srv' });
|
|
71
|
+
bridge.registerServer(config, [
|
|
72
|
+
makeMcpTool({ name: 'q_search' }),
|
|
73
|
+
makeMcpTool({ name: 'q_list' }),
|
|
74
|
+
]);
|
|
75
|
+
const tools = bridge.getServerTools('query-srv');
|
|
76
|
+
expect(tools).toHaveLength(2);
|
|
77
|
+
expect(tools.map((t) => t.name).sort()).toEqual(['q_list', 'q_search']);
|
|
78
|
+
});
|
|
79
|
+
it('getServerTools returns empty array for unknown server', () => {
|
|
80
|
+
expect(bridge.getServerTools('ghost-server')).toEqual([]);
|
|
81
|
+
});
|
|
82
|
+
/* ---------------------------------------------------------------------- */
|
|
83
|
+
/* mcpToolToDescriptor (static) */
|
|
84
|
+
/* ---------------------------------------------------------------------- */
|
|
85
|
+
describe('mcpToolToDescriptor', () => {
|
|
86
|
+
it('sets source to "mcp" and correct server id', () => {
|
|
87
|
+
const mcpTool = makeMcpTool({ name: 'convert_me' });
|
|
88
|
+
const descriptor = McpBridge.mcpToolToDescriptor(mcpTool, 'my-server');
|
|
89
|
+
expect(descriptor.name).toBe('convert_me');
|
|
90
|
+
expect(descriptor.source).toBe('mcp');
|
|
91
|
+
expect(descriptor.server).toBe('my-server');
|
|
92
|
+
});
|
|
93
|
+
it('maps inputSchema to parameters', () => {
|
|
94
|
+
const schema = {
|
|
95
|
+
type: 'object',
|
|
96
|
+
properties: {
|
|
97
|
+
path: { type: 'string' },
|
|
98
|
+
recursive: { type: 'boolean' },
|
|
99
|
+
},
|
|
100
|
+
required: ['path'],
|
|
101
|
+
};
|
|
102
|
+
const mcpTool = makeMcpTool({
|
|
103
|
+
name: 'file_list',
|
|
104
|
+
inputSchema: schema,
|
|
105
|
+
});
|
|
106
|
+
const descriptor = McpBridge.mcpToolToDescriptor(mcpTool, 'fs-server');
|
|
107
|
+
expect(descriptor.parameters).toEqual(schema);
|
|
108
|
+
});
|
|
109
|
+
it('preserves description from MCP tool definition', () => {
|
|
110
|
+
const mcpTool = makeMcpTool({
|
|
111
|
+
name: 'described_tool',
|
|
112
|
+
description: 'Does important things',
|
|
113
|
+
});
|
|
114
|
+
const descriptor = McpBridge.mcpToolToDescriptor(mcpTool, 'srv');
|
|
115
|
+
expect(descriptor.description).toBe('Does important things');
|
|
116
|
+
});
|
|
117
|
+
it('handles MCP tool with no inputSchema', () => {
|
|
118
|
+
const mcpTool = {
|
|
119
|
+
name: 'no_schema',
|
|
120
|
+
description: 'Schemaless',
|
|
121
|
+
};
|
|
122
|
+
const descriptor = McpBridge.mcpToolToDescriptor(mcpTool, 'srv');
|
|
123
|
+
expect(descriptor.name).toBe('no_schema');
|
|
124
|
+
expect(descriptor.parameters).toBeUndefined();
|
|
125
|
+
expect(descriptor.source).toBe('mcp');
|
|
126
|
+
});
|
|
127
|
+
});
|
|
128
|
+
});
|
|
129
|
+
//# sourceMappingURL=mcp-bridge.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcp-bridge.test.js","sourceRoot":"","sources":["../../src/__tests__/mcp-bridge.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE1D,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAE7C,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,SAAS,gBAAgB,CAAC,YAAsC,EAAE;IAChE,OAAO;QACL,EAAE,EAAE,WAAW;QACf,IAAI,EAAE,iBAAiB;QACvB,SAAS,EAAE,OAAO;QAClB,OAAO,EAAE,MAAM;QACf,IAAI,EAAE,CAAC,WAAW,CAAC;QACnB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,YAAwC,EAAE;IAC7D,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,WAAW,EAAE,aAAa;QAC1B,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B;SACF;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,IAAI,QAAsB,CAAC;IAC3B,IAAI,MAAiB,CAAC;IAEtB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QAC9B,MAAM,GAAG,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QACnE,MAAM,KAAK,GAAG;YACZ,WAAW,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAC/B,WAAW,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;SAC/B,CAAC;QAEF,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAErC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,WAAW,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,CAAC;QAEzE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAElD,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAEjC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QACtE,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE;YAC5B,WAAW,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACjC,WAAW,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;SAChC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,OAAO,GAAG,WAAW,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;YAEpD,MAAM,UAAU,GAAG,SAAS,CAAC,mBAAmB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YAEvE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC3C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBACxB,SAAS,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;iBAC/B;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB,CAAC;YAEF,MAAM,OAAO,GAAG,WAAW,CAAC;gBAC1B,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,MAAiC;aAC/C,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,SAAS,CAAC,mBAAmB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YAEvE,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAAG,WAAW,CAAC;gBAC1B,IAAI,EAAE,gBAAgB;gBACtB,WAAW,EAAE,uBAAuB;aACrC,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,SAAS,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAEjE,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAAsB;gBACjC,IAAI,EAAE,WAAW;gBACjB,WAAW,EAAE,YAAY;aAC1B,CAAC;YAEF,MAAM,UAAU,GAAG,SAAS,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAEjE,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/registry.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,196 @@
|
|
|
1
|
+
import { describe, expect, it, beforeEach } from 'vitest';
|
|
2
|
+
import { ToolRegistry } from '../registry.js';
|
|
3
|
+
/* -------------------------------------------------------------------------- */
|
|
4
|
+
/* Helpers */
|
|
5
|
+
/* -------------------------------------------------------------------------- */
|
|
6
|
+
function makeTool(overrides = {}) {
|
|
7
|
+
return {
|
|
8
|
+
name: 'test_tool',
|
|
9
|
+
description: 'A test tool',
|
|
10
|
+
parameters: { type: 'object', properties: {} },
|
|
11
|
+
source: 'builtin',
|
|
12
|
+
...overrides,
|
|
13
|
+
};
|
|
14
|
+
}
|
|
15
|
+
function makeServer(id, tools, overrides = {}) {
|
|
16
|
+
return {
|
|
17
|
+
id,
|
|
18
|
+
name: `Server ${id}`,
|
|
19
|
+
type: 'native',
|
|
20
|
+
tools,
|
|
21
|
+
...overrides,
|
|
22
|
+
};
|
|
23
|
+
}
|
|
24
|
+
/* ========================================================================== */
|
|
25
|
+
/* ToolRegistry */
|
|
26
|
+
/* ========================================================================== */
|
|
27
|
+
describe('ToolRegistry', () => {
|
|
28
|
+
let registry;
|
|
29
|
+
beforeEach(() => {
|
|
30
|
+
registry = new ToolRegistry();
|
|
31
|
+
});
|
|
32
|
+
/* ---------------------------------------------------------------------- */
|
|
33
|
+
/* Tool-level operations */
|
|
34
|
+
/* ---------------------------------------------------------------------- */
|
|
35
|
+
it('registers a tool and retrieves it by name', () => {
|
|
36
|
+
const tool = makeTool({ name: 'file_read' });
|
|
37
|
+
registry.register(tool);
|
|
38
|
+
const result = registry.get('file_read');
|
|
39
|
+
expect(result).toBeDefined();
|
|
40
|
+
expect(result.name).toBe('file_read');
|
|
41
|
+
expect(result.source).toBe('builtin');
|
|
42
|
+
});
|
|
43
|
+
it('registers multiple tools and lists all of them', () => {
|
|
44
|
+
registry.register(makeTool({ name: 'tool_a' }));
|
|
45
|
+
registry.register(makeTool({ name: 'tool_b' }));
|
|
46
|
+
registry.register(makeTool({ name: 'tool_c' }));
|
|
47
|
+
const all = registry.list();
|
|
48
|
+
expect(all).toHaveLength(3);
|
|
49
|
+
expect(all.map((t) => t.name).sort()).toEqual(['tool_a', 'tool_b', 'tool_c']);
|
|
50
|
+
});
|
|
51
|
+
it('registerAll registers an array of tools at once', () => {
|
|
52
|
+
registry.registerAll([
|
|
53
|
+
makeTool({ name: 'batch_a' }),
|
|
54
|
+
makeTool({ name: 'batch_b' }),
|
|
55
|
+
]);
|
|
56
|
+
expect(registry.size).toBe(2);
|
|
57
|
+
expect(registry.has('batch_a')).toBe(true);
|
|
58
|
+
expect(registry.has('batch_b')).toBe(true);
|
|
59
|
+
});
|
|
60
|
+
it('unregisters a tool and returns true when it existed', () => {
|
|
61
|
+
registry.register(makeTool({ name: 'doomed' }));
|
|
62
|
+
expect(registry.has('doomed')).toBe(true);
|
|
63
|
+
const removed = registry.unregister('doomed');
|
|
64
|
+
expect(removed).toBe(true);
|
|
65
|
+
expect(registry.has('doomed')).toBe(false);
|
|
66
|
+
expect(registry.get('doomed')).toBeUndefined();
|
|
67
|
+
});
|
|
68
|
+
it('unregister returns false for a non-existent tool', () => {
|
|
69
|
+
expect(registry.unregister('ghost')).toBe(false);
|
|
70
|
+
});
|
|
71
|
+
it('returns undefined for an unknown tool name', () => {
|
|
72
|
+
expect(registry.get('does_not_exist')).toBeUndefined();
|
|
73
|
+
});
|
|
74
|
+
it('has() returns false for missing tools', () => {
|
|
75
|
+
expect(registry.has('missing')).toBe(false);
|
|
76
|
+
});
|
|
77
|
+
it('listByServer filters tools by server id', () => {
|
|
78
|
+
registry.register(makeTool({ name: 'mcp_a', server: 'srv-1', source: 'mcp' }));
|
|
79
|
+
registry.register(makeTool({ name: 'mcp_b', server: 'srv-1', source: 'mcp' }));
|
|
80
|
+
registry.register(makeTool({ name: 'local_a', server: 'srv-2', source: 'builtin' }));
|
|
81
|
+
registry.register(makeTool({ name: 'no_server', source: 'custom' }));
|
|
82
|
+
const srv1Tools = registry.listByServer('srv-1');
|
|
83
|
+
expect(srv1Tools).toHaveLength(2);
|
|
84
|
+
expect(srv1Tools.map((t) => t.name).sort()).toEqual(['mcp_a', 'mcp_b']);
|
|
85
|
+
const srv2Tools = registry.listByServer('srv-2');
|
|
86
|
+
expect(srv2Tools).toHaveLength(1);
|
|
87
|
+
expect(srv2Tools[0].name).toBe('local_a');
|
|
88
|
+
expect(registry.listByServer('unknown')).toHaveLength(0);
|
|
89
|
+
});
|
|
90
|
+
it('duplicate name overwrites the previous descriptor', () => {
|
|
91
|
+
registry.register(makeTool({ name: 'dup', description: 'original' }));
|
|
92
|
+
expect(registry.get('dup').description).toBe('original');
|
|
93
|
+
registry.register(makeTool({ name: 'dup', description: 'replaced' }));
|
|
94
|
+
expect(registry.get('dup').description).toBe('replaced');
|
|
95
|
+
expect(registry.size).toBe(1);
|
|
96
|
+
});
|
|
97
|
+
it('keeps duplicate tool names isolated by source qualifier', () => {
|
|
98
|
+
registry.register(makeTool({ name: 'read', source: 'mcp', sourceQualifier: 'server-a', description: 'A' }));
|
|
99
|
+
registry.register(makeTool({ name: 'read', source: 'mcp', sourceQualifier: 'server-b', description: 'B' }));
|
|
100
|
+
expect(registry.size).toBe(2);
|
|
101
|
+
expect(registry.get('read', 'mcp', 'server-a').description).toBe('A');
|
|
102
|
+
expect(registry.get('read', 'mcp', 'server-b').description).toBe('B');
|
|
103
|
+
});
|
|
104
|
+
it('searches lightweight entries and lazily fetches schemas', async () => {
|
|
105
|
+
registry.registerLoader('plugin', async (entry) => ({
|
|
106
|
+
inputSchema: { type: 'object', properties: { plugin: { const: entry.sourceQualifier } } },
|
|
107
|
+
outputSchema: { type: 'object' },
|
|
108
|
+
}));
|
|
109
|
+
registry.registerTools([
|
|
110
|
+
{
|
|
111
|
+
name: 'deploy',
|
|
112
|
+
description: 'Deploy a plugin',
|
|
113
|
+
source: 'plugin',
|
|
114
|
+
sourceQualifier: 'shipper',
|
|
115
|
+
metadata: { tags: ['release'] },
|
|
116
|
+
},
|
|
117
|
+
]);
|
|
118
|
+
expect(registry.searchTools('deploy')).toEqual([
|
|
119
|
+
expect.objectContaining({ name: 'deploy', source: 'plugin', sourceQualifier: 'shipper' }),
|
|
120
|
+
]);
|
|
121
|
+
const resolved = await registry.fetchSchema('deploy', 'plugin', 'shipper');
|
|
122
|
+
expect(resolved).toEqual(expect.objectContaining({
|
|
123
|
+
name: 'deploy',
|
|
124
|
+
schema: {
|
|
125
|
+
inputSchema: { type: 'object', properties: { plugin: { const: 'shipper' } } },
|
|
126
|
+
outputSchema: { type: 'object' },
|
|
127
|
+
},
|
|
128
|
+
}));
|
|
129
|
+
expect(registry.loadedSchemaCount).toBe(1);
|
|
130
|
+
});
|
|
131
|
+
it('size reflects the current tool count', () => {
|
|
132
|
+
expect(registry.size).toBe(0);
|
|
133
|
+
registry.register(makeTool({ name: 'a' }));
|
|
134
|
+
expect(registry.size).toBe(1);
|
|
135
|
+
registry.register(makeTool({ name: 'b' }));
|
|
136
|
+
expect(registry.size).toBe(2);
|
|
137
|
+
registry.unregister('a');
|
|
138
|
+
expect(registry.size).toBe(1);
|
|
139
|
+
});
|
|
140
|
+
/* ---------------------------------------------------------------------- */
|
|
141
|
+
/* Server-level operations */
|
|
142
|
+
/* ---------------------------------------------------------------------- */
|
|
143
|
+
it('registerServer adds server entry and all its tools', () => {
|
|
144
|
+
const tools = [
|
|
145
|
+
makeTool({ name: 'srv_tool_a', source: 'mcp' }),
|
|
146
|
+
makeTool({ name: 'srv_tool_b', source: 'mcp' }),
|
|
147
|
+
];
|
|
148
|
+
const server = makeServer('s1', tools);
|
|
149
|
+
registry.registerServer(server);
|
|
150
|
+
// Server is retrievable
|
|
151
|
+
expect(registry.getServer('s1')).toBeDefined();
|
|
152
|
+
expect(registry.getServer('s1').name).toBe('Server s1');
|
|
153
|
+
// All tools are registered with server association
|
|
154
|
+
expect(registry.has('srv_tool_a')).toBe(true);
|
|
155
|
+
expect(registry.has('srv_tool_b')).toBe(true);
|
|
156
|
+
expect(registry.get('srv_tool_a').server).toBe('s1');
|
|
157
|
+
expect(registry.get('srv_tool_b').server).toBe('s1');
|
|
158
|
+
});
|
|
159
|
+
it('unregisterServer removes server and all its tools by default', () => {
|
|
160
|
+
const tools = [
|
|
161
|
+
makeTool({ name: 'remove_a', source: 'mcp' }),
|
|
162
|
+
makeTool({ name: 'remove_b', source: 'mcp' }),
|
|
163
|
+
];
|
|
164
|
+
registry.registerServer(makeServer('s2', tools));
|
|
165
|
+
expect(registry.has('remove_a')).toBe(true);
|
|
166
|
+
expect(registry.has('remove_b')).toBe(true);
|
|
167
|
+
const removed = registry.unregisterServer('s2');
|
|
168
|
+
expect(removed).toBe(true);
|
|
169
|
+
expect(registry.getServer('s2')).toBeUndefined();
|
|
170
|
+
expect(registry.has('remove_a')).toBe(false);
|
|
171
|
+
expect(registry.has('remove_b')).toBe(false);
|
|
172
|
+
});
|
|
173
|
+
it('unregisterServer with removeTools=false keeps tools', () => {
|
|
174
|
+
const tools = [makeTool({ name: 'keep_me', source: 'mcp' })];
|
|
175
|
+
registry.registerServer(makeServer('s3', tools));
|
|
176
|
+
registry.unregisterServer('s3', false);
|
|
177
|
+
expect(registry.getServer('s3')).toBeUndefined();
|
|
178
|
+
expect(registry.has('keep_me')).toBe(true);
|
|
179
|
+
});
|
|
180
|
+
it('listServers returns all registered servers', () => {
|
|
181
|
+
registry.registerServer(makeServer('a', []));
|
|
182
|
+
registry.registerServer(makeServer('b', []));
|
|
183
|
+
const servers = registry.listServers();
|
|
184
|
+
expect(servers).toHaveLength(2);
|
|
185
|
+
expect(servers.map((s) => s.id).sort()).toEqual(['a', 'b']);
|
|
186
|
+
});
|
|
187
|
+
it('clear removes all tools and servers', () => {
|
|
188
|
+
registry.register(makeTool({ name: 'solo' }));
|
|
189
|
+
registry.registerServer(makeServer('x', [makeTool({ name: 'grouped' })]));
|
|
190
|
+
registry.clear();
|
|
191
|
+
expect(registry.size).toBe(0);
|
|
192
|
+
expect(registry.list()).toHaveLength(0);
|
|
193
|
+
expect(registry.listServers()).toHaveLength(0);
|
|
194
|
+
});
|
|
195
|
+
});
|
|
196
|
+
//# sourceMappingURL=registry.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"registry.test.js","sourceRoot":"","sources":["../../src/__tests__/registry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAE1D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAG9C,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,SAAS,QAAQ,CAAC,YAAqC,EAAE;IACvD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,aAAa;QAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE;QAC9C,MAAM,EAAE,SAAS;QACjB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CACjB,EAAU,EACV,KAAuB,EACvB,YAAiC,EAAE;IAEnC,OAAO;QACL,EAAE;QACF,IAAI,EAAE,UAAU,EAAE,EAAE;QACpB,IAAI,EAAE,QAAQ;QACd,KAAK;QACL,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,QAAsB,CAAC;IAE3B,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,IAAI,GAAG,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC7C,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAExB,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,MAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAChD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAChD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAEhD,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC5B,MAAM,CAAC,GAAG,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;IAChF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,QAAQ,CAAC,WAAW,CAAC;YACnB,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC7B,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;SAC9B,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1C,MAAM,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC/E,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC/E,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACrF,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAErE,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAExE,MAAM,SAAS,GAAG,QAAQ,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAE1C,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE1D,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QACtE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC5G,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAE5G,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,UAAU,CAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,UAAU,CAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,QAAQ,CAAC,cAAc,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YAClD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,eAAe,EAAE,EAAE,EAAE;YACzF,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SACjC,CAAC,CAAC,CAAC;QACJ,QAAQ,CAAC,aAAa,CAAC;YACrB;gBACE,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,iBAAiB;gBAC9B,MAAM,EAAE,QAAQ;gBAChB,eAAe,EAAE,SAAS;gBAC1B,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,SAAS,CAAC,EAAE;aAChC;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7C,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC;SAC1F,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC3E,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAC/C,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE;gBACN,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE;gBAC7E,YAAY,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aACjC;SACF,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACzB,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,KAAK,GAAG;YACZ,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YAC/C,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;SAChD,CAAC;QACF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAEvC,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEhC,wBAAwB;QACxB,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAEzD,mDAAmD;QACnD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,KAAK,GAAG;YACZ,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YAC7C,QAAQ,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;SAC9C,CAAC;QACF,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAEjD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE5C,MAAM,OAAO,GAAG,QAAQ,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,KAAK,GAAG,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAC7D,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAEjD,QAAQ,CAAC,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7C,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;QAC9C,QAAQ,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE1E,QAAQ,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-translation.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/schema-translation.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,126 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { toToolDescriptor, fromToolDescriptor, translateTools, } from '../schema-translation.js';
|
|
3
|
+
/* -------------------------------------------------------------------------- */
|
|
4
|
+
/* Helpers */
|
|
5
|
+
/* -------------------------------------------------------------------------- */
|
|
6
|
+
function makeNormalized(overrides = {}) {
|
|
7
|
+
return {
|
|
8
|
+
name: 'test_tool',
|
|
9
|
+
description: 'A test tool',
|
|
10
|
+
parameters: { type: 'object', properties: { x: { type: 'string' } } },
|
|
11
|
+
...overrides,
|
|
12
|
+
};
|
|
13
|
+
}
|
|
14
|
+
function makeDescriptor(overrides = {}) {
|
|
15
|
+
return {
|
|
16
|
+
name: 'test_tool',
|
|
17
|
+
description: 'A test tool',
|
|
18
|
+
parameters: { type: 'object', properties: { x: { type: 'string' } } },
|
|
19
|
+
source: 'builtin',
|
|
20
|
+
...overrides,
|
|
21
|
+
};
|
|
22
|
+
}
|
|
23
|
+
/* ========================================================================== */
|
|
24
|
+
/* Schema Translation */
|
|
25
|
+
/* ========================================================================== */
|
|
26
|
+
describe('schema-translation', () => {
|
|
27
|
+
/* ---------------------------------------------------------------------- */
|
|
28
|
+
/* toToolDescriptor */
|
|
29
|
+
/* ---------------------------------------------------------------------- */
|
|
30
|
+
describe('toToolDescriptor', () => {
|
|
31
|
+
it('creates a ToolDescriptor from a NormalizedToolDefinition', () => {
|
|
32
|
+
const normalized = makeNormalized({ name: 'my_fn', description: 'Does stuff' });
|
|
33
|
+
const descriptor = toToolDescriptor(normalized, 'mcp');
|
|
34
|
+
expect(descriptor.name).toBe('my_fn');
|
|
35
|
+
expect(descriptor.description).toBe('Does stuff');
|
|
36
|
+
expect(descriptor.source).toBe('mcp');
|
|
37
|
+
expect(descriptor.parameters).toEqual(normalized.parameters);
|
|
38
|
+
});
|
|
39
|
+
it('attaches optional server, permissions, and metadata', () => {
|
|
40
|
+
const normalized = makeNormalized();
|
|
41
|
+
const descriptor = toToolDescriptor(normalized, 'plugin', {
|
|
42
|
+
server: 'srv-1',
|
|
43
|
+
permissions: ['read', 'write'],
|
|
44
|
+
metadata: { version: 2 },
|
|
45
|
+
});
|
|
46
|
+
expect(descriptor.server).toBe('srv-1');
|
|
47
|
+
expect(descriptor.permissions).toEqual(['read', 'write']);
|
|
48
|
+
expect(descriptor.metadata).toEqual({ version: 2 });
|
|
49
|
+
});
|
|
50
|
+
it('omits optional fields when extras are not provided', () => {
|
|
51
|
+
const descriptor = toToolDescriptor(makeNormalized(), 'builtin');
|
|
52
|
+
expect(descriptor.server).toBeUndefined();
|
|
53
|
+
expect(descriptor.permissions).toBeUndefined();
|
|
54
|
+
expect(descriptor.metadata).toBeUndefined();
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
/* ---------------------------------------------------------------------- */
|
|
58
|
+
/* fromToolDescriptor */
|
|
59
|
+
/* ---------------------------------------------------------------------- */
|
|
60
|
+
describe('fromToolDescriptor', () => {
|
|
61
|
+
it('strips lifecycle metadata and returns NormalizedToolDefinition', () => {
|
|
62
|
+
const descriptor = makeDescriptor({
|
|
63
|
+
name: 'stripped',
|
|
64
|
+
description: 'keep me',
|
|
65
|
+
server: 'srv-x',
|
|
66
|
+
permissions: ['admin'],
|
|
67
|
+
metadata: { extra: true },
|
|
68
|
+
});
|
|
69
|
+
const normalized = fromToolDescriptor(descriptor);
|
|
70
|
+
expect(normalized.name).toBe('stripped');
|
|
71
|
+
expect(normalized.description).toBe('keep me');
|
|
72
|
+
expect(normalized.parameters).toEqual(descriptor.parameters);
|
|
73
|
+
// Lifecycle fields should not be present
|
|
74
|
+
expect(normalized.source).toBeUndefined();
|
|
75
|
+
expect(normalized.server).toBeUndefined();
|
|
76
|
+
expect(normalized.permissions).toBeUndefined();
|
|
77
|
+
});
|
|
78
|
+
it('omits description when not present on descriptor', () => {
|
|
79
|
+
const descriptor = makeDescriptor({ description: undefined });
|
|
80
|
+
const normalized = fromToolDescriptor(descriptor);
|
|
81
|
+
expect(normalized).not.toHaveProperty('description');
|
|
82
|
+
});
|
|
83
|
+
it('omits parameters when not present on descriptor', () => {
|
|
84
|
+
const descriptor = makeDescriptor({ parameters: undefined });
|
|
85
|
+
const normalized = fromToolDescriptor(descriptor);
|
|
86
|
+
expect(normalized).not.toHaveProperty('parameters');
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
/* ---------------------------------------------------------------------- */
|
|
90
|
+
/* Round-trip */
|
|
91
|
+
/* ---------------------------------------------------------------------- */
|
|
92
|
+
describe('round-trip', () => {
|
|
93
|
+
it('toToolDescriptor -> fromToolDescriptor preserves core data', () => {
|
|
94
|
+
const original = makeNormalized({
|
|
95
|
+
name: 'round_trip',
|
|
96
|
+
description: 'survives the trip',
|
|
97
|
+
parameters: { type: 'object', properties: { q: { type: 'number' } } },
|
|
98
|
+
});
|
|
99
|
+
const descriptor = toToolDescriptor(original, 'custom', { server: 'srv' });
|
|
100
|
+
const backToNormalized = fromToolDescriptor(descriptor);
|
|
101
|
+
expect(backToNormalized.name).toBe(original.name);
|
|
102
|
+
expect(backToNormalized.description).toBe(original.description);
|
|
103
|
+
expect(backToNormalized.parameters).toEqual(original.parameters);
|
|
104
|
+
});
|
|
105
|
+
});
|
|
106
|
+
/* ---------------------------------------------------------------------- */
|
|
107
|
+
/* translateTools */
|
|
108
|
+
/* ---------------------------------------------------------------------- */
|
|
109
|
+
describe('translateTools', () => {
|
|
110
|
+
it('produces an array for a valid target format', () => {
|
|
111
|
+
const descriptors = [
|
|
112
|
+
makeDescriptor({ name: 'tool_a' }),
|
|
113
|
+
makeDescriptor({ name: 'tool_b' }),
|
|
114
|
+
];
|
|
115
|
+
const result = translateTools(descriptors, 'anthropic');
|
|
116
|
+
expect(Array.isArray(result)).toBe(true);
|
|
117
|
+
expect(result.length).toBeGreaterThan(0);
|
|
118
|
+
});
|
|
119
|
+
it('returns empty array for target format "none"', () => {
|
|
120
|
+
const descriptors = [makeDescriptor({ name: 'tool_a' })];
|
|
121
|
+
const result = translateTools(descriptors, 'none');
|
|
122
|
+
expect(result).toEqual([]);
|
|
123
|
+
});
|
|
124
|
+
});
|
|
125
|
+
});
|
|
126
|
+
//# sourceMappingURL=schema-translation.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-translation.test.js","sourceRoot":"","sources":["../../src/__tests__/schema-translation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,cAAc,GACf,MAAM,0BAA0B,CAAC;AAIlC,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,SAAS,cAAc,CACrB,YAA+C,EAAE;IAEjD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,aAAa;QAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;QACrE,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,YAAqC,EAAE;IAC7D,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,aAAa;QAC1B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;QACrE,MAAM,EAAE,SAAS;QACjB,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,gFAAgF;AAChF,gFAAgF;AAEhF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,UAAU,GAAG,cAAc,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;YAEhF,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAEvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAClD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;YAEpC,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,EAAE,QAAQ,EAAE;gBACxD,MAAM,EAAE,OAAO;gBACf,WAAW,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;gBAC9B,QAAQ,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE;aACzB,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1D,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,UAAU,GAAG,gBAAgB,CAAC,cAAc,EAAE,EAAE,SAAS,CAAC,CAAC;YAEjE,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,aAAa,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,UAAU,GAAG,cAAc,CAAC;gBAChC,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,SAAS;gBACtB,MAAM,EAAE,OAAO;gBACf,WAAW,EAAE,CAAC,OAAO,CAAC;gBACtB,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;aAC1B,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACzC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC7D,yCAAyC;YACzC,MAAM,CAAE,UAAkB,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YACnD,MAAM,CAAE,UAAkB,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YACnD,MAAM,CAAE,UAAkB,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,UAAU,GAAG,cAAc,CAAC,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;YAC9D,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,UAAU,GAAG,cAAc,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC;YAC7D,MAAM,UAAU,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,QAAQ,GAAG,cAAc,CAAC;gBAC9B,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,mBAAmB;gBAChC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;aACtE,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3E,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAExD,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAChE,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAC5E,6EAA6E;IAC7E,4EAA4E;IAE5E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,WAAW,GAAG;gBAClB,cAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAClC,cAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;aACnC,CAAC;YAEF,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,EAAE,WAAW,CAAC,CAAC;YACxD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,WAAW,GAAG,CAAC,cAAc,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"unified-contract.test.d.ts","sourceRoot":"","sources":["../../src/__tests__/unified-contract.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from 'vitest';
|
|
2
|
+
import { ToolDispatcher } from '../dispatch.js';
|
|
3
|
+
import { ToolRegistry } from '../registry.js';
|
|
4
|
+
import { ToolExecutionError, serializeToolError, } from '../types.js';
|
|
5
|
+
function makeTool(overrides = {}) {
|
|
6
|
+
return {
|
|
7
|
+
name: 'contract_tool',
|
|
8
|
+
description: 'Contract test tool',
|
|
9
|
+
parameters: { type: 'object' },
|
|
10
|
+
source: 'builtin',
|
|
11
|
+
metadata: {
|
|
12
|
+
category: 'utility',
|
|
13
|
+
tags: ['contract'],
|
|
14
|
+
cost: { unit: 'free' },
|
|
15
|
+
rateLimit: { scope: 'tool', limit: 10, windowMs: 60_000 },
|
|
16
|
+
requiresApproval: 'never',
|
|
17
|
+
cache: { read: true, write: false, ttlMs: 1_000 },
|
|
18
|
+
},
|
|
19
|
+
...overrides,
|
|
20
|
+
};
|
|
21
|
+
}
|
|
22
|
+
describe('unified tool contract', () => {
|
|
23
|
+
it('propagates AbortSignal and ordered update events through dispatch', async () => {
|
|
24
|
+
const registry = new ToolRegistry();
|
|
25
|
+
registry.register(makeTool());
|
|
26
|
+
const dispatcher = new ToolDispatcher({ registry });
|
|
27
|
+
const controller = new AbortController();
|
|
28
|
+
const events = [];
|
|
29
|
+
const onUpdate = vi.fn((event) => {
|
|
30
|
+
events.push(event);
|
|
31
|
+
});
|
|
32
|
+
const context = {
|
|
33
|
+
toolName: 'contract_tool',
|
|
34
|
+
input: { value: 1 },
|
|
35
|
+
signal: controller.signal,
|
|
36
|
+
onUpdate,
|
|
37
|
+
};
|
|
38
|
+
const result = await dispatcher.dispatch(context, async (_tool, executionContext) => {
|
|
39
|
+
expect(executionContext.signal).toBe(controller.signal);
|
|
40
|
+
await executionContext.onUpdate?.({
|
|
41
|
+
type: 'tool.stdout',
|
|
42
|
+
callId: 'contract-call',
|
|
43
|
+
chunk: 'hello',
|
|
44
|
+
sequence: 1,
|
|
45
|
+
});
|
|
46
|
+
return { ok: true, output: 'done' };
|
|
47
|
+
});
|
|
48
|
+
expect(result.error).toBeUndefined();
|
|
49
|
+
expect(result.output).toEqual({ ok: true, output: 'done' });
|
|
50
|
+
expect(onUpdate).toHaveBeenCalledOnce();
|
|
51
|
+
expect(events.map((event) => event.type)).toEqual(['tool.stdout']);
|
|
52
|
+
});
|
|
53
|
+
it('returns typed ToolExecutionError payloads instead of string-only errors', async () => {
|
|
54
|
+
const registry = new ToolRegistry();
|
|
55
|
+
registry.register(makeTool({ name: 'failing' }));
|
|
56
|
+
const dispatcher = new ToolDispatcher({ registry });
|
|
57
|
+
const result = await dispatcher.dispatch({ toolName: 'failing', input: {} }, async () => {
|
|
58
|
+
throw new ToolExecutionError('approval needed', {
|
|
59
|
+
code: 'APPROVAL_REQUIRED',
|
|
60
|
+
retryable: false,
|
|
61
|
+
details: { policy: 'always' },
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
expect(result.error).toEqual({
|
|
65
|
+
name: 'ToolExecutionError',
|
|
66
|
+
message: 'approval needed',
|
|
67
|
+
code: 'APPROVAL_REQUIRED',
|
|
68
|
+
retryable: false,
|
|
69
|
+
details: { policy: 'always' },
|
|
70
|
+
});
|
|
71
|
+
});
|
|
72
|
+
it('accepts configurable execution policy limits on the dispatcher', () => {
|
|
73
|
+
const registry = new ToolRegistry();
|
|
74
|
+
const policy = {
|
|
75
|
+
defaultTimeoutMs: 5_000,
|
|
76
|
+
defaultMaxOutputBytes: 1_024,
|
|
77
|
+
perTool: {
|
|
78
|
+
contract_tool: { timeoutMs: 250, maxOutputBytes: 64 },
|
|
79
|
+
},
|
|
80
|
+
allowStreaming: true,
|
|
81
|
+
enableReadOnlyCache: true,
|
|
82
|
+
};
|
|
83
|
+
const dispatcher = new ToolDispatcher({ registry, executionPolicy: policy });
|
|
84
|
+
expect(dispatcher.getExecutionLimits('contract_tool')).toEqual({
|
|
85
|
+
timeoutMs: 250,
|
|
86
|
+
maxOutputBytes: 64,
|
|
87
|
+
});
|
|
88
|
+
});
|
|
89
|
+
it('serializes unknown errors with a stable INTERNAL code', () => {
|
|
90
|
+
expect(serializeToolError(new Error('plain failure'))).toEqual({
|
|
91
|
+
name: 'ToolExecutionError',
|
|
92
|
+
message: 'plain failure',
|
|
93
|
+
code: 'INTERNAL',
|
|
94
|
+
retryable: false,
|
|
95
|
+
});
|
|
96
|
+
});
|
|
97
|
+
});
|
|
98
|
+
//# sourceMappingURL=unified-contract.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"unified-contract.test.js","sourceRoot":"","sources":["../../src/__tests__/unified-contract.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAElD,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EACL,kBAAkB,EAClB,kBAAkB,GACnB,MAAM,aAAa,CAAC;AAQrB,SAAS,QAAQ,CAAC,YAAqC,EAAE;IACvD,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,WAAW,EAAE,oBAAoB;QACjC,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC9B,MAAM,EAAE,SAAS;QACjB,QAAQ,EAAE;YACR,QAAQ,EAAE,SAAS;YACnB,IAAI,EAAE,CAAC,UAAU,CAAC;YAClB,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;YACtB,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE;YACzD,gBAAgB,EAAE,OAAO;YACzB,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;SAClD;QACD,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,UAAU,GAAG,IAAI,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpD,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,MAAM,GAAuB,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,KAAuB,EAAE,EAAE;YACjD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAoB;YAC/B,QAAQ,EAAE,eAAe;YACzB,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE;YACnB,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,QAAQ;SACT,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE;YAClF,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACxD,MAAM,gBAAgB,CAAC,QAAQ,EAAE,CAAC;gBAChC,IAAI,EAAE,aAAa;gBACnB,MAAM,EAAE,eAAe;gBACvB,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC;YACH,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5D,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,cAAc,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEpD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,QAAQ,CACtC,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,EAClC,KAAK,IAAI,EAAE;YACT,MAAM,IAAI,kBAAkB,CAAC,iBAAiB,EAAE;gBAC9C,IAAI,EAAE,mBAAmB;gBACzB,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;aAC9B,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;YAC3B,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,iBAAiB;YAC1B,IAAI,EAAE,mBAAmB;YACzB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;SAC9B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;QACpC,MAAM,MAAM,GAAwB;YAClC,gBAAgB,EAAE,KAAK;YACvB,qBAAqB,EAAE,KAAK;YAC5B,OAAO,EAAE;gBACP,aAAa,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,cAAc,EAAE,EAAE,EAAE;aACtD;YACD,cAAc,EAAE,IAAI;YACpB,mBAAmB,EAAE,IAAI;SAC1B,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,cAAc,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC,CAAC;QAE7E,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7D,SAAS,EAAE,GAAG;YACd,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,kBAAkB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7D,IAAI,EAAE,oBAAoB;YAC1B,OAAO,EAAE,eAAe;YACxB,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,KAAK;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|