mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/yunionio/cloudpods.git
synced 2026-09-20 08:03:53 +08:00
167682c39b
Add /v1/messages handler with Anthropic-to-OpenAI translation, upstream failover, and probe endpoints. Replace shell-based functional test scripts with pkg/aiproxy/ft and climc test commands; consolidate documentation.
35 lines
873 B
Go
35 lines
873 B
Go
package models
|
|
|
|
import (
|
|
"testing"
|
|
|
|
api "yunion.io/x/onecloud/pkg/apis/aiproxy"
|
|
)
|
|
|
|
func TestCatalogSeedModelsForPublicProviders(t *testing.T) {
|
|
publicKeys := []string{
|
|
api.ProviderKeyOpenAI,
|
|
api.ProviderKeyDeepseek,
|
|
api.ProviderKeyAnthropic,
|
|
api.ProviderKeyGemini,
|
|
}
|
|
for _, pk := range publicKeys {
|
|
if !api.HasDefaultPublicBaseURL(pk) {
|
|
t.Fatalf("%q should have default public base URL", pk)
|
|
}
|
|
entries := catalogSeedModelsForProvider(pk)
|
|
if len(entries) == 0 {
|
|
t.Fatalf("expected catalog models for public provider %q", pk)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCatalogSeedModelsSkippedForSelfHostedProviders(t *testing.T) {
|
|
selfHosted := []string{api.ProviderKeyOllama, api.ProviderKeyVLLM, api.ProviderKeySGLang}
|
|
for _, pk := range selfHosted {
|
|
if api.HasDefaultPublicBaseURL(pk) {
|
|
t.Fatalf("%q should not be treated as public SaaS provider", pk)
|
|
}
|
|
}
|
|
}
|