mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-15 03:08:56 +02:00
nathaninline
73a772520c
Un .gguf pose sur un disque externe etait invisible dans l'UI : la liste des modeles ne scannait que JEAN_HOME et un preset pointant ailleurs affichait « introuvable ». On peut desormais declarer des dossiers de modeles supplementaires (replies derriere une ligne dans l'editeur de preset, persistes dans model_dirs.json, ou via JEAN_MODEL_DIRS). Acces distant : la pastille affiche « non connecte » au lieu de rester vide quand le serveur n'est lie a aucun compte.
73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
package jean
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
// Un modèle posé hors de JEAN_HOME (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) {
|
|
home := t.TempDir()
|
|
ext := t.TempDir()
|
|
t.Setenv("JEAN_HOME", home)
|
|
t.Setenv("JEAN_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)
|
|
}
|
|
}
|
|
|
|
// JEAN_HOME garde la priorité et les non-.gguf restent refusés.
|
|
func TestResolveModelPathHomeFirst(t *testing.T) {
|
|
home := t.TempDir()
|
|
ext := t.TempDir()
|
|
t.Setenv("JEAN_HOME", home)
|
|
t.Setenv("JEAN_MODEL_DIRS", ext)
|
|
for _, d := range []string{home, 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(home, "m.gguf") {
|
|
t.Fatalf("got %q (%v), attendu le fichier de JEAN_HOME", 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)
|
|
}
|
|
}
|