summaryrefslogtreecommitdiffstats
path: root/src/client/ollama.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-04-24 07:56:24 +0800
committerGitHub <noreply@github.com>2024-04-24 07:56:24 +0800
commit4f8d895154c602fb51a7694911efd96d52129f82 (patch)
tree3c8decb301c0b77937a152af2d4f996d0960839c /src/client/ollama.rs
parent0a4c0413ef0154cde2e3485ec0415e6069596a23 (diff)
downloadaichat-4f8d895154c602fb51a7694911efd96d52129f82.tar.gz
refactor: handling of system message (#432)
Diffstat (limited to 'src/client/ollama.rs')
-rw-r--r--src/client/ollama.rs8
1 files changed, 3 insertions, 5 deletions
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()