feat: strong typing for complete endpoints

src/dto/ollama.rs

@@ -1,8 +1,6 @@
 use serde::{Deserialize, Serialize};
 use utoipa::ToSchema;
 
-use crate::errors::OllamaError;
-
 #[derive(Debug, Serialize, Deserialize)]
 pub struct OllamaModels {
     pub models: Vec<OllamaModel>,
@@ -25,3 +23,39 @@ pub struct OllamaModelDetails {
     pub parameter_size: Option<String>,
     pub quantization_level: Option<String>,
 }
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct OllamaOptions {
+    pub temperature: Option<f32>,
+    pub top_p: Option<f32>,
+    pub top_k: Option<u32>,
+    pub repeat_penalty: Option<f32>,
+    pub seed: Option<i64>,
+
+    pub num_ctx: Option<u32>,
+    pub num_predict: Option<u32>,
+}
+
+#[derive(Debug, Serialize)]
+pub struct OllamaGenerateRequest<'a> {
+    pub model: &'a str,
+    pub prompt: &'a str,
+    pub stream: bool,
+    pub options: OllamaOptions,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct OllamaGenerateResponse {
+    pub model: String,
+    pub created_at: Option<String>,
+    pub response: String,
+    pub done: bool,
+
+    #[serde(default)]
+    pub context: Option<Vec<u64>>,
+
+    pub total_duration: Option<u64>,
+    pub load_duration: Option<u64>,
+    pub prompt_eval_count: Option<u32>,
+    pub eval_count: Option<u32>,
+}