mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-16 19:59:56 +02:00
Avec LOKI_TUNE=rust (expérimental, Unix seulement), la façade sert /api/tune, /api/tune/status, /api/tune/cancel, /api/tune/apply et /api/tune/last, et déroule l'optimisation sans perte (rust/loki-rs/src/tune) : vrai moteur arrêté, essais composés à blanc puis lancés par les « loki serve » du Go (LOKI_TUNE_TRIAL), chacun dans son groupe de processus, mesurés par le bench complet à profondeur fixe, retenus au-delà du bruit ; vrai moteur relancé. Rien n'est écrit sans un clic : copie du preset, ou le preset lui-même — redémarrage, sonde, et ancienne version rétablie si elle échoue. - tune/core.rs : la partie pure (EXTRA_ARGS, réglages, essais de chaque étape, garde-fous, diff et réécriture du preset, score) ; - tune/lock.rs : le verrou tune.lock au format du Go, identité des processus comme procIdentity (la reprise du Go reconnaît la façade vivante, et écarte son verrou si elle meurt), ménage d'un verrou périmé (essai orphelin arrêté, application interrompue défaite) ; - tune/job.rs : routes, job, essais et application. Le verrou de génération reste celui du Go, pris par le bail du moteur au motif « tune » (refus de l'optimiseur pour les autres, stop du chat qui annule). Le Go prête /internal/v1/tune/context (bench, génération, jobs du mode code, état du moteur, télémétrie) et le démarrage ou l'arrêt du moteur sans le refus de l'optimiseur (service/action « os »). Hors Unix, la façade passe LOKI_TUNE vide au Go, qui garde l'optimiseur. serde_json lit désormais les flottants à l'arrondi exact (float_roundtrip), comme le Go. Parité : parity_tune_test.go tire du Go un journal (partie pure, contenu du verrou, résultat enregistré, vérification après application) rejoué par tests/tune_core.rs. tests/tune.rs déroule de vraies optimisations, le binaire de test jouant « loki serve » ; les tests du verrou du Go sont portés ; TestFacadeTune côté Go. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
112 lines
3.7 KiB
Go
112 lines
3.7 KiB
Go
package loki
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
// facade_system.go — interrupteurs système tenus par la façade Rust
|
|
// (LOKI_SYSTEM=rust, lot 17 de la migration ; expérimental, le Go reste le
|
|
// défaut) : accès internet (/api/internet), pilotage du navigateur
|
|
// (/api/computer), clé de complétion (/api/apikey) et écoute réseau du moteur
|
|
// (/api/network).
|
|
//
|
|
// Les réglages restent des clés de la base. Ce qui vit dans ce processus reste
|
|
// ici, derrière l'API interne :
|
|
//
|
|
// - le cache de joignabilité de Crawl4AI des outils web, vidé quand l'URL, la
|
|
// clé ou le moteur changent (/internal/v1/web/reach/reset) ;
|
|
// - le Chromium piloté par les outils browser_*, fermé quand l'interrupteur
|
|
// se coupe (/internal/v1/cdp/shutdown) ;
|
|
// - le redémarrage du moteur qui applique une nouvelle clé, quand la façade
|
|
// ne pilote pas le moteur (/internal/v1/service/action).
|
|
|
|
const facadeSystemEnv = "LOKI_SYSTEM"
|
|
|
|
// facadeSystem : la façade tient-elle les interrupteurs système ?
|
|
func facadeSystem() bool {
|
|
return strings.EqualFold(strings.TrimSpace(os.Getenv(facadeSystemEnv)), "rust") &&
|
|
facadeAddr() != "" && internalToken() != ""
|
|
}
|
|
|
|
// facadeSystemRoutes : les routes que la façade sert alors (system.rs).
|
|
var facadeSystemRoutes = map[string]bool{
|
|
"/api/internet": true,
|
|
"/api/computer": true,
|
|
"/api/apikey": true,
|
|
"/api/network": true,
|
|
}
|
|
|
|
// handleInternalReachReset : POST → 204 — le prochain tour reteste Crawl4AI.
|
|
func handleInternalReachReset(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
sendJSON(w, http.StatusMethodNotAllowed, map[string]any{"error": "POST attendu"})
|
|
return
|
|
}
|
|
reachMu.Lock()
|
|
reachURL = ""
|
|
reachMu.Unlock()
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}
|
|
|
|
// handleInternalCDPShutdown : POST → 204 — ferme le navigateur piloté.
|
|
func handleInternalCDPShutdown(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
sendJSON(w, http.StatusMethodNotAllowed, map[string]any{"error": "POST attendu"})
|
|
return
|
|
}
|
|
cdpShutdown()
|
|
w.WriteHeader(http.StatusNoContent)
|
|
}
|
|
|
|
// internalServiceAction : serviceAction, variable pour les tests.
|
|
var internalServiceAction = serviceAction
|
|
|
|
// internalServiceActionOS : serviceActionOS, variable pour les tests.
|
|
var internalServiceActionOS = func(a string) error { return serviceActionOSFn(a) }
|
|
|
|
// handleInternalServiceAction : POST {action, if_active} → 200 {ran, error}.
|
|
// if_active : seulement si le moteur tourne (une clé neuve ne s'applique qu'au
|
|
// redémarrage, inutile de démarrer un moteur arrêté pour elle).
|
|
func handleInternalServiceAction(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
sendJSON(w, http.StatusMethodNotAllowed, map[string]any{"error": "POST attendu"})
|
|
return
|
|
}
|
|
var in struct {
|
|
Action string `json:"action"`
|
|
IfActive bool `json:"if_active"`
|
|
// OS : sans le refus de l'optimiseur (serviceActionOS) — pour la
|
|
// façade qui tient elle-même son verrou (LOKI_TUNE=rust).
|
|
OS bool `json:"os"`
|
|
}
|
|
if err := json.NewDecoder(r.Body).Decode(&in); err != nil {
|
|
sendJSON(w, http.StatusBadRequest, map[string]any{"error": err.Error()})
|
|
return
|
|
}
|
|
switch in.Action {
|
|
case "start", "stop", "restart":
|
|
default:
|
|
sendJSON(w, http.StatusBadRequest, map[string]any{"error": "action inconnue : " + in.Action})
|
|
return
|
|
}
|
|
if in.IfActive && !serviceIsActive() {
|
|
sendJSON(w, http.StatusOK, map[string]any{"ran": false, "error": ""})
|
|
return
|
|
}
|
|
act := internalServiceAction
|
|
if in.OS {
|
|
act = internalServiceActionOS
|
|
}
|
|
msg := ""
|
|
if err := act(in.Action); err != nil {
|
|
msg = err.Error()
|
|
}
|
|
if in.OS {
|
|
gpuMonitor.invalidate()
|
|
}
|
|
sendJSON(w, http.StatusOK, map[string]any{"ran": true, "error": msg})
|
|
}
|