feat: auto-detect ollama models

This commit is contained in:
Dave
2025-12-25 12:21:58 +00:00
parent e229f2efa8
commit 6572f45422
5 changed files with 78 additions and 7 deletions

View File

@@ -18,6 +18,12 @@ pub struct ProviderConfig {
const MAX_TURNS: usize = 10;
#[tauri::command]
pub async fn get_ollama_models(base_url: Option<String>) -> Result<Vec<String>, String> {
let url = base_url.unwrap_or_else(|| "http://localhost:11434".to_string());
OllamaProvider::get_models(&url).await
}
#[tauri::command]
pub async fn chat(
messages: Vec<Message>,

View File

@@ -20,7 +20,8 @@ pub fn run() {
commands::fs::list_directory,
commands::search::search_files,
commands::shell::exec_shell,
commands::chat::chat
commands::chat::chat,
commands::chat::get_ollama_models
])
.run(tauri::generate_context!())
.expect("error while running tauri application");

View File

@@ -13,6 +13,40 @@ impl OllamaProvider {
pub fn new(base_url: String) -> Self {
Self { base_url }
}
pub async fn get_models(base_url: &str) -> Result<Vec<String>, String> {
let client = reqwest::Client::new();
let url = format!("{}/api/tags", base_url.trim_end_matches('/'));
let res = client
.get(&url)
.send()
.await
.map_err(|e| format!("Request failed: {}", e))?;
if !res.status().is_success() {
let status = res.status();
let text = res.text().await.unwrap_or_default();
return Err(format!("Ollama API error {}: {}", status, text));
}
let body: OllamaTagsResponse = res
.json()
.await
.map_err(|e| format!("Failed to parse response: {}", e))?;
Ok(body.models.into_iter().map(|m| m.name).collect())
}
}
#[derive(Deserialize)]
struct OllamaTagsResponse {
models: Vec<OllamaModelTag>,
}
#[derive(Deserialize)]
struct OllamaModelTag {
name: String,
}
// --- Request Types ---