opc-agent 1.1.1 → 1.1.3

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 (142) 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/providers/index.d.ts +1 -1
  9. package/dist/providers/index.js +148 -13
  10. package/dist/schema/oad.d.ts +3 -3
  11. package/dist/templates/code-reviewer.js +5 -5
  12. package/dist/templates/customer-service.js +2 -2
  13. package/dist/templates/data-analyst.js +5 -5
  14. package/dist/templates/knowledge-base.js +2 -2
  15. package/dist/templates/sales-assistant.js +4 -4
  16. package/dist/templates/teacher.js +6 -6
  17. package/docs/.vitepress/config.ts +103 -103
  18. package/docs/api/cli.md +48 -48
  19. package/docs/api/oad-schema.md +64 -64
  20. package/docs/api/sdk.md +80 -80
  21. package/docs/guide/concepts.md +51 -51
  22. package/docs/guide/configuration.md +79 -79
  23. package/docs/guide/deployment.md +42 -42
  24. package/docs/guide/getting-started.md +44 -44
  25. package/docs/guide/templates.md +28 -28
  26. package/docs/guide/testing.md +84 -84
  27. package/docs/index.md +27 -27
  28. package/docs/zh/api/cli.md +54 -54
  29. package/docs/zh/api/oad-schema.md +87 -87
  30. package/docs/zh/api/sdk.md +102 -102
  31. package/docs/zh/guide/concepts.md +104 -104
  32. package/docs/zh/guide/configuration.md +135 -135
  33. package/docs/zh/guide/deployment.md +81 -81
  34. package/docs/zh/guide/getting-started.md +82 -82
  35. package/docs/zh/guide/templates.md +84 -84
  36. package/docs/zh/guide/testing.md +88 -88
  37. package/docs/zh/index.md +27 -27
  38. package/examples/customer-service-demo/README.md +90 -90
  39. package/examples/customer-service-demo/oad.yaml +107 -107
  40. package/package.json +1 -1
  41. package/src/analytics/index.ts +66 -66
  42. package/src/channels/discord.ts +192 -192
  43. package/src/channels/email.ts +177 -177
  44. package/src/channels/feishu.ts +236 -236
  45. package/src/channels/index.ts +15 -15
  46. package/src/channels/slack.ts +160 -160
  47. package/src/channels/telegram.ts +90 -90
  48. package/src/channels/voice.ts +106 -106
  49. package/src/channels/web.ts +596 -596
  50. package/src/channels/webhook.ts +199 -199
  51. package/src/channels/websocket.ts +87 -87
  52. package/src/channels/wechat.ts +149 -149
  53. package/src/core/a2a.ts +143 -143
  54. package/src/core/agent.ts +152 -152
  55. package/src/core/analytics-engine.ts +186 -186
  56. package/src/core/auth.ts +57 -57
  57. package/src/core/cache.ts +141 -141
  58. package/src/core/compose.ts +77 -77
  59. package/src/core/config.ts +14 -14
  60. package/src/core/errors.ts +148 -148
  61. package/src/core/hitl.ts +138 -138
  62. package/src/core/knowledge.ts +210 -210
  63. package/src/core/logger.ts +57 -57
  64. package/src/core/orchestrator.ts +215 -215
  65. package/src/core/performance.ts +187 -187
  66. package/src/core/rate-limiter.ts +128 -128
  67. package/src/core/room.ts +109 -109
  68. package/src/core/runtime.ts +152 -152
  69. package/src/core/sandbox.ts +101 -101
  70. package/src/core/security.ts +171 -171
  71. package/src/core/types.ts +68 -68
  72. package/src/core/versioning.ts +106 -106
  73. package/src/core/watch.ts +178 -178
  74. package/src/core/workflow.ts +235 -235
  75. package/src/deploy/hermes.ts +156 -156
  76. package/src/deploy/openclaw.ts +200 -200
  77. package/src/dtv/data.ts +29 -29
  78. package/src/dtv/trust.ts +43 -43
  79. package/src/dtv/value.ts +47 -47
  80. package/src/i18n/index.ts +216 -216
  81. package/src/index.ts +110 -110
  82. package/src/marketplace/index.ts +223 -223
  83. package/src/memory/deepbrain.ts +108 -108
  84. package/src/memory/index.ts +34 -34
  85. package/src/plugins/index.ts +208 -208
  86. package/src/providers/index.ts +322 -183
  87. package/src/schema/oad.ts +155 -155
  88. package/src/skills/base.ts +16 -16
  89. package/src/skills/document.ts +100 -100
  90. package/src/skills/http.ts +35 -35
  91. package/src/skills/index.ts +27 -27
  92. package/src/skills/scheduler.ts +80 -80
  93. package/src/skills/webhook-trigger.ts +59 -59
  94. package/src/templates/code-reviewer.ts +34 -34
  95. package/src/templates/customer-service.ts +80 -80
  96. package/src/templates/data-analyst.ts +70 -70
  97. package/src/templates/executive-assistant.ts +71 -71
  98. package/src/templates/financial-advisor.ts +60 -60
  99. package/src/templates/knowledge-base.ts +31 -31
  100. package/src/templates/legal-assistant.ts +71 -71
  101. package/src/templates/sales-assistant.ts +79 -79
  102. package/src/templates/teacher.ts +79 -79
  103. package/src/testing/index.ts +181 -181
  104. package/src/tools/calculator.ts +73 -73
  105. package/src/tools/datetime.ts +149 -149
  106. package/src/tools/json-transform.ts +187 -187
  107. package/src/tools/mcp.ts +76 -76
  108. package/src/tools/text-analysis.ts +116 -116
  109. package/templates/Dockerfile +15 -15
  110. package/templates/code-reviewer/README.md +27 -27
  111. package/templates/code-reviewer/oad.yaml +41 -41
  112. package/templates/customer-service/README.md +22 -22
  113. package/templates/customer-service/oad.yaml +36 -36
  114. package/templates/docker-compose.yml +21 -21
  115. package/templates/knowledge-base/README.md +28 -28
  116. package/templates/knowledge-base/oad.yaml +38 -38
  117. package/templates/sales-assistant/README.md +26 -26
  118. package/templates/sales-assistant/oad.yaml +43 -43
  119. package/tests/a2a.test.ts +66 -66
  120. package/tests/agent.test.ts +72 -72
  121. package/tests/analytics.test.ts +50 -50
  122. package/tests/channel.test.ts +39 -39
  123. package/tests/e2e.test.ts +134 -134
  124. package/tests/errors.test.ts +83 -83
  125. package/tests/hitl.test.ts +71 -71
  126. package/tests/i18n.test.ts +41 -41
  127. package/tests/mcp.test.ts +54 -54
  128. package/tests/oad.test.ts +68 -68
  129. package/tests/performance.test.ts +115 -115
  130. package/tests/plugin.test.ts +74 -74
  131. package/tests/room.test.ts +106 -106
  132. package/tests/runtime.test.ts +42 -42
  133. package/tests/sandbox.test.ts +46 -46
  134. package/tests/security.test.ts +60 -60
  135. package/tests/templates.test.ts +77 -77
  136. package/tests/v070.test.ts +76 -76
  137. package/tests/versioning.test.ts +75 -75
  138. package/tests/voice.test.ts +61 -61
  139. package/tests/webhook.test.ts +29 -29
  140. package/tests/workflow.test.ts +143 -143
  141. package/tsconfig.json +19 -19
  142. 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
+ });