opc-agent 1.1.1 → 1.1.2

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 (139) hide show
  1. package/CHANGELOG.md +51 -51
  2. package/CONTRIBUTING.md +75 -75
  3. package/README.md +222 -126
  4. package/README.zh-CN.md +129 -80
  5. package/dist/channels/web.js +256 -256
  6. package/dist/deploy/hermes.js +22 -22
  7. package/dist/deploy/openclaw.js +31 -31
  8. package/dist/templates/code-reviewer.js +5 -5
  9. package/dist/templates/customer-service.js +2 -2
  10. package/dist/templates/data-analyst.js +5 -5
  11. package/dist/templates/knowledge-base.js +2 -2
  12. package/dist/templates/sales-assistant.js +4 -4
  13. package/dist/templates/teacher.js +6 -6
  14. package/docs/.vitepress/config.ts +103 -103
  15. package/docs/api/cli.md +48 -48
  16. package/docs/api/oad-schema.md +64 -64
  17. package/docs/api/sdk.md +80 -80
  18. package/docs/guide/concepts.md +51 -51
  19. package/docs/guide/configuration.md +79 -79
  20. package/docs/guide/deployment.md +42 -42
  21. package/docs/guide/getting-started.md +44 -44
  22. package/docs/guide/templates.md +28 -28
  23. package/docs/guide/testing.md +84 -84
  24. package/docs/index.md +27 -27
  25. package/docs/zh/api/cli.md +54 -54
  26. package/docs/zh/api/oad-schema.md +87 -87
  27. package/docs/zh/api/sdk.md +102 -102
  28. package/docs/zh/guide/concepts.md +104 -104
  29. package/docs/zh/guide/configuration.md +135 -135
  30. package/docs/zh/guide/deployment.md +81 -81
  31. package/docs/zh/guide/getting-started.md +82 -82
  32. package/docs/zh/guide/templates.md +84 -84
  33. package/docs/zh/guide/testing.md +88 -88
  34. package/docs/zh/index.md +27 -27
  35. package/examples/customer-service-demo/README.md +90 -90
  36. package/examples/customer-service-demo/oad.yaml +107 -107
  37. package/package.json +1 -1
  38. package/src/analytics/index.ts +66 -66
  39. package/src/channels/discord.ts +192 -192
  40. package/src/channels/email.ts +177 -177
  41. package/src/channels/feishu.ts +236 -236
  42. package/src/channels/index.ts +15 -15
  43. package/src/channels/slack.ts +160 -160
  44. package/src/channels/telegram.ts +90 -90
  45. package/src/channels/voice.ts +106 -106
  46. package/src/channels/web.ts +596 -596
  47. package/src/channels/webhook.ts +199 -199
  48. package/src/channels/websocket.ts +87 -87
  49. package/src/channels/wechat.ts +149 -149
  50. package/src/core/a2a.ts +143 -143
  51. package/src/core/agent.ts +152 -152
  52. package/src/core/analytics-engine.ts +186 -186
  53. package/src/core/auth.ts +57 -57
  54. package/src/core/cache.ts +141 -141
  55. package/src/core/compose.ts +77 -77
  56. package/src/core/config.ts +14 -14
  57. package/src/core/errors.ts +148 -148
  58. package/src/core/hitl.ts +138 -138
  59. package/src/core/knowledge.ts +210 -210
  60. package/src/core/logger.ts +57 -57
  61. package/src/core/orchestrator.ts +215 -215
  62. package/src/core/performance.ts +187 -187
  63. package/src/core/rate-limiter.ts +128 -128
  64. package/src/core/room.ts +109 -109
  65. package/src/core/runtime.ts +152 -152
  66. package/src/core/sandbox.ts +101 -101
  67. package/src/core/security.ts +171 -171
  68. package/src/core/types.ts +68 -68
  69. package/src/core/versioning.ts +106 -106
  70. package/src/core/watch.ts +178 -178
  71. package/src/core/workflow.ts +235 -235
  72. package/src/deploy/hermes.ts +156 -156
  73. package/src/deploy/openclaw.ts +200 -200
  74. package/src/dtv/data.ts +29 -29
  75. package/src/dtv/trust.ts +43 -43
  76. package/src/dtv/value.ts +47 -47
  77. package/src/i18n/index.ts +216 -216
  78. package/src/index.ts +110 -110
  79. package/src/marketplace/index.ts +223 -223
  80. package/src/memory/deepbrain.ts +108 -108
  81. package/src/memory/index.ts +34 -34
  82. package/src/plugins/index.ts +208 -208
  83. package/src/providers/index.ts +183 -183
  84. package/src/schema/oad.ts +155 -155
  85. package/src/skills/base.ts +16 -16
  86. package/src/skills/document.ts +100 -100
  87. package/src/skills/http.ts +35 -35
  88. package/src/skills/index.ts +27 -27
  89. package/src/skills/scheduler.ts +80 -80
  90. package/src/skills/webhook-trigger.ts +59 -59
  91. package/src/templates/code-reviewer.ts +34 -34
  92. package/src/templates/customer-service.ts +80 -80
  93. package/src/templates/data-analyst.ts +70 -70
  94. package/src/templates/executive-assistant.ts +71 -71
  95. package/src/templates/financial-advisor.ts +60 -60
  96. package/src/templates/knowledge-base.ts +31 -31
  97. package/src/templates/legal-assistant.ts +71 -71
  98. package/src/templates/sales-assistant.ts +79 -79
  99. package/src/templates/teacher.ts +79 -79
  100. package/src/testing/index.ts +181 -181
  101. package/src/tools/calculator.ts +73 -73
  102. package/src/tools/datetime.ts +149 -149
  103. package/src/tools/json-transform.ts +187 -187
  104. package/src/tools/mcp.ts +76 -76
  105. package/src/tools/text-analysis.ts +116 -116
  106. package/templates/Dockerfile +15 -15
  107. package/templates/code-reviewer/README.md +27 -27
  108. package/templates/code-reviewer/oad.yaml +41 -41
  109. package/templates/customer-service/README.md +22 -22
  110. package/templates/customer-service/oad.yaml +36 -36
  111. package/templates/docker-compose.yml +21 -21
  112. package/templates/knowledge-base/README.md +28 -28
  113. package/templates/knowledge-base/oad.yaml +38 -38
  114. package/templates/sales-assistant/README.md +26 -26
  115. package/templates/sales-assistant/oad.yaml +43 -43
  116. package/tests/a2a.test.ts +66 -66
  117. package/tests/agent.test.ts +72 -72
  118. package/tests/analytics.test.ts +50 -50
  119. package/tests/channel.test.ts +39 -39
  120. package/tests/e2e.test.ts +134 -134
  121. package/tests/errors.test.ts +83 -83
  122. package/tests/hitl.test.ts +71 -71
  123. package/tests/i18n.test.ts +41 -41
  124. package/tests/mcp.test.ts +54 -54
  125. package/tests/oad.test.ts +68 -68
  126. package/tests/performance.test.ts +115 -115
  127. package/tests/plugin.test.ts +74 -74
  128. package/tests/room.test.ts +106 -106
  129. package/tests/runtime.test.ts +42 -42
  130. package/tests/sandbox.test.ts +46 -46
  131. package/tests/security.test.ts +60 -60
  132. package/tests/templates.test.ts +77 -77
  133. package/tests/v070.test.ts +76 -76
  134. package/tests/versioning.test.ts +75 -75
  135. package/tests/voice.test.ts +61 -61
  136. package/tests/webhook.test.ts +29 -29
  137. package/tests/workflow.test.ts +143 -143
  138. package/tsconfig.json +19 -19
  139. package/vitest.config.ts +9 -9
@@ -1,41 +1,41 @@
1
- import { describe, it, expect } from 'vitest';
2
- import { t, setLocale, getLocale, detectLocale, addMessages } from '../src/i18n';
3
-
4
- describe('i18n', () => {
5
- it('should return English messages by default', () => {
6
- setLocale('en');
7
- expect(t('agent.greeting')).toBe('Hello! How can I help you?');
8
- });
9
-
10
- it('should return Chinese messages', () => {
11
- setLocale('zh-CN');
12
- expect(t('agent.greeting')).toBe('你好!有什么可以帮你的?');
13
- setLocale('en'); // reset
14
- });
15
-
16
- it('should interpolate params', () => {
17
- setLocale('en');
18
- expect(t('agent.started', { name: 'TestBot' })).toBe('Agent "TestBot" started successfully');
19
- });
20
-
21
- it('should fall back to key if not found', () => {
22
- expect(t('nonexistent.key')).toBe('nonexistent.key');
23
- });
24
-
25
- it('should get and set locale', () => {
26
- setLocale('zh-CN');
27
- expect(getLocale()).toBe('zh-CN');
28
- setLocale('en');
29
- expect(getLocale()).toBe('en');
30
- });
31
-
32
- it('should add custom messages', () => {
33
- addMessages('en', { 'custom.key': 'Custom value' });
34
- expect(t('custom.key')).toBe('Custom value');
35
- });
36
-
37
- it('should detect locale from environment', () => {
38
- const locale = detectLocale();
39
- expect(['en', 'zh-CN']).toContain(locale);
40
- });
41
- });
1
+ import { describe, it, expect } from 'vitest';
2
+ import { t, setLocale, getLocale, detectLocale, addMessages } from '../src/i18n';
3
+
4
+ describe('i18n', () => {
5
+ it('should return English messages by default', () => {
6
+ setLocale('en');
7
+ expect(t('agent.greeting')).toBe('Hello! How can I help you?');
8
+ });
9
+
10
+ it('should return Chinese messages', () => {
11
+ setLocale('zh-CN');
12
+ expect(t('agent.greeting')).toBe('你好!有什么可以帮你的?');
13
+ setLocale('en'); // reset
14
+ });
15
+
16
+ it('should interpolate params', () => {
17
+ setLocale('en');
18
+ expect(t('agent.started', { name: 'TestBot' })).toBe('Agent "TestBot" started successfully');
19
+ });
20
+
21
+ it('should fall back to key if not found', () => {
22
+ expect(t('nonexistent.key')).toBe('nonexistent.key');
23
+ });
24
+
25
+ it('should get and set locale', () => {
26
+ setLocale('zh-CN');
27
+ expect(getLocale()).toBe('zh-CN');
28
+ setLocale('en');
29
+ expect(getLocale()).toBe('en');
30
+ });
31
+
32
+ it('should add custom messages', () => {
33
+ addMessages('en', { 'custom.key': 'Custom value' });
34
+ expect(t('custom.key')).toBe('Custom value');
35
+ });
36
+
37
+ it('should detect locale from environment', () => {
38
+ const locale = detectLocale();
39
+ expect(['en', 'zh-CN']).toContain(locale);
40
+ });
41
+ });
package/tests/mcp.test.ts CHANGED
@@ -1,54 +1,54 @@
1
- import { describe, it, expect } from 'vitest';
2
- import { MCPToolRegistry, createMCPTool } from '../src/tools/mcp';
3
-
4
- describe('MCP Tool System', () => {
5
- it('should register and list tools', () => {
6
- const registry = new MCPToolRegistry();
7
- const tool = createMCPTool('calculator', 'Basic calculator', {
8
- type: 'object',
9
- properties: { expression: { type: 'string' } },
10
- }, async (input) => ({ content: `Result: ${input.expression}` }));
11
-
12
- registry.register(tool);
13
- expect(registry.has('calculator')).toBe(true);
14
- expect(registry.list().length).toBe(1);
15
- expect(registry.list()[0].name).toBe('calculator');
16
- });
17
-
18
- it('should execute a tool', async () => {
19
- const registry = new MCPToolRegistry();
20
- registry.register(createMCPTool('echo', 'Echo tool', {}, async (input) => ({
21
- content: `Echo: ${input.text}`,
22
- })));
23
-
24
- const result = await registry.execute('echo', { text: 'hello' });
25
- expect(result.content).toBe('Echo: hello');
26
- expect(result.isError).toBeUndefined();
27
- });
28
-
29
- it('should return error for missing tool', async () => {
30
- const registry = new MCPToolRegistry();
31
- const result = await registry.execute('nonexistent', {});
32
- expect(result.isError).toBe(true);
33
- expect(result.content).toContain('not found');
34
- });
35
-
36
- it('should handle tool execution errors', async () => {
37
- const registry = new MCPToolRegistry();
38
- registry.register(createMCPTool('failing', 'Fails', {}, async () => {
39
- throw new Error('boom');
40
- }));
41
-
42
- const result = await registry.execute('failing', {});
43
- expect(result.isError).toBe(true);
44
- expect(result.content).toContain('boom');
45
- });
46
-
47
- it('should unregister tools', () => {
48
- const registry = new MCPToolRegistry();
49
- registry.register(createMCPTool('temp', 'Temp', {}, async () => ({ content: '' })));
50
- expect(registry.has('temp')).toBe(true);
51
- registry.unregister('temp');
52
- expect(registry.has('temp')).toBe(false);
53
- });
54
- });
1
+ import { describe, it, expect } from 'vitest';
2
+ import { MCPToolRegistry, createMCPTool } from '../src/tools/mcp';
3
+
4
+ describe('MCP Tool System', () => {
5
+ it('should register and list tools', () => {
6
+ const registry = new MCPToolRegistry();
7
+ const tool = createMCPTool('calculator', 'Basic calculator', {
8
+ type: 'object',
9
+ properties: { expression: { type: 'string' } },
10
+ }, async (input) => ({ content: `Result: ${input.expression}` }));
11
+
12
+ registry.register(tool);
13
+ expect(registry.has('calculator')).toBe(true);
14
+ expect(registry.list().length).toBe(1);
15
+ expect(registry.list()[0].name).toBe('calculator');
16
+ });
17
+
18
+ it('should execute a tool', async () => {
19
+ const registry = new MCPToolRegistry();
20
+ registry.register(createMCPTool('echo', 'Echo tool', {}, async (input) => ({
21
+ content: `Echo: ${input.text}`,
22
+ })));
23
+
24
+ const result = await registry.execute('echo', { text: 'hello' });
25
+ expect(result.content).toBe('Echo: hello');
26
+ expect(result.isError).toBeUndefined();
27
+ });
28
+
29
+ it('should return error for missing tool', async () => {
30
+ const registry = new MCPToolRegistry();
31
+ const result = await registry.execute('nonexistent', {});
32
+ expect(result.isError).toBe(true);
33
+ expect(result.content).toContain('not found');
34
+ });
35
+
36
+ it('should handle tool execution errors', async () => {
37
+ const registry = new MCPToolRegistry();
38
+ registry.register(createMCPTool('failing', 'Fails', {}, async () => {
39
+ throw new Error('boom');
40
+ }));
41
+
42
+ const result = await registry.execute('failing', {});
43
+ expect(result.isError).toBe(true);
44
+ expect(result.content).toContain('boom');
45
+ });
46
+
47
+ it('should unregister tools', () => {
48
+ const registry = new MCPToolRegistry();
49
+ registry.register(createMCPTool('temp', 'Temp', {}, async () => ({ content: '' })));
50
+ expect(registry.has('temp')).toBe(true);
51
+ registry.unregister('temp');
52
+ expect(registry.has('temp')).toBe(false);
53
+ });
54
+ });
package/tests/oad.test.ts CHANGED
@@ -1,68 +1,68 @@
1
- import { describe, it, expect } from 'vitest';
2
- import { OADSchema } from '../src/schema/oad';
3
-
4
- describe('OAD Schema', () => {
5
- it('should parse a valid OAD document', () => {
6
- const doc = OADSchema.parse({
7
- apiVersion: 'opc/v1',
8
- kind: 'Agent',
9
- metadata: { name: 'test-agent' },
10
- spec: {},
11
- });
12
- expect(doc.metadata.name).toBe('test-agent');
13
- expect(doc.metadata.version).toBe('1.0.0');
14
- expect(doc.spec.model).toBe('deepseek-chat');
15
- });
16
-
17
- it('should reject invalid apiVersion', () => {
18
- expect(() =>
19
- OADSchema.parse({
20
- apiVersion: 'v2',
21
- kind: 'Agent',
22
- metadata: { name: 'test' },
23
- spec: {},
24
- })
25
- ).toThrow();
26
- });
27
-
28
- it('should parse full document with all fields', () => {
29
- const doc = OADSchema.parse({
30
- apiVersion: 'opc/v1',
31
- kind: 'Agent',
32
- metadata: {
33
- name: 'full-agent',
34
- version: '2.0.0',
35
- description: 'A full agent',
36
- author: 'Test',
37
- license: 'MIT',
38
- marketplace: { certified: true, category: 'support' },
39
- },
40
- spec: {
41
- provider: { default: 'openai', allowed: ['openai'] },
42
- model: 'gpt-4',
43
- systemPrompt: 'You are helpful.',
44
- skills: [{ name: 'faq', description: 'FAQ skill' }],
45
- channels: [{ type: 'web', port: 8080 }],
46
- memory: { shortTerm: true, longTerm: true },
47
- dtv: {
48
- trust: { level: 'certified' },
49
- value: { metrics: ['response_time'] },
50
- },
51
- },
52
- });
53
- expect(doc.metadata.version).toBe('2.0.0');
54
- expect(doc.spec.channels[0].port).toBe(8080);
55
- expect(doc.spec.dtv?.trust?.level).toBe('certified');
56
- });
57
-
58
- it('should reject missing metadata.name', () => {
59
- expect(() =>
60
- OADSchema.parse({
61
- apiVersion: 'opc/v1',
62
- kind: 'Agent',
63
- metadata: {},
64
- spec: {},
65
- })
66
- ).toThrow();
67
- });
68
- });
1
+ import { describe, it, expect } from 'vitest';
2
+ import { OADSchema } from '../src/schema/oad';
3
+
4
+ describe('OAD Schema', () => {
5
+ it('should parse a valid OAD document', () => {
6
+ const doc = OADSchema.parse({
7
+ apiVersion: 'opc/v1',
8
+ kind: 'Agent',
9
+ metadata: { name: 'test-agent' },
10
+ spec: {},
11
+ });
12
+ expect(doc.metadata.name).toBe('test-agent');
13
+ expect(doc.metadata.version).toBe('1.0.0');
14
+ expect(doc.spec.model).toBe('deepseek-chat');
15
+ });
16
+
17
+ it('should reject invalid apiVersion', () => {
18
+ expect(() =>
19
+ OADSchema.parse({
20
+ apiVersion: 'v2',
21
+ kind: 'Agent',
22
+ metadata: { name: 'test' },
23
+ spec: {},
24
+ })
25
+ ).toThrow();
26
+ });
27
+
28
+ it('should parse full document with all fields', () => {
29
+ const doc = OADSchema.parse({
30
+ apiVersion: 'opc/v1',
31
+ kind: 'Agent',
32
+ metadata: {
33
+ name: 'full-agent',
34
+ version: '2.0.0',
35
+ description: 'A full agent',
36
+ author: 'Test',
37
+ license: 'MIT',
38
+ marketplace: { certified: true, category: 'support' },
39
+ },
40
+ spec: {
41
+ provider: { default: 'openai', allowed: ['openai'] },
42
+ model: 'gpt-4',
43
+ systemPrompt: 'You are helpful.',
44
+ skills: [{ name: 'faq', description: 'FAQ skill' }],
45
+ channels: [{ type: 'web', port: 8080 }],
46
+ memory: { shortTerm: true, longTerm: true },
47
+ dtv: {
48
+ trust: { level: 'certified' },
49
+ value: { metrics: ['response_time'] },
50
+ },
51
+ },
52
+ });
53
+ expect(doc.metadata.version).toBe('2.0.0');
54
+ expect(doc.spec.channels[0].port).toBe(8080);
55
+ expect(doc.spec.dtv?.trust?.level).toBe('certified');
56
+ });
57
+
58
+ it('should reject missing metadata.name', () => {
59
+ expect(() =>
60
+ OADSchema.parse({
61
+ apiVersion: 'opc/v1',
62
+ kind: 'Agent',
63
+ metadata: {},
64
+ spec: {},
65
+ })
66
+ ).toThrow();
67
+ });
68
+ });
@@ -1,115 +1,115 @@
1
- import { describe, it, expect } from 'vitest';
2
- import { ConnectionPool, RequestBatcher, LazyLoader } from '../src/core/performance';
3
-
4
- describe('ConnectionPool', () => {
5
- it('should acquire and release connections', () => {
6
- const pool = new ConnectionPool(3);
7
- const conn = pool.acquire('openai');
8
- expect(conn.inUse).toBe(true);
9
- pool.release(conn.id);
10
- const stats = pool.getStats();
11
- expect(stats['openai'].total).toBe(1);
12
- expect(stats['openai'].inUse).toBe(0);
13
- });
14
-
15
- it('should reuse released connections', () => {
16
- const pool = new ConnectionPool(2);
17
- const c1 = pool.acquire('openai');
18
- pool.release(c1.id);
19
- const c2 = pool.acquire('openai');
20
- expect(c2.id).toBe(c1.id);
21
- });
22
-
23
- it('should drain all connections', () => {
24
- const pool = new ConnectionPool();
25
- pool.acquire('openai');
26
- pool.acquire('deepseek');
27
- pool.drain();
28
- expect(pool.getStats()).toEqual({});
29
- });
30
- });
31
-
32
- describe('RequestBatcher', () => {
33
- it('should batch requests', async () => {
34
- const batcher = new RequestBatcher<string>(
35
- async (batch) => batch.map(s => s.toUpperCase()),
36
- 2,
37
- 10,
38
- );
39
-
40
- const [r1, r2] = await Promise.all([
41
- batcher.add('hello'),
42
- batcher.add('world'),
43
- ]);
44
- expect(r1).toBe('HELLO');
45
- expect(r2).toBe('WORLD');
46
- });
47
-
48
- it('should flush on timer', async () => {
49
- const batcher = new RequestBatcher<number>(
50
- async (batch) => batch.map(n => n * 2),
51
- 10,
52
- 20,
53
- );
54
-
55
- const result = await batcher.add(5);
56
- expect(result).toBe(10);
57
- });
58
-
59
- it('should track pending count', () => {
60
- const batcher = new RequestBatcher<string>(
61
- async (batch) => batch,
62
- 100,
63
- 10000,
64
- );
65
- batcher.add('a');
66
- batcher.add('b');
67
- expect(batcher.pending).toBe(2);
68
- });
69
- });
70
-
71
- describe('LazyLoader', () => {
72
- it('should lazily load items', async () => {
73
- const loader = new LazyLoader<string>();
74
- let loadCount = 0;
75
- loader.register('greeting', async () => { loadCount++; return 'hello'; });
76
-
77
- expect(loader.isLoaded('greeting')).toBe(false);
78
- const val = await loader.get('greeting');
79
- expect(val).toBe('hello');
80
- expect(loader.isLoaded('greeting')).toBe(true);
81
-
82
- // Second call should use cache
83
- await loader.get('greeting');
84
- expect(loadCount).toBe(1);
85
- });
86
-
87
- it('should throw for unregistered items', async () => {
88
- const loader = new LazyLoader();
89
- await expect(loader.get('unknown')).rejects.toThrow('No loader registered');
90
- });
91
-
92
- it('should evict and reload', async () => {
93
- const loader = new LazyLoader<number>();
94
- let count = 0;
95
- loader.register('counter', async () => ++count);
96
-
97
- await loader.get('counter');
98
- expect(loader.loadedCount).toBe(1);
99
- loader.evict('counter');
100
- expect(loader.loadedCount).toBe(0);
101
- await loader.get('counter');
102
- expect(count).toBe(2);
103
- });
104
-
105
- it('should clear all', async () => {
106
- const loader = new LazyLoader<string>();
107
- loader.register('a', async () => 'a');
108
- loader.register('b', async () => 'b');
109
- await loader.get('a');
110
- await loader.get('b');
111
- loader.clear();
112
- expect(loader.loadedCount).toBe(0);
113
- expect(loader.registeredCount).toBe(2);
114
- });
115
- });
1
+ import { describe, it, expect } from 'vitest';
2
+ import { ConnectionPool, RequestBatcher, LazyLoader } from '../src/core/performance';
3
+
4
+ describe('ConnectionPool', () => {
5
+ it('should acquire and release connections', () => {
6
+ const pool = new ConnectionPool(3);
7
+ const conn = pool.acquire('openai');
8
+ expect(conn.inUse).toBe(true);
9
+ pool.release(conn.id);
10
+ const stats = pool.getStats();
11
+ expect(stats['openai'].total).toBe(1);
12
+ expect(stats['openai'].inUse).toBe(0);
13
+ });
14
+
15
+ it('should reuse released connections', () => {
16
+ const pool = new ConnectionPool(2);
17
+ const c1 = pool.acquire('openai');
18
+ pool.release(c1.id);
19
+ const c2 = pool.acquire('openai');
20
+ expect(c2.id).toBe(c1.id);
21
+ });
22
+
23
+ it('should drain all connections', () => {
24
+ const pool = new ConnectionPool();
25
+ pool.acquire('openai');
26
+ pool.acquire('deepseek');
27
+ pool.drain();
28
+ expect(pool.getStats()).toEqual({});
29
+ });
30
+ });
31
+
32
+ describe('RequestBatcher', () => {
33
+ it('should batch requests', async () => {
34
+ const batcher = new RequestBatcher<string>(
35
+ async (batch) => batch.map(s => s.toUpperCase()),
36
+ 2,
37
+ 10,
38
+ );
39
+
40
+ const [r1, r2] = await Promise.all([
41
+ batcher.add('hello'),
42
+ batcher.add('world'),
43
+ ]);
44
+ expect(r1).toBe('HELLO');
45
+ expect(r2).toBe('WORLD');
46
+ });
47
+
48
+ it('should flush on timer', async () => {
49
+ const batcher = new RequestBatcher<number>(
50
+ async (batch) => batch.map(n => n * 2),
51
+ 10,
52
+ 20,
53
+ );
54
+
55
+ const result = await batcher.add(5);
56
+ expect(result).toBe(10);
57
+ });
58
+
59
+ it('should track pending count', () => {
60
+ const batcher = new RequestBatcher<string>(
61
+ async (batch) => batch,
62
+ 100,
63
+ 10000,
64
+ );
65
+ batcher.add('a');
66
+ batcher.add('b');
67
+ expect(batcher.pending).toBe(2);
68
+ });
69
+ });
70
+
71
+ describe('LazyLoader', () => {
72
+ it('should lazily load items', async () => {
73
+ const loader = new LazyLoader<string>();
74
+ let loadCount = 0;
75
+ loader.register('greeting', async () => { loadCount++; return 'hello'; });
76
+
77
+ expect(loader.isLoaded('greeting')).toBe(false);
78
+ const val = await loader.get('greeting');
79
+ expect(val).toBe('hello');
80
+ expect(loader.isLoaded('greeting')).toBe(true);
81
+
82
+ // Second call should use cache
83
+ await loader.get('greeting');
84
+ expect(loadCount).toBe(1);
85
+ });
86
+
87
+ it('should throw for unregistered items', async () => {
88
+ const loader = new LazyLoader();
89
+ await expect(loader.get('unknown')).rejects.toThrow('No loader registered');
90
+ });
91
+
92
+ it('should evict and reload', async () => {
93
+ const loader = new LazyLoader<number>();
94
+ let count = 0;
95
+ loader.register('counter', async () => ++count);
96
+
97
+ await loader.get('counter');
98
+ expect(loader.loadedCount).toBe(1);
99
+ loader.evict('counter');
100
+ expect(loader.loadedCount).toBe(0);
101
+ await loader.get('counter');
102
+ expect(count).toBe(2);
103
+ });
104
+
105
+ it('should clear all', async () => {
106
+ const loader = new LazyLoader<string>();
107
+ loader.register('a', async () => 'a');
108
+ loader.register('b', async () => 'b');
109
+ await loader.get('a');
110
+ await loader.get('b');
111
+ loader.clear();
112
+ expect(loader.loadedCount).toBe(0);
113
+ expect(loader.registeredCount).toBe(2);
114
+ });
115
+ });