mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
MichaelandClaude Opus 4.8
a34f997950
Sur un long tour, les gros résultats d'outils (MCP 8 Ko, shell 4 Ko) saturaient num_ctx et Ollama tronquait silencieusement le DÉBUT de la conversation : le modèle perdait la consigne et « oubliait » sa tâche. - élagage : les résultats d'outils des itérations passées sont compactés (350 c. + marqueur), le dernier lot reste intact - num_ctx 16384 par défaut (migration v6 : les profils restés sur les anciens défauts 4096/8192 sont montés, les valeurs perso respectées) - serveur MCP avec paramètre requis manquant (SEARXNG_URL) : refus clair au démarrage au lieu d'un échec cryptique à chaque appel Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
import os
|
|
import tempfile
|
|
|
|
os.environ.setdefault("DATA_DIR", tempfile.mkdtemp())
|
|
os.environ.setdefault("WORKSPACE_DIR", tempfile.mkdtemp())
|
|
|
|
import pytest # noqa: E402
|
|
|
|
from app import agent, db, mcp_client # noqa: E402
|
|
|
|
db.init_db()
|
|
|
|
|
|
def test_prune_compacte_les_anciens_resultats():
|
|
big = "x" * 5000
|
|
convo = [
|
|
{"role": "system", "content": "consigne"},
|
|
{"role": "user", "content": "tâche"},
|
|
{"role": "assistant", "content": "", "tool_calls": [{}]},
|
|
{"role": "tool", "tool_name": "read_file", "content": big},
|
|
{"role": "assistant", "content": "", "tool_calls": [{}]},
|
|
{"role": "tool", "tool_name": "grep_search", "content": big},
|
|
]
|
|
# Le dernier lot (index 5) commence à 5 : seul l'index 3 est compacté.
|
|
agent._prune_old_tool_results(convo, before_index=5)
|
|
assert len(convo[3]["content"]) < 500
|
|
assert "archivé" in convo[3]["content"]
|
|
assert convo[5]["content"] == big # lot courant intact
|
|
assert convo[0]["content"] == "consigne" # système intact
|
|
|
|
|
|
def test_prune_ignore_les_petits_resultats():
|
|
convo = [{"role": "tool", "tool_name": "list_dir", "content": "court"}]
|
|
agent._prune_old_tool_results(convo, before_index=1)
|
|
assert convo[0]["content"] == "court"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_searxng_sans_url_erreur_claire():
|
|
mgr = mcp_client.McpManager()
|
|
try:
|
|
result = await mgr.test_server("searxng")
|
|
assert result["ok"] is False
|
|
assert "SEARXNG_URL" in (result["error"] or "")
|
|
assert "requis" in (result["error"] or "")
|
|
finally:
|
|
await mgr.aclose()
|
|
|
|
|
|
def test_migration_v6_monte_les_anciens_defauts():
|
|
from app import agent_config
|
|
profiles = db.get_config_value(agent_config.MODEL_PROFILES_KEY) or {}
|
|
profiles["testmodel:7b"] = {"num_ctx": 8192}
|
|
profiles["custom:7b"] = {"num_ctx": 5000}
|
|
db.set_config_value(agent_config.MODEL_PROFILES_KEY, profiles)
|
|
db.set_config_value(agent_config.PROFILE_STATE_KEY, {"version": 5})
|
|
|
|
agent_config._migrate_profiles()
|
|
|
|
migrated = db.get_config_value(agent_config.MODEL_PROFILES_KEY)
|
|
assert migrated["testmodel:7b"]["num_ctx"] == 16384 # ancien défaut monté
|
|
assert migrated["custom:7b"]["num_ctx"] == 5000 # valeur perso respectée
|