51 lines
1.6 KiB
Python
Executable File
51 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
import json
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
ROSTER_PATH = ROOT / 'references' / 'roster.json'
|
|
|
|
|
|
def load_roster():
|
|
return json.loads(ROSTER_PATH.read_text(encoding='utf-8'))
|
|
|
|
|
|
def resolve_plan(mode: str, tier: str):
|
|
roster = load_roster()
|
|
if mode not in roster:
|
|
raise SystemExit(f'Unknown mode: {mode}')
|
|
if tier not in roster['tierPolicy']:
|
|
raise SystemExit(f'Unknown tier: {tier}')
|
|
|
|
roles = roster[mode]['roles']
|
|
tier_policy = roster['tierPolicy'][tier]
|
|
plan = {
|
|
'mode': mode,
|
|
'tier': tier,
|
|
'roles': {},
|
|
'notes': tier_policy.get('notes', '')
|
|
}
|
|
|
|
for role, cfg in roles.items():
|
|
mission = cfg.get('mission', 'advisor')
|
|
override = tier_policy.get('synthesisModel') if mission == 'synthesis' else tier_policy.get('advisorModel')
|
|
plan['roles'][role] = {
|
|
'agentId': cfg['agentId'],
|
|
'mission': mission,
|
|
'defaultModel': cfg.get('defaultModel'),
|
|
'modelOverride': override,
|
|
'fallbacks': cfg.get('fallbacks', [])
|
|
}
|
|
return plan
|
|
|
|
|
|
if __name__ == '__main__':
|
|
p = argparse.ArgumentParser(description='Resolve council role/agent/model plan')
|
|
p.add_argument('--mode', default='personality', choices=['personality', 'dp'])
|
|
p.add_argument('--tier', default='light', choices=['light', 'medium', 'heavy'])
|
|
p.add_argument('--pretty', action='store_true')
|
|
args = p.parse_args()
|
|
plan = resolve_plan(args.mode, args.tier)
|
|
print(json.dumps(plan, indent=2 if args.pretty else None))
|