Skip to main content

vtcode_config/constants/models/
nvidia.rs

1//! Curated NVIDIA NIM models exposed by VT Code.
2
3pub const NEMOTRON_3_ULTRA_550B_A55B: &str = "nvidia/nemotron-3-ultra-550b-a55b";
4pub const NEMOTRON_3_SUPER_120B_A12B: &str = "nvidia/nemotron-3-super-120b-a12b";
5pub const NEMOTRON_3_NANO_30B_A3B: &str = "nvidia/nemotron-3-nano-30b-a3b";
6
7pub const DEFAULT_MODEL: &str = NEMOTRON_3_ULTRA_550B_A55B;
8
9/// Curated agent-oriented NVIDIA models. Explicit model IDs outside this list
10/// remain valid because NVIDIA's catalog is larger than VT Code's picker.
11pub const SUPPORTED_MODELS: &[&str] = &[
12    NEMOTRON_3_ULTRA_550B_A55B,
13    NEMOTRON_3_SUPER_120B_A12B,
14    NEMOTRON_3_NANO_30B_A3B,
15];
16
17pub const REASONING_MODELS: &[&str] = SUPPORTED_MODELS;