xinmi/internal/models/embedding/config_from_model_test.go

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)
	}
}