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::ClaudeOpus46,
38 ModelId::ClaudeSonnet46,
39 ModelId::ClaudeHaiku45,
40 ModelId::DeepSeekChat,
42 ModelId::DeepSeekReasoner,
43 ModelId::ZaiGlm5,
45 ModelId::MoonshotKimiK25,
47 ModelId::OllamaGptOss20b,
49 ModelId::OllamaGptOss20bCloud,
50 ModelId::OllamaGptOss120bCloud,
51 ModelId::OllamaQwen317b,
52 ModelId::OllamaQwen3CoderNext,
53 ModelId::OllamaDeepseekV32Cloud,
54 ModelId::OllamaQwen3Next80bCloud,
55 ModelId::OllamaGlm5Cloud,
56 ModelId::OllamaGemini3FlashPreviewCloud,
57 ModelId::OllamaMinimaxM2Cloud,
58 ModelId::OllamaMinimaxM25Cloud,
59 ModelId::OllamaNemotron3SuperCloud,
60 ModelId::MinimaxM27,
62 ModelId::MinimaxM25,
63 ModelId::HuggingFaceDeepseekV32,
65 ModelId::HuggingFaceOpenAIGptOss20b,
66 ModelId::HuggingFaceOpenAIGptOss120b,
67 ModelId::HuggingFaceMinimaxM25Novita,
68 ModelId::HuggingFaceDeepseekV32Novita,
69 ModelId::HuggingFaceXiaomiMimoV2FlashNovita,
70 ModelId::HuggingFaceGlm5Novita,
71 ModelId::HuggingFaceQwen3CoderNextNovita,
72 ModelId::HuggingFaceQwen35397BA17BTogether,
73 ModelId::HuggingFaceStep35Flash,
74 ModelId::OpenRouterNvidiaNemotron3Super120bA12bFree,
75 ModelId::OpenRouterMinimaxM25,
76 ModelId::OpenRouterQwen3CoderNext,
77 ];
78 models.extend(Self::openrouter_models());
79 let mut seen = HashSet::new();
80 models.retain(|model| seen.insert(*model));
81 models
82 }
83
84 pub fn models_for_provider(provider: Provider) -> Vec<ModelId> {
86 Self::all_models()
87 .into_iter()
88 .filter(|model| model.provider() == provider)
89 .collect()
90 }
91}