mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
MichaelandClaude Opus 4.8
bddab1c619
« Corrige les bugs » partait à Aider sans contexte (quel fichier ? quel projet ?) et le chemin agent ignorait quels fichiers existaient — le modèle collait alors le code en chat au lieu de modifier le fichier. - moteur code : récap de session (demande initiale + dernières demandes + fichiers touchés) préfixé à l'instruction ; fichiers de la session passés à Aider quand aucun n'est cité - chemin agent : liste des fichiers du workspace + fichiers de la tâche injectée chaque tour en système - directive renforcée : « ne colle JAMAIS le code corrigé dans ta réponse sans l'avoir écrit dans le fichier » Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
import os
|
|
import tempfile
|
|
|
|
os.environ.setdefault("DATA_DIR", tempfile.mkdtemp())
|
|
os.environ.setdefault("WORKSPACE_DIR", tempfile.mkdtemp())
|
|
|
|
from app import router # noqa: E402
|
|
from app.routes.chat import ( # noqa: E402
|
|
_mentioned_files,
|
|
_prev_was_code,
|
|
_session_code_context,
|
|
_workspace_listing,
|
|
)
|
|
from app.config import settings # noqa: E402
|
|
|
|
|
|
def test_message_code_explicite():
|
|
assert router.is_code_task("crée une page html avec un formulaire") is True
|
|
|
|
|
|
def test_message_banal_pas_code():
|
|
assert router.is_code_task("quelle heure est-il ?") is False
|
|
|
|
|
|
def test_followup_courts():
|
|
# Suites typiques d'un travail de code, sans vocabulaire code explicite.
|
|
assert router.is_code_followup("ajoute un bouton rouge en haut") is True
|
|
assert router.is_code_followup("continue") is True
|
|
assert router.is_code_followup("enlève le menu et agrandis le titre") is True
|
|
assert router.is_code_followup("merci beaucoup !") is False
|
|
|
|
|
|
def test_prev_was_code_via_engine():
|
|
history = [
|
|
{"role": "user", "content": "crée un site", "meta": None},
|
|
{"role": "assistant", "content": "fait", "meta": {"engine": "code"}},
|
|
]
|
|
assert _prev_was_code(history) is True
|
|
|
|
|
|
def test_prev_was_code_via_write_file():
|
|
history = [
|
|
{"role": "assistant", "content": "ok",
|
|
"meta": {"tools": [{"name": "write_file", "args": {}}]}},
|
|
]
|
|
assert _prev_was_code(history) is True
|
|
|
|
|
|
def test_prev_was_code_discussion():
|
|
history = [
|
|
{"role": "assistant", "content": "voici la réponse", "meta": None},
|
|
]
|
|
assert _prev_was_code(history) is False
|
|
|
|
|
|
def test_mentioned_files_existants_seulement():
|
|
root = os.path.abspath(settings.workspace_dir)
|
|
os.makedirs(root, exist_ok=True)
|
|
with open(os.path.join(root, "index.html"), "w", encoding="utf-8") as f:
|
|
f.write("<html></html>")
|
|
found = _mentioned_files("modifie index.html et style.css")
|
|
assert found == ["index.html"] # style.css n'existe pas
|
|
|
|
|
|
def test_session_code_context_recap_et_fichiers():
|
|
root = os.path.abspath(settings.workspace_dir)
|
|
os.makedirs(root, exist_ok=True)
|
|
with open(os.path.join(root, "jeu.html"), "w", encoding="utf-8") as f:
|
|
f.write("<html>jeu</html>")
|
|
history = [
|
|
{"role": "user", "content": "crée un jeu snake dans jeu.html",
|
|
"meta": None},
|
|
{"role": "assistant", "content": "fait",
|
|
"meta": {"engine": "code",
|
|
"tools": [{"name": "write_file",
|
|
"args": {"path": "jeu.html"}}]}},
|
|
{"role": "user", "content": "il y a des bugs", "meta": None},
|
|
]
|
|
recap, files = _session_code_context(history)
|
|
assert "crée un jeu snake" in recap
|
|
assert files == ["jeu.html"]
|
|
|
|
|
|
def test_session_code_context_vide_sans_historique():
|
|
recap, files = _session_code_context([])
|
|
assert recap == ""
|
|
assert files == []
|
|
|
|
|
|
def test_workspace_listing_contient_fichiers():
|
|
root = os.path.abspath(settings.workspace_dir)
|
|
os.makedirs(root, exist_ok=True)
|
|
with open(os.path.join(root, "jeu.html"), "w", encoding="utf-8") as f:
|
|
f.write("x")
|
|
listing = _workspace_listing()
|
|
assert "jeu.html" in listing
|