From 4f8d895154c602fb51a7694911efd96d52129f82 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 24 Apr 2024 07:56:24 +0800
Subject: refactor: handling of system message (#432)

---
 src/client/ollama.rs | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'src/client/ollama.rs')

diff --git a/src/client/ollama.rs b/src/client/ollama.rs
index de658c1..403712f 100644
--- a/src/client/ollama.rs
+++ b/src/client/ollama.rs
@@ -1,6 +1,6 @@
 use super::{
-    message::*, patch_system_message, Client, ExtraConfig, Model, ModelConfig, OllamaClient,
-    PromptType, ReplyHandler, SendData,
+    message::*, Client, ExtraConfig, Model, ModelConfig, OllamaClient, PromptType, ReplyHandler,
+    SendData,
 };
 
 use crate::utils::PromptKind;
@@ -121,13 +121,11 @@ async fn send_message_streaming(builder: RequestBuilder, handler: &mut ReplyHand
 
 fn build_body(data: SendData, model: &Model) -> Result<Value> {
     let SendData {
-        mut messages,
+        messages,
         temperature,
         stream,
     } = data;
 
-    patch_system_message(&mut messages);
-
     let mut network_image_urls = vec![];
     let messages: Vec<Value> = messages
         .into_iter()
-- 
cgit v1.2.3