feat: unify installation model catalog

2026-09-02 18:45:33 +02:00
parent ae053961a3
commit 7b7927bfe5
169 changed files with 3696 additions and 4572 deletions
@@ -0,0 +1,91 @@
+package modelprojection
+
+import (
+	"bytes"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+
+	"github.com/aritmolab/thothii/tools/tht/internal/config"
+)
+
+func TestRenderProducesDeterministicCatalogPiAndComposeProjections(t *testing.T) {
+	installation := projectionFixture(t)
+
+	first, err := Render(installation)
+	if err != nil {
+		t.Fatalf("Render() error = %v", err)
+	}
+	second, err := Render(installation)
+	if err != nil {
+		t.Fatalf("Render() second error = %v", err)
+	}
+	if len(first) != 4 {
+		t.Fatalf("artifact count = %d, want 4", len(first))
+	}
+	for path, contents := range first {
+		if !bytes.Equal(contents, second[path]) {
+			t.Fatalf("artifact %q is not deterministic", path)
+		}
+	}
+
+	catalog := string(first[CatalogFile])
+	for _, expected := range []string{`"id": "local/qwen"`, `"defaultSession": "zai/glm-5.3"`, `"embedding"`} {
+		if !strings.Contains(catalog, expected) {
+			t.Fatalf("catalog projection missing %s:\n%s", expected, catalog)
+		}
+	}
+	piModels := string(first[PiModelsFile])
+	if !strings.Contains(piModels, `"zai"`) || strings.Contains(piModels, `"deepseek"`) {
+		t.Fatalf("Pi models projection must contain only custom providers:\n%s", piModels)
+	}
+	settings := string(first[PiSettingsFile])
+	if !strings.Contains(settings, `"deepseek/deepseek-v4-pro"`) || !strings.Contains(settings, `"local/qwen"`) {
+		t.Fatalf("Pi settings projection missing enabled session models:\n%s", settings)
+	}
+	compose := string(first[ComposeFile])
+	for _, expected := range []string{"THT_MODEL_CATALOG_FILE", "THT_DEFAULT_SESSION_MODEL", "zai/glm-5.3", "THT_INTERNAL_EMBEDDING_ID", "THT_INTERNAL_EMBEDDING_MODEL", "OLLAMA_MODEL", installation.GeneratedModelCatalogPath()} {
+		if !strings.Contains(compose, expected) {
+			t.Fatalf("Compose projection missing %q:\n%s", expected, compose)
+		}
+	}
+}
+
+func TestGeneratePublishesAllArtifactsAndCheckDetectsDrift(t *testing.T) {
+	installation := projectionFixture(t)
+	if err := Generate(installation); err != nil {
+		t.Fatalf("Generate() error = %v", err)
+	}
+	for _, path := range []string{installation.GeneratedModelCatalogPath(), installation.GeneratedPiModelsPath(), installation.GeneratedPiSettingsPath(), installation.ModelProjectionComposePath()} {
+		if info, err := os.Stat(path); err != nil || !info.Mode().IsRegular() {
+			t.Fatalf("generated artifact %q: info=%v err=%v", path, info, err)
+		}
+	}
+	if drift, err := Check(installation); err != nil || len(drift) != 0 {
+		t.Fatalf("Check() = %v, %v; want no drift", drift, err)
+	}
+	if err := os.WriteFile(installation.GeneratedPiSettingsPath(), []byte("{}\n"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	if drift, err := Check(installation); err != nil || len(drift) != 1 || drift[0] != PiSettingsFile {
+		t.Fatalf("Check() = %v, %v; want [%s]", drift, err, PiSettingsFile)
+	}
+}
+
+func projectionFixture(t *testing.T) config.Installation {
+	t.Helper()
+	root := t.TempDir()
+	return config.Installation{
+		Path: filepath.Join(root, "thothii-installation.yaml"),
+		ModelCatalog: config.ModelCatalog{
+			Defaults:  config.ModelCatalogDefaults{Session: "zai/glm-5.3", MetadataGeneration: "local/qwen"},
+			Embedding: config.ModelCatalogEmbedding{ID: "ollama/qwen3-embedding:0.6b", Dimensions: 1024},
+			Providers: map[string]config.ModelProvider{
+				"deepseek": {Authentication: config.ModelAuthentication{Mode: "pi_auth"}, Session: &config.ModelSessionAdapter{Mode: "pi_builtin"}, Models: map[string]config.CatalogModel{"deepseek-v4-pro": {Session: &config.SessionModel{}}}},
+				"zai":      {Endpoint: &config.ModelEndpoint{BaseURL: "https://api.z.ai/v1"}, Authentication: config.ModelAuthentication{Mode: "none"}, Session: &config.ModelSessionAdapter{Mode: "openai_compatible"}, Models: map[string]config.CatalogModel{"glm-5.3": {Label: "GLM-5.3", Session: &config.SessionModel{Reasoning: true, ContextWindow: 200000, MaxTokens: 131072}}}},
+				"local":    {Endpoint: &config.ModelEndpoint{BaseURL: "http://ollama:11434/v1"}, Authentication: config.ModelAuthentication{Mode: "none"}, Session: &config.ModelSessionAdapter{Mode: "openai_compatible"}, MetadataGeneration: &config.ModelMetadataAdapter{LiteLLMProvider: "openai"}, Models: map[string]config.CatalogModel{"qwen": {Session: &config.SessionModel{ContextWindow: 32768, MaxTokens: 8192}, MetadataGeneration: &config.MetadataGenerationModel{DisableThinking: true}}}},
+			},
+		},
+	}
+}