245 lines
9.0 KiB
TypeScript
245 lines
9.0 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
import { MindDB } from '@waggle/core';
|
|
import { Orchestrator } from '../src/orchestrator.js';
|
|
import { MockEmbedder } from '../../hive-mind-core/tests/mind/helpers/mock-embedder.js';
|
|
|
|
describe('Agent Loop with Custom Tools', () => {
|
|
let db: MindDB;
|
|
let orchestrator: Orchestrator;
|
|
|
|
beforeEach(() => {
|
|
db = new MindDB(':memory:');
|
|
orchestrator = new Orchestrator({
|
|
db,
|
|
embedder: new MockEmbedder(),
|
|
});
|
|
});
|
|
|
|
afterEach(() => {
|
|
db.close();
|
|
});
|
|
|
|
describe('Tool definitions', () => {
|
|
it('creates all required tool definitions', () => {
|
|
const tools = orchestrator.getTools();
|
|
const names = tools.map(t => t.name);
|
|
expect(names).toContain('search_memory');
|
|
expect(names).toContain('save_memory');
|
|
expect(names).toContain('get_identity');
|
|
expect(names).toContain('get_awareness');
|
|
expect(names).toContain('query_knowledge');
|
|
expect(names).toContain('add_task');
|
|
});
|
|
|
|
it('each tool has name, description, parameters, and execute function', () => {
|
|
for (const tool of orchestrator.getTools()) {
|
|
expect(tool.name).toBeTruthy();
|
|
expect(tool.description).toBeTruthy();
|
|
expect(tool.parameters).toBeDefined();
|
|
expect(typeof tool.execute).toBe('function');
|
|
}
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: get_identity', () => {
|
|
it('returns identity context', async () => {
|
|
orchestrator.getIdentity().create({
|
|
name: 'Waggle',
|
|
role: 'Personal Assistant',
|
|
department: '',
|
|
personality: 'Helpful',
|
|
capabilities: 'Memory, search',
|
|
system_prompt: 'You are Waggle.',
|
|
});
|
|
const result = await orchestrator.executeTool('get_identity', {});
|
|
expect(result).toContain('Waggle');
|
|
expect(result).toContain('Personal Assistant');
|
|
});
|
|
|
|
it('returns message when no identity configured', async () => {
|
|
const result = await orchestrator.executeTool('get_identity', {});
|
|
expect(result).toContain('No identity configured');
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: get_awareness', () => {
|
|
it('returns awareness context', async () => {
|
|
orchestrator.getAwareness().add('task', 'Review PR #42', 10);
|
|
const result = await orchestrator.executeTool('get_awareness', {});
|
|
expect(result).toContain('Review PR #42');
|
|
});
|
|
|
|
it('returns empty message when no items', async () => {
|
|
const result = await orchestrator.executeTool('get_awareness', {});
|
|
expect(result).toContain('No active awareness items');
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: save_memory', () => {
|
|
it('saves a memory to the current session', async () => {
|
|
const result = await orchestrator.executeTool('save_memory', {
|
|
content: 'User prefers dark mode',
|
|
importance: 'normal',
|
|
});
|
|
expect(result).toContain('Memory saved');
|
|
});
|
|
|
|
it('creates a session if none active', async () => {
|
|
await orchestrator.executeTool('save_memory', { content: 'First memory' });
|
|
const active = orchestrator.getSessions().getActive();
|
|
expect(active.length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it('saved memory is searchable', async () => {
|
|
await orchestrator.executeTool('save_memory', { content: 'The meeting about quantum computing was postponed' });
|
|
|
|
// CognifyPipeline now auto-indexes frames for vector search
|
|
const searchResult = await orchestrator.executeTool('search_memory', { query: 'quantum' });
|
|
expect(searchResult).toContain('quantum computing');
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: search_memory', () => {
|
|
it('returns no results for empty memory', async () => {
|
|
const result = await orchestrator.executeTool('search_memory', { query: 'anything' });
|
|
expect(result).toContain('No relevant memories');
|
|
});
|
|
|
|
it('searches with different profiles', async () => {
|
|
await orchestrator.executeTool('save_memory', { content: 'Important meeting notes from today' });
|
|
|
|
// CognifyPipeline now auto-indexes frames for vector search
|
|
const result = await orchestrator.executeTool('search_memory', {
|
|
query: 'meeting',
|
|
profile: 'recent',
|
|
limit: 5,
|
|
});
|
|
expect(result).toContain('meeting notes');
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: query_knowledge', () => {
|
|
it('queries entities from knowledge graph', async () => {
|
|
orchestrator.getKnowledge().createEntity('person', 'Alice', { role: 'engineer' });
|
|
orchestrator.getKnowledge().createEntity('person', 'Bob', { role: 'designer' });
|
|
|
|
const result = await orchestrator.executeTool('query_knowledge', { query: 'Alice' });
|
|
expect(result).toContain('Alice');
|
|
});
|
|
|
|
it('shows relations for found entities', async () => {
|
|
const alice = orchestrator.getKnowledge().createEntity('person', 'Alice', {});
|
|
const waggle = orchestrator.getKnowledge().createEntity('project', 'Waggle', {});
|
|
orchestrator.getKnowledge().createRelation(alice.id, waggle.id, 'works_on', 0.95);
|
|
|
|
const result = await orchestrator.executeTool('query_knowledge', { query: 'Alice' });
|
|
expect(result).toContain('works_on');
|
|
expect(result).toContain('Waggle');
|
|
});
|
|
|
|
it('returns empty message for no matches', async () => {
|
|
const result = await orchestrator.executeTool('query_knowledge', { query: 'NonexistentPerson' });
|
|
expect(result).toContain('No entities found');
|
|
});
|
|
});
|
|
|
|
describe('Tool execution: add_task', () => {
|
|
it('adds a task to awareness', async () => {
|
|
const result = await orchestrator.executeTool('add_task', {
|
|
content: 'Deploy v2.1',
|
|
priority: 8,
|
|
});
|
|
expect(result).toContain('Deploy v2.1');
|
|
expect(result).toContain('priority: 8');
|
|
|
|
const tasks = orchestrator.getAwareness().getByCategory('task');
|
|
expect(tasks).toHaveLength(1);
|
|
});
|
|
});
|
|
|
|
describe('System prompt builder', () => {
|
|
it('builds system prompt from identity + awareness', () => {
|
|
orchestrator.getIdentity().create({
|
|
name: 'Waggle',
|
|
role: 'Assistant',
|
|
department: '',
|
|
personality: '',
|
|
capabilities: '',
|
|
system_prompt: '',
|
|
});
|
|
orchestrator.getAwareness().add('task', 'Check emails', 5);
|
|
|
|
const prompt = orchestrator.buildSystemPrompt();
|
|
expect(prompt).toContain('Waggle');
|
|
expect(prompt).toContain('Check emails');
|
|
// Self-awareness section present
|
|
expect(prompt).toContain('# Self-Awareness');
|
|
});
|
|
|
|
it('builds prompt without identity', () => {
|
|
const prompt = orchestrator.buildSystemPrompt();
|
|
expect(prompt).toContain('# Self-Awareness');
|
|
expect(prompt).not.toContain('# Identity');
|
|
});
|
|
});
|
|
|
|
describe('Unknown tool handling', () => {
|
|
it('throws for unknown tool name', async () => {
|
|
await expect(orchestrator.executeTool('nonexistent', {})).rejects.toThrow('Unknown tool: nonexistent');
|
|
});
|
|
});
|
|
|
|
describe('E4: Auto-save style detection', () => {
|
|
it('saves explicit user preferences to personal mind', async () => {
|
|
const saved = await orchestrator.autoSaveFromExchange(
|
|
'I prefer bullet-point summaries for everything',
|
|
'Got it! I will use bullet points going forward.'
|
|
);
|
|
expect(saved.length).toBeGreaterThan(0);
|
|
expect(saved[0]).toContain('User preference');
|
|
|
|
// Verify it was saved in memory
|
|
const searchResult = await orchestrator.executeTool('search_memory', { query: 'preference' });
|
|
expect(searchResult).toContain('bullet');
|
|
});
|
|
|
|
it('detects implicit style signals from user messages', async () => {
|
|
// Use phrasing that triggers implicit style detection, not explicit preference patterns
|
|
const saved = await orchestrator.autoSaveFromExchange(
|
|
'Can you present this data in a table format with columns for each category?',
|
|
'Here is the table: ...'
|
|
);
|
|
expect(saved.length).toBeGreaterThan(0);
|
|
expect(saved.some(s => s.includes('Style note'))).toBe(true);
|
|
});
|
|
|
|
it('does not duplicate style notes on repeated detection', async () => {
|
|
await orchestrator.autoSaveFromExchange(
|
|
'Just show me the code examples please',
|
|
'Here is the code: ...'
|
|
);
|
|
const saved2 = await orchestrator.autoSaveFromExchange(
|
|
'Can you just show me the code for the other function too?',
|
|
'Here is the other code: ...'
|
|
);
|
|
// Second save should not duplicate the style note
|
|
const styleNotes = saved2.filter(s => s.includes('Style note'));
|
|
expect(styleNotes.length).toBe(0);
|
|
});
|
|
});
|
|
|
|
describe('E4: Personal preferences in context', () => {
|
|
it('includes personal preferences in loadRecentContext when workspace is active', async () => {
|
|
// Save a preference first
|
|
await orchestrator.autoSaveFromExchange(
|
|
'I always want concise answers from now on',
|
|
'Understood, I will keep things concise.'
|
|
);
|
|
|
|
// Without workspace, loadRecentContext just shows personal frames
|
|
const ctx = orchestrator.loadRecentContext();
|
|
expect(ctx).toContain('preference');
|
|
});
|
|
});
|
|
});
|