@a5c-ai/tools-adapter 5.1.1-staging.0c6199708314

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.
Files changed (54) hide show
  1. package/README.md +57 -0
  2. package/dist/__tests__/dispatch.test.d.ts +2 -0
  3. package/dist/__tests__/dispatch.test.d.ts.map +1 -0
  4. package/dist/__tests__/dispatch.test.js +251 -0
  5. package/dist/__tests__/dispatch.test.js.map +1 -0
  6. package/dist/__tests__/hooks.test.d.ts +2 -0
  7. package/dist/__tests__/hooks.test.d.ts.map +1 -0
  8. package/dist/__tests__/hooks.test.js +172 -0
  9. package/dist/__tests__/hooks.test.js.map +1 -0
  10. package/dist/__tests__/mcp-bridge.test.d.ts +2 -0
  11. package/dist/__tests__/mcp-bridge.test.d.ts.map +1 -0
  12. package/dist/__tests__/mcp-bridge.test.js +129 -0
  13. package/dist/__tests__/mcp-bridge.test.js.map +1 -0
  14. package/dist/__tests__/registry.test.d.ts +2 -0
  15. package/dist/__tests__/registry.test.d.ts.map +1 -0
  16. package/dist/__tests__/registry.test.js +196 -0
  17. package/dist/__tests__/registry.test.js.map +1 -0
  18. package/dist/__tests__/schema-translation.test.d.ts +2 -0
  19. package/dist/__tests__/schema-translation.test.d.ts.map +1 -0
  20. package/dist/__tests__/schema-translation.test.js +126 -0
  21. package/dist/__tests__/schema-translation.test.js.map +1 -0
  22. package/dist/__tests__/unified-contract.test.d.ts +2 -0
  23. package/dist/__tests__/unified-contract.test.d.ts.map +1 -0
  24. package/dist/__tests__/unified-contract.test.js +98 -0
  25. package/dist/__tests__/unified-contract.test.js.map +1 -0
  26. package/dist/dispatch.d.ts +43 -0
  27. package/dist/dispatch.d.ts.map +1 -0
  28. package/dist/dispatch.js +157 -0
  29. package/dist/dispatch.js.map +1 -0
  30. package/dist/hooks.d.ts +99 -0
  31. package/dist/hooks.d.ts.map +1 -0
  32. package/dist/hooks.js +110 -0
  33. package/dist/hooks.js.map +1 -0
  34. package/dist/index.d.ts +13 -0
  35. package/dist/index.d.ts.map +1 -0
  36. package/dist/index.js +25 -0
  37. package/dist/index.js.map +1 -0
  38. package/dist/mcp-bridge.d.ts +54 -0
  39. package/dist/mcp-bridge.d.ts.map +1 -0
  40. package/dist/mcp-bridge.js +78 -0
  41. package/dist/mcp-bridge.js.map +1 -0
  42. package/dist/registry.d.ts +70 -0
  43. package/dist/registry.d.ts.map +1 -0
  44. package/dist/registry.js +221 -0
  45. package/dist/registry.js.map +1 -0
  46. package/dist/schema-translation.d.ts +38 -0
  47. package/dist/schema-translation.d.ts.map +1 -0
  48. package/dist/schema-translation.js +61 -0
  49. package/dist/schema-translation.js.map +1 -0
  50. package/dist/types.d.ts +159 -0
  51. package/dist/types.d.ts.map +1 -0
  52. package/dist/types.js +39 -0
  53. package/dist/types.js.map +1 -0
  54. 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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=registry.test.d.ts.map
@@ -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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=schema-translation.test.d.ts.map
@@ -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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=unified-contract.test.d.ts.map
@@ -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"}