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::Gemini3FlashPreview => Provider::Gemini,
16            ModelId::GPT
17            | ModelId::GPT5
18            | ModelId::GPT52
19            | ModelId::GPT54
20            | ModelId::GPT54Pro
21            | ModelId::GPT5Mini
22            | ModelId::GPT5Nano
23            | ModelId::GPT53Codex
24            | ModelId::OpenAIGptOss20b
25            | ModelId::OpenAIGptOss120b => Provider::OpenAI,
26            ModelId::ClaudeOpus46 | ModelId::ClaudeSonnet46 | ModelId::ClaudeHaiku45 => {
27                Provider::Anthropic
28            }
29            ModelId::DeepSeekChat | ModelId::DeepSeekReasoner => Provider::DeepSeek,
30            ModelId::HuggingFaceDeepseekV32
31            | ModelId::HuggingFaceOpenAIGptOss20b
32            | ModelId::HuggingFaceOpenAIGptOss120b
33            | ModelId::HuggingFaceMinimaxM25Novita
34            | ModelId::HuggingFaceDeepseekV32Novita
35            | ModelId::HuggingFaceXiaomiMimoV2FlashNovita
36            | ModelId::HuggingFaceGlm5Novita
37            | ModelId::HuggingFaceQwen3CoderNextNovita
38            | ModelId::HuggingFaceQwen35397BA17BTogether
39            | ModelId::HuggingFaceStep35Flash => Provider::HuggingFace,
40            ModelId::ZaiGlm5 => Provider::ZAI,
41            ModelId::MoonshotKimiK25 => Provider::Moonshot,
42            ModelId::OllamaGptOss20b
43            | ModelId::OllamaGptOss20bCloud
44            | ModelId::OllamaGptOss120bCloud
45            | ModelId::OllamaQwen317b
46            | ModelId::OllamaQwen3CoderNext
47            | ModelId::OllamaDeepseekV32Cloud
48            | ModelId::OllamaQwen3Next80bCloud
49            | ModelId::OllamaGemini3FlashPreviewCloud
50            | ModelId::OllamaMinimaxM2Cloud
51            | ModelId::OllamaMinimaxM25Cloud
52            | ModelId::OllamaNemotron3SuperCloud
53            | ModelId::OllamaGlm5Cloud => Provider::Ollama,
54            ModelId::MinimaxM25 => Provider::Minimax,
55            ModelId::OpenRouterMinimaxM25 | ModelId::OpenRouterQwen3CoderNext => {
56                Provider::OpenRouter
57            }
58            // OpenRouter models - explicitly handled even if openrouter_metadata() returns Some
59            ModelId::OpenRouterQwen332b
60            | ModelId::OpenRouterQwen330bA3b
61            | ModelId::OpenRouterQwen330bA3bInstruct2507
62            | ModelId::OpenRouterQwen330bA3bThinking2507
63            | ModelId::OpenRouterQwen314b
64            | ModelId::OpenRouterQwen38b
65            | ModelId::OpenRouterQwen3Next80bA3bInstruct
66            | ModelId::OpenRouterQwen3Next80bA3bThinking
67            | ModelId::OpenRouterQwen35Plus0215
68            | ModelId::OpenRouterQwen3Coder
69            | ModelId::OpenRouterQwen3CoderPlus
70            | ModelId::OpenRouterQwen3CoderFlash
71            | ModelId::OpenRouterQwen3Coder30bA3bInstruct
72            | ModelId::OpenRouterDeepseekChat
73            | ModelId::OpenRouterDeepSeekV32
74            | ModelId::OpenRouterDeepseekReasoner
75            | ModelId::OpenRouterDeepSeekV32Speciale
76            | ModelId::OpenRouterDeepSeekV32Exp
77            | ModelId::OpenRouterDeepSeekChatV31
78            | ModelId::OpenRouterDeepSeekR1
79            | ModelId::OpenRouterOpenAIGptOss120b
80            | ModelId::OpenRouterOpenAIGptOss120bFree
81            | ModelId::OpenRouterOpenAIGptOss20b
82            | ModelId::OpenRouterOpenAIGpt5
83            | ModelId::OpenRouterOpenAIGpt5Chat
84            | ModelId::OpenRouterGoogleGemini31ProPreview
85            | ModelId::OpenRouterAnthropicClaudeSonnet45
86            | ModelId::OpenRouterAnthropicClaudeSonnet46
87            | ModelId::OpenRouterAnthropicClaudeHaiku45
88            | ModelId::OpenRouterMistralaiMistralLarge2512
89            | ModelId::OpenRouterNexAgiDeepseekV31NexN1
90            | ModelId::OpenRouterStepfunStep35FlashFree
91            | ModelId::OpenRouterNvidiaNemotron3Super120bA12bFree
92            | ModelId::OpenRouterZaiGlm5 => Provider::OpenRouter,
93        }
94    }
95
96    /// Whether this model supports configurable reasoning effort levels
97    pub fn supports_reasoning_effort(&self) -> bool {
98        self.provider().supports_reasoning_effort(self.as_str())
99    }
100}