feat: update test

src/providers/ollama/client.rs

@@ -36,6 +36,10 @@ impl OllamaProvider {
         Ok(res.models.iter().any(|m| m.name == model))
     }
 
+    fn has_user_message(&self, messages: &[api::Message]) -> bool {
+        messages.iter().any(|m| matches!(m.role, api::Role::User))
+    }
+
     fn extract_completion_params<'a>(
         &self,
         body: &'a api::CompletionRequest,
@@ -56,6 +60,28 @@ impl OllamaProvider {
         Ok((&body.messages, model))
     }
 
+    pub fn parse_keep_alive(&self, s: &str) -> Result<(), OllamaError> {
+        let s = s.trim();
+
+        if s == "-1" || s.parse::<u64>().is_ok() {
+            return Ok(());
+        }
+
+        let split = s
+            .find(|c: char| c.is_alphabetic())
+            .ok_or_else(|| OllamaError::InvalidKeepAlive(s.to_string()))?;
+
+        let (num, unit) = s.split_at(split);
+
+        num.parse::<u64>()
+            .map_err(|_| OllamaError::InvalidKeepAlive(s.to_string()))?;
+
+        match unit {
+            "s" | "m" | "h" => Ok(()),
+            _ => Err(OllamaError::InvalidKeepAlive(s.to_string())),
+        }
+    }
+
     // // ── public endpoints ─────────────────────────────────────────────────────
 
     pub async fn list_models(&self) -> Result<api::ModelsResponse, OllamaError> {
@@ -77,10 +103,14 @@ impl OllamaProvider {
     pub async fn load_model(
         &self,
         model: &str,
-        keep_alive: &str,
+        keep_alive: Option<&str>,
     ) -> Result<api::LoadModelResponse, OllamaError> {
         let url = format!("{}/api/generate", self.base_url);
 
+        let keep_alive = keep_alive.ok_or(OllamaError::MissingKeepAlive)?;
+
+        self.parse_keep_alive(keep_alive)?;
+
         let exists = self.model_exists(model).await?;
         if !exists {
             return Err(OllamaError::ModelNotFound(model.to_string()));
@@ -99,7 +129,7 @@ impl OllamaProvider {
             .json(&payload)
             .send()
             .await?
-            .json::<ollama::OllamaGenerateResponse>()
+            .text()
             .await?;
 
         Ok(api::LoadModelResponse {
@@ -130,7 +160,7 @@ impl OllamaProvider {
             .json(&payload)
             .send()
             .await?
-            .json::<ollama::OllamaGenerateResponse>()
+            .text()
             .await?;
 
         Ok(api::UnloadModelResponse {
@@ -277,6 +307,10 @@ impl OllamaProvider {
             return Err(OllamaError::MissingMessages);
         }
 
+        if !self.has_user_message(&body.messages) {
+            return Err(OllamaError::MissingMessages);
+        }
+
         if model.is_empty() {
             return Err(OllamaError::MissingModel);
         }