Skip to content

Commit 6c3796a

Browse files
committed
feat: add nvidia nim provider preset
1 parent 955a99a commit 6c3796a

1 file changed

Lines changed: 18 additions & 0 deletions

File tree

crates/rexos-kernel/src/config.rs

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,15 @@ impl Default for RexosConfig {
195195
default_model: "MiniMax-M2.5".to_string(),
196196
},
197197
);
198+
providers.insert(
199+
"nvidia".to_string(),
200+
ProviderConfig {
201+
kind: ProviderKind::OpenAiCompatible,
202+
base_url: "https://integrate.api.nvidia.com/v1".to_string(),
203+
api_key_env: "NVIDIA_API_KEY".to_string(),
204+
default_model: "meta/llama-3.2-3b-instruct".to_string(),
205+
},
206+
);
198207
providers.insert(
199208
"minimax_anthropic".to_string(),
200209
ProviderConfig {
@@ -348,6 +357,7 @@ mod tests {
348357
assert!(toml_str.contains("[providers.glm_native]"));
349358
assert!(toml_str.contains("[providers.minimax]"));
350359
assert!(toml_str.contains("[providers.minimax_native]"));
360+
assert!(toml_str.contains("[providers.nvidia]"));
351361
assert!(toml_str.contains("[providers.anthropic]"));
352362
assert!(toml_str.contains("[providers.gemini]"));
353363
assert!(toml_str.contains("kind = \"openai_compatible\""));
@@ -374,4 +384,12 @@ mod tests {
374384
let minimax_native = cfg.providers.get("minimax_native").unwrap();
375385
assert_eq!(minimax_native.base_url, "https://api.minimax.chat/v1");
376386
}
387+
388+
#[test]
389+
fn nvidia_preset_uses_nim_base_url() {
390+
let cfg = RexosConfig::default();
391+
392+
let nvidia = cfg.providers.get("nvidia").unwrap();
393+
assert_eq!(nvidia.base_url, "https://integrate.api.nvidia.com/v1");
394+
}
377395
}

0 commit comments

Comments
 (0)