diff --git a/pkg/providers/http_provider.go b/pkg/providers/http_provider.go index 17eb6214c..78fff13b5 100644 --- a/pkg/providers/http_provider.go +++ b/pkg/providers/http_provider.go @@ -322,7 +322,15 @@ func CreateProvider(cfg *config.Config) (LLMProvider, error) { apiBase = "localhost:4321" } 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" + } + } } }