mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
Loki
de6551a153
- module github.com/R0m1k3/Loki, cmd/loki, internal/loki (package loki) - LOKI_HOME, LOKI_MODEL_DIRS, LOKI_SERVICE, LOKI_DL_CONNS ; /etc/loki ; units loki-engine / loki-ui ; binaire et aide CLI - updateRepo pointe sur R0m1k3/Loki (l'auto-update ne tirera plus les binaires AJEAN amont) Conservé à l'identique : le domaine ajean.link (service de tunnel amont), les littéraux de migration 0.7.x (migrate_07.go), RELEASE_NOTES.md et LICENSE (historique et licence de l'amont). go build/vet/test : verts.
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
package loki
|
|
|
|
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/loki/tpl.jinja"`: `--jinja --chat-template-file "/etc/loki/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/loki/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
|
|
}
|