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::OllamaQwen3Coder480bCloud
73            | ModelId::OllamaGlm46Cloud
74            | ModelId::OllamaGemini3ProPreviewLatestCloud
75            | ModelId::OllamaGemini3FlashPreviewCloud
76            | ModelId::OllamaDevstral2123bCloud
77            | ModelId::OllamaMinimaxM2Cloud
78            | ModelId::OllamaMinimaxM21Cloud
79            | ModelId::OllamaNemotron3Nano30bCloud
80            | ModelId::OllamaGlm47Cloud => Provider::Ollama,
81            ModelId::LmStudioMetaLlama38BInstruct
82            | ModelId::LmStudioMetaLlama318BInstruct
83            | ModelId::LmStudioQwen257BInstruct
84            | ModelId::LmStudioGemma22BIt
85            | ModelId::LmStudioGemma29BIt
86            | ModelId::LmStudioPhi31Mini4kInstruct => Provider::LmStudio,
87            ModelId::MinimaxM21 | ModelId::MinimaxM21Lightning | ModelId::MinimaxM2 => {
88                Provider::Minimax
89            }
90            _ => unreachable!(),
91        }
92    }
93
94    /// Whether this model supports configurable reasoning effort levels
95    pub fn supports_reasoning_effort(&self) -> bool {
96        self.provider().supports_reasoning_effort(self.as_str())
97    }
98}