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::Gemini35Flash,
20 ModelId::Gemini3FlashPreview,
21 ModelId::GPT55,
23 ModelId::GPT54,
24 ModelId::GPT54Pro,
25 ModelId::GPT54Nano,
26 ModelId::GPT54Mini,
27 ModelId::GPT53Codex,
28 ModelId::GPT52Codex,
29 ModelId::GPT51Codex,
30 ModelId::GPT51CodexMax,
31 ModelId::GPT5Codex,
32 ModelId::GPT52,
33 ModelId::GPT5,
34 ModelId::GPT5Mini,
35 ModelId::GPT5Nano,
36 ModelId::OpenAIGptOss20b,
37 ModelId::OpenAIGptOss120b,
38 ModelId::ClaudeOpus47,
40 ModelId::ClaudeOpus46,
41 ModelId::ClaudeSonnet46,
42 ModelId::ClaudeHaiku45,
43 ModelId::ClaudeMythosPreview,
44 ModelId::CopilotAuto,
45 ModelId::CopilotGPT52Codex,
46 ModelId::CopilotGPT51CodexMax,
47 ModelId::CopilotGPT54,
48 ModelId::CopilotGPT54Mini,
49 ModelId::CopilotClaudeSonnet46,
50 ModelId::DeepSeekV4Pro,
52 ModelId::DeepSeekV4Flash,
53 ModelId::ZaiGlm5,
55 ModelId::ZaiGlm51,
56 ModelId::MoonshotKimiK26,
58 ModelId::MoonshotKimiK25,
59 ModelId::OpenCodeZenGPT54,
61 ModelId::OpenCodeZenGPT54Mini,
62 ModelId::OpenCodeZenClaudeSonnet46,
63 ModelId::OpenCodeZenGlm51,
64 ModelId::OpenCodeZenKimiK25,
65 ModelId::OpenCodeGoGlm51,
67 ModelId::OpenCodeGoKimiK25,
68 ModelId::OpenCodeGoMinimaxM25,
69 ModelId::OpenCodeGoMinimaxM27,
70 ModelId::OllamaGptOss20b,
72 ModelId::OllamaGptOss20bCloud,
73 ModelId::OllamaGptOss120bCloud,
74 ModelId::OllamaQwen317b,
75 ModelId::OllamaQwen3CoderNext,
76 ModelId::OllamaDeepseekV32Cloud,
77 ModelId::OllamaDeepseekV4FlashCloud,
78 ModelId::OllamaQwen3Next80bCloud,
79 ModelId::OllamaDeepseekV4ProCloud,
80 ModelId::OllamaGlm5Cloud,
81 ModelId::OllamaGlm51Cloud,
82 ModelId::OllamaGemini3FlashPreviewCloud,
83 ModelId::OllamaMinimaxM2Cloud,
84 ModelId::OllamaMinimaxM27Cloud,
85 ModelId::OllamaMinimaxM25Cloud,
86 ModelId::OllamaKimiK26Cloud,
87 ModelId::OllamaNemotron3SuperCloud,
88 ModelId::MinimaxM27,
90 ModelId::MinimaxM25,
91 ModelId::HuggingFaceDeepseekV32,
93 ModelId::HuggingFaceOpenAIGptOss20b,
94 ModelId::HuggingFaceOpenAIGptOss120b,
95 ModelId::HuggingFaceMinimaxM25Novita,
96 ModelId::HuggingFaceDeepseekV32Novita,
97 ModelId::HuggingFaceXiaomiMimoV2FlashNovita,
98 ModelId::HuggingFaceGlm5Novita,
99 ModelId::HuggingFaceGlm51ZaiOrg,
100 ModelId::HuggingFaceQwen3CoderNextNovita,
101 ModelId::HuggingFaceQwen35397BA17BTogether,
102 ModelId::HuggingFaceKimiK26Novita,
103 ModelId::HuggingFaceStep35Flash,
104 ModelId::OpenRouterNvidiaNemotron3Super120bA12bFree,
105 ModelId::OpenRouterMinimaxM25,
106 ModelId::OpenRouterQwen3CoderNext,
107 ModelId::OpenRouterZaiGlm47,
108 ModelId::OpenRouterMoonshotaiKimiK26,
109 ];
110 models.extend(Self::openrouter_models());
111 let mut seen = HashSet::new();
112 models.retain(|model| seen.insert(*model));
113 models
114 }
115
116 pub fn models_for_provider(provider: Provider) -> Vec<ModelId> {
118 Self::all_models()
119 .into_iter()
120 .filter(|model| model.provider() == provider)
121 .collect()
122 }
123}