@timmeck/brain-core 2.36.57 → 2.36.59

package/dist/agent-training/__tests__/agent-trainer.test.js

@@ -0,0 +1,158 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import Database from 'better-sqlite3';
+vi.mock('../../utils/logger.js', () => ({
+    getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
+}));
+import { AgentTrainer, runTrainerMigration } from '../agent-trainer.js';
+import { BenchmarkSuite, runBenchmarkMigration } from '../benchmark-suite.js';
+describe('AgentTrainer', () => {
+    let db;
+    let suite;
+    const easyCases = [
+        { input: '2+2', expected: '4', category: 'math', difficulty: 'easy' },
+        { input: '3+1', expected: '4', category: 'math', difficulty: 'easy' },
+    ];
+    const mediumCases = [
+        { input: '10*5', expected: '50', category: 'math', difficulty: 'medium' },
+        { input: '100/4', expected: '25', category: 'math', difficulty: 'medium' },
+    ];
+    const hardCases = [
+        { input: 'sqrt(144)', expected: '12', category: 'math', difficulty: 'hard' },
+        { input: '2^10', expected: '1024', category: 'math', difficulty: 'hard' },
+    ];
+    beforeEach(() => {
+        db = new Database(':memory:');
+        runBenchmarkMigration(db);
+        runTrainerMigration(db);
+        suite = new BenchmarkSuite(db);
+        suite.addCases([...easyCases, ...mediumCases, ...hardCases]);
+    });
+    afterEach(() => { db.close(); });
+    it('creates trainer without error', () => {
+        const trainer = new AgentTrainer(db);
+        expect(trainer).toBeInstanceOf(AgentTrainer);
+    });
+    it('getStatus returns zeros when no training has occurred', () => {
+        const trainer = new AgentTrainer(db);
+        const status = trainer.getStatus();
+        expect(status.totalTrainingSessions).toBe(0);
+        expect(status.bestAccuracy).toBe(0);
+        expect(status.lastAccuracy).toBeNull();
+        expect(status.totalEpochsRun).toBe(0);
+        expect(status.avgImprovement).toBe(0);
+    });
+    it('setBenchmarkSuite sets suite for training', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        // Verifying it doesn't throw when training after setting suite
+        const report = await trainer.train(async () => 'ignored', { epochs: 1 });
+        expect(report.totalEpochs).toBe(1);
+    });
+    it('throws if train is called without setBenchmarkSuite', async () => {
+        const trainer = new AgentTrainer(db);
+        await expect(trainer.train(async () => 'x')).rejects.toThrow('BenchmarkSuite not set');
+    });
+    it('trains with evalFn that returns correct answers', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const answers = {
+            '2+2': '4', '3+1': '4',
+            '10*5': '50', '100/4': '25',
+            'sqrt(144)': '12', '2^10': '1024',
+        };
+        const report = await trainer.train(async (input) => answers[input] ?? 'unknown', { epochs: 3, name: 'perfect-run' });
+        expect(report.name).toBe('perfect-run');
+        expect(report.totalEpochs).toBe(3);
+        expect(report.bestAccuracy).toBe(1);
+        expect(report.finalAccuracy).toBe(1);
+        expect(report.passed).toBe(true);
+        expect(report.id).toMatch(/^train-/);
+        expect(report.durationMs).toBeGreaterThanOrEqual(0);
+    });
+    it('trains with evalFn that returns wrong answers', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const report = await trainer.train(async () => 'wrong', { epochs: 1, name: 'fail-run', curriculumLearning: false });
+        expect(report.finalAccuracy).toBe(0);
+        expect(report.passed).toBe(false);
+        expect(report.epochs[0].passed).toBe(false);
+    });
+    it('getHistory returns persisted sessions', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        await trainer.train(async () => '4', { epochs: 1, name: 'session-1' });
+        await trainer.train(async () => '50', { epochs: 1, name: 'session-2' });
+        const history = trainer.getHistory();
+        expect(history).toHaveLength(2);
+        expect(history.map(h => h.name)).toContain('session-1');
+        expect(history.map(h => h.name)).toContain('session-2');
+        expect(typeof history[0].finalAccuracy).toBe('number');
+        expect(typeof history[0].passed).toBe('boolean');
+    });
+    it('getSession returns full report by id', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const report = await trainer.train(async () => '4', { epochs: 1, name: 'retrievable' });
+        const session = trainer.getSession(report.id);
+        expect(session).not.toBeNull();
+        expect(session.name).toBe('retrievable');
+        expect(session.epochs).toHaveLength(1);
+        expect(session.id).toBe(report.id);
+    });
+    it('getSession returns null for unknown id', () => {
+        const trainer = new AgentTrainer(db);
+        expect(trainer.getSession('nonexistent-id')).toBeNull();
+    });
+    it('uses curriculum learning with ascending difficulty', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const report = await trainer.train(async () => 'any', { epochs: 3, curriculumLearning: true, name: 'curriculum-test' });
+        expect(report.epochs).toHaveLength(3);
+        expect(report.epochs[0].difficulty).toBe('easy');
+        expect(report.epochs[1].difficulty).toBe('medium');
+        expect(report.epochs[2].difficulty).toBe('hard');
+    });
+    it('skips curriculum when curriculumLearning is false', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const report = await trainer.train(async () => 'any', { epochs: 2, curriculumLearning: false, name: 'no-curriculum' });
+        expect(report.epochs).toHaveLength(2);
+        expect(report.epochs[0].difficulty).toBe('all');
+        expect(report.epochs[1].difficulty).toBe('all');
+    });
+    it('triggers early stopping when accuracy drops significantly', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        // Easy cases get correct answers, medium/hard get wrong answers
+        // Epoch 1 (easy): 100% accuracy, Epoch 2 (medium): 0% accuracy → drop > 0.2 → stop
+        const answers = { '2+2': '4', '3+1': '4' };
+        const report = await trainer.train(async (input) => answers[input] ?? 'wrong', { epochs: 3, curriculumLearning: true, earlyStop: true, name: 'early-stop-test' });
+        // Should stop after epoch 2 due to accuracy drop from 1.0 to 0.0 (> 0.2)
+        expect(report.totalEpochs).toBe(2);
+        expect(report.epochs).toHaveLength(2);
+        expect(report.epochs[0].difficulty).toBe('easy');
+        expect(report.epochs[1].difficulty).toBe('medium');
+    });
+    it('getStatus reflects training history', async () => {
+        const trainer = new AgentTrainer(db);
+        trainer.setBenchmarkSuite(suite);
+        const answers = {
+            '2+2': '4', '3+1': '4',
+            '10*5': '50', '100/4': '25',
+            'sqrt(144)': '12', '2^10': '1024',
+        };
+        await trainer.train(async (input) => answers[input] ?? 'unknown', { epochs: 1, name: 'status-check' });
+        const status = trainer.getStatus();
+        expect(status.totalTrainingSessions).toBe(1);
+        expect(status.bestAccuracy).toBeGreaterThan(0);
+        expect(status.lastAccuracy).not.toBeNull();
+        expect(status.totalEpochsRun).toBe(1);
+    });
+    it('migration is idempotent', () => {
+        runTrainerMigration(db);
+        runTrainerMigration(db);
+        const trainer = new AgentTrainer(db);
+        expect(trainer.getHistory()).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=agent-trainer.test.js.map

package/dist/agent-training/__tests__/agent-trainer.test.js.map

@@ -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"}

package/dist/agent-training/__tests__/sub-agent-factory.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/agent-training/__tests__/sub-agent-factory.test.js

@@ -0,0 +1,100 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import Database from 'better-sqlite3';
+vi.mock('../../utils/logger.js', () => ({
+    getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
+}));
+import { SubAgentFactory } from '../sub-agent-factory.js';
+describe('SubAgentFactory', () => {
+    let db;
+    beforeEach(() => { db = new Database(':memory:'); });
+    afterEach(() => { db.close(); });
+    it('creates a factory with no agents', () => {
+        const factory = new SubAgentFactory(db);
+        expect(factory.list()).toHaveLength(0);
+        expect(factory.getStatus().totalAgents).toBe(0);
+    });
+    it('creates an agent with config', () => {
+        const factory = new SubAgentFactory(db);
+        const agent = factory.create({
+            name: 'analyst',
+            specialization: 'crypto',
+            description: 'Crypto analyst',
+            systemPrompt: 'Analyze crypto.',
+            tools: ['market.data'],
+        });
+        expect(agent.config.name).toBe('analyst');
+        expect(factory.list()).toHaveLength(1);
+    });
+    it('prevents duplicate agent names', () => {
+        const factory = new SubAgentFactory(db);
+        factory.create({ name: 'dup', specialization: 'x', description: '', systemPrompt: '', tools: [] });
+        expect(() => factory.create({ name: 'dup', specialization: 'y', description: '', systemPrompt: '', tools: [] })).toThrow('already exists');
+    });
+    it('gets agent by name', () => {
+        const factory = new SubAgentFactory(db);
+        factory.create({ name: 'writer', specialization: 'content', description: '', systemPrompt: '', tools: [] });
+        const agent = factory.get('writer');
+        expect(agent).not.toBeNull();
+        expect(agent.config.specialization).toBe('content');
+    });
+    it('returns null for unknown agent', () => {
+        const factory = new SubAgentFactory(db);
+        expect(factory.get('nonexistent')).toBeNull();
+    });
+    it('removes an agent', () => {
+        const factory = new SubAgentFactory(db);
+        factory.create({ name: 'temp', specialization: 'x', description: '', systemPrompt: '', tools: [] });
+        expect(factory.remove('temp')).toBe(true);
+        expect(factory.list()).toHaveLength(0);
+        expect(factory.remove('temp')).toBe(false);
+    });
+    it('creates agent from preset', () => {
+        const factory = new SubAgentFactory(db);
+        const agent = factory.createFromPreset('crypto-analyst');
+        expect(agent.config.specialization).toBe('cryptocurrency');
+        expect(agent.config.tools.length).toBeGreaterThan(0);
+    });
+    it('creates agent from preset with custom name', () => {
+        const factory = new SubAgentFactory(db);
+        const agent = factory.createFromPreset('content-writer', 'my-writer');
+        expect(agent.config.name).toBe('my-writer');
+        expect(agent.config.specialization).toBe('content');
+    });
+    it('throws for unknown preset', () => {
+        const factory = new SubAgentFactory(db);
+        expect(() => factory.createFromPreset('nonexistent')).toThrow('Unknown preset');
+    });
+    it('lists available presets', () => {
+        const factory = new SubAgentFactory(db);
+        const presets = factory.getPresets();
+        expect(presets).toContain('crypto-analyst');
+        expect(presets).toContain('content-writer');
+        expect(presets).toContain('code-reviewer');
+        expect(presets).toContain('research-agent');
+    });
+    it('sets executor on all agents', async () => {
+        const factory = new SubAgentFactory(db);
+        factory.setExecutor(async (input) => `echo: ${input}`);
+        const agent = factory.create({ name: 'runner', specialization: 'x', description: '', systemPrompt: '', tools: [] });
+        const task = await agent.execute('hello');
+        expect(task.output).toBe('echo: hello');
+    });
+    it('persists agents across factory instances', () => {
+        const factory1 = new SubAgentFactory(db);
+        factory1.create({ name: 'persistent', specialization: 'db', description: 'test', systemPrompt: 'test', tools: ['a'] });
+        const factory2 = new SubAgentFactory(db);
+        expect(factory2.list()).toHaveLength(1);
+        expect(factory2.get('persistent').config.specialization).toBe('db');
+    });
+    it('returns comprehensive status', async () => {
+        const factory = new SubAgentFactory(db);
+        factory.setExecutor(async () => 'done');
+        const agent = factory.create({ name: 'a1', specialization: 'x', description: '', systemPrompt: '', tools: [] });
+        await agent.execute('task');
+        const status = factory.getStatus();
+        expect(status.totalAgents).toBe(1);
+        expect(status.agents[0].name).toBe('a1');
+        expect(status.agents[0].completedTasks).toBe(1);
+    });
+});
+//# sourceMappingURL=sub-agent-factory.test.js.map

package/dist/agent-training/__tests__/sub-agent-factory.test.js.map

@@ -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"}

package/dist/agent-training/__tests__/sub-agent.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/agent-training/__tests__/sub-agent.test.js

@@ -0,0 +1,102 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import Database from 'better-sqlite3';
+vi.mock('../../utils/logger.js', () => ({
+    getLogger: () => ({ info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() }),
+}));
+import { SubAgent, runSubAgentMigration } from '../sub-agent.js';
+describe('SubAgent', () => {
+    let db;
+    const config = {
+        name: 'test-agent',
+        specialization: 'testing',
+        description: 'Test agent',
+        systemPrompt: 'You are a test agent.',
+        tools: ['tool1', 'tool2'],
+    };
+    beforeEach(() => {
+        db = new Database(':memory:');
+        runSubAgentMigration(db);
+        // Insert agent row
+        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());
+    });
+    afterEach(() => { db.close(); });
+    it('creates a sub-agent with config', () => {
+        const agent = new SubAgent(db, 1, config);
+        expect(agent.id).toBe(1);
+        expect(agent.config.name).toBe('test-agent');
+        expect(agent.config.specialization).toBe('testing');
+    });
+    it('submits a task', () => {
+        const agent = new SubAgent(db, 1, config);
+        const taskId = agent.submit('analyze something');
+        expect(taskId).toMatch(/^task-/);
+    });
+    it('lists tasks after submission', () => {
+        const agent = new SubAgent(db, 1, config);
+        agent.submit('task 1');
+        agent.submit('task 2');
+        const tasks = agent.getTasks();
+        expect(tasks).toHaveLength(2);
+        expect(tasks[0].status).toBe('pending');
+    });
+    it('executes a task with executor', async () => {
+        const agent = new SubAgent(db, 1, config);
+        agent.setExecutor(async (input) => `Result: ${input}`);
+        const task = await agent.execute('analyze BTC');
+        expect(task.status).toBe('completed');
+        expect(task.output).toBe('Result: analyze BTC');
+    });
+    it('handles executor failure', async () => {
+        const agent = new SubAgent(db, 1, config);
+        agent.setExecutor(async () => { throw new Error('LLM timeout'); });
+        const task = await agent.execute('fail me');
+        expect(task.status).toBe('failed');
+        expect(task.error).toBe('LLM timeout');
+    });
+    it('throws if no executor set', async () => {
+        const agent = new SubAgent(db, 1, config);
+        agent.submit('task');
+        await expect(agent.run('nonexistent')).rejects.toThrow('No executor set');
+    });
+    it('returns status with counts', async () => {
+        const agent = new SubAgent(db, 1, config);
+        agent.setExecutor(async (input) => `done: ${input}`);
+        await agent.execute('task 1');
+        await agent.execute('task 2');
+        const status = agent.getStatus();
+        expect(status.name).toBe('test-agent');
+        expect(status.totalTasks).toBe(2);
+        expect(status.completedTasks).toBe(2);
+        expect(status.failedTasks).toBe(0);
+        expect(status.lastRunAt).toBeTruthy();
+    });
+    it('tracks mixed task outcomes in status', async () => {
+        const agent = new SubAgent(db, 1, config);
+        let callCount = 0;
+        agent.setExecutor(async () => {
+            callCount++;
+            if (callCount === 2)
+                throw new Error('fail');
+            return 'ok';
+        });
+        await agent.execute('task 1');
+        await agent.execute('task 2');
+        const status = agent.getStatus();
+        expect(status.completedTasks).toBe(1);
+        expect(status.failedTasks).toBe(1);
+    });
+    it('passes system prompt and tools to executor', async () => {
+        const agent = new SubAgent(db, 1, config);
+        const executor = vi.fn().mockResolvedValue('result');
+        agent.setExecutor(executor);
+        await agent.execute('input text');
+        expect(executor).toHaveBeenCalledWith('input text', 'You are a test agent.', ['tool1', 'tool2']);
+    });
+    it('migration is idempotent', () => {
+        runSubAgentMigration(db);
+        runSubAgentMigration(db);
+        const agent = new SubAgent(db, 1, config);
+        expect(agent.getTasks()).toHaveLength(0);
+    });
+});
+//# sourceMappingURL=sub-agent.test.js.map

package/dist/agent-training/__tests__/sub-agent.test.js.map

@@ -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"}

package/dist/agent-training/index.d.ts

@@ -2,3 +2,7 @@ export { BenchmarkSuite, runBenchmarkMigration } from './benchmark-suite.js';
 export type { EvalCase, EvalResult, BenchmarkReport, BenchmarkSuiteStatus, EvalFunction, ScoreFunction, } from './benchmark-suite.js';
 export { AgentTrainer, runTrainerMigration } from './agent-trainer.js';
 export type { TrainingConfig, EpochResult, TrainingReport, AgentTrainerStatus, } from './agent-trainer.js';
+export { SubAgent, runSubAgentMigration } from './sub-agent.js';
+export type { SubAgentConfig, SubAgentTask, SubAgentStatus, } from './sub-agent.js';
+export { SubAgentFactory } from './sub-agent-factory.js';
+export type { SubAgentFactoryStatus } from './sub-agent-factory.js';

package/dist/agent-training/index.js

@@ -1,3 +1,5 @@
 export { BenchmarkSuite, runBenchmarkMigration } from './benchmark-suite.js';
 export { AgentTrainer, runTrainerMigration } from './agent-trainer.js';
+export { SubAgent, runSubAgentMigration } from './sub-agent.js';
+export { SubAgentFactory } from './sub-agent-factory.js';
 //# sourceMappingURL=index.js.map

package/dist/agent-training/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent-training/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAM7E,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent-training/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAM7E,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAKvE,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAKhE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC"}

package/dist/agent-training/sub-agent-factory.d.ts

@@ -0,0 +1,36 @@
+import type Database from 'better-sqlite3';
+import { SubAgent } from './sub-agent.js';
+import type { SubAgentConfig, SubAgentStatus } from './sub-agent.js';
+export interface SubAgentFactoryStatus {
+    totalAgents: number;
+    agents: SubAgentStatus[];
+}
+type TaskExecutor = (input: string, systemPrompt: string, tools: string[]) => Promise<string>;
+export declare class SubAgentFactory {
+    private readonly db;
+    private readonly agents;
+    private executor;
+    private readonly stmtInsert;
+    private readonly stmtGetByName;
+    private readonly stmtGetAll;
+    private readonly stmtDelete;
+    constructor(db: Database.Database);
+    /** Set the executor used to run all sub-agent tasks */
+    setExecutor(executor: TaskExecutor): void;
+    /** Create a new sub-agent from config */
+    create(config: SubAgentConfig): SubAgent;
+    /** Create a sub-agent from a preset template */
+    createFromPreset(presetName: string, name?: string): SubAgent;
+    /** Get a sub-agent by name */
+    get(name: string): SubAgent | null;
+    /** List all sub-agents */
+    list(): SubAgent[];
+    /** Remove a sub-agent */
+    remove(name: string): boolean;
+    /** Get available preset names */
+    getPresets(): string[];
+    /** Get factory status */
+    getStatus(): SubAgentFactoryStatus;
+    private loadFromDb;
+}
+export {};

package/dist/agent-training/sub-agent-factory.js

@@ -0,0 +1,128 @@
+import { getLogger } from '../utils/logger.js';
+import { SubAgent, runSubAgentMigration } from './sub-agent.js';
+// ── Preset Templates ──────────────────────────────────────
+const PRESETS = {
+    'crypto-analyst': {
+        specialization: 'cryptocurrency',
+        description: 'Analyzes crypto markets, trends, and trading signals',
+        systemPrompt: 'You are a crypto market analyst. Analyze the given data and provide actionable insights about market trends, price movements, and trading signals.',
+        tools: ['market.data', 'signal.cross.status', 'paper.status'],
+    },
+    'content-writer': {
+        specialization: 'content',
+        description: 'Writes and optimizes social media content',
+        systemPrompt: 'You are a content strategist. Create engaging, informative social media posts about the given topics. Focus on clarity, engagement, and value.',
+        tools: ['content.generate', 'content.best', 'insight.list'],
+    },
+    'code-reviewer': {
+        specialization: 'code',
+        description: 'Reviews code patterns and suggests improvements',
+        systemPrompt: 'You are a senior code reviewer. Analyze the given code patterns, identify issues, and suggest improvements based on best practices.',
+        tools: ['codeforge.patterns', 'codeforge.products', 'code.health'],
+    },
+    'research-agent': {
+        specialization: 'research',
+        description: 'Conducts deep research on given topics',
+        systemPrompt: 'You are a research agent. Investigate the given topic thoroughly, gather evidence, form hypotheses, and synthesize findings.',
+        tools: ['insight.list', 'memory.query', 'knowledge.search'],
+    },
+};
+// ── Factory ──────────────────────────────────────────────
+const log = getLogger();
+export class SubAgentFactory {
+    db;
+    agents = new Map();
+    executor = null;
+    stmtInsert;
+    stmtGetByName;
+    stmtGetAll;
+    stmtDelete;
+    constructor(db) {
+        this.db = db;
+        runSubAgentMigration(db);
+        this.stmtInsert = db.prepare(`INSERT INTO sub_agents (name, specialization, description, system_prompt, tools, max_concurrent, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)`);
+        this.stmtGetByName = db.prepare(`SELECT * FROM sub_agents WHERE name = ?`);
+        this.stmtGetAll = db.prepare(`SELECT * FROM sub_agents ORDER BY created_at ASC`);
+        this.stmtDelete = db.prepare(`DELETE FROM sub_agents WHERE name = ?`);
+        // Load existing agents from DB
+        this.loadFromDb();
+    }
+    /** Set the executor used to run all sub-agent tasks */
+    setExecutor(executor) {
+        this.executor = executor;
+        for (const agent of this.agents.values()) {
+            agent.setExecutor(executor);
+        }
+    }
+    /** Create a new sub-agent from config */
+    create(config) {
+        if (this.agents.has(config.name)) {
+            throw new Error(`Agent "${config.name}" already exists`);
+        }
+        const result = this.stmtInsert.run(config.name, config.specialization, config.description, config.systemPrompt, JSON.stringify(config.tools), config.maxConcurrent ?? 1, Date.now());
+        const agent = new SubAgent(this.db, Number(result.lastInsertRowid), config);
+        if (this.executor)
+            agent.setExecutor(this.executor);
+        this.agents.set(config.name, agent);
+        log.info(`[sub-agent-factory] Created agent: ${config.name} (${config.specialization})`);
+        return agent;
+    }
+    /** Create a sub-agent from a preset template */
+    createFromPreset(presetName, name) {
+        const preset = PRESETS[presetName];
+        if (!preset) {
+            throw new Error(`Unknown preset: "${presetName}". Available: ${Object.keys(PRESETS).join(', ')}`);
+        }
+        return this.create({ name: name ?? presetName, ...preset });
+    }
+    /** Get a sub-agent by name */
+    get(name) {
+        return this.agents.get(name) ?? null;
+    }
+    /** List all sub-agents */
+    list() {
+        return Array.from(this.agents.values());
+    }
+    /** Remove a sub-agent */
+    remove(name) {
+        if (!this.agents.has(name))
+            return false;
+        this.stmtDelete.run(name);
+        this.agents.delete(name);
+        log.info(`[sub-agent-factory] Removed agent: ${name}`);
+        return true;
+    }
+    /** Get available preset names */
+    getPresets() {
+        return Object.keys(PRESETS);
+    }
+    /** Get factory status */
+    getStatus() {
+        return {
+            totalAgents: this.agents.size,
+            agents: Array.from(this.agents.values()).map(a => a.getStatus()),
+        };
+    }
+    // ── Private ──────────────────────────────────────────────
+    loadFromDb() {
+        const rows = this.stmtGetAll.all();
+        for (const row of rows) {
+            const config = {
+                name: row.name,
+                specialization: row.specialization,
+                description: row.description,
+                systemPrompt: row.system_prompt,
+                tools: JSON.parse(row.tools),
+                maxConcurrent: row.max_concurrent,
+            };
+            const agent = new SubAgent(this.db, row.id, config);
+            if (this.executor)
+                agent.setExecutor(this.executor);
+            this.agents.set(row.name, agent);
+        }
+        if (rows.length > 0) {
+            log.info(`[sub-agent-factory] Loaded ${rows.length} agents from DB`);
+        }
+    }
+}
+//# sourceMappingURL=sub-agent-factory.js.map

package/dist/agent-training/sub-agent-factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sub-agent-factory.js","sourceRoot":"","sources":["../../src/agent-training/sub-agent-factory.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAYhE,6DAA6D;AAE7D,MAAM,OAAO,GAAiD;IAC5D,gBAAgB,EAAE;QAChB,cAAc,EAAE,gBAAgB;QAChC,WAAW,EAAE,sDAAsD;QACnE,YAAY,EAAE,oJAAoJ;QAClK,KAAK,EAAE,CAAC,aAAa,EAAE,qBAAqB,EAAE,cAAc,CAAC;KAC9D;IACD,gBAAgB,EAAE;QAChB,cAAc,EAAE,SAAS;QACzB,WAAW,EAAE,2CAA2C;QACxD,YAAY,EAAE,gJAAgJ;QAC9J,KAAK,EAAE,CAAC,kBAAkB,EAAE,cAAc,EAAE,cAAc,CAAC;KAC5D;IACD,eAAe,EAAE;QACf,cAAc,EAAE,MAAM;QACtB,WAAW,EAAE,iDAAiD;QAC9D,YAAY,EAAE,qIAAqI;QACnJ,KAAK,EAAE,CAAC,oBAAoB,EAAE,oBAAoB,EAAE,aAAa,CAAC;KACnE;IACD,gBAAgB,EAAE;QAChB,cAAc,EAAE,UAAU;QAC1B,WAAW,EAAE,wCAAwC;QACrD,YAAY,EAAE,8HAA8H;QAC5I,KAAK,EAAE,CAAC,cAAc,EAAE,cAAc,EAAE,kBAAkB,CAAC;KAC5D;CACF,CAAC;AAEF,4DAA4D;AAE5D,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;AAExB,MAAM,OAAO,eAAe;IACT,EAAE,CAAoB;IACtB,MAAM,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC9C,QAAQ,GAAwB,IAAI,CAAC;IAE5B,UAAU,CAAC;IACX,aAAa,CAAC;IACd,UAAU,CAAC;IACX,UAAU,CAAC;IAE5B,YAAY,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;QACb,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAEzB,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,2IAA2I,CAAC,CAAC;QAC1K,IAAI,CAAC,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC;QAC3E,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,kDAAkD,CAAC,CAAC;QACjF,IAAI,CAAC,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,uCAAuC,CAAC,CAAC;QAEtE,+BAA+B;QAC/B,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,uDAAuD;IACvD,WAAW,CAAC,QAAsB;QAChC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;YACzC,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,yCAAyC;IACzC,MAAM,CAAC,MAAsB;QAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,UAAU,MAAM,CAAC,IAAI,kBAAkB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAChC,MAAM,CAAC,IAAI,EACX,MAAM,CAAC,cAAc,EACrB,MAAM,CAAC,WAAW,EAClB,MAAM,CAAC,YAAY,EACnB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,EAC5B,MAAM,CAAC,aAAa,IAAI,CAAC,EACzB,IAAI,CAAC,GAAG,EAAE,CACX,CAAC;QAEF,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,EAAE,MAAM,CAAC,CAAC;QAC5E,IAAI,IAAI,CAAC,QAAQ;YAAE,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAEpC,GAAG,CAAC,IAAI,CAAC,sCAAsC,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,cAAc,GAAG,CAAC,CAAC;QACzF,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gDAAgD;IAChD,gBAAgB,CAAC,UAAkB,EAAE,IAAa;QAChD,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,UAAU,iBAAiB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpG,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,UAAU,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED,8BAA8B;IAC9B,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC;IACvC,CAAC;IAED,0BAA0B;IAC1B,IAAI;QACF,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,yBAAyB;IACzB,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,OAAO,KAAK,CAAC;QACzC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,GAAG,CAAC,IAAI,CAAC,sCAAsC,IAAI,EAAE,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iCAAiC;IACjC,UAAU;QACR,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAED,yBAAyB;IACzB,SAAS;QACP,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YAC7B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;SACjE,CAAC;IACJ,CAAC;IAED,4DAA4D;IAEpD,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAgB,CAAC;QACjD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,MAAM,GAAmB;gBAC7B,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,cAAc,EAAE,GAAG,CAAC,cAAc;gBAClC,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,YAAY,EAAE,GAAG,CAAC,aAAa;gBAC/B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC;gBAC5B,aAAa,EAAE,GAAG,CAAC,cAAc;aAClC,CAAC;YACF,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YACpD,IAAI,IAAI,CAAC,QAAQ;gBAAE,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,GAAG,CAAC,IAAI,CAAC,8BAA8B,IAAI,CAAC,MAAM,iBAAiB,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;CACF"}