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