feat: config loader with validation and tilde expansion

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-04-04 14:16:49 +02:00
parent deb732b12a
commit 10a05ba55a
2 changed files with 56 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
import os
import yaml
CONFIG_PATH = os.path.expanduser('~/.claude/autoresearch.yaml')
DEFAULTS = {'time_limit_minutes': 10, 'benchmarks': []}
class ConfigError(Exception):
pass
def load_config(path=CONFIG_PATH):
with open(path) as f:
cfg = yaml.safe_load(f)
if not cfg or 'projects' not in cfg or not cfg['projects']:
raise ConfigError(f"'projects' list is required in {path}")
for p in cfg['projects']:
p.setdefault('benchmarks', DEFAULTS['benchmarks'])
p.setdefault('time_limit_minutes', DEFAULTS['time_limit_minutes'])
p['path'] = os.path.expanduser(p['path'])
cfg.setdefault('token_threshold', {'context_remaining_pct': 60, 'api_budget_usd': 5.0})
return cfg
+36
View File
@@ -0,0 +1,36 @@
import sys
sys.path.insert(0, 'bin')
from config import load_config, ConfigError
import pytest, tempfile, os, yaml
def test_load_valid_config():
data = {
'projects': [{'path': '/tmp', 'benchmarks': ['echo ok'], 'time_limit_minutes': 5}],
'token_threshold': {'context_remaining_pct': 60, 'api_budget_usd': 5.0}
}
with tempfile.NamedTemporaryFile('w', suffix='.yaml', delete=False) as f:
yaml.dump(data, f)
name = f.name
cfg = load_config(name)
assert cfg['projects'][0]['time_limit_minutes'] == 5
os.unlink(name)
def test_missing_projects_raises():
with tempfile.NamedTemporaryFile('w', suffix='.yaml', delete=False) as f:
yaml.dump({}, f)
name = f.name
with pytest.raises(ConfigError):
load_config(name)
os.unlink(name)
def test_expands_tilde_in_path():
data = {
'projects': [{'path': '~/.claude', 'benchmarks': [], 'time_limit_minutes': 5}],
'token_threshold': {'context_remaining_pct': 60, 'api_budget_usd': 5.0}
}
with tempfile.NamedTemporaryFile('w', suffix='.yaml', delete=False) as f:
yaml.dump(data, f)
name = f.name
cfg = load_config(name)
assert '~' not in cfg['projects'][0]['path']
os.unlink(name)