mirror of
https://github.com/coder/coder.git
synced 2026-06-02 20:48:20 +00:00
Danny Kopping
282ab7de34
Replace the env-based `BuildProviders` with a DB-backed loader. The database is now the single source of truth for runtime provider configuration; env config arrives via `SeedAIProvidersFromEnv` (run at boot) and `BuildProviders` reads it back as `aibridge.Provider` instances. `cli/server.go` and `enterprise/cli/server.go` both call the same path, so aibridged and aibridgeproxyd see the same provider set. Per-provider `DumpDir` is replaced by a top-level `CODER_AI_GATEWAY_DUMP_DIR` base; each provider's effective dump path is `<base>/<provider name>`.
73 lines
2.1 KiB
Go
73 lines
2.1 KiB
Go
//go:build !slim
|
|
|
|
package cli
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/coder/coder/v2/aibridge"
|
|
)
|
|
|
|
func TestDomainsFromProviders(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("ExtractsHostnames", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
providers := []aibridge.Provider{
|
|
aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{Name: "openai", BaseURL: "https://api.openai.com/v1/"}),
|
|
aibridge.NewAnthropicProvider(aibridge.AnthropicConfig{Name: "anthropic", BaseURL: "https://api.anthropic.com/"}, nil),
|
|
aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{Name: "custom", BaseURL: "https://custom-llm.example.com:8443/api"}),
|
|
}
|
|
|
|
domains, mapping := domainsFromProviders(providers)
|
|
|
|
assert.Contains(t, domains, "api.openai.com")
|
|
assert.Contains(t, domains, "api.anthropic.com")
|
|
assert.Contains(t, domains, "custom-llm.example.com")
|
|
|
|
assert.Equal(t, "openai", mapping("api.openai.com"))
|
|
assert.Equal(t, "anthropic", mapping("api.anthropic.com"))
|
|
assert.Equal(t, "custom", mapping("custom-llm.example.com"))
|
|
assert.Empty(t, mapping("unknown.com"))
|
|
})
|
|
|
|
t.Run("DeduplicatesSameHost", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
providers := []aibridge.Provider{
|
|
aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{Name: "first", BaseURL: "https://api.example.com/v1"}),
|
|
aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{Name: "second", BaseURL: "https://api.example.com/v2"}),
|
|
}
|
|
|
|
domains, mapping := domainsFromProviders(providers)
|
|
|
|
// Count occurrences of api.example.com.
|
|
count := 0
|
|
for _, d := range domains {
|
|
if d == "api.example.com" {
|
|
count++
|
|
}
|
|
}
|
|
assert.Equal(t, 1, count)
|
|
// First provider wins.
|
|
assert.Equal(t, "first", mapping("api.example.com"))
|
|
})
|
|
|
|
t.Run("CaseInsensitive", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
providers := []aibridge.Provider{
|
|
aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{Name: "provider", BaseURL: "https://API.Example.COM/v1"}),
|
|
}
|
|
|
|
domains, mapping := domainsFromProviders(providers)
|
|
|
|
assert.Contains(t, domains, "api.example.com")
|
|
assert.Equal(t, "provider", mapping("API.Example.COM"))
|
|
assert.Equal(t, "provider", mapping("api.example.com"))
|
|
})
|
|
}
|