feat: update test

src/dto/api.rs

@@ -1,8 +1,6 @@
 use serde::{Deserialize, Serialize};
 use utoipa::ToSchema;
 
-use crate::errors::OllamaError;
-
 #[derive(Debug, Serialize, Deserialize, ToSchema)]
 pub struct ModelsResponse {
     pub models: Vec<ModelInfo>,
@@ -28,37 +26,13 @@ pub struct LoadModelBody {
     pub keep_alive: Option<String>,
 }
 
-impl LoadModelBody {
-    pub fn parse_keep_alive(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())),
-        }
-    }
-}
-
 #[derive(Debug, Serialize, Deserialize, ToSchema)]
 pub struct UnloadModelResponse {
     pub model: String,
     pub status: String,
 }
 
-#[derive(Debug, Deserialize, Serialize, ToSchema)]
+#[derive(Debug, Deserialize, Serialize, ToSchema, Default)]
 pub struct BaseLLMRequest {
     pub model: String,
 
@@ -89,13 +63,13 @@ pub struct CompletionRequest {
     pub prompt: String,
 }
 
-#[derive(Debug, Serialize, Deserialize, ToSchema)]
+#[derive(Debug, Serialize, Deserialize, ToSchema, PartialEq, Eq)]
 #[serde(rename_all = "snake_case")]
 pub enum CompletionObject {
     TextCompletion,
 }
 
-#[derive(Debug, Serialize, Deserialize, ToSchema)]
+#[derive(Debug, Serialize, Deserialize, ToSchema, PartialEq, Eq)]
 #[serde(rename_all = "snake_case")]
 pub enum FinishReason {
     Stop,
@@ -149,7 +123,7 @@ pub struct Message {
     pub content: String,
 }
 
-#[derive(Debug, Deserialize, Serialize, ToSchema)]
+#[derive(Debug, Deserialize, Serialize, ToSchema, PartialEq, Eq)]
 #[serde(rename_all = "lowercase")]
 pub enum Role {
     System,