mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-21 14:08:11 +02:00
Rebaptise AJEAN en Loki (fork, lignée conservée)
- 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.
This commit is contained in:
157 files changed
+928
-928
No files matched your search
@@ -0,0 +1,73 @@
|
||||
package loki
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// Un modèle posé hors de models/ (disque externe) doit être utilisable dès
|
||||
// que son dossier est déclaré — et refusé tant qu'il ne l'est pas.
|
||||
func TestResolveModelPathExtraDir(t *testing.T) {
|
||||
testHome(t)
|
||||
ext := t.TempDir()
|
||||
t.Setenv("LOKI_MODEL_DIRS", "")
|
||||
|
||||
extModel := filepath.Join(ext, "gros.gguf")
|
||||
if err := os.WriteFile(extModel, []byte("x"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if _, err := resolveModelPath(extModel); err == nil {
|
||||
t.Fatal("un dossier non déclaré devrait être refusé")
|
||||
}
|
||||
|
||||
if err := saveExtraModelDirs([]string{ext}); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
got, err := resolveModelPath(extModel)
|
||||
if err != nil {
|
||||
t.Fatalf("chemin absolu déclaré refusé : %v", err)
|
||||
}
|
||||
if got != filepath.Clean(extModel) {
|
||||
t.Fatalf("chemin = %s, attendu %s", got, extModel)
|
||||
}
|
||||
// Le simple nom de fichier doit aussi être retrouvé dans le dossier ajouté.
|
||||
if got, err := resolveModelPath("gros.gguf"); err != nil || got != extModel {
|
||||
t.Fatalf("nom seul = %q (%v), attendu %s", got, err, extModel)
|
||||
}
|
||||
}
|
||||
|
||||
// models/ garde la priorité et les non-.gguf restent refusés.
|
||||
func TestResolveModelPathHomeFirst(t *testing.T) {
|
||||
testHome(t)
|
||||
ext := t.TempDir()
|
||||
t.Setenv("LOKI_MODEL_DIRS", ext)
|
||||
if err := os.MkdirAll(modelsDir(), 0o755); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
for _, d := range []string{modelsDir(), ext} {
|
||||
if err := os.WriteFile(filepath.Join(d, "m.gguf"), []byte("x"), 0o644); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
got, err := resolveModelPath("m.gguf")
|
||||
if err != nil || got != filepath.Join(modelsDir(), "m.gguf") {
|
||||
t.Fatalf("got %q (%v), attendu le fichier de models/", got, err)
|
||||
}
|
||||
if _, err := resolveModelPath("/etc/passwd"); err == nil {
|
||||
t.Fatal("un fichier non-.gguf devrait être refusé")
|
||||
}
|
||||
}
|
||||
|
||||
// La ligne MODEL= garde son chemin complet, mais la quantization se déduit
|
||||
// toujours du nom de fichier.
|
||||
func TestPresetModelKeepsPath(t *testing.T) {
|
||||
content := "MODEL=\"/mnt/ext/Qwen3-30B-Q5_K_M.gguf\"\nCTX=8192\n"
|
||||
if got := modelFromPresetContent(content); got != "/mnt/ext/Qwen3-30B-Q5_K_M.gguf" {
|
||||
t.Fatalf("MODEL = %q, chemin tronqué", got)
|
||||
}
|
||||
if got := detectQuant(content); got != "Q5_K_M" {
|
||||
t.Fatalf("quant = %q, attendu Q5_K_M", got)
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user