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::GPT5,
22 ModelId::GPT52,
23 ModelId::GPT5Mini,
24 ModelId::GPT5Nano,
25 ModelId::GPT53Codex,
26 ModelId::OpenAIGptOss20b,
27 ModelId::OpenAIGptOss120b,
28 ModelId::ClaudeOpus46,
30 ModelId::ClaudeSonnet46,
31 ModelId::ClaudeHaiku45,
32 ModelId::DeepSeekChat,
34 ModelId::DeepSeekReasoner,
35 ModelId::ZaiGlm5,
37 ModelId::MoonshotKimiK25,
39 ModelId::OllamaGptOss20b,
41 ModelId::OllamaGptOss20bCloud,
42 ModelId::OllamaGptOss120bCloud,
43 ModelId::OllamaQwen317b,
44 ModelId::OllamaQwen3CoderNext,
45 ModelId::OllamaDeepseekV32Cloud,
46 ModelId::OllamaQwen3Next80bCloud,
47 ModelId::OllamaGlm5Cloud,
48 ModelId::OllamaGemini3FlashPreviewCloud,
49 ModelId::OllamaMinimaxM2Cloud,
50 ModelId::OllamaMinimaxM25Cloud,
51 ModelId::MinimaxM25,
53 ModelId::HuggingFaceDeepseekV32,
55 ModelId::HuggingFaceOpenAIGptOss20b,
56 ModelId::HuggingFaceOpenAIGptOss120b,
57 ModelId::HuggingFaceMinimaxM25Novita,
58 ModelId::HuggingFaceDeepseekV32Novita,
59 ModelId::HuggingFaceXiaomiMimoV2FlashNovita,
60 ModelId::HuggingFaceGlm5Novita,
61 ModelId::HuggingFaceQwen3CoderNextNovita,
62 ModelId::HuggingFaceQwen35397BA17BTogether,
63 ModelId::HuggingFaceStep35Flash,
64 ModelId::OpenRouterMinimaxM25,
65 ModelId::OpenRouterQwen3CoderNext,
66 ];
67 models.extend(Self::openrouter_models());
68 let mut seen = HashSet::new();
69 models.retain(|model| seen.insert(*model));
70 models
71 }
72
73 pub fn models_for_provider(provider: Provider) -> Vec<ModelId> {
75 Self::all_models()
76 .into_iter()
77 .filter(|model| model.provider() == provider)
78 .collect()
79 }
80}