feat: add AgentConfigRegistry for named agent configurations

This commit is contained in:
William Valentin
2026-02-06 15:52:58 -08:00
parent 430cb3f96e
commit d65ce078b7
2 changed files with 121 additions and 0 deletions
+64
View File
@@ -0,0 +1,64 @@
import { describe, it, expect } from 'vitest';
import { AgentConfigRegistry, type AgentConfig } from './registry.js';
describe('AgentConfigRegistry', () => {
describe('register()', () => {
it('registers a named agent config', () => {
const registry = new AgentConfigRegistry();
const config: AgentConfig = { name: 'assistant', systemPrompt: 'Be helpful.' };
registry.register(config);
expect(registry.get('assistant')).toEqual(config);
});
it('throws on duplicate name', () => {
const registry = new AgentConfigRegistry();
registry.register({ name: 'assistant' });
expect(() => registry.register({ name: 'assistant' })).toThrow('already registered');
});
});
describe('get()', () => {
it('returns undefined for unknown name', () => {
const registry = new AgentConfigRegistry();
expect(registry.get('nonexistent')).toBeUndefined();
});
});
describe('list()', () => {
it('returns all registered configs', () => {
const registry = new AgentConfigRegistry();
registry.register({ name: 'a' });
registry.register({ name: 'b' });
expect(registry.list().map(c => c.name).sort()).toEqual(['a', 'b']);
});
});
describe('loadFromConfig()', () => {
it('loads configs from a raw config object', () => {
const registry = new AgentConfigRegistry();
registry.loadFromConfig({
assistant: {
system_prompt: 'Be helpful.',
model_tier: 'default',
tool_profile: 'messaging',
sandbox: false,
},
coder: {
model_tier: 'complex',
tool_profile: 'coding',
sandbox: true,
},
});
expect(registry.list()).toHaveLength(2);
const assistant = registry.get('assistant')!;
expect(assistant.systemPrompt).toBe('Be helpful.');
expect(assistant.modelTier).toBe('default');
expect(assistant.toolProfile).toBe('messaging');
const coder = registry.get('coder')!;
expect(coder.sandbox).toBe(true);
});
});
});
+57
View File
@@ -0,0 +1,57 @@
import type { ToolProfile, ToolOverrideConfig } from '../config/schema.js';
import type { ModelTier } from '../models/router.js';
export interface AgentConfig {
name: string;
systemPrompt?: string;
modelTier?: ModelTier;
toolProfile?: ToolProfile;
toolOverrides?: ToolOverrideConfig;
sandbox?: boolean;
}
/**
* AgentConfigRegistry — stores named agent configurations.
* Loaded from YAML config at startup.
*/
export class AgentConfigRegistry {
private configs = new Map<string, AgentConfig>();
register(config: AgentConfig): void {
if (this.configs.has(config.name)) {
throw new Error(`Agent config '${config.name}' is already registered`);
}
this.configs.set(config.name, config);
}
get(name: string): AgentConfig | undefined {
return this.configs.get(name);
}
list(): AgentConfig[] {
return Array.from(this.configs.values());
}
/**
* Load agent configs from the parsed YAML config.
* Maps from the config schema format to the internal AgentConfig format.
*/
loadFromConfig(rawConfigs: Record<string, {
system_prompt?: string;
model_tier?: string;
tool_profile?: string;
tool_overrides?: ToolOverrideConfig;
sandbox?: boolean;
}>): void {
for (const [name, raw] of Object.entries(rawConfigs)) {
this.register({
name,
systemPrompt: raw.system_prompt,
modelTier: raw.model_tier as ModelTier | undefined,
toolProfile: raw.tool_profile as ToolProfile | undefined,
toolOverrides: raw.tool_overrides,
sandbox: raw.sandbox,
});
}
}
}