vtcode_config/models/model_id/
collection.rs1use crate::models::Provider;
2use hashbrown::HashSet;
3
4use super::ModelId;
5
6impl ModelId {
7 pub fn openrouter_vendor(&self) -> Option<&'static str> {
9 self.openrouter_metadata().map(|meta| meta.vendor)
10 }
11
12 pub fn all_models() -> Vec<ModelId> {
14 let mut models = vec![
15 ModelId::Gemini31ProPreview,
17 ModelId::Gemini31ProPreviewCustomTools,
18 ModelId::Gemini31FlashLitePreview,
19 ModelId::Gemini3FlashPreview,
20 ModelId::GPT55,
22 ModelId::GPT54,
23 ModelId::GPT54Pro,
24 ModelId::GPT54Nano,
25 ModelId::GPT54Mini,
26 ModelId::GPT53Codex,
27 ModelId::GPT52Codex,
28 ModelId::GPT51Codex,
29 ModelId::GPT51CodexMax,
30 ModelId::GPT5Codex,
31 ModelId::GPT52,
32 ModelId::GPT5,
33 ModelId::GPT5Mini,
34 ModelId::GPT5Nano,
35 ModelId::OpenAIGptOss20b,
36 ModelId::OpenAIGptOss120b,
37 ModelId::ClaudeOpus47,
39 ModelId::ClaudeOpus46,
40 ModelId::ClaudeSonnet46,
41 ModelId::ClaudeHaiku45,
42 ModelId::ClaudeMythosPreview,
43 ModelId::CopilotAuto,
44 ModelId::CopilotGPT52Codex,
45 ModelId::CopilotGPT51CodexMax,
46 ModelId::CopilotGPT54,
47 ModelId::CopilotGPT54Mini,
48 ModelId::CopilotClaudeSonnet46,
49 ModelId::DeepSeekV4Pro,
51 ModelId::DeepSeekV4Flash,
52 ModelId::ZaiGlm5,
54 ModelId::ZaiGlm51,
55 ModelId::MoonshotKimiK26,
57 ModelId::MoonshotKimiK25,
58 ModelId::OpenCodeZenGPT54,
60 ModelId::OpenCodeZenGPT54Mini,
61 ModelId::OpenCodeZenClaudeSonnet46,
62 ModelId::OpenCodeZenGlm51,
63 ModelId::OpenCodeZenKimiK25,
64 ModelId::OpenCodeGoGlm51,
66 ModelId::OpenCodeGoKimiK25,
67 ModelId::OpenCodeGoMinimaxM25,
68 ModelId::OpenCodeGoMinimaxM27,
69 ModelId::OllamaGptOss20b,
71 ModelId::OllamaGptOss20bCloud,
72 ModelId::OllamaGptOss120bCloud,
73 ModelId::OllamaQwen317b,
74 ModelId::OllamaQwen3CoderNext,
75 ModelId::OllamaDeepseekV32Cloud,
76 ModelId::OllamaDeepseekV4FlashCloud,
77 ModelId::OllamaQwen3Next80bCloud,
78 ModelId::OllamaDeepseekV4ProCloud,
79 ModelId::OllamaGlm5Cloud,
80 ModelId::OllamaGlm51Cloud,
81 ModelId::OllamaGemini3FlashPreviewCloud,
82 ModelId::OllamaMinimaxM2Cloud,
83 ModelId::OllamaMinimaxM27Cloud,
84 ModelId::OllamaMinimaxM25Cloud,
85 ModelId::OllamaKimiK26Cloud,
86 ModelId::OllamaNemotron3SuperCloud,
87 ModelId::MinimaxM27,
89 ModelId::MinimaxM25,
90 ModelId::HuggingFaceDeepseekV32,
92 ModelId::HuggingFaceOpenAIGptOss20b,
93 ModelId::HuggingFaceOpenAIGptOss120b,
94 ModelId::HuggingFaceMinimaxM25Novita,
95 ModelId::HuggingFaceDeepseekV32Novita,
96 ModelId::HuggingFaceXiaomiMimoV2FlashNovita,
97 ModelId::HuggingFaceGlm5Novita,
98 ModelId::HuggingFaceGlm51ZaiOrg,
99 ModelId::HuggingFaceQwen3CoderNextNovita,
100 ModelId::HuggingFaceQwen35397BA17BTogether,
101 ModelId::HuggingFaceKimiK26Novita,
102 ModelId::HuggingFaceStep35Flash,
103 ModelId::OpenRouterNvidiaNemotron3Super120bA12bFree,
104 ModelId::OpenRouterMinimaxM25,
105 ModelId::OpenRouterQwen3CoderNext,
106 ModelId::OpenRouterZaiGlm47,
107 ModelId::OpenRouterMoonshotaiKimiK26,
108 ];
109 models.extend(Self::openrouter_models());
110 let mut seen = HashSet::new();
111 models.retain(|model| seen.insert(*model));
112 models
113 }
114
115 pub fn models_for_provider(provider: Provider) -> Vec<ModelId> {
117 Self::all_models()
118 .into_iter()
119 .filter(|model| model.provider() == provider)
120 .collect()
121 }
122}