Files
Loki/backend/app/router.py
T
MichaelandClaude Opus 4.8 584d28a9d0 fix(router): reprise de code — routage contextuel + directive d'action
Sur une reprise (« ajoute un bouton », « continue »), le routeur ne voyait
que le message courant : score trop bas, jamais le moteur code, et le
modèle décrivait les changements sans les appliquer.

- routage contextuel : si le tour précédent était du code (engine ou
  outils write_file/edit_file/code_task), les suites courtes partent au
  moteur code (is_code_followup)
- chemin agent sur session code : directive système « AGIS avec les
  outils, ne décris jamais sans appliquer »
- fichiers cités dans le message et présents au workspace transmis à
  Aider (travail direct, sans deviner via la repo map)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-07-19 00:42:06 +02:00

70 lines
2.5 KiB
Python

"""Routeur automatique : moteur code (Aider) ou boucle agent classique ?
Invisible pour l'utilisateur : chaque message est classé par une heuristique
lexicale instantanée. Les cas ambigus partent vers la boucle agent, qui garde
l'outil `code_task` en secours — plus d'appel LLM bloquant avant le premier
token (l'ancien micro-classifieur coûtait un aller-retour modèle complet et
chargeait un runner divergent).
"""
from __future__ import annotations
import re
# Vocabulaire fortement lié au code / au développement.
_STRONG = re.compile(
r"\b(code|coder?|script|fonction|classe|refactor|bug|d[ée]bug|html|css|"
r"javascript|js|python|typescript|react|api|composant|page|site|landing|"
r"formulaire|d[ée]veloppe|impl[ée]mente|programme|widget|frontend|backend)\b",
re.I,
)
# Extensions de fichiers mentionnées explicitement.
_FILE_EXT = re.compile(
r"\.(py|js|ts|tsx|jsx|html?|css|json|md|sh|sql|ya?ml|go|rs|java|php|vue)\b", re.I
)
# Verbes d'action de création/modification.
_ACTION = re.compile(
r"\b(cr[ée]e[rs]?|modifie[rs]?|corrige[rs]?|ajoute[rs]?|[ée]cri[st]|refai[st]|"
r"am[ée]liore[rs]?|fixe?|update|change[rs]?|construis|g[ée]n[èe]re[rs]?)\b",
re.I,
)
def score_code_task(message: str) -> int:
"""Score heuristique : >= 3 -> code, <= 2 -> agent."""
score = 0
if _STRONG.search(message):
score += 2
if _FILE_EXT.search(message):
score += 2
if _ACTION.search(message):
score += 1
if "```" in message:
score += 2
return score
def is_code_task(message: str) -> bool:
"""Décision : heuristique pure, aucune requête modèle."""
return score_code_task(message) >= 3
# Formulations de « suite de travail » : courtes, sans vocabulaire code
# explicite, mais qui prolongent clairement la tâche en cours.
_FOLLOWUP = re.compile(
r"\b(continue[rs]?|reprend[s]?|poursui[st]|termine[rs]?|finis|"
r"rajoute[rs]?|enl[èe]ve[rs]?|retire[rs]?|supprime[rs]?|d[ée]place[rs]?|"
r"agrandi[st]|r[ée]dui[st]|remet[s]?|remplace[rs]?|inverse[rs]?|"
r"plut[ôo]t|aussi|encore|pareil|m[êe]me chose|comme avant)\b",
re.I,
)
def is_code_followup(message: str) -> bool:
"""Reprise probable d'un travail de code en cours.
À n'utiliser que si le tour précédent de la session était du code : un
« ajoute un bouton rouge » isolé n'est pas du code, mais après un
code_task, si.
"""
return bool(_FOLLOWUP.search(message)) or score_code_task(message) >= 1