feat: /brain-compose/tiers endpoint — données tiers live depuis brain-compose.yml

This commit is contained in:
2026-03-20 21:44:43 +01:00
parent de4cd85798
commit f97e970650

View File

@@ -125,7 +125,7 @@ TIER_RANK = {'free': 0, 'featured': 1, 'pro': 2, 'owner': 3, 'full': 3} # chaî
# ── Tier cache ──────────────────────────────────────────────────────────────────
KEYS_API = os.getenv('BRAIN_KEYS_API', '') # URL key server — vide = tier free par défaut
KEYS_API = os.getenv('BRAIN_KEYS_API', '')
TIER_TTL = 3600 # 1h TTL normal
TIER_GRACE = 7 * 86400 # 7 jours grace offline
@@ -213,14 +213,6 @@ BRAIN_ROOT = Path(__file__).parent.parent
app = FastAPI(title='Brain-as-a-Service', version='BE-4', docs_url='/api-docs')
# ── Montage brain-ui static (si build disponible) ────────────────────────────
_UI_DIST = BRAIN_ROOT / 'brain-ui' / 'dist'
if _UI_DIST.is_dir():
from fastapi.staticfiles import StaticFiles
app.mount('/ui', StaticFiles(directory=str(_UI_DIST), html=True), name='brain-ui')
log.info('brain-ui monté sur /ui depuis %s', _UI_DIST)
# ── Level 2 — localhost frictionless ───────────────────────────────────────────
@@ -275,6 +267,89 @@ def health():
return JSONResponse(status_code=503, content={'status': 'error', 'detail': str(e), 'uptime': uptime})
# ── Brain-compose live — données tiers depuis brain-compose.yml ─────────────────
@app.get('/brain-compose/tiers')
def brain_compose_tiers():
"""Retourne les feature_sets structurés depuis brain-compose.yml — source de vérité."""
compose_path = BRAIN_ROOT / 'brain-compose.yml'
if not compose_path.exists():
raise HTTPException(status_code=404, detail='brain-compose.yml introuvable')
# Parse custom — brain-compose.yml utilise `extends:` inline qui n'est pas du YAML standard
raw = compose_path.read_text(encoding='utf-8')
# Retirer les lignes `extends: X` qui cassent le parser YAML
cleaned = re.sub(r'^\s+extends:\s+\w+\s*$', '', raw, flags=re.MULTILINE)
try:
data = yaml.safe_load(cleaned) if _YAML_AVAILABLE else {}
except Exception as e:
raise HTTPException(status_code=500, detail=f'Erreur parsing brain-compose.yml: {e}')
feature_sets = data.get('feature_sets', {})
version = data.get('version', 'unknown')
# Résoudre l'héritage (extends) et compter les agents cumulés
resolved = {}
tier_chain = ['free', 'featured', 'pro', 'full']
cumulative_agents: list[str] = []
cumulative_sessions: list[str] = []
for tier_name in tier_chain:
tier_data = feature_sets.get(tier_name, {})
if not tier_data:
continue
# Agents de ce tier
tier_agents = tier_data.get('agents', [])
if tier_agents == '*':
# full = tous les agents
agents_dir = BRAIN_ROOT / 'agents'
if agents_dir.is_dir():
all_agents = sorted([
f.stem for f in agents_dir.glob('*.md')
if f.stem not in ('AGENTS', 'CATALOG', '_template', '_template-orchestrator')
])
cumulative_agents = all_agents
tier_agents_list = cumulative_agents[:]
else:
for a in tier_agents:
if a not in cumulative_agents:
cumulative_agents.append(a)
tier_agents_list = cumulative_agents[:]
# Sessions de ce tier
tier_sessions = tier_data.get('sessions', [])
if tier_sessions == '*':
cumulative_sessions = [
'navigate', 'work', 'debug', 'brainstorm', 'brain', 'handoff',
'coach', 'capital', 'audit', 'deploy', 'infra', 'urgence',
'kernel', 'edit-brain', 'pilote'
]
elif isinstance(tier_sessions, list):
for s in tier_sessions:
if s != 'extends' and not isinstance(s, dict) and s not in cumulative_sessions:
cumulative_sessions.append(s)
resolved[tier_name] = {
'description': tier_data.get('description', ''),
'coach_level': tier_data.get('coach_level', ''),
'distillation': tier_data.get('distillation', False),
'agents_new': [a for a in (tier_agents if isinstance(tier_agents, list) else []) if a != '*'],
'agents_total': tier_agents_list,
'agents_count': len(tier_agents_list),
'sessions_new': [s for s in (tier_sessions if isinstance(tier_sessions, list) else []) if s not in ('extends',) and not isinstance(s, dict)],
'sessions_total': cumulative_sessions[:],
'sessions_count': len(cumulative_sessions),
}
return {
'version': version,
'tiers': resolved,
'tier_chain': tier_chain,
}
# ── Docs live — sert docs/*.md depuis le filesystem ────────────────────────────
@app.get('/docs')