mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
MichaelandClaude Opus 4.8
de2befe31e
Backend : - tools.py : web_search (DuckDuckGo HTML sans clé, ou SearxNG via SEARX_URL) et run_shell (confiné au workspace, sortie bornée, timeout) - agent.py : run_shell sensible -> émet tool_confirm et ne s'exécute pas tant que l'utilisateur n'a pas validé (confirm_shell) - agent_config.py : web_search/run_shell désactivés par défaut, flag confirm_shell - routes/shell.py : POST /api/shell/run exécute une commande validée - routes/chat.py : relaie l'événement tool_confirm, passe confirm_shell Frontend : - streamChat : événement tool_confirm - store : pendingShell + approveShell/rejectShell (réinjecte le résultat à l'agent) - ChatPanel : carte de validation de commande (Approuver / Refuser) - SettingsView : toggles web_search/run_shell, marqueur sensible, switch confirm_shell - ToolCard : glyphes web_search/run_shell, statut 'à valider', aperçu d'argument Tests : run_shell confiné, gate de confirmation (commande dangereuse non exécutée), route shell, parser DuckDuckGo. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01SVay7z3y7q2gEe54ByAE6N
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
"""Profil de configuration de l'agent : invite système, génération, outils.
|
|
|
|
Persisté en base sous la clé `agent`. Fournit les valeurs par défaut et la
|
|
fusion avec ce qui est stocké, pour rester robuste aux montées de version.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from . import db
|
|
|
|
CONFIG_KEY = "agent"
|
|
|
|
DEFAULT_SYSTEM_PROMPT = (
|
|
"Tu es Loki, un assistant de développement local agentique. Tu disposes "
|
|
"d'outils pour lire, écrire et lister des fichiers dans le workspace. "
|
|
"Utilise-les pour accomplir les tâches concrètement, puis réponds de façon "
|
|
"concise en français. Après avoir écrit un fichier, propose un aperçu."
|
|
)
|
|
|
|
# Outils disponibles. Les sensibles (web_search, run_shell) sont désactivés
|
|
# par défaut, conformément à la maquette.
|
|
AVAILABLE_TOOLS = ["read_file", "write_file", "list_dir", "web_search", "run_shell"]
|
|
SENSITIVE_TOOLS = {"run_shell"}
|
|
DEFAULT_TOOL_STATE = {
|
|
"read_file": True,
|
|
"write_file": True,
|
|
"list_dir": True,
|
|
"web_search": False,
|
|
"run_shell": False,
|
|
}
|
|
|
|
DEFAULT_CONFIG: dict = {
|
|
"system_prompt": DEFAULT_SYSTEM_PROMPT,
|
|
"temperature": 0.7,
|
|
"top_p": 0.9,
|
|
"top_k": 40,
|
|
"max_tokens": 2048,
|
|
"tools": dict(DEFAULT_TOOL_STATE),
|
|
# Demander une validation utilisateur avant toute commande shell.
|
|
"confirm_shell": True,
|
|
}
|
|
|
|
|
|
def get_config() -> dict:
|
|
"""Config courante = défauts fusionnés avec le stockage."""
|
|
stored = db.get_config_value(CONFIG_KEY) or {}
|
|
cfg = {**DEFAULT_CONFIG, **stored}
|
|
cfg["tools"] = {
|
|
name: bool(stored.get("tools", {}).get(name, DEFAULT_TOOL_STATE[name]))
|
|
for name in AVAILABLE_TOOLS
|
|
}
|
|
return cfg
|
|
|
|
|
|
def save_config(patch: dict) -> dict:
|
|
"""Applique une mise à jour partielle et renvoie la config complète."""
|
|
cfg = {**get_config(), **{k: v for k, v in patch.items() if v is not None}}
|
|
if "tools" in patch and patch["tools"]:
|
|
cfg["tools"] = {
|
|
name: bool(
|
|
patch["tools"].get(
|
|
name, cfg["tools"].get(name, DEFAULT_TOOL_STATE[name])
|
|
)
|
|
)
|
|
for name in AVAILABLE_TOOLS
|
|
}
|
|
db.set_config_value(CONFIG_KEY, cfg)
|
|
return get_config()
|
|
|
|
|
|
def ollama_options(cfg: dict) -> dict:
|
|
"""Traduit la config en options de génération Ollama."""
|
|
return {
|
|
"temperature": cfg["temperature"],
|
|
"top_p": cfg["top_p"],
|
|
"top_k": cfg["top_k"],
|
|
"num_predict": cfg["max_tokens"],
|
|
}
|
|
|
|
|
|
def enabled_tool_names(cfg: dict) -> list[str]:
|
|
return [name for name, on in cfg["tools"].items() if on]
|