Default LiteLLM base for explicit provider selection
This commit is contained in:
parent
e7d6d532f1
commit
69f581f54b
2 changed files with 12 additions and 0 deletions
|
|
@ -107,6 +107,9 @@ func resolveProviderSelection(cfg *config.Config) (providerSelection, error) {
|
||||||
sel.apiKey = cfg.Providers.LiteLLM.APIKey
|
sel.apiKey = cfg.Providers.LiteLLM.APIKey
|
||||||
sel.apiBase = cfg.Providers.LiteLLM.APIBase
|
sel.apiBase = cfg.Providers.LiteLLM.APIBase
|
||||||
sel.proxy = cfg.Providers.LiteLLM.Proxy
|
sel.proxy = cfg.Providers.LiteLLM.Proxy
|
||||||
|
if sel.apiBase == "" {
|
||||||
|
sel.apiBase = "http://localhost:4000/v1"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
case "zhipu", "glm":
|
case "zhipu", "glm":
|
||||||
if cfg.Providers.Zhipu.APIKey != "" {
|
if cfg.Providers.Zhipu.APIKey != "" {
|
||||||
|
|
|
||||||
|
|
@ -29,6 +29,15 @@ func TestResolveProviderSelection(t *testing.T) {
|
||||||
wantAPIBase: "http://localhost:4000/v1",
|
wantAPIBase: "http://localhost:4000/v1",
|
||||||
wantProxy: "http://127.0.0.1:7890",
|
wantProxy: "http://127.0.0.1:7890",
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
name: "explicit litellm provider defaults base when only key is configured",
|
||||||
|
setup: func(cfg *config.Config) {
|
||||||
|
cfg.Agents.Defaults.Provider = "litellm"
|
||||||
|
cfg.Providers.LiteLLM.APIKey = "litellm-key"
|
||||||
|
},
|
||||||
|
wantType: providerTypeHTTPCompat,
|
||||||
|
wantAPIBase: "http://localhost:4000/v1",
|
||||||
|
},
|
||||||
{
|
{
|
||||||
name: "explicit claude-cli provider routes to cli provider type",
|
name: "explicit claude-cli provider routes to cli provider type",
|
||||||
setup: func(cfg *config.Config) {
|
setup: func(cfg *config.Config) {
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue