opc-agent 1.2.0 → 1.2.1

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