Add Serialize Support for Further Stucts

src/generation/chat/mod.rs

@@ -154,7 +154,7 @@ impl Ollama {
     }
 }
 
-#[derive(Debug, Clone, Deserialize)]
+#[derive(Debug, Clone, Deserialize, Serialize)]
 pub struct ChatMessageResponse {
     /// The name of the model used for the completion.
     pub model: String,
@@ -168,7 +168,7 @@ pub struct ChatMessageResponse {
     pub final_data: Option<ChatMessageFinalResponseData>,
 }
 
-#[derive(Debug, Clone, Deserialize)]
+#[derive(Debug, Clone, Deserialize, Serialize)]
 pub struct ChatMessageFinalResponseData {
     /// Time spent generating the response
     pub total_duration: u64,

src/generation/chat/request.rs

@@ -1,11 +1,11 @@
-use serde::Serialize;
+use serde::{Deserialize, Serialize};
 
 use crate::generation::{options::GenerationOptions, parameters::FormatType};
 
 use super::ChatMessage;
 
 /// A chat message request to Ollama.
-#[derive(Debug, Clone, Serialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct ChatMessageRequest {
     #[serde(rename = "model")]
     pub model_name: String,

src/models.rs

@@ -6,18 +6,18 @@ pub mod pull;
 pub mod push;
 pub mod show_info;
 
-use serde::Deserialize;
+use serde::{Deserialize, Serialize};
 
 /// A local model pulled from Ollama.
-#[derive(Debug, Clone, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct LocalModel {
     pub name: String,
     pub modified_at: String,
     pub size: u64,
 }
 
 /// A model's info.
-#[derive(Debug, Clone, Deserialize)]
+#[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct ModelInfo {
     pub license: String,
     pub modelfile: String,