diff --git a/internal/llm/providers.go b/internal/llm/providers.go index abd6a65..1e5f8b9 100644 --- a/internal/llm/providers.go +++ b/internal/llm/providers.go @@ -173,6 +173,21 @@ var registry = []Provider{ "MiniMax-M2.5-highspeed", }, }, + { + Name: "baidu-qianfan", + DisplayName: "Baidu Qianfan API", + Protocol: "openai", + BaseURL: "https://qianfan.baidubce.com/v2", + EnvVar: "QIANFAN_API_KEY", + Models: []string{ + "ernie-5.1", + "ernie-5.0", + "ernie-x1.1", + "ernie-x1-turbo-32k-preview", + "deepseek-v4-pro", + "deepseek-v4-flash", + }, + }, } var registryMap map[string]Provider diff --git a/internal/llm/providers_test.go b/internal/llm/providers_test.go index 09962e3..c4f5e07 100644 --- a/internal/llm/providers_test.go +++ b/internal/llm/providers_test.go @@ -40,7 +40,7 @@ func TestListProviders_Order(t *testing.T) { if len(providers) < 3 { t.Fatalf("expected at least 3 providers, got %d", len(providers)) } - expected := []string{"anthropic", "dashscope", "dashscope-tokenplan", "deepseek", "hy-tokenplan", "kimi", "mimo", "minimax", "openai", "tencent-tokenhub", "volcengine", "z-ai"} + expected := []string{"anthropic", "baidu-qianfan", "dashscope", "dashscope-tokenplan", "deepseek", "hy-tokenplan", "kimi", "mimo", "minimax", "openai", "tencent-tokenhub", "volcengine", "z-ai"} if len(providers) != len(expected) { t.Fatalf("expected %d providers, got %d", len(expected), len(providers)) }