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::Gemini31ProPreview
13            | ModelId::Gemini31ProPreviewCustomTools
14            | ModelId::Gemini31FlashLitePreview
15            | ModelId::Gemini35Flash
16            | ModelId::Gemini3FlashPreview => Provider::Gemini,
17            ModelId::GPT55
18            | ModelId::GPT5
19            | ModelId::GPT52
20            | ModelId::GPT52Codex
21            | ModelId::GPT54
22            | ModelId::GPT54Pro
23            | ModelId::GPT54Nano
24            | ModelId::GPT54Mini
25            | ModelId::GPT5Mini
26            | ModelId::GPT5Nano
27            | ModelId::GPT53Codex
28            | ModelId::GPT51Codex
29            | ModelId::GPT51CodexMax
30            | ModelId::GPT5Codex
31            | ModelId::OpenAIGptOss20b
32            | ModelId::OpenAIGptOss120b => Provider::OpenAI,
33            ModelId::ClaudeOpus47
34            | ModelId::ClaudeOpus46
35            | ModelId::ClaudeSonnet46
36            | ModelId::ClaudeHaiku45
37            | ModelId::ClaudeMythosPreview => Provider::Anthropic,
38            ModelId::CopilotAuto
39            | ModelId::CopilotGPT52Codex
40            | ModelId::CopilotGPT51CodexMax
41            | ModelId::CopilotGPT54
42            | ModelId::CopilotGPT54Mini
43            | ModelId::CopilotClaudeSonnet46 => Provider::Copilot,
44            ModelId::DeepSeekV4Pro | ModelId::DeepSeekV4Flash => Provider::DeepSeek,
45            ModelId::MistralLarge3 => Provider::Mistral,
46            ModelId::MiMoV25Pro
47            | ModelId::MiMoV25
48            | ModelId::MiMoV2Pro
49            | ModelId::MiMoV2Omni
50            | ModelId::MiMoV2Flash => Provider::MiMo,
51            ModelId::HuggingFaceDeepseekV32
52            | ModelId::HuggingFaceOpenAIGptOss20b
53            | ModelId::HuggingFaceOpenAIGptOss120b
54            | ModelId::HuggingFaceMinimaxM25Novita
55            | ModelId::HuggingFaceDeepseekV32Novita
56            | ModelId::HuggingFaceXiaomiMimoV2FlashNovita
57            | ModelId::HuggingFaceGlm5Novita
58            | ModelId::HuggingFaceGlm51ZaiOrg
59            | ModelId::HuggingFaceQwen3CoderNextNovita
60            | ModelId::HuggingFaceQwen35397BA17BTogether
61            | ModelId::HuggingFaceKimiK26Novita
62            | ModelId::HuggingFaceDeepseekV4FlashNovita
63            | ModelId::HuggingFaceDeepseekV4ProTogether
64            | ModelId::HuggingFaceStep35Flash => Provider::HuggingFace,
65            ModelId::ZaiGlm5 | ModelId::ZaiGlm51 => Provider::ZAI,
66            ModelId::MoonshotKimiK26 | ModelId::MoonshotKimiK25 => Provider::Moonshot,
67            ModelId::OpenCodeZenGPT54
68            | ModelId::OpenCodeZenGPT54Mini
69            | ModelId::OpenCodeZenClaudeSonnet46
70            | ModelId::OpenCodeZenGlm51
71            | ModelId::OpenCodeZenKimiK25 => Provider::OpenCodeZen,
72            ModelId::OpenCodeGoGlm51
73            | ModelId::OpenCodeGoKimiK25
74            | ModelId::OpenCodeGoMinimaxM25
75            | ModelId::OpenCodeGoMinimaxM27 => Provider::OpenCodeGo,
76            ModelId::OllamaGptOss20b
77            | ModelId::OllamaGptOss20bCloud
78            | ModelId::OllamaGptOss120bCloud
79            | ModelId::OllamaQwen317b
80            | ModelId::OllamaQwen3CoderNext
81            | ModelId::OllamaDeepseekV32Cloud
82            | ModelId::OllamaDeepseekV4FlashCloud
83            | ModelId::OllamaDeepseekV4ProCloud
84            | ModelId::OllamaQwen3Next80bCloud
85            | ModelId::OllamaGemini3FlashPreviewCloud
86            | ModelId::OllamaMinimaxM2Cloud
87            | ModelId::OllamaMinimaxM27Cloud
88            | ModelId::OllamaMinimaxM25Cloud
89            | ModelId::OllamaNemotron3SuperCloud
90            | ModelId::OllamaKimiK26Cloud
91            | ModelId::OllamaGlm5Cloud
92            | ModelId::OllamaGlm51Cloud => Provider::Ollama,
93            ModelId::MinimaxM27 | ModelId::MinimaxM25 => Provider::Minimax,
94            ModelId::OpenRouterMinimaxM25 | ModelId::OpenRouterQwen3CoderNext => {
95                Provider::OpenRouter
96            }
97            // OpenRouter models - explicitly handled even if openrouter_metadata() returns Some
98            ModelId::OpenRouterQwen332b
99            | ModelId::OpenRouterQwen330bA3b
100            | ModelId::OpenRouterQwen330bA3bInstruct2507
101            | ModelId::OpenRouterQwen330bA3bThinking2507
102            | ModelId::OpenRouterQwen314b
103            | ModelId::OpenRouterQwen38b
104            | ModelId::OpenRouterQwen3Next80bA3bInstruct
105            | ModelId::OpenRouterQwen3Next80bA3bThinking
106            | ModelId::OpenRouterQwen35Plus0215
107            | ModelId::OpenRouterQwen3Coder
108            | ModelId::OpenRouterQwen3CoderPlus
109            | ModelId::OpenRouterQwen3CoderFlash
110            | ModelId::OpenRouterQwen3Coder30bA3bInstruct
111            | ModelId::OpenRouterDeepSeekV4Pro
112            | ModelId::OpenRouterDeepSeekV4Flash
113            | ModelId::OpenRouterDeepSeekR1
114            | ModelId::OpenRouterDeepSeekChatV31
115            | ModelId::OpenRouterOpenAIGptOss120b
116            | ModelId::OpenRouterOpenAIGptOss120bFree
117            | ModelId::OpenRouterOpenAIGptOss20b
118            | ModelId::OpenRouterOpenAIGpt5
119            | ModelId::OpenRouterOpenAIGpt55
120            | ModelId::OpenRouterOpenAIGpt5Chat
121            | ModelId::OpenRouterGoogleGemini31ProPreview
122            | ModelId::OpenRouterAnthropicClaudeSonnet45
123            | ModelId::OpenRouterAnthropicClaudeSonnet46
124            | ModelId::OpenRouterAnthropicClaudeHaiku45
125            | ModelId::OpenRouterMistralaiMistralLarge2512
126            | ModelId::OpenRouterNexAgiDeepseekV31NexN1
127            | ModelId::OpenRouterStepfunStep35FlashFree
128            | ModelId::OpenRouterNvidiaNemotron3Super120bA12bFree
129            | ModelId::OpenRouterZaiGlm5
130            | ModelId::OpenRouterZaiGlm47
131            | ModelId::OpenRouterZaiGlm51
132            | ModelId::OpenRouterMoonshotaiKimiK26 => Provider::OpenRouter,
133        }
134    }
135
136    /// Whether this model supports configurable reasoning effort levels
137    pub fn supports_reasoning_effort(&self) -> bool {
138        self.provider().supports_reasoning_effort(self.as_str())
139    }
140}