From e118e864d5183710dc9892d1b20853a2f59f7713 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 14 Nov 2023 10:03:01 +0800
Subject: refactor: ernie client system message (#244)

---
 src/client/ernie.rs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

(limited to 'src/client')

diff --git a/src/client/ernie.rs b/src/client/ernie.rs
index 084933c..200433c 100644
--- a/src/client/ernie.rs
+++ b/src/client/ernie.rs
@@ -196,18 +196,17 @@ fn build_body(data: SendData, _model: String) -> Value {
         stream,
     } = data;
 
-    let mut system = None;
     if messages[0].role.is_system() {
-        let message = messages.remove(0);
-        system = Some(message.content);
+        let system_message = messages.remove(0);
+        if let Some(message) = messages.get_mut(0) {
+            message.content = format!("{}\n\n{}", system_message.content, message.content)
+        }
     }
 
     let mut body = json!({
         "messages": messages,
     });
-    if let Some(system) = system {
-        body["system"] = system.into();
-    };
+
     if let Some(temperature) = temperature {
         body["temperature"] = (temperature / 2.0).into();
     }
-- 
cgit v1.2.3