diff --git a/crates/openfang-cli/src/tui/screens/init_wizard.rs b/crates/openfang-cli/src/tui/screens/init_wizard.rs index 5a076c23..279b1e6c 100644 --- a/crates/openfang-cli/src/tui/screens/init_wizard.rs +++ b/crates/openfang-cli/src/tui/screens/init_wizard.rs @@ -188,6 +188,14 @@ const PROVIDERS: &[ProviderInfo] = &[ needs_key: true, hint: "", }, + ProviderInfo { + name: "nvidia", + display: "NVIDIA NIM", + env_var: "NVIDIA_API_KEY", + default_model: "nvidia/llama-3.1-nemotron-70b-instruct", + needs_key: true, + hint: "", + }, ProviderInfo { name: "claude-code", display: "Claude Code", diff --git a/crates/openfang-cli/src/tui/screens/wizard.rs b/crates/openfang-cli/src/tui/screens/wizard.rs index 338e4d85..f15b8f8c 100644 --- a/crates/openfang-cli/src/tui/screens/wizard.rs +++ b/crates/openfang-cli/src/tui/screens/wizard.rs @@ -127,6 +127,12 @@ const PROVIDERS: &[ProviderInfo] = &[ default_model: "codegeex-4", needs_key: true, }, + ProviderInfo { + name: "nvidia", + env_var: "NVIDIA_API_KEY", + default_model: "nvidia/llama-3.1-nemotron-70b-instruct", + needs_key: true, + }, ProviderInfo { name: "claude-code", env_var: "",