1 files changed, 5 insertions, 3 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs
index 200433c..4bb3435 100644
--- a/src/client/ernie.rs
+++ b/src/client/ernie.rs
@@ -1,4 +1,4 @@
-use super::{ErnieClient, Client, ExtraConfig, PromptType, SendData, Model};
+use super::{ErnieClient, Client, ExtraConfig, PromptType, SendData, Model, MessageContent};
 
 use crate::{
     config::GlobalConfig,
@@ -198,8 +198,10 @@ fn build_body(data: SendData, _model: String) -> Value {
 
     if messages[0].role.is_system() {
         let system_message = messages.remove(0);
-        if let Some(message) = messages.get_mut(0) {
-            message.content = format!("{}\n\n{}", system_message.content, message.content)
+        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))
+            }
         }
     }