diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-24 07:56:24 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-24 07:56:24 +0800 |
| commit | 4f8d895154c602fb51a7694911efd96d52129f82 (patch) | |
| tree | 3c8decb301c0b77937a152af2d4f996d0960839c /src/client/ollama.rs | |
| parent | 0a4c0413ef0154cde2e3485ec0415e6069596a23 (diff) | |
| download | aichat-4f8d895154c602fb51a7694911efd96d52129f82.tar.gz | |
refactor: handling of system message (#432)
Diffstat (limited to 'src/client/ollama.rs')
| -rw-r--r-- | src/client/ollama.rs | 8 |
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() |
