add config to own_assist_common

2026-07-28 17:48:31 +02:00
parent b776cafab7
commit c5b594e1da
5 changed files with 108 additions and 57 deletions
@@ -1550,7 +1550,7 @@ dependencies = [
 "hyper-util",
 "icalendar",
 "libdav",
- "own_assist_common 0.3.0 (sparse+https://git.mboemer.de/api/packages/milan/cargo/)",
+ "own_assist_common 0.3.0",
 "rmcp",
 "serde",
 "thiserror 2.0.19",
@@ -1576,7 +1576,7 @@ dependencies = [
 "hyper-util",
 "icalendar",
 "libdav",
- "own_assist_common 0.3.0 (sparse+https://git.mboemer.de/api/packages/milan/cargo/)",
+ "own_assist_common 0.3.0",
 "rmcp",
 "serde",
 "thiserror 2.0.19",
@@ -1998,7 +1998,7 @@ dependencies = [
 "log",
 "mcp_server_collection 0.2.2 (sparse+https://git.mboemer.de/api/packages/milan/cargo/)",
 "ollama-rs",
- "own_assist_common 0.3.0 (sparse+https://git.mboemer.de/api/packages/milan/cargo/)",
+ "own_assist_common 0.3.0",
 "own_mcp 0.1.1 (sparse+https://git.mboemer.de/api/packages/milan/cargo/)",
 "rmcp",
 "rodio",
@@ -2012,6 +2012,8 @@ dependencies = [
 [[package]]
 name = "own_assist_common"
 version = "0.3.0"
+source = "sparse+https://git.mboemer.de/api/packages/milan/cargo/"
+checksum = "e4f114c598d4f9b1c26def3a047fbba893a202bb2a91be6f21d80881320ad872"
 dependencies = [
 "log",
 "serde",
@@ -2022,15 +2024,15 @@ dependencies = [

 [[package]]
 name = "own_assist_common"
-version = "0.3.0"
-source = "sparse+https://git.mboemer.de/api/packages/milan/cargo/"
-checksum = "e4f114c598d4f9b1c26def3a047fbba893a202bb2a91be6f21d80881320ad872"
+version = "0.4.0"
 dependencies = [
 "log",
+ "ollama-rs",
 "serde",
 "thiserror 2.0.19",
 "toml",
 "tracing-subscriber",
+ "url",
 ]

 [[package]]
@@ -19,7 +19,7 @@ pub struct Config {
    ollama: OllamaConfig,
    audio: Option<AudioConfig>,
    #[serde(rename = "mcp-servers")]
-    mcp_servers: Vec<MCPServerConfig>,
+    mcp_servers: Vec<MCPClientConfig>,
 }

 impl Config {
@@ -31,23 +31,6 @@ impl Config {
        &self.ollama
    }

-    pub fn ollama_instance(&self) -> Ollama {
-        let mut ollama_headers = HeaderMap::new();
-
-        if let Some(authorization_header) = &self.ollama.authorization {
-            ollama_headers.append(
-                "Authorization",
-                HeaderValue::from_str(authorization_header).unwrap(),
-            );
-        }
-
-        let ollama_default_url = Url::parse("http://127.0.0.1:11434").unwrap();
-        let mut ollama = Ollama::from_url(self.ollama.url.clone().unwrap_or(ollama_default_url));
-        ollama.set_headers(Some(ollama_headers));
-
-        ollama
-    }
-
    pub fn audio_client(&self) -> Option<AudioClient> {
        if let Some(audio_config) = &self.audio {
            let audio_server = &audio_config.server;
@@ -118,36 +101,6 @@ pub struct AudioConfig {
    server: AudioServerConfig,
 }

-#[derive(Debug, Deserialize)]
-pub struct AudioServerConfig {
-    url: Url,
-    authorization: Option<String>,
-}
-
-#[derive(Debug, Deserialize)]
-pub struct OllamaConfig {
-    pub url: Option<Url>,
-    pub model: OllamaModelConfig,
-    pub authorization: Option<String>,
-}
-
-#[derive(Debug, Deserialize)]
-pub struct OllamaModelConfig {
-    #[serde(rename = "from-model")]
-    pub from_model: String,
-    pub name: String,
-    pub system_prompt: Option<String>,
-    pub context_size: Option<u64>,
-    pub temperature: Option<f32>,
-}
-
-#[derive(Debug, Deserialize)]
-pub struct MCPServerConfig {
-    pub name: Option<String>,
-    pub url: Url,
-    pub authorization: Option<String>, // should probably implement oauth some time // actually, fuck oauth
-}
-
 #[derive(Debug, Deserialize)]
 pub struct PermissionConfig {
    #[serde(rename = "tool-name")]
@@ -1,12 +1,14 @@
 [package]
 name = "own_assist_common"
-version = "0.3.0"
+version = "0.4.0"
 edition = "2024"
 license = "GPL-3.0"

 [dependencies]
 thiserror = "2.0.18"
 toml = "1.1.2"
-serde = "1.0.228"
+serde = { version = "1.0.228", features = ["derive"] }
 log = "0.4.29"
 tracing-subscriber = { version = "0.3.23", features = ["env-filter"]}
+url = { version = "2.5.8", features = ["serde"] }
+ollama-rs = { version = "0.3.6", features = ["headers"]}
@@ -0,0 +1,93 @@
+use ollama_rs::error::OllamaError;
+use ollama_rs::models::ModelOptions;
+use ollama_rs::models::create::{CreateModelRequest, CreateModelStatus};
+use serde::Deserialize;
+use url::Url;
+
+#[derive(Debug, Deserialize)]
+pub struct AudioClientConfig {
+    pub url: Url,
+    pub authorization: Option<String>,
+}
+
+fn ollama_default_url() -> Url {
+    Url::parse("http://127.0.0.1:11434").unwrap()
+}
+
+#[derive(Debug, Deserialize)]
+pub struct OllamaConfig {
+    #[serde(default = "ollama_default_url")]
+    pub url: Url,
+    #[serde(rename = "model")]
+    pub model_config: OllamaModelConfig,
+    pub authorization: Option<String>,
+}
+
+impl OllamaConfig {
+    pub fn ollama(&self) -> ollama_rs::Ollama {
+        let mut ollama_headers = ollama_rs::headers::HeaderMap::new();
+
+        if let Some(authorization_header) = &self.authorization {
+            ollama_headers.append(
+                "Authorization",
+                ollama_rs::headers::HeaderValue::from_str(authorization_header.as_str()).unwrap(),
+            );
+        }
+
+        let ollama_default_url = Url::parse("http://127.0.0.1:11434").unwrap();
+        let mut ollama = ollama_rs::Ollama::from_url(self.url.clone());
+        ollama.set_headers(Some(ollama_headers));
+
+        ollama
+    }
+
+    pub async fn create_model(
+        &self,
+        download_needed_message: Option<String>,
+    ) -> Result<CreateModelStatus, OllamaError> {
+        let ollama = self.ollama();
+        let model_config = &self.model_config;
+
+        let model_options = ModelOptions::default()
+            .num_ctx(model_config.context_size.unwrap_or(2048)) // 2048 is the ollama default.
+            .temperature(model_config.temperature.unwrap_or(0.8)); // 0.8 is the ollama default
+
+        // print message to warn user of long waiting times
+        if let Ok(models) = ollama.list_local_models().await
+            && !models.iter().any(|m| m.name == model_config.from_model)
+            && let Some(download_needed_message) = download_needed_message
+        {
+            log::info!("{}", download_needed_message);
+            println!("{}", download_needed_message);
+        }
+
+        let mut create_model_request = CreateModelRequest::new(model_config.name.clone())
+            .from_model(model_config.from_model.clone())
+            .parameters(model_options);
+
+        if let Some(system_prompt) = &model_config.system_prompt {
+            create_model_request = create_model_request.system(system_prompt.clone());
+        }
+
+        ollama
+            .create_model(create_model_request)
+            .await
+    }
+}
+
+#[derive(Debug, Deserialize)]
+pub struct OllamaModelConfig {
+    #[serde(rename = "from-model")]
+    pub from_model: String,
+    pub name: String,
+    pub system_prompt: Option<String>,
+    pub context_size: Option<u64>,
+    pub temperature: Option<f32>,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct MCPClientConfig {
+    pub name: Option<String>,
+    pub url: Url,
+    pub authorization: Option<String>, // should probably implement oauth some time // actually, fuck oauth
+}
@@ -1,3 +1,4 @@
+pub mod config;
 pub mod config_loader;
 mod exit_error;
 mod tracing_init;