diff options
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 11 |
1 files changed, 2 insertions, 9 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 75a362a..0844b39 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -1,4 +1,4 @@ -use super::{ErnieClient, Client, ExtraConfig, PromptType, SendData, Model, MessageContent}; +use super::{ErnieClient, Client, ExtraConfig, PromptType, SendData, Model, patch_system_message}; use crate::{ config::GlobalConfig, @@ -197,14 +197,7 @@ fn build_body(data: SendData, _model: String) -> Value { stream, } = data; - if messages[0].role.is_system() { - let system_message = messages.remove(0); - if let (Some(message), MessageContent::Text(system_text)) = (messages.get_mut(0), system_message.content) { - if let MessageContent::Text(text) = message.content.clone() { - message.content = MessageContent::Text(format!("{}\n\n{}", system_text, text)) - } - } - } + patch_system_message(&mut messages); let mut body = json!({ "messages": messages, |