mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-16 19:59:56 +02:00
Claude
ed9b2e8c1c
Port de backend_strata.go (rust/loki-rs/src/strata) : la machine (cartes par nvidia-smi, Python 3.10+ et venv, RAM totale et disponible, RAM du moteur Strata en marche, place libre), ce qui y tient et la recommandation, la configuration de lancement en clair des modèles installés (placement des experts drop/ram/arena/disk, carte d'aide, réglages du preset), les réglages d'un modèle installé et l'installation (job des moteurs : paquet figé vérifié par SHA-256, installeur lancé et journalisé comme le sinkWriter du Go, preset écrit puis activé par le Go). Tenu seulement avec LOKI_BACKENDS=rust ; le lancement (loki serve) reste au Go, qui relaie les routes du tunnel. Correction du Go : /api/strata/settings décidait « en service » après avoir écrit les nouveaux réglages, donc jamais pour un réglage changé ; décidé désormais avant l'écriture, comme SavePresetApplying. Le port suit. Parité : parity_strata_test.go (crochets de test : release, meminfo, place libre, action de service) tire du Go un journal rejoué par tests/strata.rs ; les tests du Go portés dans strata/tests.rs ; TestFacadeStrata côté Go. Mutations : 71 tuées sur 71. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016tcGfJWefW5kUGoC2wL8cv
68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
package loki
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
// Strata part à la façade avec LOKI_STRATA=rust (lot 25) ; le reste du
|
|
// moteur, non.
|
|
func TestFacadeStrata(t *testing.T) {
|
|
testHome(t)
|
|
var mu sync.Mutex
|
|
var calls []string
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Header.Get(internalTokenHeader) != jetonTest {
|
|
w.WriteHeader(403)
|
|
return
|
|
}
|
|
mu.Lock()
|
|
calls = append(calls, r.Method+" "+r.URL.Path)
|
|
mu.Unlock()
|
|
io.WriteString(w, `{"ok":true,"de":"facade"}`)
|
|
}))
|
|
defer srv.Close()
|
|
t.Setenv(internalTokenEnv, jetonTest)
|
|
t.Setenv(facadeAddrEnv, strings.TrimPrefix(srv.URL, "http://"))
|
|
local := func(w http.ResponseWriter, r *http.Request) { io.WriteString(w, "go") }
|
|
|
|
t.Setenv(facadeStrataEnv, "")
|
|
for p := range facadeStrataRoutes {
|
|
w := httptest.NewRecorder()
|
|
withFacadeChat(p, local)(w, httptest.NewRequest("POST", p, strings.NewReader("{}")))
|
|
if w.Body.String() != "go" {
|
|
t.Fatalf("sans LOKI_STRATA, %s : %q", p, w.Body.String())
|
|
}
|
|
}
|
|
t.Setenv(facadeStrataEnv, " RUST")
|
|
if len(facadeStrataRoutes) != 3 {
|
|
t.Fatalf("routes : %v", facadeStrataRoutes)
|
|
}
|
|
for p := range facadeStrataRoutes {
|
|
w := httptest.NewRecorder()
|
|
withFacadeChat(p, local)(w, httptest.NewRequest("POST", p, strings.NewReader("{}")))
|
|
if !strings.Contains(w.Body.String(), "facade") {
|
|
t.Fatalf("%s non relayé : %q", p, w.Body.String())
|
|
}
|
|
}
|
|
mu.Lock()
|
|
n := len(calls)
|
|
mu.Unlock()
|
|
if n != len(facadeStrataRoutes) {
|
|
t.Fatalf("relais : %v", calls)
|
|
}
|
|
w := httptest.NewRecorder()
|
|
withFacadeChat("/api/llamacpp/install", local)(w, httptest.NewRequest("POST", "/api/llamacpp/install", nil))
|
|
if w.Body.String() != "go" {
|
|
t.Fatalf("la compilation reste au Go : %q", w.Body.String())
|
|
}
|
|
t.Setenv(internalTokenEnv, "")
|
|
if facadeStrata() {
|
|
t.Fatal("tenu sans jeton interne")
|
|
}
|
|
}
|