mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-17 12:16:21 +02:00
nathaninline
db47c72240
- start/restart : preflight (BIN, MODEL, .gguf) et fin du faux « [ok] activating » sur un service qui redemarre en boucle (SubState auto-restart). - MODEL en chemin absolu accepte hors des dossiers declares (lecture seule) ; l'allowlist reste sur la suppression de fichiers. - ajean edit : squelette commente des cles au lieu d'un fichier quasi vide. - install : les prochaines etapes n'oublient plus « ajean llamacpp install ». - Presets : un nouveau preset ne recopie plus la config active (issue #17), seulement BIN/HOST/PORT. - Presets : plus de guillemet interne mange a la relecture (Go et interface), et EXTRA_ARGS decoupe en respectant les guillemets.
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
package ajean
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
// Un guillemet INTERNE (argument cité dans EXTRA_ARGS) ne doit pas être mangé :
|
|
// l'ancien Trim retirait le guillemet final et renvoyait une valeur déséquilibrée
|
|
// dans le preset (issue #17).
|
|
func TestParseEnvKeepsInnerQuotes(t *testing.T) {
|
|
cases := map[string]string{
|
|
`EXTRA_ARGS=--jinja --chat-template-file "/etc/ajean/tpl.jinja"`: `--jinja --chat-template-file "/etc/ajean/tpl.jinja"`,
|
|
`EXTRA_ARGS="--jinja --flash-attn"`: `--jinja --flash-attn`,
|
|
`MODEL='/mnt/d/x.gguf'`: `/mnt/d/x.gguf`,
|
|
`MODEL=/mnt/d/x.gguf`: `/mnt/d/x.gguf`,
|
|
}
|
|
for line, want := range cases {
|
|
m := parseEnv(line)
|
|
var got string
|
|
for _, v := range m {
|
|
got = v
|
|
}
|
|
if got != want {
|
|
t.Errorf("parseEnv(%q) = %q, veut %q", line, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Écrire puis relire doit rendre exactement la même configuration.
|
|
func TestFormatEnvRoundTrip(t *testing.T) {
|
|
in := map[string]string{
|
|
"MODEL": "/mnt/mes modèles/x.gguf",
|
|
"EXTRA_ARGS": `--jinja --chat-template-file "/etc/ajean/tpl.jinja"`,
|
|
"CTX": "32768",
|
|
}
|
|
out := parseEnv(formatEnv(in))
|
|
if !reflect.DeepEqual(in, out) {
|
|
t.Fatalf("aller-retour cassé :\n%v\n%v", in, out)
|
|
}
|
|
}
|
|
|
|
func TestSplitArgs(t *testing.T) {
|
|
got := splitArgs(`--jinja --chat-template-file "/etc/mes modèles/tpl.jinja" --flash-attn`)
|
|
want := []string{"--jinja", "--chat-template-file", "/etc/mes modèles/tpl.jinja", "--flash-attn"}
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("splitArgs = %q, veut %q", got, want)
|
|
}
|
|
if len(splitArgs("")) != 0 {
|
|
t.Fatal("EXTRA_ARGS vide doit donner zéro argument")
|
|
}
|
|
}
|
|
|
|
// L'éditeur doit proposer les clés utiles même quand la configuration est vide,
|
|
// et ne jamais perdre une clé déjà définie.
|
|
func TestConfigEditorText(t *testing.T) {
|
|
txt := configEditorText(map[string]string{"MODEL": "x.gguf", "INCONNUE": "1"})
|
|
round := parseEnv(txt)
|
|
if round["MODEL"] != "x.gguf" || round["INCONNUE"] != "1" {
|
|
t.Fatalf("clés perdues : %v", round)
|
|
}
|
|
if len(round) != 2 {
|
|
t.Fatalf("les clés non renseignées doivent rester commentées : %v", round)
|
|
}
|
|
if !contains(txt, "#BIN=") {
|
|
t.Fatal("le squelette doit documenter BIN")
|
|
}
|
|
}
|
|
|
|
func contains(s, sub string) bool {
|
|
for i := 0; i+len(sub) <= len(s); i++ {
|
|
if s[i:i+len(sub)] == sub {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|