WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

internal/models/embedding/config_from_model_test.go

@@ -0,0 +1,47 @@
+package embedding
+
+import (
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+func TestConfigFromModel(t *testing.T) {
+	m := &types.Model{
+		ID:     "emb-1",
+		Name:   "text-embedding-3-small",
+		Source: types.ModelSourceRemote,
+		Parameters: types.ModelParameters{
+			BaseURL:  "https://api.example.com/v1",
+			APIKey:   "sk-xxx",
+			Provider: "openai",
+			EmbeddingParameters: types.EmbeddingParameters{
+				Dimension:                 1536,
+				TruncatePromptTokens:      512,
+				SupportsDimensionOverride: true,
+			},
+			ExtraConfig:   map[string]string{"region": "us-east"},
+			CustomHeaders: map[string]string{"X-Gateway": "g1"},
+		},
+	}
+
+	cfg := ConfigFromModel(m, "app", "secret")
+	if cfg.ModelID != "emb-1" || cfg.ModelName != "text-embedding-3-small" {
+		t.Errorf("identity mismatch: %+v", cfg)
+	}
+	if cfg.Dimensions != 1536 || cfg.TruncatePromptTokens != 512 {
+		t.Errorf("embedding params mismatch: %+v", cfg)
+	}
+	if !cfg.SupportsDimensionOverride {
+		t.Errorf("SupportsDimensionOverride not propagated: %+v", cfg)
+	}
+	if cfg.CustomHeaders["X-Gateway"] != "g1" {
+		t.Errorf("CustomHeaders not propagated: %+v", cfg.CustomHeaders)
+	}
+	if cfg.ExtraConfig["region"] != "us-east" {
+		t.Errorf("ExtraConfig not propagated: %+v", cfg.ExtraConfig)
+	}
+	if cfg.AppID != "app" || cfg.AppSecret != "secret" {
+		t.Errorf("cloud creds mismatch: %+v", cfg)
+	}
+}