diff --git a/crates/language_models/src/provider/ollama.rs b/crates/language_models/src/provider/ollama.rs index df9f8f383e0a28a15bd9dec03f0297a3146e7f74..ea3a6f65035200b48d58f752123abbce02b7d472 100644 --- a/crates/language_models/src/provider/ollama.rs +++ b/crates/language_models/src/provider/ollama.rs @@ -269,13 +269,15 @@ impl LanguageModelProvider for OllamaLanguageModelProvider { let mut models: HashMap = HashMap::new(); let settings = OllamaLanguageModelProvider::settings(cx); - // Add models from the Ollama API - for model in self.state.read(cx).fetched_models.iter() { - let mut model = model.clone(); - if let Some(context_window) = settings.context_window { - model.max_tokens = context_window; + if settings.auto_discover { + // Add models from the Ollama API + for model in self.state.read(cx).fetched_models.iter() { + let mut model = model.clone(); + if let Some(context_window) = settings.context_window { + model.max_tokens = context_window; + } + models.insert(model.name.clone(), model); } - models.insert(model.name.clone(), model); } // Override with available models from settings