1 files changed, 5 insertions, 3 deletions
diff --git a/src/client/palm.rs b/src/client/palm.rs
index a2aec8a..37ed0ae 100644
--- a/src/client/palm.rs
+++ b/src/client/palm.rs
@@ -1,4 +1,4 @@
-use super::{PaLMClient, Client, ExtraConfig, Model, PromptType, SendData, TokensCountFactors, send_message_as_streaming};
+use super::{PaLMClient, Client, ExtraConfig, Model, PromptType, SendData, TokensCountFactors, send_message_as_streaming, MessageContent};
 
 use crate::{config::GlobalConfig, render::ReplyHandler, utils::PromptKind};
 
@@ -115,8 +115,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))
+            }
         }
     }