opc-agent 4.1.0 → 4.1.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.
- package/.github/ISSUE_TEMPLATE/bug_report.md +20 -20
- package/.github/ISSUE_TEMPLATE/feature_request.md +14 -14
- package/.github/PULL_REQUEST_TEMPLATE.md +13 -13
- package/CHANGELOG.md +48 -48
- package/CONTRIBUTING.md +36 -36
- package/README.zh-CN.md +497 -497
- package/dist/channels/wechat.js +6 -6
- package/dist/deploy/index.js +56 -56
- package/dist/studio/server.js +30 -1
- package/dist/studio-ui/index.html +230 -10
- package/dist/ui/components.js +105 -105
- package/examples/README.md +22 -22
- package/examples/basic-agent.ts +90 -90
- package/examples/brain-integration.ts +71 -71
- package/examples/multi-channel.ts +74 -74
- package/fix-sidebar.mjs +188 -188
- package/install.ps1 +154 -154
- package/install.sh +164 -164
- package/package.json +1 -1
- package/scripts/install.ps1 +31 -31
- package/scripts/install.sh +40 -40
- package/serve-studio.js +13 -13
- package/serve-test.js +25 -25
- package/src/channels/dingtalk.ts +46 -46
- package/src/channels/email.ts +351 -351
- package/src/channels/feishu.ts +349 -349
- package/src/channels/googlechat.ts +42 -42
- package/src/channels/imessage.ts +31 -31
- package/src/channels/irc.ts +82 -82
- package/src/channels/line.ts +32 -32
- package/src/channels/matrix.ts +33 -33
- package/src/channels/mattermost.ts +57 -57
- package/src/channels/msteams.ts +32 -32
- package/src/channels/nostr.ts +32 -32
- package/src/channels/qq.ts +33 -33
- package/src/channels/signal.ts +32 -32
- package/src/channels/sms.ts +33 -33
- package/src/channels/telegram.ts +616 -616
- package/src/channels/twitch.ts +65 -65
- package/src/channels/voice-call.ts +100 -100
- package/src/channels/websocket.ts +399 -399
- package/src/channels/wechat.ts +329 -329
- package/src/channels/whatsapp.ts +32 -32
- package/src/cli/chat.ts +99 -99
- package/src/cli/setup.ts +314 -314
- package/src/core/agent.ts +476 -476
- package/src/core/api-server.ts +277 -277
- package/src/core/audio.ts +98 -98
- package/src/core/collaboration.ts +275 -275
- package/src/core/context-discovery.ts +85 -85
- package/src/core/context-refs.ts +140 -140
- package/src/core/gateway.ts +106 -106
- package/src/core/heartbeat.ts +51 -51
- package/src/core/hooks.ts +105 -105
- package/src/core/ide-bridge.ts +133 -133
- package/src/core/node-network.ts +86 -86
- package/src/core/profiles.ts +122 -122
- package/src/core/scheduler.ts +187 -187
- package/src/core/session-manager.ts +137 -137
- package/src/core/subagent.ts +98 -98
- package/src/core/vision.ts +180 -180
- package/src/core/workflow-graph.ts +365 -365
- package/src/daemon.ts +96 -96
- package/src/deploy/index.ts +255 -255
- package/src/doctor.ts +156 -156
- package/src/eval/index.ts +211 -211
- package/src/eval/suites/basic.json +16 -16
- package/src/eval/suites/memory.json +12 -12
- package/src/eval/suites/safety.json +14 -14
- package/src/hub/brain-seed.ts +54 -54
- package/src/hub/client.ts +60 -60
- package/src/mcp/servers/calculator-mcp.ts +65 -65
- package/src/mcp/servers/crypto-mcp.ts +73 -73
- package/src/mcp/servers/database-mcp.ts +72 -72
- package/src/mcp/servers/datetime-mcp.ts +69 -69
- package/src/mcp/servers/filesystem.ts +66 -66
- package/src/mcp/servers/github-mcp.ts +58 -58
- package/src/mcp/servers/index.ts +63 -63
- package/src/mcp/servers/json-mcp.ts +102 -102
- package/src/mcp/servers/memory-mcp.ts +56 -56
- package/src/mcp/servers/regex-mcp.ts +53 -53
- package/src/mcp/servers/web-mcp.ts +49 -49
- package/src/memory/context-compressor.ts +189 -189
- package/src/memory/seed-loader.ts +212 -212
- package/src/memory/user-profiler.ts +215 -215
- package/src/plugins/content-filter.ts +23 -23
- package/src/plugins/logger.ts +18 -18
- package/src/plugins/rate-limiter.ts +38 -38
- package/src/protocols/a2a/client.ts +132 -132
- package/src/protocols/a2a/index.ts +8 -8
- package/src/protocols/a2a/server.ts +333 -333
- package/src/protocols/a2a/types.ts +88 -88
- package/src/protocols/a2a/utils.ts +50 -50
- package/src/protocols/agui/client.ts +83 -83
- package/src/protocols/agui/index.ts +4 -4
- package/src/protocols/agui/server.ts +218 -218
- package/src/protocols/agui/types.ts +153 -153
- package/src/protocols/index.ts +2 -2
- package/src/protocols/mcp/agent-tools.ts +134 -134
- package/src/protocols/mcp/index.ts +8 -8
- package/src/protocols/mcp/server.ts +262 -262
- package/src/protocols/mcp/types.ts +69 -69
- package/src/providers/index.ts +632 -632
- package/src/publish/index.ts +376 -376
- package/src/scheduler/cron-engine.ts +191 -191
- package/src/scheduler/index.ts +2 -2
- package/src/schema/oad.ts +217 -217
- package/src/security/approval.ts +131 -131
- package/src/security/approvals.ts +143 -143
- package/src/security/elevated.ts +105 -105
- package/src/security/guardrails.ts +248 -248
- package/src/security/index.ts +9 -9
- package/src/security/keys.ts +87 -87
- package/src/security/secrets.ts +129 -129
- package/src/skills/builtin/index.ts +408 -408
- package/src/skills/marketplace.ts +113 -113
- package/src/skills/types.ts +42 -42
- package/src/studio/server.ts +31 -1
- package/src/studio/templates-data.ts +178 -178
- package/src/studio-ui/index.html +230 -10
- package/src/telemetry/index.ts +324 -324
- package/src/tools/builtin/browser.ts +299 -299
- package/src/tools/builtin/datetime.ts +41 -41
- package/src/tools/builtin/file.ts +107 -107
- package/src/tools/builtin/home-assistant.ts +116 -116
- package/src/tools/builtin/rl-tools.ts +243 -243
- package/src/tools/builtin/shell.ts +43 -43
- package/src/tools/builtin/vision.ts +64 -64
- package/src/tools/builtin/web-search.ts +126 -126
- package/src/tools/builtin/web.ts +35 -35
- package/src/tools/document-processor.ts +213 -213
- package/src/tools/image-generator.ts +150 -150
- package/src/tools/integrations/calendar.ts +73 -73
- package/src/tools/integrations/code-exec.ts +39 -39
- package/src/tools/integrations/csv-analyzer.ts +92 -92
- package/src/tools/integrations/database.ts +44 -44
- package/src/tools/integrations/email-send.ts +76 -76
- package/src/tools/integrations/git-tool.ts +42 -42
- package/src/tools/integrations/github-tool.ts +76 -76
- package/src/tools/integrations/image-gen.ts +56 -56
- package/src/tools/integrations/index.ts +92 -92
- package/src/tools/integrations/jira.ts +83 -83
- package/src/tools/integrations/notion.ts +71 -71
- package/src/tools/integrations/npm-tool.ts +48 -48
- package/src/tools/integrations/pdf-reader.ts +58 -58
- package/src/tools/integrations/slack.ts +65 -65
- package/src/tools/integrations/summarizer.ts +49 -49
- package/src/tools/integrations/translator.ts +48 -48
- package/src/tools/integrations/trello.ts +60 -60
- package/src/tools/integrations/vector-search.ts +42 -42
- package/src/tools/integrations/web-scraper.ts +47 -47
- package/src/tools/integrations/web-search.ts +58 -58
- package/src/tools/integrations/webhook.ts +38 -38
- package/src/tools/mcp-client.ts +131 -131
- package/src/tools/web-scraper.ts +179 -179
- package/src/tools/web-search.ts +180 -180
- package/src/ui/components.ts +127 -127
- package/srv-out.txt +1 -1
- package/templates/ecommerce-assistant/README.md +45 -45
- package/templates/ecommerce-assistant/oad.yaml +47 -47
- package/templates/tech-support/README.md +43 -43
- package/templates/tech-support/oad.yaml +45 -45
- package/test-agent/Dockerfile +9 -9
- package/test-agent/README.md +50 -50
- package/test-agent/agent.yaml +23 -23
- package/test-agent/docker-compose.yml +11 -11
- package/test-agent/oad.yaml +31 -31
- package/test-agent/package-lock.json +1492 -1492
- package/test-agent/package.json +17 -17
- package/test-agent/src/index.ts +24 -24
- package/test-agent/src/skills/echo.ts +15 -15
- package/test-agent/tsconfig.json +24 -24
- package/test-full.js +43 -43
- package/test-sidebar.js +22 -22
- package/test-studio3.js +75 -75
- package/test-studio4.js +41 -41
- package/tests/a2a-protocol.test.ts +285 -285
- package/tests/agui-protocol.test.ts +246 -246
- package/tests/api-server.test.ts +148 -148
- package/tests/approvals.test.ts +89 -89
- package/tests/audio.test.ts +40 -40
- package/tests/brain-seed-extended.test.ts +490 -490
- package/tests/brain-seed.test.ts +239 -239
- package/tests/browser.test.ts +179 -179
- package/tests/channels/discord.test.ts +79 -79
- package/tests/channels/email.test.ts +148 -148
- package/tests/channels/feishu.test.ts +123 -123
- package/tests/channels/telegram.test.ts +129 -129
- package/tests/channels/websocket.test.ts +53 -53
- package/tests/channels/wechat.test.ts +170 -170
- package/tests/channels-extra.test.ts +45 -45
- package/tests/chat-cli.test.ts +160 -160
- package/tests/cli.test.ts +46 -46
- package/tests/context-compressor.test.ts +172 -172
- package/tests/context-refs.test.ts +121 -121
- package/tests/cron-engine.test.ts +101 -101
- package/tests/daemon.test.ts +135 -135
- package/tests/deepbrain-wire.test.ts +234 -234
- package/tests/deploy-and-dag.test.ts +196 -196
- package/tests/doctor.test.ts +38 -38
- package/tests/document-processor.test.ts +69 -69
- package/tests/e2e-nocode.test.ts +442 -442
- package/tests/elevated.test.ts +69 -69
- package/tests/eval.test.ts +173 -173
- package/tests/gateway.test.ts +63 -63
- package/tests/guardrails.test.ts +177 -177
- package/tests/home-assistant.test.ts +40 -40
- package/tests/hooks.test.ts +79 -79
- package/tests/ide-bridge.test.ts +38 -38
- package/tests/image-generator.test.ts +84 -84
- package/tests/init-role.test.ts +124 -124
- package/tests/integrations.test.ts +249 -249
- package/tests/mcp-client.test.ts +92 -92
- package/tests/mcp-server.test.ts +178 -178
- package/tests/mcp-servers.test.ts +260 -260
- package/tests/node-network.test.ts +74 -74
- package/tests/plugin-a2a-enhanced.test.ts +230 -230
- package/tests/profiles.test.ts +61 -61
- package/tests/publish.test.ts +231 -231
- package/tests/rl-tools.test.ts +93 -93
- package/tests/sandbox-manager.test.ts +46 -46
- package/tests/scheduler.test.ts +200 -200
- package/tests/secrets.test.ts +107 -107
- package/tests/security-enhanced.test.ts +233 -233
- package/tests/settings-api.test.ts +148 -148
- package/tests/setup.test.ts +73 -73
- package/tests/subagent.test.ts +193 -193
- package/tests/telegram-discord.test.ts +60 -60
- package/tests/telemetry.test.ts +186 -186
- package/tests/user-profiler.test.ts +169 -169
- package/tests/v090-features.test.ts +254 -254
- package/tests/vision.test.ts +61 -61
- package/tests/voice-call.test.ts +47 -47
- package/tests/voice-enhanced.test.ts +169 -169
- package/tests/voice-interaction.test.ts +38 -38
- package/tests/web-search.test.ts +155 -155
- package/tests/workflow-graph.test.ts +279 -279
- package/tutorial/customer-service-agent/README.md +612 -612
- package/tutorial/customer-service-agent/SOUL.md +26 -26
- package/tutorial/customer-service-agent/agent.yaml +63 -63
- package/tutorial/customer-service-agent/package.json +19 -19
- package/tutorial/customer-service-agent/src/index.ts +69 -69
- package/tutorial/customer-service-agent/src/skills/faq.ts +27 -27
- package/tutorial/customer-service-agent/src/skills/ticket.ts +22 -22
- package/tutorial/customer-service-agent/tsconfig.json +14 -14
|
@@ -1,254 +1,254 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import { DingTalkChannel } from '../src/channels/dingtalk';
|
|
3
|
-
import { MattermostChannel } from '../src/channels/mattermost';
|
|
4
|
-
import { GoogleChatChannel } from '../src/channels/googlechat';
|
|
5
|
-
import { TwitchChannel } from '../src/channels/twitch';
|
|
6
|
-
import { IRCChannel } from '../src/channels/irc';
|
|
7
|
-
import { ContextDiscovery } from '../src/core/context-discovery';
|
|
8
|
-
import { SessionManager } from '../src/core/session-manager';
|
|
9
|
-
import { HeartbeatManager } from '../src/core/heartbeat';
|
|
10
|
-
import * as fs from 'fs';
|
|
11
|
-
import * as path from 'path';
|
|
12
|
-
import * as os from 'os';
|
|
13
|
-
|
|
14
|
-
// ── Channel Tests (10) ──
|
|
15
|
-
|
|
16
|
-
describe('DingTalkChannel', () => {
|
|
17
|
-
it('should require webhookUrl', () => {
|
|
18
|
-
expect(() => new DingTalkChannel({ webhookUrl: '' })).toThrow('webhookUrl');
|
|
19
|
-
});
|
|
20
|
-
it('should throw when sending before start', async () => {
|
|
21
|
-
const ch = new DingTalkChannel({ webhookUrl: 'https://example.com/hook' });
|
|
22
|
-
await expect(ch.send('chat', 'hi')).rejects.toThrow('not started');
|
|
23
|
-
});
|
|
24
|
-
});
|
|
25
|
-
|
|
26
|
-
describe('MattermostChannel', () => {
|
|
27
|
-
it('should require serverUrl and token', () => {
|
|
28
|
-
expect(() => new MattermostChannel({ serverUrl: '', token: '' })).toThrow('serverUrl');
|
|
29
|
-
});
|
|
30
|
-
it('should throw when sending before start', async () => {
|
|
31
|
-
const ch = new MattermostChannel({ serverUrl: 'https://mm.example.com', token: 'tok' });
|
|
32
|
-
await expect(ch.send('ch1', 'hi')).rejects.toThrow('not started');
|
|
33
|
-
});
|
|
34
|
-
});
|
|
35
|
-
|
|
36
|
-
describe('GoogleChatChannel', () => {
|
|
37
|
-
it('should require webhookUrl', () => {
|
|
38
|
-
expect(() => new GoogleChatChannel({ webhookUrl: '' })).toThrow('webhookUrl');
|
|
39
|
-
});
|
|
40
|
-
it('should throw when sending before start', async () => {
|
|
41
|
-
const ch = new GoogleChatChannel({ webhookUrl: 'https://chat.googleapis.com/v1/spaces/x/messages?key=y' });
|
|
42
|
-
await expect(ch.send('space', 'hi')).rejects.toThrow('not started');
|
|
43
|
-
});
|
|
44
|
-
});
|
|
45
|
-
|
|
46
|
-
describe('TwitchChannel', () => {
|
|
47
|
-
it('should require username, oauthToken, channels', () => {
|
|
48
|
-
expect(() => new TwitchChannel({ username: '', oauthToken: '', channels: [] })).toThrow('username');
|
|
49
|
-
});
|
|
50
|
-
it('should throw when starting without tmi.js', async () => {
|
|
51
|
-
const ch = new TwitchChannel({ username: 'bot', oauthToken: 'oauth:xxx', channels: ['#test'] });
|
|
52
|
-
await expect(ch.start()).rejects.toThrow('tmi.js');
|
|
53
|
-
});
|
|
54
|
-
});
|
|
55
|
-
|
|
56
|
-
describe('IRCChannel', () => {
|
|
57
|
-
it('should require host, nick, channels', () => {
|
|
58
|
-
expect(() => new IRCChannel({ host: '', nick: '', channels: [] })).toThrow('host');
|
|
59
|
-
});
|
|
60
|
-
it('should throw when starting without irc-framework', async () => {
|
|
61
|
-
const ch = new IRCChannel({ host: 'irc.libera.chat', nick: 'bot', channels: ['#test'] });
|
|
62
|
-
await expect(ch.start()).rejects.toThrow('irc-framework');
|
|
63
|
-
});
|
|
64
|
-
});
|
|
65
|
-
|
|
66
|
-
// ── Context Discovery Tests (8) ──
|
|
67
|
-
|
|
68
|
-
describe('ContextDiscovery', () => {
|
|
69
|
-
let tmpDir: string;
|
|
70
|
-
let cd: ContextDiscovery;
|
|
71
|
-
|
|
72
|
-
beforeEach(() => {
|
|
73
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ctx-'));
|
|
74
|
-
cd = new ContextDiscovery();
|
|
75
|
-
});
|
|
76
|
-
|
|
77
|
-
afterEach(() => {
|
|
78
|
-
cd.stopWatching();
|
|
79
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
80
|
-
});
|
|
81
|
-
|
|
82
|
-
it('should discover AGENTS.md', () => {
|
|
83
|
-
fs.writeFileSync(path.join(tmpDir, 'AGENTS.md'), '# Agents');
|
|
84
|
-
const files = cd.discover(tmpDir);
|
|
85
|
-
expect(files.length).toBe(1);
|
|
86
|
-
expect(files[0].type).toBe('agents');
|
|
87
|
-
});
|
|
88
|
-
|
|
89
|
-
it('should discover multiple standard files', () => {
|
|
90
|
-
fs.writeFileSync(path.join(tmpDir, 'SOUL.md'), '# Soul');
|
|
91
|
-
fs.writeFileSync(path.join(tmpDir, 'USER.md'), '# User');
|
|
92
|
-
const files = cd.discover(tmpDir);
|
|
93
|
-
expect(files.length).toBe(2);
|
|
94
|
-
});
|
|
95
|
-
|
|
96
|
-
it('should return empty for no context files', () => {
|
|
97
|
-
expect(cd.discover(tmpDir)).toEqual([]);
|
|
98
|
-
});
|
|
99
|
-
|
|
100
|
-
it('should load and merge files into prompt', () => {
|
|
101
|
-
fs.writeFileSync(path.join(tmpDir, 'SOUL.md'), 'Be nice');
|
|
102
|
-
fs.writeFileSync(path.join(tmpDir, 'TOOLS.md'), 'Use web');
|
|
103
|
-
const files = cd.discover(tmpDir);
|
|
104
|
-
const prompt = cd.load(files);
|
|
105
|
-
expect(prompt).toContain('Be nice');
|
|
106
|
-
expect(prompt).toContain('Use web');
|
|
107
|
-
});
|
|
108
|
-
|
|
109
|
-
it('should add and discover custom files', () => {
|
|
110
|
-
const customPath = path.join(tmpDir, 'CUSTOM.md');
|
|
111
|
-
fs.writeFileSync(customPath, '# Custom');
|
|
112
|
-
cd.addCustomFile(customPath);
|
|
113
|
-
const files = cd.discover(tmpDir);
|
|
114
|
-
expect(files.some(f => f.type === 'custom' && f.content.includes('Custom'))).toBe(true);
|
|
115
|
-
});
|
|
116
|
-
|
|
117
|
-
it('should not duplicate custom files', () => {
|
|
118
|
-
cd.addCustomFile('test.md');
|
|
119
|
-
cd.addCustomFile('test.md');
|
|
120
|
-
// Internal check - discover won't find it but custom list should have 1
|
|
121
|
-
expect(cd.discover(tmpDir).filter(f => f.path.includes('test.md')).length).toBeLessThanOrEqual(1);
|
|
122
|
-
});
|
|
123
|
-
|
|
124
|
-
it('should assign correct types', () => {
|
|
125
|
-
fs.writeFileSync(path.join(tmpDir, 'MEMORY.md'), 'mem');
|
|
126
|
-
fs.writeFileSync(path.join(tmpDir, 'IDENTITY.md'), 'id');
|
|
127
|
-
const files = cd.discover(tmpDir);
|
|
128
|
-
expect(files.find(f => f.type === 'memory')).toBeTruthy();
|
|
129
|
-
expect(files.find(f => f.type === 'identity')).toBeTruthy();
|
|
130
|
-
});
|
|
131
|
-
|
|
132
|
-
it('should handle watch without errors', () => {
|
|
133
|
-
// Just ensure no throw
|
|
134
|
-
cd.watch(tmpDir, () => {});
|
|
135
|
-
cd.stopWatching();
|
|
136
|
-
});
|
|
137
|
-
});
|
|
138
|
-
|
|
139
|
-
// ── Session Manager Tests (8) ──
|
|
140
|
-
|
|
141
|
-
describe('SessionManager', () => {
|
|
142
|
-
let sm: SessionManager;
|
|
143
|
-
let tmpDir: string;
|
|
144
|
-
|
|
145
|
-
beforeEach(() => {
|
|
146
|
-
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'sess-'));
|
|
147
|
-
sm = new SessionManager(tmpDir);
|
|
148
|
-
});
|
|
149
|
-
|
|
150
|
-
afterEach(() => {
|
|
151
|
-
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
152
|
-
});
|
|
153
|
-
|
|
154
|
-
it('should create a session', () => {
|
|
155
|
-
const s = sm.create('agent1', 'telegram');
|
|
156
|
-
expect(s.id).toBeTruthy();
|
|
157
|
-
expect(s.agentId).toBe('agent1');
|
|
158
|
-
});
|
|
159
|
-
|
|
160
|
-
it('should get a session by id', () => {
|
|
161
|
-
const s = sm.create('agent1', 'telegram');
|
|
162
|
-
expect(sm.get(s.id)).toEqual(s);
|
|
163
|
-
expect(sm.get('nonexistent')).toBeNull();
|
|
164
|
-
});
|
|
165
|
-
|
|
166
|
-
it('should add messages', () => {
|
|
167
|
-
const s = sm.create('a', 'c');
|
|
168
|
-
sm.addMessage(s.id, { id: '1', role: 'user', content: 'hi', timestamp: Date.now() });
|
|
169
|
-
expect(sm.get(s.id)!.messages.length).toBe(1);
|
|
170
|
-
});
|
|
171
|
-
|
|
172
|
-
it('should list with filters', () => {
|
|
173
|
-
sm.create('a1', 'telegram');
|
|
174
|
-
sm.create('a2', 'discord');
|
|
175
|
-
expect(sm.list({ agentId: 'a1' }).length).toBe(1);
|
|
176
|
-
expect(sm.list({ channel: 'discord' }).length).toBe(1);
|
|
177
|
-
});
|
|
178
|
-
|
|
179
|
-
it('should compact sessions', async () => {
|
|
180
|
-
const s = sm.create('a', 'c');
|
|
181
|
-
for (let i = 0; i < 15; i++) {
|
|
182
|
-
sm.addMessage(s.id, { id: `${i}`, role: 'user', content: `msg ${i}`, timestamp: Date.now() });
|
|
183
|
-
}
|
|
184
|
-
await sm.compact(s.id);
|
|
185
|
-
expect(sm.get(s.id)!.messages.length).toBeLessThan(15);
|
|
186
|
-
expect(sm.get(s.id)!.compactedAt).toBeTruthy();
|
|
187
|
-
});
|
|
188
|
-
|
|
189
|
-
it('should prune old sessions', () => {
|
|
190
|
-
const s = sm.create('a', 'c');
|
|
191
|
-
// Manually set old lastActivity
|
|
192
|
-
(sm.get(s.id) as any).lastActivity = Date.now() - 999999;
|
|
193
|
-
expect(sm.prune(1000)).toBe(1);
|
|
194
|
-
});
|
|
195
|
-
|
|
196
|
-
it('should fork and get lineage', () => {
|
|
197
|
-
const parent = sm.create('a', 'c');
|
|
198
|
-
const child = sm.fork(parent.id);
|
|
199
|
-
expect(child.parentId).toBe(parent.id);
|
|
200
|
-
const lineage = sm.getLineage(child.id);
|
|
201
|
-
expect(lineage.length).toBe(2);
|
|
202
|
-
expect(lineage[0].id).toBe(parent.id);
|
|
203
|
-
});
|
|
204
|
-
|
|
205
|
-
it('should export as markdown and save/load', () => {
|
|
206
|
-
const s = sm.create('a', 'c');
|
|
207
|
-
sm.addMessage(s.id, { id: '1', role: 'user', content: 'hello', timestamp: Date.now() });
|
|
208
|
-
const md = sm.export(s.id);
|
|
209
|
-
expect(md).toContain('hello');
|
|
210
|
-
sm.save();
|
|
211
|
-
const sm2 = new SessionManager(tmpDir);
|
|
212
|
-
sm2.load();
|
|
213
|
-
expect(sm2.get(s.id)).toBeTruthy();
|
|
214
|
-
});
|
|
215
|
-
});
|
|
216
|
-
|
|
217
|
-
// ── Heartbeat Tests (4) ──
|
|
218
|
-
|
|
219
|
-
describe('HeartbeatManager', () => {
|
|
220
|
-
it('should require valid config', () => {
|
|
221
|
-
expect(() => new HeartbeatManager({ interval: 0, checkFn: async () => 'ok' })).toThrow('interval');
|
|
222
|
-
expect(() => new HeartbeatManager({ interval: 1000, checkFn: null as any })).toThrow('checkFn');
|
|
223
|
-
});
|
|
224
|
-
|
|
225
|
-
it('should start and stop', () => {
|
|
226
|
-
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'ok' });
|
|
227
|
-
hb.start();
|
|
228
|
-
hb.stop();
|
|
229
|
-
// Double stop should be safe
|
|
230
|
-
hb.stop();
|
|
231
|
-
});
|
|
232
|
-
|
|
233
|
-
it('should fire callbacks on beat', async () => {
|
|
234
|
-
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'alive' });
|
|
235
|
-
const statuses: string[] = [];
|
|
236
|
-
hb.onBeat(s => statuses.push(s));
|
|
237
|
-
hb.start();
|
|
238
|
-
await new Promise(r => setTimeout(r, 350));
|
|
239
|
-
hb.stop();
|
|
240
|
-
expect(statuses.length).toBeGreaterThanOrEqual(2);
|
|
241
|
-
expect(statuses[0]).toBe('alive');
|
|
242
|
-
});
|
|
243
|
-
|
|
244
|
-
it('should report last beat', async () => {
|
|
245
|
-
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'ok' });
|
|
246
|
-
expect(hb.getLastBeat()).toBeNull();
|
|
247
|
-
hb.start();
|
|
248
|
-
await new Promise(r => setTimeout(r, 200));
|
|
249
|
-
hb.stop();
|
|
250
|
-
const beat = hb.getLastBeat();
|
|
251
|
-
expect(beat).toBeTruthy();
|
|
252
|
-
expect(beat!.status).toBe('ok');
|
|
253
|
-
});
|
|
254
|
-
});
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { DingTalkChannel } from '../src/channels/dingtalk';
|
|
3
|
+
import { MattermostChannel } from '../src/channels/mattermost';
|
|
4
|
+
import { GoogleChatChannel } from '../src/channels/googlechat';
|
|
5
|
+
import { TwitchChannel } from '../src/channels/twitch';
|
|
6
|
+
import { IRCChannel } from '../src/channels/irc';
|
|
7
|
+
import { ContextDiscovery } from '../src/core/context-discovery';
|
|
8
|
+
import { SessionManager } from '../src/core/session-manager';
|
|
9
|
+
import { HeartbeatManager } from '../src/core/heartbeat';
|
|
10
|
+
import * as fs from 'fs';
|
|
11
|
+
import * as path from 'path';
|
|
12
|
+
import * as os from 'os';
|
|
13
|
+
|
|
14
|
+
// ── Channel Tests (10) ──
|
|
15
|
+
|
|
16
|
+
describe('DingTalkChannel', () => {
|
|
17
|
+
it('should require webhookUrl', () => {
|
|
18
|
+
expect(() => new DingTalkChannel({ webhookUrl: '' })).toThrow('webhookUrl');
|
|
19
|
+
});
|
|
20
|
+
it('should throw when sending before start', async () => {
|
|
21
|
+
const ch = new DingTalkChannel({ webhookUrl: 'https://example.com/hook' });
|
|
22
|
+
await expect(ch.send('chat', 'hi')).rejects.toThrow('not started');
|
|
23
|
+
});
|
|
24
|
+
});
|
|
25
|
+
|
|
26
|
+
describe('MattermostChannel', () => {
|
|
27
|
+
it('should require serverUrl and token', () => {
|
|
28
|
+
expect(() => new MattermostChannel({ serverUrl: '', token: '' })).toThrow('serverUrl');
|
|
29
|
+
});
|
|
30
|
+
it('should throw when sending before start', async () => {
|
|
31
|
+
const ch = new MattermostChannel({ serverUrl: 'https://mm.example.com', token: 'tok' });
|
|
32
|
+
await expect(ch.send('ch1', 'hi')).rejects.toThrow('not started');
|
|
33
|
+
});
|
|
34
|
+
});
|
|
35
|
+
|
|
36
|
+
describe('GoogleChatChannel', () => {
|
|
37
|
+
it('should require webhookUrl', () => {
|
|
38
|
+
expect(() => new GoogleChatChannel({ webhookUrl: '' })).toThrow('webhookUrl');
|
|
39
|
+
});
|
|
40
|
+
it('should throw when sending before start', async () => {
|
|
41
|
+
const ch = new GoogleChatChannel({ webhookUrl: 'https://chat.googleapis.com/v1/spaces/x/messages?key=y' });
|
|
42
|
+
await expect(ch.send('space', 'hi')).rejects.toThrow('not started');
|
|
43
|
+
});
|
|
44
|
+
});
|
|
45
|
+
|
|
46
|
+
describe('TwitchChannel', () => {
|
|
47
|
+
it('should require username, oauthToken, channels', () => {
|
|
48
|
+
expect(() => new TwitchChannel({ username: '', oauthToken: '', channels: [] })).toThrow('username');
|
|
49
|
+
});
|
|
50
|
+
it('should throw when starting without tmi.js', async () => {
|
|
51
|
+
const ch = new TwitchChannel({ username: 'bot', oauthToken: 'oauth:xxx', channels: ['#test'] });
|
|
52
|
+
await expect(ch.start()).rejects.toThrow('tmi.js');
|
|
53
|
+
});
|
|
54
|
+
});
|
|
55
|
+
|
|
56
|
+
describe('IRCChannel', () => {
|
|
57
|
+
it('should require host, nick, channels', () => {
|
|
58
|
+
expect(() => new IRCChannel({ host: '', nick: '', channels: [] })).toThrow('host');
|
|
59
|
+
});
|
|
60
|
+
it('should throw when starting without irc-framework', async () => {
|
|
61
|
+
const ch = new IRCChannel({ host: 'irc.libera.chat', nick: 'bot', channels: ['#test'] });
|
|
62
|
+
await expect(ch.start()).rejects.toThrow('irc-framework');
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
|
|
66
|
+
// ── Context Discovery Tests (8) ──
|
|
67
|
+
|
|
68
|
+
describe('ContextDiscovery', () => {
|
|
69
|
+
let tmpDir: string;
|
|
70
|
+
let cd: ContextDiscovery;
|
|
71
|
+
|
|
72
|
+
beforeEach(() => {
|
|
73
|
+
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ctx-'));
|
|
74
|
+
cd = new ContextDiscovery();
|
|
75
|
+
});
|
|
76
|
+
|
|
77
|
+
afterEach(() => {
|
|
78
|
+
cd.stopWatching();
|
|
79
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
80
|
+
});
|
|
81
|
+
|
|
82
|
+
it('should discover AGENTS.md', () => {
|
|
83
|
+
fs.writeFileSync(path.join(tmpDir, 'AGENTS.md'), '# Agents');
|
|
84
|
+
const files = cd.discover(tmpDir);
|
|
85
|
+
expect(files.length).toBe(1);
|
|
86
|
+
expect(files[0].type).toBe('agents');
|
|
87
|
+
});
|
|
88
|
+
|
|
89
|
+
it('should discover multiple standard files', () => {
|
|
90
|
+
fs.writeFileSync(path.join(tmpDir, 'SOUL.md'), '# Soul');
|
|
91
|
+
fs.writeFileSync(path.join(tmpDir, 'USER.md'), '# User');
|
|
92
|
+
const files = cd.discover(tmpDir);
|
|
93
|
+
expect(files.length).toBe(2);
|
|
94
|
+
});
|
|
95
|
+
|
|
96
|
+
it('should return empty for no context files', () => {
|
|
97
|
+
expect(cd.discover(tmpDir)).toEqual([]);
|
|
98
|
+
});
|
|
99
|
+
|
|
100
|
+
it('should load and merge files into prompt', () => {
|
|
101
|
+
fs.writeFileSync(path.join(tmpDir, 'SOUL.md'), 'Be nice');
|
|
102
|
+
fs.writeFileSync(path.join(tmpDir, 'TOOLS.md'), 'Use web');
|
|
103
|
+
const files = cd.discover(tmpDir);
|
|
104
|
+
const prompt = cd.load(files);
|
|
105
|
+
expect(prompt).toContain('Be nice');
|
|
106
|
+
expect(prompt).toContain('Use web');
|
|
107
|
+
});
|
|
108
|
+
|
|
109
|
+
it('should add and discover custom files', () => {
|
|
110
|
+
const customPath = path.join(tmpDir, 'CUSTOM.md');
|
|
111
|
+
fs.writeFileSync(customPath, '# Custom');
|
|
112
|
+
cd.addCustomFile(customPath);
|
|
113
|
+
const files = cd.discover(tmpDir);
|
|
114
|
+
expect(files.some(f => f.type === 'custom' && f.content.includes('Custom'))).toBe(true);
|
|
115
|
+
});
|
|
116
|
+
|
|
117
|
+
it('should not duplicate custom files', () => {
|
|
118
|
+
cd.addCustomFile('test.md');
|
|
119
|
+
cd.addCustomFile('test.md');
|
|
120
|
+
// Internal check - discover won't find it but custom list should have 1
|
|
121
|
+
expect(cd.discover(tmpDir).filter(f => f.path.includes('test.md')).length).toBeLessThanOrEqual(1);
|
|
122
|
+
});
|
|
123
|
+
|
|
124
|
+
it('should assign correct types', () => {
|
|
125
|
+
fs.writeFileSync(path.join(tmpDir, 'MEMORY.md'), 'mem');
|
|
126
|
+
fs.writeFileSync(path.join(tmpDir, 'IDENTITY.md'), 'id');
|
|
127
|
+
const files = cd.discover(tmpDir);
|
|
128
|
+
expect(files.find(f => f.type === 'memory')).toBeTruthy();
|
|
129
|
+
expect(files.find(f => f.type === 'identity')).toBeTruthy();
|
|
130
|
+
});
|
|
131
|
+
|
|
132
|
+
it('should handle watch without errors', () => {
|
|
133
|
+
// Just ensure no throw
|
|
134
|
+
cd.watch(tmpDir, () => {});
|
|
135
|
+
cd.stopWatching();
|
|
136
|
+
});
|
|
137
|
+
});
|
|
138
|
+
|
|
139
|
+
// ── Session Manager Tests (8) ──
|
|
140
|
+
|
|
141
|
+
describe('SessionManager', () => {
|
|
142
|
+
let sm: SessionManager;
|
|
143
|
+
let tmpDir: string;
|
|
144
|
+
|
|
145
|
+
beforeEach(() => {
|
|
146
|
+
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'sess-'));
|
|
147
|
+
sm = new SessionManager(tmpDir);
|
|
148
|
+
});
|
|
149
|
+
|
|
150
|
+
afterEach(() => {
|
|
151
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
152
|
+
});
|
|
153
|
+
|
|
154
|
+
it('should create a session', () => {
|
|
155
|
+
const s = sm.create('agent1', 'telegram');
|
|
156
|
+
expect(s.id).toBeTruthy();
|
|
157
|
+
expect(s.agentId).toBe('agent1');
|
|
158
|
+
});
|
|
159
|
+
|
|
160
|
+
it('should get a session by id', () => {
|
|
161
|
+
const s = sm.create('agent1', 'telegram');
|
|
162
|
+
expect(sm.get(s.id)).toEqual(s);
|
|
163
|
+
expect(sm.get('nonexistent')).toBeNull();
|
|
164
|
+
});
|
|
165
|
+
|
|
166
|
+
it('should add messages', () => {
|
|
167
|
+
const s = sm.create('a', 'c');
|
|
168
|
+
sm.addMessage(s.id, { id: '1', role: 'user', content: 'hi', timestamp: Date.now() });
|
|
169
|
+
expect(sm.get(s.id)!.messages.length).toBe(1);
|
|
170
|
+
});
|
|
171
|
+
|
|
172
|
+
it('should list with filters', () => {
|
|
173
|
+
sm.create('a1', 'telegram');
|
|
174
|
+
sm.create('a2', 'discord');
|
|
175
|
+
expect(sm.list({ agentId: 'a1' }).length).toBe(1);
|
|
176
|
+
expect(sm.list({ channel: 'discord' }).length).toBe(1);
|
|
177
|
+
});
|
|
178
|
+
|
|
179
|
+
it('should compact sessions', async () => {
|
|
180
|
+
const s = sm.create('a', 'c');
|
|
181
|
+
for (let i = 0; i < 15; i++) {
|
|
182
|
+
sm.addMessage(s.id, { id: `${i}`, role: 'user', content: `msg ${i}`, timestamp: Date.now() });
|
|
183
|
+
}
|
|
184
|
+
await sm.compact(s.id);
|
|
185
|
+
expect(sm.get(s.id)!.messages.length).toBeLessThan(15);
|
|
186
|
+
expect(sm.get(s.id)!.compactedAt).toBeTruthy();
|
|
187
|
+
});
|
|
188
|
+
|
|
189
|
+
it('should prune old sessions', () => {
|
|
190
|
+
const s = sm.create('a', 'c');
|
|
191
|
+
// Manually set old lastActivity
|
|
192
|
+
(sm.get(s.id) as any).lastActivity = Date.now() - 999999;
|
|
193
|
+
expect(sm.prune(1000)).toBe(1);
|
|
194
|
+
});
|
|
195
|
+
|
|
196
|
+
it('should fork and get lineage', () => {
|
|
197
|
+
const parent = sm.create('a', 'c');
|
|
198
|
+
const child = sm.fork(parent.id);
|
|
199
|
+
expect(child.parentId).toBe(parent.id);
|
|
200
|
+
const lineage = sm.getLineage(child.id);
|
|
201
|
+
expect(lineage.length).toBe(2);
|
|
202
|
+
expect(lineage[0].id).toBe(parent.id);
|
|
203
|
+
});
|
|
204
|
+
|
|
205
|
+
it('should export as markdown and save/load', () => {
|
|
206
|
+
const s = sm.create('a', 'c');
|
|
207
|
+
sm.addMessage(s.id, { id: '1', role: 'user', content: 'hello', timestamp: Date.now() });
|
|
208
|
+
const md = sm.export(s.id);
|
|
209
|
+
expect(md).toContain('hello');
|
|
210
|
+
sm.save();
|
|
211
|
+
const sm2 = new SessionManager(tmpDir);
|
|
212
|
+
sm2.load();
|
|
213
|
+
expect(sm2.get(s.id)).toBeTruthy();
|
|
214
|
+
});
|
|
215
|
+
});
|
|
216
|
+
|
|
217
|
+
// ── Heartbeat Tests (4) ──
|
|
218
|
+
|
|
219
|
+
describe('HeartbeatManager', () => {
|
|
220
|
+
it('should require valid config', () => {
|
|
221
|
+
expect(() => new HeartbeatManager({ interval: 0, checkFn: async () => 'ok' })).toThrow('interval');
|
|
222
|
+
expect(() => new HeartbeatManager({ interval: 1000, checkFn: null as any })).toThrow('checkFn');
|
|
223
|
+
});
|
|
224
|
+
|
|
225
|
+
it('should start and stop', () => {
|
|
226
|
+
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'ok' });
|
|
227
|
+
hb.start();
|
|
228
|
+
hb.stop();
|
|
229
|
+
// Double stop should be safe
|
|
230
|
+
hb.stop();
|
|
231
|
+
});
|
|
232
|
+
|
|
233
|
+
it('should fire callbacks on beat', async () => {
|
|
234
|
+
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'alive' });
|
|
235
|
+
const statuses: string[] = [];
|
|
236
|
+
hb.onBeat(s => statuses.push(s));
|
|
237
|
+
hb.start();
|
|
238
|
+
await new Promise(r => setTimeout(r, 350));
|
|
239
|
+
hb.stop();
|
|
240
|
+
expect(statuses.length).toBeGreaterThanOrEqual(2);
|
|
241
|
+
expect(statuses[0]).toBe('alive');
|
|
242
|
+
});
|
|
243
|
+
|
|
244
|
+
it('should report last beat', async () => {
|
|
245
|
+
const hb = new HeartbeatManager({ interval: 100, checkFn: async () => 'ok' });
|
|
246
|
+
expect(hb.getLastBeat()).toBeNull();
|
|
247
|
+
hb.start();
|
|
248
|
+
await new Promise(r => setTimeout(r, 200));
|
|
249
|
+
hb.stop();
|
|
250
|
+
const beat = hb.getLastBeat();
|
|
251
|
+
expect(beat).toBeTruthy();
|
|
252
|
+
expect(beat!.status).toBe('ok');
|
|
253
|
+
});
|
|
254
|
+
});
|
package/tests/vision.test.ts
CHANGED
|
@@ -1,61 +1,61 @@
|
|
|
1
|
-
import { describe, it, expect } from 'vitest';
|
|
2
|
-
import { VisionManager, detectMimeType } from '../src/core/vision';
|
|
3
|
-
import type { ImageInput, VisionResult } from '../src/core/vision';
|
|
4
|
-
|
|
5
|
-
describe('VisionManager', () => {
|
|
6
|
-
it('should create VisionManager with defaults', () => {
|
|
7
|
-
const vm = new VisionManager();
|
|
8
|
-
expect(vm).toBeDefined();
|
|
9
|
-
});
|
|
10
|
-
|
|
11
|
-
it('should create VisionManager with custom config', () => {
|
|
12
|
-
const vm = new VisionManager({ model: 'gpt-4o-mini', maxImageSize: 5 * 1024 * 1024 });
|
|
13
|
-
expect(vm).toBeDefined();
|
|
14
|
-
});
|
|
15
|
-
|
|
16
|
-
it('should detect PNG mime type', () => {
|
|
17
|
-
const buf = Buffer.from([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]);
|
|
18
|
-
expect(detectMimeType(buf)).toBe('image/png');
|
|
19
|
-
});
|
|
20
|
-
|
|
21
|
-
it('should detect JPEG mime type', () => {
|
|
22
|
-
const buf = Buffer.from([0xFF, 0xD8, 0xFF, 0xE0]);
|
|
23
|
-
expect(detectMimeType(buf)).toBe('image/jpeg');
|
|
24
|
-
});
|
|
25
|
-
|
|
26
|
-
it('should detect GIF mime type', () => {
|
|
27
|
-
const buf = Buffer.from([0x47, 0x49, 0x46, 0x38, 0x39, 0x61]);
|
|
28
|
-
expect(detectMimeType(buf)).toBe('image/gif');
|
|
29
|
-
});
|
|
30
|
-
|
|
31
|
-
it('should detect WebP mime type', () => {
|
|
32
|
-
const buf = Buffer.from([0x52, 0x49, 0x46, 0x46, 0x00, 0x00, 0x00, 0x00]);
|
|
33
|
-
expect(detectMimeType(buf)).toBe('image/webp');
|
|
34
|
-
});
|
|
35
|
-
|
|
36
|
-
it('should return octet-stream for unknown', () => {
|
|
37
|
-
const buf = Buffer.from([0x00, 0x01, 0x02]);
|
|
38
|
-
expect(detectMimeType(buf)).toBe('application/octet-stream');
|
|
39
|
-
});
|
|
40
|
-
|
|
41
|
-
it('should prepare multimodal message with URL images', () => {
|
|
42
|
-
const vm = new VisionManager();
|
|
43
|
-
const images: ImageInput[] = [{ type: 'url', data: 'https://example.com/img.png' }];
|
|
44
|
-
const msgs = vm.prepareMessage(images, 'Describe this');
|
|
45
|
-
expect(msgs).toHaveLength(1);
|
|
46
|
-
expect((msgs[0] as any).content).toHaveLength(2);
|
|
47
|
-
expect((msgs[0] as any).content[0].type).toBe('text');
|
|
48
|
-
expect((msgs[0] as any).content[1].type).toBe('image_url');
|
|
49
|
-
});
|
|
50
|
-
|
|
51
|
-
it('should throw on analyze without API key', async () => {
|
|
52
|
-
const vm = new VisionManager({ apiKey: '' });
|
|
53
|
-
const input: ImageInput = { type: 'url', data: 'https://example.com/img.png' };
|
|
54
|
-
await expect(vm.analyze(input)).rejects.toThrow('Vision API key not configured');
|
|
55
|
-
});
|
|
56
|
-
|
|
57
|
-
it('should throw on compareImages with less than 2 images', async () => {
|
|
58
|
-
const vm = new VisionManager({ apiKey: 'test' });
|
|
59
|
-
await expect(vm.compareImages([{ type: 'url', data: 'https://example.com/1.png' }])).rejects.toThrow('at least 2');
|
|
60
|
-
});
|
|
61
|
-
});
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { VisionManager, detectMimeType } from '../src/core/vision';
|
|
3
|
+
import type { ImageInput, VisionResult } from '../src/core/vision';
|
|
4
|
+
|
|
5
|
+
describe('VisionManager', () => {
|
|
6
|
+
it('should create VisionManager with defaults', () => {
|
|
7
|
+
const vm = new VisionManager();
|
|
8
|
+
expect(vm).toBeDefined();
|
|
9
|
+
});
|
|
10
|
+
|
|
11
|
+
it('should create VisionManager with custom config', () => {
|
|
12
|
+
const vm = new VisionManager({ model: 'gpt-4o-mini', maxImageSize: 5 * 1024 * 1024 });
|
|
13
|
+
expect(vm).toBeDefined();
|
|
14
|
+
});
|
|
15
|
+
|
|
16
|
+
it('should detect PNG mime type', () => {
|
|
17
|
+
const buf = Buffer.from([0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A]);
|
|
18
|
+
expect(detectMimeType(buf)).toBe('image/png');
|
|
19
|
+
});
|
|
20
|
+
|
|
21
|
+
it('should detect JPEG mime type', () => {
|
|
22
|
+
const buf = Buffer.from([0xFF, 0xD8, 0xFF, 0xE0]);
|
|
23
|
+
expect(detectMimeType(buf)).toBe('image/jpeg');
|
|
24
|
+
});
|
|
25
|
+
|
|
26
|
+
it('should detect GIF mime type', () => {
|
|
27
|
+
const buf = Buffer.from([0x47, 0x49, 0x46, 0x38, 0x39, 0x61]);
|
|
28
|
+
expect(detectMimeType(buf)).toBe('image/gif');
|
|
29
|
+
});
|
|
30
|
+
|
|
31
|
+
it('should detect WebP mime type', () => {
|
|
32
|
+
const buf = Buffer.from([0x52, 0x49, 0x46, 0x46, 0x00, 0x00, 0x00, 0x00]);
|
|
33
|
+
expect(detectMimeType(buf)).toBe('image/webp');
|
|
34
|
+
});
|
|
35
|
+
|
|
36
|
+
it('should return octet-stream for unknown', () => {
|
|
37
|
+
const buf = Buffer.from([0x00, 0x01, 0x02]);
|
|
38
|
+
expect(detectMimeType(buf)).toBe('application/octet-stream');
|
|
39
|
+
});
|
|
40
|
+
|
|
41
|
+
it('should prepare multimodal message with URL images', () => {
|
|
42
|
+
const vm = new VisionManager();
|
|
43
|
+
const images: ImageInput[] = [{ type: 'url', data: 'https://example.com/img.png' }];
|
|
44
|
+
const msgs = vm.prepareMessage(images, 'Describe this');
|
|
45
|
+
expect(msgs).toHaveLength(1);
|
|
46
|
+
expect((msgs[0] as any).content).toHaveLength(2);
|
|
47
|
+
expect((msgs[0] as any).content[0].type).toBe('text');
|
|
48
|
+
expect((msgs[0] as any).content[1].type).toBe('image_url');
|
|
49
|
+
});
|
|
50
|
+
|
|
51
|
+
it('should throw on analyze without API key', async () => {
|
|
52
|
+
const vm = new VisionManager({ apiKey: '' });
|
|
53
|
+
const input: ImageInput = { type: 'url', data: 'https://example.com/img.png' };
|
|
54
|
+
await expect(vm.analyze(input)).rejects.toThrow('Vision API key not configured');
|
|
55
|
+
});
|
|
56
|
+
|
|
57
|
+
it('should throw on compareImages with less than 2 images', async () => {
|
|
58
|
+
const vm = new VisionManager({ apiKey: 'test' });
|
|
59
|
+
await expect(vm.compareImages([{ type: 'url', data: 'https://example.com/1.png' }])).rejects.toThrow('at least 2');
|
|
60
|
+
});
|
|
61
|
+
});
|