fix: add missing nvidia case in explicit provider switch
This commit is contained in:
parent
9a3f3611c3
commit
0e0ea3d05a
1 changed files with 9 additions and 1 deletions
|
|
@ -322,7 +322,15 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) {
|
||||||
apiBase = "localhost:4321"
|
apiBase = "localhost:4321"
|
||||||
}
|
}
|
||||||
return NewGitHubCopilotProvider(apiBase, cfg.Providers.GitHubCopilot.ConnectMode, model)
|
return NewGitHubCopilotProvider(apiBase, cfg.Providers.GitHubCopilot.ConnectMode, model)
|
||||||
|
case "nvidia":
|
||||||
|
if cfg.Providers.Nvidia.APIKey != "" {
|
||||||
|
apiKey = cfg.Providers.Nvidia.APIKey
|
||||||
|
apiBase = cfg.Providers.Nvidia.APIBase
|
||||||
|
proxy = cfg.Providers.Nvidia.Proxy
|
||||||
|
if apiBase == "" {
|
||||||
|
apiBase = "https://integrate.api.nvidia.com/v1"
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue