你çnepiedg
12619aa4ab
- Implemented repairSmokeAIAdviceIndexes to manage the unique index for fa_smoke_ai_advice. - Added unit tests for the new function to ensure correct index recreation and validation. - Updated AutoMigrate to include the new index repair function.
29 lines
746 B
Go
29 lines
746 B
Go
package service
|
|
|
|
import "testing"
|
|
|
|
func TestPreferredSmokeAIModel(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
name string
|
|
input string
|
|
want string
|
|
}{
|
|
{name: "deepseek reasoner fallback", input: "deepseek-reasoner", want: "deepseek-chat"},
|
|
{name: "trim whitespace", input: " deepseek-reasoner ", want: "deepseek-chat"},
|
|
{name: "keep normal model", input: "gpt-4o-mini", want: "gpt-4o-mini"},
|
|
{name: "keep deepseek chat", input: "deepseek-chat", want: "deepseek-chat"},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
if got := preferredSmokeAIModel(tc.input); got != tc.want {
|
|
t.Fatalf("preferredSmokeAIModel(%q)=%q, want=%q", tc.input, got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|