Skip to main content

vtcode_config/models/model_id/
provider.rs

1use crate::models::Provider;
2
3use super::ModelId;
4
5impl ModelId {
6    /// Get the provider for this model
7    pub fn provider(&self) -> Provider {
8        if self.openrouter_metadata().is_some() {
9            return Provider::OpenRouter;
10        }
11        match self {
12            ModelId::Gemini25FlashPreview
13            | ModelId::Gemini25Flash
14            | ModelId::Gemini25FlashLite
15            | ModelId::Gemini25Pro
16            | ModelId::Gemini3ProPreview => Provider::Gemini,
17            ModelId::GPT5
18            | ModelId::GPT5Codex
19            | ModelId::GPT5Mini
20            | ModelId::GPT5Nano
21            | ModelId::CodexMiniLatest
22            | ModelId::OpenAIGptOss20b
23            | ModelId::OpenAIGptOss120b => Provider::OpenAI,
24            ModelId::ClaudeOpus45
25            | ModelId::ClaudeOpus41
26            | ModelId::ClaudeSonnet45
27            | ModelId::ClaudeHaiku45
28            | ModelId::ClaudeSonnet4 => Provider::Anthropic,
29            ModelId::DeepSeekChat | ModelId::DeepSeekReasoner => Provider::DeepSeek,
30            ModelId::HuggingFaceDeepseekV32
31            | ModelId::HuggingFaceOpenAIGptOss20b
32            | ModelId::HuggingFaceOpenAIGptOss120b
33            | ModelId::HuggingFaceGlm47
34            | ModelId::HuggingFaceGlm47FlashNovita
35            | ModelId::HuggingFaceKimiK2Thinking
36            | ModelId::HuggingFaceKimiK25Novita
37            | ModelId::HuggingFaceMinimaxM21Novita
38            | ModelId::HuggingFaceDeepseekV32Novita
39            | ModelId::HuggingFaceXiaomiMimoV2FlashNovita => Provider::HuggingFace,
40            ModelId::XaiGrok4
41            | ModelId::XaiGrok4Mini
42            | ModelId::XaiGrok4Code
43            | ModelId::XaiGrok4CodeLatest
44            | ModelId::XaiGrok4Vision => Provider::XAI,
45            ModelId::ZaiGlm4Plus
46            | ModelId::ZaiGlm4PlusDeepThinking
47            | ModelId::ZaiGlm47
48            | ModelId::ZaiGlm47DeepThinking
49            | ModelId::ZaiGlm47Flash
50            | ModelId::ZaiGlm46
51            | ModelId::ZaiGlm46DeepThinking
52            | ModelId::ZaiGlm46V
53            | ModelId::ZaiGlm46VFlash
54            | ModelId::ZaiGlm46VFlashX
55            | ModelId::ZaiGlm45
56            | ModelId::ZaiGlm45DeepThinking
57            | ModelId::ZaiGlm45Air
58            | ModelId::ZaiGlm45X
59            | ModelId::ZaiGlm45Airx
60            | ModelId::ZaiGlm45Flash
61            | ModelId::ZaiGlm45V
62            | ModelId::ZaiGlm432b0414128k => Provider::ZAI,
63            ModelId::MoonshotKimiK25 => Provider::Moonshot,
64            ModelId::OllamaGptOss20b
65            | ModelId::OllamaGptOss20bCloud
66            | ModelId::OllamaGptOss120bCloud
67            | ModelId::OllamaQwen317b
68            | ModelId::OllamaDeepseekV32Cloud
69            | ModelId::OllamaQwen3Next80bCloud
70            | ModelId::OllamaMistralLarge3675bCloud
71            | ModelId::OllamaKimiK2ThinkingCloud
72            | ModelId::OllamaKimiK25Cloud
73            | ModelId::OllamaQwen3Coder480bCloud
74            | ModelId::OllamaGlm46Cloud
75            | ModelId::OllamaGemini3ProPreviewLatestCloud
76            | ModelId::OllamaGemini3FlashPreviewCloud
77            | ModelId::OllamaDevstral2123bCloud
78            | ModelId::OllamaMinimaxM2Cloud
79            | ModelId::OllamaMinimaxM21Cloud
80            | ModelId::OllamaNemotron3Nano30bCloud
81            | ModelId::OllamaGlm47Cloud => Provider::Ollama,
82            ModelId::LmStudioMetaLlama38BInstruct
83            | ModelId::LmStudioMetaLlama318BInstruct
84            | ModelId::LmStudioQwen257BInstruct
85            | ModelId::LmStudioGemma22BIt
86            | ModelId::LmStudioGemma29BIt
87            | ModelId::LmStudioPhi31Mini4kInstruct => Provider::LmStudio,
88            ModelId::MinimaxM21 | ModelId::MinimaxM21Lightning | ModelId::MinimaxM2 => {
89                Provider::Minimax
90            }
91            _ => unreachable!(),
92        }
93    }
94
95    /// Whether this model supports configurable reasoning effort levels
96    pub fn supports_reasoning_effort(&self) -> bool {
97        self.provider().supports_reasoning_effort(self.as_str())
98    }
99}