Skip to content

Commit 7f579d3

Browse files
committed
Fix NVIDIA provider model name
1 parent d8c3a0c commit 7f579d3

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

nvidia.go

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,11 @@ func (p *nvidiaProvider) Name() string {
2727
}
2828

2929
func (p *nvidiaProvider) Complete(ctx context.Context, messages []Message) (string, error) {
30-
url := p.cfg.BaseURL + "/chat/completions"
31-
if p.cfg.BaseURL == "" {
30+
url := p.cfg.BaseURL
31+
if url == "" {
3232
url = "https://integrate.api.nvidia.com/v1/chat/completions"
33+
} else {
34+
url = url + "/chat/completions"
3335
}
3436

3537
reqBody := openaiRequest{

provider.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ func NewProvider(providerName string, apiKey ...string) (Provider, error) {
104104
}
105105
return NewNvidia(OpenAICompatConfig{
106106
BaseURL: getEnvOr("NVIDIA_BASE_URL", "https://integrate.api.nvidia.com/v1"),
107-
Model: getEnvOr("ITERATE_MODEL", "nvidia/llama-3.3-nemotron-70b-instruct"),
107+
Model: getEnvOr("ITERATE_MODEL", "meta/llama-3.3-70b-instruct"),
108108
APIKey: key,
109109
}), nil
110110

0 commit comments

Comments
 (0)