102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
package config
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
// 注意：这些测试函数测试的是未导出的私有函数
|
|
// 如果编译失败，说明函数未定义或包名不匹配
|
|
|
|
func TestIsRerankModelName(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
model string
|
|
expected bool
|
|
}{
|
|
{"gte-rerank-v2", "gte-rerank-v2", true},
|
|
{"qwen3-rerank", "qwen3-rerank", true},
|
|
{"bge-rerank-large", "bge-rerank-large", true},
|
|
{"text-embedding-v4", "text-embedding-v4", false},
|
|
{"text-embedding-v3", "text-embedding-v3", false},
|
|
{"qwen-turbo", "qwen-turbo", false},
|
|
{"empty string", "", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result := isRerankModelName(tt.model)
|
|
if result != tt.expected {
|
|
t.Errorf("isRerankModelName(%q) = %v, want %v", tt.model, result, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestIsEmbeddingModelName(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
model string
|
|
expected bool
|
|
}{
|
|
{"text-embedding-v4", "text-embedding-v4", true},
|
|
{"text-embedding-v3", "text-embedding-v3", true},
|
|
{"bge-large-zh", "bge-large-zh", true},
|
|
{"gte-large", "gte-large", true},
|
|
{"gte-rerank-v2", "gte-rerank-v2", false},
|
|
{"qwen3-rerank", "qwen3-rerank", false},
|
|
{"qwen-turbo", "qwen-turbo", false},
|
|
{"empty string", "", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result := isEmbeddingModelName(tt.model)
|
|
if result != tt.expected {
|
|
t.Errorf("isEmbeddingModelName(%q) = %v, want %v", tt.model, result, tt.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyDefaultsFixesWrongModelConfig(t *testing.T) {
|
|
// 测试错误配置会被自动修正
|
|
cfg := NewDefaultConfig()
|
|
|
|
// 模拟用户错误地将 Rerank 模型填到 Embedding 字段
|
|
cfg.AutoReplyConfig.Retrieval.EmbeddingModel = "gte-rerank-v2"
|
|
cfg.AutoReplyConfig.Retrieval.RerankModel = "qwen3-rerank"
|
|
|
|
cfg.ApplyDefaults()
|
|
|
|
// 验证错误的 Embedding 模型被修正为默认值
|
|
if cfg.AutoReplyConfig.Retrieval.EmbeddingModel != "text-embedding-v4" {
|
|
t.Errorf("Expected embedding model to be corrected to 'text-embedding-v4', got %q", cfg.AutoReplyConfig.Retrieval.EmbeddingModel)
|
|
}
|
|
|
|
// 验证 Rerank 模型保持不变
|
|
if cfg.AutoReplyConfig.Retrieval.RerankModel != "qwen3-rerank" {
|
|
t.Errorf("Expected rerank model to remain 'qwen3-rerank', got %q", cfg.AutoReplyConfig.Retrieval.RerankModel)
|
|
}
|
|
}
|
|
|
|
func TestApplyDefaultsFixesWrongRerankConfig(t *testing.T) {
|
|
// 测试错误配置会被自动修正
|
|
cfg := NewDefaultConfig()
|
|
|
|
// 模拟用户错误地将 Embedding 模型填到 Rerank 字段
|
|
cfg.AutoReplyConfig.Retrieval.EmbeddingModel = "text-embedding-v4"
|
|
cfg.AutoReplyConfig.Retrieval.RerankModel = "text-embedding-v3"
|
|
|
|
cfg.ApplyDefaults()
|
|
|
|
// 验证 Embedding 模型保持不变
|
|
if cfg.AutoReplyConfig.Retrieval.EmbeddingModel != "text-embedding-v4" {
|
|
t.Errorf("Expected embedding model to remain 'text-embedding-v4', got %q", cfg.AutoReplyConfig.Retrieval.EmbeddingModel)
|
|
}
|
|
|
|
// 验证错误的 Rerank 模型被修正为默认值
|
|
if cfg.AutoReplyConfig.Retrieval.RerankModel != "qwen3-rerank" {
|
|
t.Errorf("Expected rerank model to be corrected to 'qwen3-rerank', got %q", cfg.AutoReplyConfig.Retrieval.RerankModel)
|
|
}
|
|
}
|