diff --git a/crates/language_models/src/provider/ollama.rs b/crates/language_models/src/provider/ollama.rs index ed5f10da23..2f39680acd 100644 --- a/crates/language_models/src/provider/ollama.rs +++ b/crates/language_models/src/provider/ollama.rs @@ -19,7 +19,7 @@ use serde::{Deserialize, Serialize}; use settings::{Settings, SettingsStore}; use std::pin::Pin; use std::sync::atomic::{AtomicU64, Ordering}; -use std::{collections::BTreeMap, sync::Arc}; +use std::{collections::HashMap, sync::Arc}; use ui::{ButtonLike, Indicator, List, prelude::*}; use util::ResultExt; @@ -201,7 +201,7 @@ impl LanguageModelProvider for OllamaLanguageModelProvider { } fn provided_models(&self, cx: &App) -> Vec> { - let mut models: BTreeMap = BTreeMap::default(); + let mut models: HashMap = HashMap::new(); // Add models from the Ollama API for model in self.state.read(cx).available_models.iter() { @@ -228,7 +228,7 @@ impl LanguageModelProvider for OllamaLanguageModelProvider { ); } - models + let mut models = models .into_values() .map(|model| { Arc::new(OllamaLanguageModel { @@ -238,7 +238,9 @@ impl LanguageModelProvider for OllamaLanguageModelProvider { request_limiter: RateLimiter::new(4), }) as Arc }) - .collect() + .collect::>(); + models.sort_by_key(|model| model.name()); + models } fn load_model(&self, model: Arc, cx: &App) {