@timmeck/brain-core 2.36.62 → 2.36.63
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/dist/guardrails/guardrail-engine.d.ts +2 -0
- package/dist/guardrails/guardrail-engine.js +17 -9
- package/dist/guardrails/guardrail-engine.js.map +1 -1
- package/dist/index.d.ts +2 -0
- package/dist/index.js +1 -0
- package/dist/index.js.map +1 -1
- package/dist/utils/memory-watchdog.d.ts +33 -0
- package/dist/utils/memory-watchdog.js +88 -0
- package/dist/utils/memory-watchdog.js.map +1 -0
- package/package.json +1 -1
- package/dist/action/__tests__/action-bridge.test.d.ts +0 -1
- package/dist/action/__tests__/action-bridge.test.js +0 -175
- package/dist/action/__tests__/action-bridge.test.js.map +0 -1
- package/dist/action/__tests__/adjust-parameter-handler.test.d.ts +0 -1
- package/dist/action/__tests__/adjust-parameter-handler.test.js +0 -138
- package/dist/action/__tests__/adjust-parameter-handler.test.js.map +0 -1
- package/dist/action/__tests__/content-handler.test.d.ts +0 -1
- package/dist/action/__tests__/content-handler.test.js +0 -100
- package/dist/action/__tests__/content-handler.test.js.map +0 -1
- package/dist/action/__tests__/creative-seed-handler.test.d.ts +0 -1
- package/dist/action/__tests__/creative-seed-handler.test.js +0 -96
- package/dist/action/__tests__/creative-seed-handler.test.js.map +0 -1
- package/dist/action/__tests__/trade-handler.test.d.ts +0 -1
- package/dist/action/__tests__/trade-handler.test.js +0 -165
- package/dist/action/__tests__/trade-handler.test.js.map +0 -1
- package/dist/active-learning/__tests__/active-learning.test.d.ts +0 -1
- package/dist/active-learning/__tests__/active-learning.test.js +0 -132
- package/dist/active-learning/__tests__/active-learning.test.js.map +0 -1
- package/dist/agent-training/__tests__/agent-trainer.test.d.ts +0 -1
- package/dist/agent-training/__tests__/agent-trainer.test.js +0 -158
- package/dist/agent-training/__tests__/agent-trainer.test.js.map +0 -1
- package/dist/agent-training/__tests__/sub-agent-factory.test.d.ts +0 -1
- package/dist/agent-training/__tests__/sub-agent-factory.test.js +0 -100
- package/dist/agent-training/__tests__/sub-agent-factory.test.js.map +0 -1
- package/dist/agent-training/__tests__/sub-agent.test.d.ts +0 -1
- package/dist/agent-training/__tests__/sub-agent.test.js +0 -102
- package/dist/agent-training/__tests__/sub-agent.test.js.map +0 -1
- package/dist/chat/__tests__/chat-engine.test.d.ts +0 -1
- package/dist/chat/__tests__/chat-engine.test.js +0 -209
- package/dist/chat/__tests__/chat-engine.test.js.map +0 -1
- package/dist/checkpoint/__tests__/checkpoint-manager.test.d.ts +0 -1
- package/dist/checkpoint/__tests__/checkpoint-manager.test.js +0 -160
- package/dist/checkpoint/__tests__/checkpoint-manager.test.js.map +0 -1
- package/dist/code-health/__tests__/code-health.test.d.ts +0 -1
- package/dist/code-health/__tests__/code-health.test.js +0 -123
- package/dist/code-health/__tests__/code-health.test.js.map +0 -1
- package/dist/codegen/__tests__/code-forge.test.d.ts +0 -1
- package/dist/codegen/__tests__/code-forge.test.js +0 -105
- package/dist/codegen/__tests__/code-forge.test.js.map +0 -1
- package/dist/config/__tests__/loader.test.d.ts +0 -1
- package/dist/config/__tests__/loader.test.js +0 -86
- package/dist/config/__tests__/loader.test.js.map +0 -1
- package/dist/consciousness/__tests__/thought-stream.test.d.ts +0 -1
- package/dist/consciousness/__tests__/thought-stream.test.js +0 -62
- package/dist/consciousness/__tests__/thought-stream.test.js.map +0 -1
- package/dist/consensus/__tests__/consensus.test.d.ts +0 -1
- package/dist/consensus/__tests__/consensus.test.js +0 -159
- package/dist/consensus/__tests__/consensus.test.js.map +0 -1
- package/dist/content/__tests__/auto-publisher.test.d.ts +0 -1
- package/dist/content/__tests__/auto-publisher.test.js +0 -125
- package/dist/content/__tests__/auto-publisher.test.js.map +0 -1
- package/dist/content/__tests__/content-forge.test.d.ts +0 -1
- package/dist/content/__tests__/content-forge.test.js +0 -117
- package/dist/content/__tests__/content-forge.test.js.map +0 -1
- package/dist/creative/__tests__/creative-engine.test.d.ts +0 -1
- package/dist/creative/__tests__/creative-engine.test.js +0 -229
- package/dist/creative/__tests__/creative-engine.test.js.map +0 -1
- package/dist/cross-brain/__tests__/borg-sync-engine.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/borg-sync-engine.test.js +0 -240
- package/dist/cross-brain/__tests__/borg-sync-engine.test.js.map +0 -1
- package/dist/cross-brain/__tests__/client.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/client.test.js +0 -31
- package/dist/cross-brain/__tests__/client.test.js.map +0 -1
- package/dist/cross-brain/__tests__/notifications.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/notifications.test.js +0 -52
- package/dist/cross-brain/__tests__/notifications.test.js.map +0 -1
- package/dist/cross-brain/__tests__/signal-router.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/signal-router.test.js +0 -218
- package/dist/cross-brain/__tests__/signal-router.test.js.map +0 -1
- package/dist/cross-brain/__tests__/subscription.test.d.ts +0 -1
- package/dist/cross-brain/__tests__/subscription.test.js +0 -75
- package/dist/cross-brain/__tests__/subscription.test.js.map +0 -1
- package/dist/dashboard/__tests__/command-center-server.test.d.ts +0 -1
- package/dist/dashboard/__tests__/command-center-server.test.js +0 -522
- package/dist/dashboard/__tests__/command-center-server.test.js.map +0 -1
- package/dist/dashboard/__tests__/server.test.d.ts +0 -1
- package/dist/dashboard/__tests__/server.test.js +0 -105
- package/dist/dashboard/__tests__/server.test.js.map +0 -1
- package/dist/db/__tests__/connection.test.d.ts +0 -1
- package/dist/db/__tests__/connection.test.js +0 -135
- package/dist/db/__tests__/connection.test.js.map +0 -1
- package/dist/dream/__tests__/dream-engine.test.d.ts +0 -1
- package/dist/dream/__tests__/dream-engine.test.js +0 -184
- package/dist/dream/__tests__/dream-engine.test.js.map +0 -1
- package/dist/embeddings/__tests__/engine.test.d.ts +0 -1
- package/dist/embeddings/__tests__/engine.test.js +0 -82
- package/dist/embeddings/__tests__/engine.test.js.map +0 -1
- package/dist/feedback/__tests__/feedback-engine.test.d.ts +0 -1
- package/dist/feedback/__tests__/feedback-engine.test.js +0 -156
- package/dist/feedback/__tests__/feedback-engine.test.js.map +0 -1
- package/dist/feedback/__tests__/feedback-router.test.d.ts +0 -1
- package/dist/feedback/__tests__/feedback-router.test.js +0 -173
- package/dist/feedback/__tests__/feedback-router.test.js.map +0 -1
- package/dist/goals/__tests__/goal-engine.test.d.ts +0 -1
- package/dist/goals/__tests__/goal-engine.test.js +0 -203
- package/dist/goals/__tests__/goal-engine.test.js.map +0 -1
- package/dist/guardrails/__tests__/guardrail-engine.test.d.ts +0 -1
- package/dist/guardrails/__tests__/guardrail-engine.test.js +0 -343
- package/dist/guardrails/__tests__/guardrail-engine.test.js.map +0 -1
- package/dist/ipc/__tests__/integration.test.d.ts +0 -1
- package/dist/ipc/__tests__/integration.test.js +0 -352
- package/dist/ipc/__tests__/integration.test.js.map +0 -1
- package/dist/ipc/__tests__/protocol.test.d.ts +0 -1
- package/dist/ipc/__tests__/protocol.test.js +0 -69
- package/dist/ipc/__tests__/protocol.test.js.map +0 -1
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.d.ts +0 -1
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.js +0 -114
- package/dist/knowledge-graph/__tests__/contradiction-resolver.test.js.map +0 -1
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.d.ts +0 -1
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.js +0 -215
- package/dist/knowledge-graph/__tests__/knowledge-graph.test.js.map +0 -1
- package/dist/learning/__tests__/base-engine.test.d.ts +0 -1
- package/dist/learning/__tests__/base-engine.test.js +0 -49
- package/dist/learning/__tests__/base-engine.test.js.map +0 -1
- package/dist/llm/__tests__/anthropic-provider.test.d.ts +0 -1
- package/dist/llm/__tests__/anthropic-provider.test.js +0 -121
- package/dist/llm/__tests__/anthropic-provider.test.js.map +0 -1
- package/dist/llm/__tests__/llm-service.test.d.ts +0 -1
- package/dist/llm/__tests__/llm-service.test.js +0 -507
- package/dist/llm/__tests__/llm-service.test.js.map +0 -1
- package/dist/llm/__tests__/ollama-embedding.test.d.ts +0 -1
- package/dist/llm/__tests__/ollama-embedding.test.js +0 -128
- package/dist/llm/__tests__/ollama-embedding.test.js.map +0 -1
- package/dist/llm/__tests__/ollama-provider.test.d.ts +0 -1
- package/dist/llm/__tests__/ollama-provider.test.js +0 -213
- package/dist/llm/__tests__/ollama-provider.test.js.map +0 -1
- package/dist/llm/__tests__/provider.test.d.ts +0 -1
- package/dist/llm/__tests__/provider.test.js +0 -126
- package/dist/llm/__tests__/provider.test.js.map +0 -1
- package/dist/math/__tests__/time-decay.test.d.ts +0 -1
- package/dist/math/__tests__/time-decay.test.js +0 -37
- package/dist/math/__tests__/time-decay.test.js.map +0 -1
- package/dist/math/__tests__/wilson-score.test.d.ts +0 -1
- package/dist/math/__tests__/wilson-score.test.js +0 -43
- package/dist/math/__tests__/wilson-score.test.js.map +0 -1
- package/dist/memory/__tests__/base-memory-engine.test.d.ts +0 -1
- package/dist/memory/__tests__/base-memory-engine.test.js +0 -233
- package/dist/memory/__tests__/base-memory-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/auto-experiment-engine.test.d.ts +0 -1
- package/dist/metacognition/__tests__/auto-experiment-engine.test.js +0 -218
- package/dist/metacognition/__tests__/auto-experiment-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/evolution-engine.test.d.ts +0 -1
- package/dist/metacognition/__tests__/evolution-engine.test.js +0 -473
- package/dist/metacognition/__tests__/evolution-engine.test.js.map +0 -1
- package/dist/metacognition/__tests__/parameter-registry.test.d.ts +0 -1
- package/dist/metacognition/__tests__/parameter-registry.test.js +0 -223
- package/dist/metacognition/__tests__/parameter-registry.test.js.map +0 -1
- package/dist/missions/__tests__/mission-engine.test.d.ts +0 -1
- package/dist/missions/__tests__/mission-engine.test.js +0 -67
- package/dist/missions/__tests__/mission-engine.test.js.map +0 -1
- package/dist/notifications/__tests__/notification-service.test.d.ts +0 -1
- package/dist/notifications/__tests__/notification-service.test.js +0 -176
- package/dist/notifications/__tests__/notification-service.test.js.map +0 -1
- package/dist/plugin/__tests__/plugin-registry.test.d.ts +0 -1
- package/dist/plugin/__tests__/plugin-registry.test.js +0 -166
- package/dist/plugin/__tests__/plugin-registry.test.js.map +0 -1
- package/dist/prediction/__tests__/prediction-engine.test.d.ts +0 -1
- package/dist/prediction/__tests__/prediction-engine.test.js +0 -252
- package/dist/prediction/__tests__/prediction-engine.test.js.map +0 -1
- package/dist/proactive/__tests__/proactive-engine.test.d.ts +0 -1
- package/dist/proactive/__tests__/proactive-engine.test.js +0 -183
- package/dist/proactive/__tests__/proactive-engine.test.js.map +0 -1
- package/dist/rag/__tests__/rag-engine.test.d.ts +0 -1
- package/dist/rag/__tests__/rag-engine.test.js +0 -235
- package/dist/rag/__tests__/rag-engine.test.js.map +0 -1
- package/dist/research/__tests__/autonomy-features.test.d.ts +0 -1
- package/dist/research/__tests__/autonomy-features.test.js +0 -155
- package/dist/research/__tests__/autonomy-features.test.js.map +0 -1
- package/dist/research/__tests__/base-engine.test.d.ts +0 -1
- package/dist/research/__tests__/base-engine.test.js +0 -56
- package/dist/research/__tests__/base-engine.test.js.map +0 -1
- package/dist/research/__tests__/semantic-compressor.test.d.ts +0 -1
- package/dist/research/__tests__/semantic-compressor.test.js +0 -153
- package/dist/research/__tests__/semantic-compressor.test.js.map +0 -1
- package/dist/research/adapters/__tests__/playwright-adapter.test.d.ts +0 -1
- package/dist/research/adapters/__tests__/playwright-adapter.test.js +0 -64
- package/dist/research/adapters/__tests__/playwright-adapter.test.js.map +0 -1
- package/dist/research/adapters/__tests__/web-adapters.test.d.ts +0 -1
- package/dist/research/adapters/__tests__/web-adapters.test.js +0 -106
- package/dist/research/adapters/__tests__/web-adapters.test.js.map +0 -1
- package/dist/sandbox/__tests__/code-sandbox.test.d.ts +0 -1
- package/dist/sandbox/__tests__/code-sandbox.test.js +0 -182
- package/dist/sandbox/__tests__/code-sandbox.test.js.map +0 -1
- package/dist/scanner/__tests__/signal-scanner.test.d.ts +0 -1
- package/dist/scanner/__tests__/signal-scanner.test.js +0 -142
- package/dist/scanner/__tests__/signal-scanner.test.js.map +0 -1
- package/dist/self-modification/__tests__/self-modification-engine.test.d.ts +0 -1
- package/dist/self-modification/__tests__/self-modification-engine.test.js +0 -313
- package/dist/self-modification/__tests__/self-modification-engine.test.js.map +0 -1
- package/dist/strategy/__tests__/strategy-forge.test.d.ts +0 -1
- package/dist/strategy/__tests__/strategy-forge.test.js +0 -190
- package/dist/strategy/__tests__/strategy-forge.test.js.map +0 -1
- package/dist/strategy/__tests__/strategy-mutator.test.d.ts +0 -1
- package/dist/strategy/__tests__/strategy-mutator.test.js +0 -141
- package/dist/strategy/__tests__/strategy-mutator.test.js.map +0 -1
- package/dist/synapses/__tests__/activation.test.d.ts +0 -1
- package/dist/synapses/__tests__/activation.test.js +0 -87
- package/dist/synapses/__tests__/activation.test.js.map +0 -1
- package/dist/synapses/__tests__/decay.test.d.ts +0 -1
- package/dist/synapses/__tests__/decay.test.js +0 -73
- package/dist/synapses/__tests__/decay.test.js.map +0 -1
- package/dist/synapses/__tests__/hebbian.test.d.ts +0 -1
- package/dist/synapses/__tests__/hebbian.test.js +0 -95
- package/dist/synapses/__tests__/hebbian.test.js.map +0 -1
- package/dist/synapses/__tests__/pathfinder.test.d.ts +0 -1
- package/dist/synapses/__tests__/pathfinder.test.js +0 -74
- package/dist/synapses/__tests__/pathfinder.test.js.map +0 -1
- package/dist/synapses/__tests__/synapse-manager.test.d.ts +0 -1
- package/dist/synapses/__tests__/synapse-manager.test.js +0 -94
- package/dist/synapses/__tests__/synapse-manager.test.js.map +0 -1
- package/dist/teaching/__tests__/teaching.test.d.ts +0 -1
- package/dist/teaching/__tests__/teaching.test.js +0 -252
- package/dist/teaching/__tests__/teaching.test.js.map +0 -1
- package/dist/techradar/__tests__/techradar-engine.test.d.ts +0 -1
- package/dist/techradar/__tests__/techradar-engine.test.js +0 -246
- package/dist/techradar/__tests__/techradar-engine.test.js.map +0 -1
- package/dist/tool-learning/__tests__/tool-learning.test.d.ts +0 -1
- package/dist/tool-learning/__tests__/tool-learning.test.js +0 -187
- package/dist/tool-learning/__tests__/tool-learning.test.js.map +0 -1
- package/dist/user-model/__tests__/user-model.test.d.ts +0 -1
- package/dist/user-model/__tests__/user-model.test.js +0 -142
- package/dist/user-model/__tests__/user-model.test.js.map +0 -1
- package/dist/utils/__tests__/batch-queue.test.d.ts +0 -1
- package/dist/utils/__tests__/batch-queue.test.js +0 -85
- package/dist/utils/__tests__/batch-queue.test.js.map +0 -1
- package/dist/utils/__tests__/events.test.d.ts +0 -1
- package/dist/utils/__tests__/events.test.js +0 -38
- package/dist/utils/__tests__/events.test.js.map +0 -1
- package/dist/utils/__tests__/hash.test.d.ts +0 -1
- package/dist/utils/__tests__/hash.test.js +0 -25
- package/dist/utils/__tests__/hash.test.js.map +0 -1
- package/dist/utils/__tests__/logger.test.d.ts +0 -1
- package/dist/utils/__tests__/logger.test.js +0 -55
- package/dist/utils/__tests__/logger.test.js.map +0 -1
- package/dist/utils/__tests__/paths.test.d.ts +0 -1
- package/dist/utils/__tests__/paths.test.js +0 -50
- package/dist/utils/__tests__/paths.test.js.map +0 -1
- package/dist/utils/__tests__/retry.test.d.ts +0 -1
- package/dist/utils/__tests__/retry.test.js +0 -91
- package/dist/utils/__tests__/retry.test.js.map +0 -1
- package/dist/watchdog/__tests__/watchdog-service.test.d.ts +0 -1
- package/dist/watchdog/__tests__/watchdog-service.test.js +0 -113
- package/dist/watchdog/__tests__/watchdog-service.test.js.map +0 -1
|
@@ -1,132 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import Database from 'better-sqlite3';
|
|
3
|
-
import { ActiveLearner, runActiveLearningMigration } from '../active-learner.js';
|
|
4
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
5
|
-
getLogger: () => ({
|
|
6
|
-
info: vi.fn(),
|
|
7
|
-
warn: vi.fn(),
|
|
8
|
-
error: vi.fn(),
|
|
9
|
-
debug: vi.fn(),
|
|
10
|
-
}),
|
|
11
|
-
}));
|
|
12
|
-
describe('ActiveLearner', () => {
|
|
13
|
-
let db;
|
|
14
|
-
let learner;
|
|
15
|
-
beforeEach(() => {
|
|
16
|
-
db = new Database(':memory:');
|
|
17
|
-
learner = new ActiveLearner(db, { brainName: 'brain' });
|
|
18
|
-
});
|
|
19
|
-
afterEach(() => {
|
|
20
|
-
db.close();
|
|
21
|
-
});
|
|
22
|
-
it('addGap creates a learning gap', () => {
|
|
23
|
-
const gap = learner.addGap('knowledge_void', 'quantum computing', 'No data on quantum algorithms', 0.8, 0.3);
|
|
24
|
-
expect(gap.id).toBeDefined();
|
|
25
|
-
expect(gap.gapType).toBe('knowledge_void');
|
|
26
|
-
expect(gap.topic).toBe('quantum computing');
|
|
27
|
-
expect(gap.description).toBe('No data on quantum algorithms');
|
|
28
|
-
expect(gap.impact).toBe(0.8);
|
|
29
|
-
expect(gap.ease).toBe(0.3);
|
|
30
|
-
expect(gap.status).toBe('open');
|
|
31
|
-
});
|
|
32
|
-
it('getOpenGaps returns gaps sorted by priority', () => {
|
|
33
|
-
learner.addGap('knowledge_void', 'topic-a', undefined, 0.9, 0.9); // priority = 0.81
|
|
34
|
-
learner.addGap('low_confidence', 'topic-b', undefined, 0.5, 0.5); // priority = 0.25
|
|
35
|
-
learner.addGap('unanswered', 'topic-c', undefined, 0.8, 0.7); // priority = 0.56
|
|
36
|
-
const gaps = learner.getOpenGaps();
|
|
37
|
-
expect(gaps.length).toBe(3);
|
|
38
|
-
// Sorted by impact * ease descending
|
|
39
|
-
expect(gaps[0].topic).toBe('topic-a');
|
|
40
|
-
expect(gaps[1].topic).toBe('topic-c');
|
|
41
|
-
expect(gaps[2].topic).toBe('topic-b');
|
|
42
|
-
});
|
|
43
|
-
it('planLearning selects strategy based on gap type', () => {
|
|
44
|
-
const voidGap = learner.addGap('knowledge_void', 'void-topic');
|
|
45
|
-
const lowConf = learner.addGap('low_confidence', 'low-conf-topic');
|
|
46
|
-
const unanswered = learner.addGap('unanswered', 'unanswered-topic');
|
|
47
|
-
const crossBrain = learner.addGap('cross_brain', 'cross-topic');
|
|
48
|
-
expect(learner.planLearning(voidGap.id).strategy).toBe('research_mission');
|
|
49
|
-
expect(learner.planLearning(lowConf.id).strategy).toBe('experiment');
|
|
50
|
-
expect(learner.planLearning(unanswered.id).strategy).toBe('ask_user');
|
|
51
|
-
expect(learner.planLearning(crossBrain.id).strategy).toBe('teach_request');
|
|
52
|
-
});
|
|
53
|
-
it('prioritize sorts gaps by impact * ease descending', () => {
|
|
54
|
-
const gaps = [
|
|
55
|
-
{ gapType: 'knowledge_void', topic: 'a', impact: 0.3, ease: 0.3, status: 'open' },
|
|
56
|
-
{ gapType: 'low_confidence', topic: 'b', impact: 0.9, ease: 0.9, status: 'open' },
|
|
57
|
-
{ gapType: 'unanswered', topic: 'c', impact: 0.5, ease: 0.8, status: 'open' },
|
|
58
|
-
];
|
|
59
|
-
const sorted = learner.prioritize(gaps);
|
|
60
|
-
expect(sorted[0].topic).toBe('b'); // 0.81
|
|
61
|
-
expect(sorted[1].topic).toBe('c'); // 0.40
|
|
62
|
-
expect(sorted[2].topic).toBe('a'); // 0.09
|
|
63
|
-
});
|
|
64
|
-
it('closeGap records outcome and marks closed', () => {
|
|
65
|
-
const gap = learner.addGap('knowledge_void', 'test-close');
|
|
66
|
-
learner.planLearning(gap.id);
|
|
67
|
-
const closed = learner.closeGap(gap.id, 'success');
|
|
68
|
-
expect(closed).toBe(true);
|
|
69
|
-
// Verify it's no longer in open gaps
|
|
70
|
-
const openGaps = learner.getOpenGaps();
|
|
71
|
-
const found = openGaps.find(g => g.id === gap.id);
|
|
72
|
-
expect(found).toBeUndefined();
|
|
73
|
-
});
|
|
74
|
-
it('strategy success tracking in getStatus', () => {
|
|
75
|
-
const g1 = learner.addGap('knowledge_void', 'g1');
|
|
76
|
-
const g2 = learner.addGap('low_confidence', 'g2');
|
|
77
|
-
const g3 = learner.addGap('unanswered', 'g3');
|
|
78
|
-
learner.planLearning(g1.id);
|
|
79
|
-
learner.planLearning(g2.id);
|
|
80
|
-
learner.planLearning(g3.id);
|
|
81
|
-
learner.closeGap(g1.id, 'success');
|
|
82
|
-
learner.closeGap(g2.id, 'failure');
|
|
83
|
-
learner.closeGap(g3.id, 'success');
|
|
84
|
-
const status = learner.getStatus();
|
|
85
|
-
// 3 plan strategies (pending) + 3 close strategies = 6 total
|
|
86
|
-
// 2 success out of 6
|
|
87
|
-
expect(status.strategySuccessRate).toBeCloseTo(2 / 6, 2);
|
|
88
|
-
});
|
|
89
|
-
it('identifyGaps returns empty for no sources', () => {
|
|
90
|
-
const gaps = learner.identifyGaps();
|
|
91
|
-
expect(gaps).toEqual([]);
|
|
92
|
-
});
|
|
93
|
-
it('identifyGaps finds gaps from sources', () => {
|
|
94
|
-
const gaps = learner.identifyGaps({
|
|
95
|
-
knowledgeVoids: [{ topic: 'blockchain', description: 'No data' }],
|
|
96
|
-
lowConfidenceFacts: [{ topic: 'AI safety', confidence: 0.1 }],
|
|
97
|
-
ragMisses: [{ query: 'quantum entanglement' }],
|
|
98
|
-
});
|
|
99
|
-
expect(gaps.length).toBe(3);
|
|
100
|
-
expect(gaps[0].gapType).toBe('knowledge_void');
|
|
101
|
-
expect(gaps[1].gapType).toBe('low_confidence');
|
|
102
|
-
expect(gaps[2].gapType).toBe('unanswered');
|
|
103
|
-
});
|
|
104
|
-
it('gap limit enforcement', () => {
|
|
105
|
-
const smallLearner = new ActiveLearner(db, { brainName: 'brain', maxOpenGaps: 3 });
|
|
106
|
-
smallLearner.addGap('knowledge_void', 'g1');
|
|
107
|
-
smallLearner.addGap('knowledge_void', 'g2');
|
|
108
|
-
smallLearner.addGap('knowledge_void', 'g3');
|
|
109
|
-
expect(() => smallLearner.addGap('knowledge_void', 'g4'))
|
|
110
|
-
.toThrow(/Maximum open gaps/);
|
|
111
|
-
});
|
|
112
|
-
it('migration is idempotent', () => {
|
|
113
|
-
runActiveLearningMigration(db);
|
|
114
|
-
runActiveLearningMigration(db);
|
|
115
|
-
const tables = db
|
|
116
|
-
.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name LIKE 'learning%'")
|
|
117
|
-
.all();
|
|
118
|
-
expect(tables.length).toBe(2); // learning_gaps + learning_strategies
|
|
119
|
-
});
|
|
120
|
-
it('getStatus returns correct summary', () => {
|
|
121
|
-
learner.addGap('knowledge_void', 'open-gap');
|
|
122
|
-
const g2 = learner.addGap('low_confidence', 'to-close');
|
|
123
|
-
learner.planLearning(g2.id);
|
|
124
|
-
learner.closeGap(g2.id, 'success');
|
|
125
|
-
const status = learner.getStatus();
|
|
126
|
-
expect(status.totalGaps).toBe(2);
|
|
127
|
-
expect(status.openGaps).toBe(1);
|
|
128
|
-
expect(status.closedGaps).toBe(1);
|
|
129
|
-
expect(status.strategySuccessRate).toBeGreaterThan(0);
|
|
130
|
-
});
|
|
131
|
-
});
|
|
132
|
-
//# sourceMappingURL=active-learning.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"active-learning.test.js","sourceRoot":"","sources":["../../../src/active-learning/__tests__/active-learning.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAEjF,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;QAChB,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;QACb,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,EAAqB,CAAC;IAC1B,IAAI,OAAsB,CAAC;IAE3B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9B,OAAO,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,EAAE,+BAA+B,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7G,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;QAC9D,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,kBAAkB;QACpF,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,kBAAkB;QACpF,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,kBAAkB;QAEhF,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,qCAAqC;QACrC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,CAAC;QACnE,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;QAEhE,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,EAAG,CAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC7E,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,EAAG,CAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,EAAG,CAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxE,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,EAAG,CAAE,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,IAAI,GAAG;YACX,EAAE,OAAO,EAAE,gBAAyB,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,MAAe,EAAE;YACnG,EAAE,OAAO,EAAE,gBAAyB,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,MAAe,EAAE;YACnG,EAAE,OAAO,EAAE,YAAqB,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,MAAe,EAAE;SAChG,CAAC;QAEF,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;QAC3D,OAAO,CAAC,YAAY,CAAC,GAAG,CAAC,EAAG,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAG,EAAE,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE1B,qCAAqC;QACrC,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAClD,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QAE9C,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,EAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,EAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,EAAG,CAAC,CAAC;QAE7B,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAG,EAAE,SAAS,CAAC,CAAC;QACpC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAG,EAAE,SAAS,CAAC,CAAC;QACpC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAG,EAAE,SAAS,CAAC,CAAC;QAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,6DAA6D;QAC7D,qBAAqB;QACrB,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,IAAI,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,IAAI,GAAG,OAAO,CAAC,YAAY,CAAC;YAChC,cAAc,EAAE,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC;YACjE,kBAAkB,EAAE,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;YAC7D,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;SAC/C,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;QAEnF,YAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAC5C,YAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAC5C,YAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QAE5C,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;aACtD,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,0BAA0B,CAAC,EAAE,CAAC,CAAC;QAC/B,0BAA0B,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,MAAM,GAAG,EAAE;aACd,OAAO,CAAC,6EAA6E,CAAC;aACtF,GAAG,EAAE,CAAC;QACT,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,sCAAsC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QAC7C,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;QACxD,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC,EAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAG,EAAE,SAAS,CAAC,CAAC;QAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,158 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import Database from 'better-sqlite3';
|
|
3
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
4
|
-
getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
|
|
5
|
-
}));
|
|
6
|
-
import { AgentTrainer, runTrainerMigration } from '../agent-trainer.js';
|
|
7
|
-
import { BenchmarkSuite, runBenchmarkMigration } from '../benchmark-suite.js';
|
|
8
|
-
describe('AgentTrainer', () => {
|
|
9
|
-
let db;
|
|
10
|
-
let suite;
|
|
11
|
-
const easyCases = [
|
|
12
|
-
{ input: '2+2', expected: '4', category: 'math', difficulty: 'easy' },
|
|
13
|
-
{ input: '3+1', expected: '4', category: 'math', difficulty: 'easy' },
|
|
14
|
-
];
|
|
15
|
-
const mediumCases = [
|
|
16
|
-
{ input: '10*5', expected: '50', category: 'math', difficulty: 'medium' },
|
|
17
|
-
{ input: '100/4', expected: '25', category: 'math', difficulty: 'medium' },
|
|
18
|
-
];
|
|
19
|
-
const hardCases = [
|
|
20
|
-
{ input: 'sqrt(144)', expected: '12', category: 'math', difficulty: 'hard' },
|
|
21
|
-
{ input: '2^10', expected: '1024', category: 'math', difficulty: 'hard' },
|
|
22
|
-
];
|
|
23
|
-
beforeEach(() => {
|
|
24
|
-
db = new Database(':memory:');
|
|
25
|
-
runBenchmarkMigration(db);
|
|
26
|
-
runTrainerMigration(db);
|
|
27
|
-
suite = new BenchmarkSuite(db);
|
|
28
|
-
suite.addCases([...easyCases, ...mediumCases, ...hardCases]);
|
|
29
|
-
});
|
|
30
|
-
afterEach(() => { db.close(); });
|
|
31
|
-
it('creates trainer without error', () => {
|
|
32
|
-
const trainer = new AgentTrainer(db);
|
|
33
|
-
expect(trainer).toBeInstanceOf(AgentTrainer);
|
|
34
|
-
});
|
|
35
|
-
it('getStatus returns zeros when no training has occurred', () => {
|
|
36
|
-
const trainer = new AgentTrainer(db);
|
|
37
|
-
const status = trainer.getStatus();
|
|
38
|
-
expect(status.totalTrainingSessions).toBe(0);
|
|
39
|
-
expect(status.bestAccuracy).toBe(0);
|
|
40
|
-
expect(status.lastAccuracy).toBeNull();
|
|
41
|
-
expect(status.totalEpochsRun).toBe(0);
|
|
42
|
-
expect(status.avgImprovement).toBe(0);
|
|
43
|
-
});
|
|
44
|
-
it('setBenchmarkSuite sets suite for training', async () => {
|
|
45
|
-
const trainer = new AgentTrainer(db);
|
|
46
|
-
trainer.setBenchmarkSuite(suite);
|
|
47
|
-
// Verifying it doesn't throw when training after setting suite
|
|
48
|
-
const report = await trainer.train(async () => 'ignored', { epochs: 1 });
|
|
49
|
-
expect(report.totalEpochs).toBe(1);
|
|
50
|
-
});
|
|
51
|
-
it('throws if train is called without setBenchmarkSuite', async () => {
|
|
52
|
-
const trainer = new AgentTrainer(db);
|
|
53
|
-
await expect(trainer.train(async () => 'x')).rejects.toThrow('BenchmarkSuite not set');
|
|
54
|
-
});
|
|
55
|
-
it('trains with evalFn that returns correct answers', async () => {
|
|
56
|
-
const trainer = new AgentTrainer(db);
|
|
57
|
-
trainer.setBenchmarkSuite(suite);
|
|
58
|
-
const answers = {
|
|
59
|
-
'2+2': '4', '3+1': '4',
|
|
60
|
-
'10*5': '50', '100/4': '25',
|
|
61
|
-
'sqrt(144)': '12', '2^10': '1024',
|
|
62
|
-
};
|
|
63
|
-
const report = await trainer.train(async (input) => answers[input] ?? 'unknown', { epochs: 3, name: 'perfect-run' });
|
|
64
|
-
expect(report.name).toBe('perfect-run');
|
|
65
|
-
expect(report.totalEpochs).toBe(3);
|
|
66
|
-
expect(report.bestAccuracy).toBe(1);
|
|
67
|
-
expect(report.finalAccuracy).toBe(1);
|
|
68
|
-
expect(report.passed).toBe(true);
|
|
69
|
-
expect(report.id).toMatch(/^train-/);
|
|
70
|
-
expect(report.durationMs).toBeGreaterThanOrEqual(0);
|
|
71
|
-
});
|
|
72
|
-
it('trains with evalFn that returns wrong answers', async () => {
|
|
73
|
-
const trainer = new AgentTrainer(db);
|
|
74
|
-
trainer.setBenchmarkSuite(suite);
|
|
75
|
-
const report = await trainer.train(async () => 'wrong', { epochs: 1, name: 'fail-run', curriculumLearning: false });
|
|
76
|
-
expect(report.finalAccuracy).toBe(0);
|
|
77
|
-
expect(report.passed).toBe(false);
|
|
78
|
-
expect(report.epochs[0].passed).toBe(false);
|
|
79
|
-
});
|
|
80
|
-
it('getHistory returns persisted sessions', async () => {
|
|
81
|
-
const trainer = new AgentTrainer(db);
|
|
82
|
-
trainer.setBenchmarkSuite(suite);
|
|
83
|
-
await trainer.train(async () => '4', { epochs: 1, name: 'session-1' });
|
|
84
|
-
await trainer.train(async () => '50', { epochs: 1, name: 'session-2' });
|
|
85
|
-
const history = trainer.getHistory();
|
|
86
|
-
expect(history).toHaveLength(2);
|
|
87
|
-
expect(history.map(h => h.name)).toContain('session-1');
|
|
88
|
-
expect(history.map(h => h.name)).toContain('session-2');
|
|
89
|
-
expect(typeof history[0].finalAccuracy).toBe('number');
|
|
90
|
-
expect(typeof history[0].passed).toBe('boolean');
|
|
91
|
-
});
|
|
92
|
-
it('getSession returns full report by id', async () => {
|
|
93
|
-
const trainer = new AgentTrainer(db);
|
|
94
|
-
trainer.setBenchmarkSuite(suite);
|
|
95
|
-
const report = await trainer.train(async () => '4', { epochs: 1, name: 'retrievable' });
|
|
96
|
-
const session = trainer.getSession(report.id);
|
|
97
|
-
expect(session).not.toBeNull();
|
|
98
|
-
expect(session.name).toBe('retrievable');
|
|
99
|
-
expect(session.epochs).toHaveLength(1);
|
|
100
|
-
expect(session.id).toBe(report.id);
|
|
101
|
-
});
|
|
102
|
-
it('getSession returns null for unknown id', () => {
|
|
103
|
-
const trainer = new AgentTrainer(db);
|
|
104
|
-
expect(trainer.getSession('nonexistent-id')).toBeNull();
|
|
105
|
-
});
|
|
106
|
-
it('uses curriculum learning with ascending difficulty', async () => {
|
|
107
|
-
const trainer = new AgentTrainer(db);
|
|
108
|
-
trainer.setBenchmarkSuite(suite);
|
|
109
|
-
const report = await trainer.train(async () => 'any', { epochs: 3, curriculumLearning: true, name: 'curriculum-test' });
|
|
110
|
-
expect(report.epochs).toHaveLength(3);
|
|
111
|
-
expect(report.epochs[0].difficulty).toBe('easy');
|
|
112
|
-
expect(report.epochs[1].difficulty).toBe('medium');
|
|
113
|
-
expect(report.epochs[2].difficulty).toBe('hard');
|
|
114
|
-
});
|
|
115
|
-
it('skips curriculum when curriculumLearning is false', async () => {
|
|
116
|
-
const trainer = new AgentTrainer(db);
|
|
117
|
-
trainer.setBenchmarkSuite(suite);
|
|
118
|
-
const report = await trainer.train(async () => 'any', { epochs: 2, curriculumLearning: false, name: 'no-curriculum' });
|
|
119
|
-
expect(report.epochs).toHaveLength(2);
|
|
120
|
-
expect(report.epochs[0].difficulty).toBe('all');
|
|
121
|
-
expect(report.epochs[1].difficulty).toBe('all');
|
|
122
|
-
});
|
|
123
|
-
it('triggers early stopping when accuracy drops significantly', async () => {
|
|
124
|
-
const trainer = new AgentTrainer(db);
|
|
125
|
-
trainer.setBenchmarkSuite(suite);
|
|
126
|
-
// Easy cases get correct answers, medium/hard get wrong answers
|
|
127
|
-
// Epoch 1 (easy): 100% accuracy, Epoch 2 (medium): 0% accuracy → drop > 0.2 → stop
|
|
128
|
-
const answers = { '2+2': '4', '3+1': '4' };
|
|
129
|
-
const report = await trainer.train(async (input) => answers[input] ?? 'wrong', { epochs: 3, curriculumLearning: true, earlyStop: true, name: 'early-stop-test' });
|
|
130
|
-
// Should stop after epoch 2 due to accuracy drop from 1.0 to 0.0 (> 0.2)
|
|
131
|
-
expect(report.totalEpochs).toBe(2);
|
|
132
|
-
expect(report.epochs).toHaveLength(2);
|
|
133
|
-
expect(report.epochs[0].difficulty).toBe('easy');
|
|
134
|
-
expect(report.epochs[1].difficulty).toBe('medium');
|
|
135
|
-
});
|
|
136
|
-
it('getStatus reflects training history', async () => {
|
|
137
|
-
const trainer = new AgentTrainer(db);
|
|
138
|
-
trainer.setBenchmarkSuite(suite);
|
|
139
|
-
const answers = {
|
|
140
|
-
'2+2': '4', '3+1': '4',
|
|
141
|
-
'10*5': '50', '100/4': '25',
|
|
142
|
-
'sqrt(144)': '12', '2^10': '1024',
|
|
143
|
-
};
|
|
144
|
-
await trainer.train(async (input) => answers[input] ?? 'unknown', { epochs: 1, name: 'status-check' });
|
|
145
|
-
const status = trainer.getStatus();
|
|
146
|
-
expect(status.totalTrainingSessions).toBe(1);
|
|
147
|
-
expect(status.bestAccuracy).toBeGreaterThan(0);
|
|
148
|
-
expect(status.lastAccuracy).not.toBeNull();
|
|
149
|
-
expect(status.totalEpochsRun).toBe(1);
|
|
150
|
-
});
|
|
151
|
-
it('migration is idempotent', () => {
|
|
152
|
-
runTrainerMigration(db);
|
|
153
|
-
runTrainerMigration(db);
|
|
154
|
-
const trainer = new AgentTrainer(db);
|
|
155
|
-
expect(trainer.getHistory()).toHaveLength(0);
|
|
156
|
-
});
|
|
157
|
-
});
|
|
158
|
-
//# sourceMappingURL=agent-trainer.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"agent-trainer.test.js","sourceRoot":"","sources":["../../../src/agent-training/__tests__/agent-trainer.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;CACpF,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AACxE,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAG9E,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,EAAqB,CAAC;IAC1B,IAAI,KAAqB,CAAC;IAE1B,MAAM,SAAS,GAAe;QAC5B,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;QACrE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;KACtE,CAAC;IAEF,MAAM,WAAW,GAAe;QAC9B,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE;QACzE,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE;KAC3E,CAAC;IAEF,MAAM,SAAS,GAAe;QAC5B,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;QAC5E,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE;KAC1E,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9B,qBAAqB,CAAC,EAAE,CAAC,CAAC;QAC1B,mBAAmB,CAAC,EAAE,CAAC,CAAC;QACxB,KAAK,GAAG,IAAI,cAAc,CAAC,EAAE,CAAC,CAAC;QAC/B,KAAK,CAAC,QAAQ,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,WAAW,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACjC,+DAA+D;QAC/D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;IACzF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,OAAO,GAA2B;YACtC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG;YACtB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI;YAC3B,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM;SAClC,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAChC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,SAAS,EAC5C,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CACnC,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAChC,KAAK,IAAI,EAAE,CAAC,OAAO,EACnB,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,kBAAkB,EAAE,KAAK,EAAE,CAC3D,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QACvE,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAExE,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,GAAG,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QACxF,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE9C,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,OAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAQ,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,OAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAChC,KAAK,IAAI,EAAE,CAAC,KAAK,EACjB,EAAE,MAAM,EAAE,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,CACjE,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAChC,KAAK,IAAI,EAAE,CAAC,KAAK,EACjB,EAAE,MAAM,EAAE,CAAC,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,EAAE,eAAe,EAAE,CAChE,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,gEAAgE;QAChE,mFAAmF;QACnF,MAAM,OAAO,GAA2B,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;QAEnE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,CAChC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,EAC1C,EAAE,MAAM,EAAE,CAAC,EAAE,kBAAkB,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAClF,CAAC;QAEF,yEAAyE;QACzE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEjC,MAAM,OAAO,GAA2B;YACtC,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG;YACtB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI;YAC3B,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM;SAClC,CAAC;QACF,MAAM,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QAEvG,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC3C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,mBAAmB,CAAC,EAAE,CAAC,CAAC;QACxB,mBAAmB,CAAC,EAAE,CAAC,CAAC;QACxB,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,100 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import Database from 'better-sqlite3';
|
|
3
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
4
|
-
getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
|
|
5
|
-
}));
|
|
6
|
-
import { SubAgentFactory } from '../sub-agent-factory.js';
|
|
7
|
-
describe('SubAgentFactory', () => {
|
|
8
|
-
let db;
|
|
9
|
-
beforeEach(() => { db = new Database(':memory:'); });
|
|
10
|
-
afterEach(() => { db.close(); });
|
|
11
|
-
it('creates a factory with no agents', () => {
|
|
12
|
-
const factory = new SubAgentFactory(db);
|
|
13
|
-
expect(factory.list()).toHaveLength(0);
|
|
14
|
-
expect(factory.getStatus().totalAgents).toBe(0);
|
|
15
|
-
});
|
|
16
|
-
it('creates an agent with config', () => {
|
|
17
|
-
const factory = new SubAgentFactory(db);
|
|
18
|
-
const agent = factory.create({
|
|
19
|
-
name: 'analyst',
|
|
20
|
-
specialization: 'crypto',
|
|
21
|
-
description: 'Crypto analyst',
|
|
22
|
-
systemPrompt: 'Analyze crypto.',
|
|
23
|
-
tools: ['market.data'],
|
|
24
|
-
});
|
|
25
|
-
expect(agent.config.name).toBe('analyst');
|
|
26
|
-
expect(factory.list()).toHaveLength(1);
|
|
27
|
-
});
|
|
28
|
-
it('prevents duplicate agent names', () => {
|
|
29
|
-
const factory = new SubAgentFactory(db);
|
|
30
|
-
factory.create({ name: 'dup', specialization: 'x', description: '', systemPrompt: '', tools: [] });
|
|
31
|
-
expect(() => factory.create({ name: 'dup', specialization: 'y', description: '', systemPrompt: '', tools: [] })).toThrow('already exists');
|
|
32
|
-
});
|
|
33
|
-
it('gets agent by name', () => {
|
|
34
|
-
const factory = new SubAgentFactory(db);
|
|
35
|
-
factory.create({ name: 'writer', specialization: 'content', description: '', systemPrompt: '', tools: [] });
|
|
36
|
-
const agent = factory.get('writer');
|
|
37
|
-
expect(agent).not.toBeNull();
|
|
38
|
-
expect(agent.config.specialization).toBe('content');
|
|
39
|
-
});
|
|
40
|
-
it('returns null for unknown agent', () => {
|
|
41
|
-
const factory = new SubAgentFactory(db);
|
|
42
|
-
expect(factory.get('nonexistent')).toBeNull();
|
|
43
|
-
});
|
|
44
|
-
it('removes an agent', () => {
|
|
45
|
-
const factory = new SubAgentFactory(db);
|
|
46
|
-
factory.create({ name: 'temp', specialization: 'x', description: '', systemPrompt: '', tools: [] });
|
|
47
|
-
expect(factory.remove('temp')).toBe(true);
|
|
48
|
-
expect(factory.list()).toHaveLength(0);
|
|
49
|
-
expect(factory.remove('temp')).toBe(false);
|
|
50
|
-
});
|
|
51
|
-
it('creates agent from preset', () => {
|
|
52
|
-
const factory = new SubAgentFactory(db);
|
|
53
|
-
const agent = factory.createFromPreset('crypto-analyst');
|
|
54
|
-
expect(agent.config.specialization).toBe('cryptocurrency');
|
|
55
|
-
expect(agent.config.tools.length).toBeGreaterThan(0);
|
|
56
|
-
});
|
|
57
|
-
it('creates agent from preset with custom name', () => {
|
|
58
|
-
const factory = new SubAgentFactory(db);
|
|
59
|
-
const agent = factory.createFromPreset('content-writer', 'my-writer');
|
|
60
|
-
expect(agent.config.name).toBe('my-writer');
|
|
61
|
-
expect(agent.config.specialization).toBe('content');
|
|
62
|
-
});
|
|
63
|
-
it('throws for unknown preset', () => {
|
|
64
|
-
const factory = new SubAgentFactory(db);
|
|
65
|
-
expect(() => factory.createFromPreset('nonexistent')).toThrow('Unknown preset');
|
|
66
|
-
});
|
|
67
|
-
it('lists available presets', () => {
|
|
68
|
-
const factory = new SubAgentFactory(db);
|
|
69
|
-
const presets = factory.getPresets();
|
|
70
|
-
expect(presets).toContain('crypto-analyst');
|
|
71
|
-
expect(presets).toContain('content-writer');
|
|
72
|
-
expect(presets).toContain('code-reviewer');
|
|
73
|
-
expect(presets).toContain('research-agent');
|
|
74
|
-
});
|
|
75
|
-
it('sets executor on all agents', async () => {
|
|
76
|
-
const factory = new SubAgentFactory(db);
|
|
77
|
-
factory.setExecutor(async (input) => `echo: ${input}`);
|
|
78
|
-
const agent = factory.create({ name: 'runner', specialization: 'x', description: '', systemPrompt: '', tools: [] });
|
|
79
|
-
const task = await agent.execute('hello');
|
|
80
|
-
expect(task.output).toBe('echo: hello');
|
|
81
|
-
});
|
|
82
|
-
it('persists agents across factory instances', () => {
|
|
83
|
-
const factory1 = new SubAgentFactory(db);
|
|
84
|
-
factory1.create({ name: 'persistent', specialization: 'db', description: 'test', systemPrompt: 'test', tools: ['a'] });
|
|
85
|
-
const factory2 = new SubAgentFactory(db);
|
|
86
|
-
expect(factory2.list()).toHaveLength(1);
|
|
87
|
-
expect(factory2.get('persistent').config.specialization).toBe('db');
|
|
88
|
-
});
|
|
89
|
-
it('returns comprehensive status', async () => {
|
|
90
|
-
const factory = new SubAgentFactory(db);
|
|
91
|
-
factory.setExecutor(async () => 'done');
|
|
92
|
-
const agent = factory.create({ name: 'a1', specialization: 'x', description: '', systemPrompt: '', tools: [] });
|
|
93
|
-
await agent.execute('task');
|
|
94
|
-
const status = factory.getStatus();
|
|
95
|
-
expect(status.totalAgents).toBe(1);
|
|
96
|
-
expect(status.agents[0].name).toBe('a1');
|
|
97
|
-
expect(status.agents[0].completedTasks).toBe(1);
|
|
98
|
-
});
|
|
99
|
-
});
|
|
100
|
-
//# sourceMappingURL=sub-agent-factory.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"sub-agent-factory.test.js","sourceRoot":"","sources":["../../../src/agent-training/__tests__/sub-agent-factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;CACpF,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAE1D,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,EAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC;YAC3B,IAAI,EAAE,SAAS;YACf,cAAc,EAAE,QAAQ;YACxB,WAAW,EAAE,gBAAgB;YAC7B,YAAY,EAAE,iBAAiB;YAC/B,KAAK,EAAE,CAAC,aAAa,CAAC;SACvB,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACnG,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAC7I,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5G,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC7B,MAAM,CAAC,KAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC1B,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACpG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;QACzD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC3D,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,gBAAgB,CAAC,gBAAgB,EAAE,WAAW,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,OAAO,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACpH,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACzC,QAAQ,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,cAAc,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAEvH,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAE,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC,EAAE,CAAC,CAAC;QACxC,OAAO,CAAC,WAAW,CAAC,KAAK,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,GAAG,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAChH,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAE5B,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|
|
@@ -1,102 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
-
import Database from 'better-sqlite3';
|
|
3
|
-
vi.mock('../../utils/logger.js', () => ({
|
|
4
|
-
getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
|
|
5
|
-
}));
|
|
6
|
-
import { SubAgent, runSubAgentMigration } from '../sub-agent.js';
|
|
7
|
-
describe('SubAgent', () => {
|
|
8
|
-
let db;
|
|
9
|
-
const config = {
|
|
10
|
-
name: 'test-agent',
|
|
11
|
-
specialization: 'testing',
|
|
12
|
-
description: 'Test agent',
|
|
13
|
-
systemPrompt: 'You are a test agent.',
|
|
14
|
-
tools: ['tool1', 'tool2'],
|
|
15
|
-
};
|
|
16
|
-
beforeEach(() => {
|
|
17
|
-
db = new Database(':memory:');
|
|
18
|
-
runSubAgentMigration(db);
|
|
19
|
-
// Insert agent row
|
|
20
|
-
db.prepare(`INSERT INTO sub_agents (name, specialization, description, system_prompt, tools, max_concurrent, created_at) VALUES (?, ?, ?, ?, ?, 1, ?)`).run(config.name, config.specialization, config.description, config.systemPrompt, JSON.stringify(config.tools), Date.now());
|
|
21
|
-
});
|
|
22
|
-
afterEach(() => { db.close(); });
|
|
23
|
-
it('creates a sub-agent with config', () => {
|
|
24
|
-
const agent = new SubAgent(db, 1, config);
|
|
25
|
-
expect(agent.id).toBe(1);
|
|
26
|
-
expect(agent.config.name).toBe('test-agent');
|
|
27
|
-
expect(agent.config.specialization).toBe('testing');
|
|
28
|
-
});
|
|
29
|
-
it('submits a task', () => {
|
|
30
|
-
const agent = new SubAgent(db, 1, config);
|
|
31
|
-
const taskId = agent.submit('analyze something');
|
|
32
|
-
expect(taskId).toMatch(/^task-/);
|
|
33
|
-
});
|
|
34
|
-
it('lists tasks after submission', () => {
|
|
35
|
-
const agent = new SubAgent(db, 1, config);
|
|
36
|
-
agent.submit('task 1');
|
|
37
|
-
agent.submit('task 2');
|
|
38
|
-
const tasks = agent.getTasks();
|
|
39
|
-
expect(tasks).toHaveLength(2);
|
|
40
|
-
expect(tasks[0].status).toBe('pending');
|
|
41
|
-
});
|
|
42
|
-
it('executes a task with executor', async () => {
|
|
43
|
-
const agent = new SubAgent(db, 1, config);
|
|
44
|
-
agent.setExecutor(async (input) => `Result: ${input}`);
|
|
45
|
-
const task = await agent.execute('analyze BTC');
|
|
46
|
-
expect(task.status).toBe('completed');
|
|
47
|
-
expect(task.output).toBe('Result: analyze BTC');
|
|
48
|
-
});
|
|
49
|
-
it('handles executor failure', async () => {
|
|
50
|
-
const agent = new SubAgent(db, 1, config);
|
|
51
|
-
agent.setExecutor(async () => { throw new Error('LLM timeout'); });
|
|
52
|
-
const task = await agent.execute('fail me');
|
|
53
|
-
expect(task.status).toBe('failed');
|
|
54
|
-
expect(task.error).toBe('LLM timeout');
|
|
55
|
-
});
|
|
56
|
-
it('throws if no executor set', async () => {
|
|
57
|
-
const agent = new SubAgent(db, 1, config);
|
|
58
|
-
agent.submit('task');
|
|
59
|
-
await expect(agent.run('nonexistent')).rejects.toThrow('No executor set');
|
|
60
|
-
});
|
|
61
|
-
it('returns status with counts', async () => {
|
|
62
|
-
const agent = new SubAgent(db, 1, config);
|
|
63
|
-
agent.setExecutor(async (input) => `done: ${input}`);
|
|
64
|
-
await agent.execute('task 1');
|
|
65
|
-
await agent.execute('task 2');
|
|
66
|
-
const status = agent.getStatus();
|
|
67
|
-
expect(status.name).toBe('test-agent');
|
|
68
|
-
expect(status.totalTasks).toBe(2);
|
|
69
|
-
expect(status.completedTasks).toBe(2);
|
|
70
|
-
expect(status.failedTasks).toBe(0);
|
|
71
|
-
expect(status.lastRunAt).toBeTruthy();
|
|
72
|
-
});
|
|
73
|
-
it('tracks mixed task outcomes in status', async () => {
|
|
74
|
-
const agent = new SubAgent(db, 1, config);
|
|
75
|
-
let callCount = 0;
|
|
76
|
-
agent.setExecutor(async () => {
|
|
77
|
-
callCount++;
|
|
78
|
-
if (callCount === 2)
|
|
79
|
-
throw new Error('fail');
|
|
80
|
-
return 'ok';
|
|
81
|
-
});
|
|
82
|
-
await agent.execute('task 1');
|
|
83
|
-
await agent.execute('task 2');
|
|
84
|
-
const status = agent.getStatus();
|
|
85
|
-
expect(status.completedTasks).toBe(1);
|
|
86
|
-
expect(status.failedTasks).toBe(1);
|
|
87
|
-
});
|
|
88
|
-
it('passes system prompt and tools to executor', async () => {
|
|
89
|
-
const agent = new SubAgent(db, 1, config);
|
|
90
|
-
const executor = vi.fn().mockResolvedValue('result');
|
|
91
|
-
agent.setExecutor(executor);
|
|
92
|
-
await agent.execute('input text');
|
|
93
|
-
expect(executor).toHaveBeenCalledWith('input text', 'You are a test agent.', ['tool1', 'tool2']);
|
|
94
|
-
});
|
|
95
|
-
it('migration is idempotent', () => {
|
|
96
|
-
runSubAgentMigration(db);
|
|
97
|
-
runSubAgentMigration(db);
|
|
98
|
-
const agent = new SubAgent(db, 1, config);
|
|
99
|
-
expect(agent.getTasks()).toHaveLength(0);
|
|
100
|
-
});
|
|
101
|
-
});
|
|
102
|
-
//# sourceMappingURL=sub-agent.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"sub-agent.test.js","sourceRoot":"","sources":["../../../src/agent-training/__tests__/sub-agent.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AAEtC,EAAE,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;CACpF,CAAC,CAAC,CAAC;AAEJ,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAGjE,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,EAAqB,CAAC;IAC1B,MAAM,MAAM,GAAmB;QAC7B,IAAI,EAAE,YAAY;QAClB,cAAc,EAAE,SAAS;QACzB,WAAW,EAAE,YAAY;QACzB,YAAY,EAAE,uBAAuB;QACrC,KAAK,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC;KAC1B,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC9B,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACzB,mBAAmB;QACnB,EAAE,CAAC,OAAO,CAAC,2IAA2I,CAAC,CAAC,GAAG,CACzJ,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,GAAG,EAAE,CACtH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAEjC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;QACxB,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvB,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACvB,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,KAAK,EAAE,CAAC,CAAC;QACvD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACtC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,KAAK,CAAC,WAAW,CAAC,KAAK,IAAI,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACrB,MAAM,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC;QACrD,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,KAAK,CAAC,WAAW,CAAC,KAAK,IAAI,EAAE;YAC3B,SAAS,EAAE,CAAC;YACZ,IAAI,SAAS,KAAK,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QACH,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9B,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAE9B,MAAM,MAAM,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QACrD,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC5B,MAAM,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAElC,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACnG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACzB,oBAAoB,CAAC,EAAE,CAAC,CAAC;QACzB,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC,EAAE,MAAM,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
export {};
|