mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-22 22:48:11 +02:00
Migration : reecrire les chemins absolus pointant vers l'ancien dossier
Bug trouve en testant la migration sur une vraie machine (5,4 Go, 51 351 fichiers). Le rename du dossier reussissait, mais config.env contenait BIN=C:\ProgramData\jean\backends\llama.cpp\build\bin\Release\llama-server.exe soit un chemin ABSOLU vers le dossier qu'on venait de renommer. Resultat : le dossier migre intact, et llama-server ne demarre plus. Ca touche tous ceux qui ont fait un `llamacpp install`, donc le cas nominal, pas un cas limite. Les presets de configs/ portent le meme BIN et etaient touches pareil (deux presets concernes sur la machine de test). Apres un rename reussi, on reecrit donc les references a l'ancien chemin dans les fichiers de config qu'on ecrit nous-memes : config.env, model_dirs.json, mcp.json, webprefs.json et les presets de configs/. Les deux ecritures de separateur sont couvertes, Windows acceptant \ et / dans une meme valeur. Restreint a ces fichiers texte : on ne reecrit pas a l'aveugle les 51 000 fichiers d'un dossier de donnees. Les fichiers d'etat du service sont repris au passage (jean.pid/jean.log -> ajean.pid/ajean.log), meme raison que pour le worker de lien : le PID dit si le service tourne, l'ignorer ferait demarrer un second service. Non traite volontairement : l'arborescence de build llama.cpp (CMakeCache.txt et consorts) contient aussi des chemins absolus. Elle est regenerable par `ajean llamacpp update` et la reecrire en masse serait plus risque qu'utile. Verifie de bout en bout sur la machine de test : service demarre, llama-server chargé depuis le nouveau chemin, /health ok, completion reelle ok. Deux tests ajoutes reproduisant le cas.
This commit is contained in:
1 parent
993be881d5
commit
8a2b5c6687
3 files changed
+156
-2
No files matched your search
@@ -4,6 +4,7 @@ import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
@@ -100,3 +101,74 @@ func TestMigrateHomeFallsBackWhenRenameFails(t *testing.T) {
|
||||
t.Fatalf("l'ancien dossier a été abîmé par une migration ratée: %q (%v)", b, err)
|
||||
}
|
||||
}
|
||||
|
||||
// Le cas qui a casse une vraie machine : config.env contient un chemin ABSOLU
|
||||
// vers le dossier de donnees (BIN=<home>\backends\...\llama-server.exe). Apres
|
||||
// migration ce chemin doit suivre, sinon llama-server ne demarre plus — et ca
|
||||
// concerne tous ceux qui ont fait un `llamacpp install`, donc le cas nominal.
|
||||
func TestMigrateHomeRewritesAbsolutePaths(t *testing.T) {
|
||||
root := t.TempDir()
|
||||
legacy := filepath.Join(root, "jean")
|
||||
target := filepath.Join(root, "ajean")
|
||||
mkHome(t, legacy)
|
||||
if err := os.MkdirAll(filepath.Join(legacy, "configs"), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
bin := filepath.Join(legacy, "backends", "llama.cpp", "build", "bin", "llama-server")
|
||||
write := func(rel, body string) {
|
||||
if err := os.WriteFile(filepath.Join(legacy, rel), []byte(body), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
write("config.env", "BIN="+bin+"\nMODEL=\"x.gguf\"\n")
|
||||
// Les presets portent le meme BIN et doivent suivre aussi.
|
||||
write(filepath.Join("configs", "gros.env"), "BIN="+bin+"\n")
|
||||
// Ecriture avec des slashs : Windows accepte les deux formes.
|
||||
write("model_dirs.json", `{"dirs":["`+filepath.ToSlash(legacy)+`/models"]}`)
|
||||
|
||||
if got := migrateHome(target, legacy); got != target {
|
||||
t.Fatalf("migration echouee: %q", got)
|
||||
}
|
||||
|
||||
for _, rel := range []string{"config.env", filepath.Join("configs", "gros.env"), "model_dirs.json"} {
|
||||
b, err := os.ReadFile(filepath.Join(target, rel))
|
||||
if err != nil {
|
||||
t.Fatalf("%s illisible: %v", rel, err)
|
||||
}
|
||||
if strings.Contains(string(b), legacy) || strings.Contains(string(b), filepath.ToSlash(legacy)) {
|
||||
t.Errorf("%s reference encore l'ancien dossier:\n%s", rel, b)
|
||||
}
|
||||
if !strings.Contains(string(b), target) && !strings.Contains(string(b), filepath.ToSlash(target)) {
|
||||
t.Errorf("%s ne pointe pas vers le nouveau dossier:\n%s", rel, b)
|
||||
}
|
||||
}
|
||||
|
||||
// Le BIN reecrit doit designer un chemin reellement atteignable.
|
||||
newBin := filepath.Join(target, "backends", "llama.cpp", "build", "bin", "llama-server")
|
||||
if b, _ := os.ReadFile(filepath.Join(target, "config.env")); !strings.Contains(string(b), newBin) {
|
||||
t.Errorf("BIN ne pointe pas sur %s:\n%s", newBin, b)
|
||||
}
|
||||
}
|
||||
|
||||
// Les fichiers d'etat du service (jean.pid/jean.log) doivent etre repris : le
|
||||
// PID dit si le service tourne, l'ignorer ferait demarrer un second service.
|
||||
func TestMigrateHomeAdoptsServiceStateFiles(t *testing.T) {
|
||||
root := t.TempDir()
|
||||
legacy := filepath.Join(root, "jean")
|
||||
target := filepath.Join(root, "ajean")
|
||||
mkHome(t, legacy)
|
||||
if err := os.WriteFile(filepath.Join(legacy, "jean.pid"), []byte("4242\n"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
migrateHome(target, legacy)
|
||||
|
||||
b, err := os.ReadFile(filepath.Join(target, "ajean.pid"))
|
||||
if err != nil {
|
||||
t.Fatalf("ajean.pid absent apres migration: %v", err)
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "4242" {
|
||||
t.Fatalf("PID perdu: %q", b)
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user