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.
- package/CHANGELOG.md +51 -51
- package/CONTRIBUTING.md +75 -75
- package/README.md +222 -126
- package/README.zh-CN.md +129 -80
- package/dist/channels/web.js +256 -256
- package/dist/deploy/hermes.js +22 -22
- package/dist/deploy/openclaw.js +31 -31
- package/dist/templates/code-reviewer.js +5 -5
- package/dist/templates/customer-service.js +2 -2
- package/dist/templates/data-analyst.js +5 -5
- package/dist/templates/knowledge-base.js +2 -2
- package/dist/templates/sales-assistant.js +4 -4
- package/dist/templates/teacher.js +6 -6
- package/docs/.vitepress/config.ts +103 -103
- package/docs/api/cli.md +48 -48
- package/docs/api/oad-schema.md +64 -64
- package/docs/api/sdk.md +80 -80
- package/docs/guide/concepts.md +51 -51
- package/docs/guide/configuration.md +79 -79
- package/docs/guide/deployment.md +42 -42
- package/docs/guide/getting-started.md +44 -44
- package/docs/guide/templates.md +28 -28
- package/docs/guide/testing.md +84 -84
- package/docs/index.md +27 -27
- package/docs/zh/api/cli.md +54 -54
- package/docs/zh/api/oad-schema.md +87 -87
- package/docs/zh/api/sdk.md +102 -102
- package/docs/zh/guide/concepts.md +104 -104
- package/docs/zh/guide/configuration.md +135 -135
- package/docs/zh/guide/deployment.md +81 -81
- package/docs/zh/guide/getting-started.md +82 -82
- package/docs/zh/guide/templates.md +84 -84
- package/docs/zh/guide/testing.md +88 -88
- package/docs/zh/index.md +27 -27
- package/examples/customer-service-demo/README.md +90 -90
- package/examples/customer-service-demo/oad.yaml +107 -107
- package/package.json +1 -1
- package/src/analytics/index.ts +66 -66
- package/src/channels/discord.ts +192 -192
- package/src/channels/email.ts +177 -177
- package/src/channels/feishu.ts +236 -236
- package/src/channels/index.ts +15 -15
- package/src/channels/slack.ts +160 -160
- package/src/channels/telegram.ts +90 -90
- package/src/channels/voice.ts +106 -106
- package/src/channels/web.ts +596 -596
- package/src/channels/webhook.ts +199 -199
- package/src/channels/websocket.ts +87 -87
- package/src/channels/wechat.ts +149 -149
- package/src/core/a2a.ts +143 -143
- package/src/core/agent.ts +152 -152
- package/src/core/analytics-engine.ts +186 -186
- package/src/core/auth.ts +57 -57
- package/src/core/cache.ts +141 -141
- package/src/core/compose.ts +77 -77
- package/src/core/config.ts +14 -14
- package/src/core/errors.ts +148 -148
- package/src/core/hitl.ts +138 -138
- package/src/core/knowledge.ts +210 -210
- package/src/core/logger.ts +57 -57
- package/src/core/orchestrator.ts +215 -215
- package/src/core/performance.ts +187 -187
- package/src/core/rate-limiter.ts +128 -128
- package/src/core/room.ts +109 -109
- package/src/core/runtime.ts +152 -152
- package/src/core/sandbox.ts +101 -101
- package/src/core/security.ts +171 -171
- package/src/core/types.ts +68 -68
- package/src/core/versioning.ts +106 -106
- package/src/core/watch.ts +178 -178
- package/src/core/workflow.ts +235 -235
- package/src/deploy/hermes.ts +156 -156
- package/src/deploy/openclaw.ts +200 -200
- package/src/dtv/data.ts +29 -29
- package/src/dtv/trust.ts +43 -43
- package/src/dtv/value.ts +47 -47
- package/src/i18n/index.ts +216 -216
- package/src/index.ts +110 -110
- package/src/marketplace/index.ts +223 -223
- package/src/memory/deepbrain.ts +108 -108
- package/src/memory/index.ts +34 -34
- package/src/plugins/index.ts +208 -208
- package/src/providers/index.ts +183 -183
- package/src/schema/oad.ts +155 -155
- package/src/skills/base.ts +16 -16
- package/src/skills/document.ts +100 -100
- package/src/skills/http.ts +35 -35
- package/src/skills/index.ts +27 -27
- package/src/skills/scheduler.ts +80 -80
- package/src/skills/webhook-trigger.ts +59 -59
- package/src/templates/code-reviewer.ts +34 -34
- package/src/templates/customer-service.ts +80 -80
- package/src/templates/data-analyst.ts +70 -70
- package/src/templates/executive-assistant.ts +71 -71
- package/src/templates/financial-advisor.ts +60 -60
- package/src/templates/knowledge-base.ts +31 -31
- package/src/templates/legal-assistant.ts +71 -71
- package/src/templates/sales-assistant.ts +79 -79
- package/src/templates/teacher.ts +79 -79
- package/src/testing/index.ts +181 -181
- package/src/tools/calculator.ts +73 -73
- package/src/tools/datetime.ts +149 -149
- package/src/tools/json-transform.ts +187 -187
- package/src/tools/mcp.ts +76 -76
- package/src/tools/text-analysis.ts +116 -116
- package/templates/Dockerfile +15 -15
- package/templates/code-reviewer/README.md +27 -27
- package/templates/code-reviewer/oad.yaml +41 -41
- package/templates/customer-service/README.md +22 -22
- package/templates/customer-service/oad.yaml +36 -36
- package/templates/docker-compose.yml +21 -21
- package/templates/knowledge-base/README.md +28 -28
- package/templates/knowledge-base/oad.yaml +38 -38
- package/templates/sales-assistant/README.md +26 -26
- package/templates/sales-assistant/oad.yaml +43 -43
- package/tests/a2a.test.ts +66 -66
- package/tests/agent.test.ts +72 -72
- package/tests/analytics.test.ts +50 -50
- package/tests/channel.test.ts +39 -39
- package/tests/e2e.test.ts +134 -134
- package/tests/errors.test.ts +83 -83
- package/tests/hitl.test.ts +71 -71
- package/tests/i18n.test.ts +41 -41
- package/tests/mcp.test.ts +54 -54
- package/tests/oad.test.ts +68 -68
- package/tests/performance.test.ts +115 -115
- package/tests/plugin.test.ts +74 -74
- package/tests/room.test.ts +106 -106
- package/tests/runtime.test.ts +42 -42
- package/tests/sandbox.test.ts +46 -46
- package/tests/security.test.ts +60 -60
- package/tests/templates.test.ts +77 -77
- package/tests/v070.test.ts +76 -76
- package/tests/versioning.test.ts +75 -75
- package/tests/voice.test.ts +61 -61
- package/tests/webhook.test.ts +29 -29
- package/tests/workflow.test.ts +143 -143
- package/tsconfig.json +19 -19
- package/vitest.config.ts +9 -9
package/tests/i18n.test.ts
CHANGED
|
@@ -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
|
+
});
|