refactor: handling of system message (#432)

1 files changed, 8 insertions, 0 deletions

diff --git a/src/client/common.rs b/src/client/common.rs
index b6ccce7..62c5f92 100644
--- a/src/client/common.rs
+++ b/src/client/common.rs
@@ -414,6 +414,14 @@ pub fn patch_system_message(messages: &mut Vec<Message>) {
     }
 }
 
+pub fn extract_sytem_message(messages: &mut Vec<Message>) -> Option<String> {
+    if messages[0].role.is_system() {
+        let system_message = messages.remove(0);
+        return Some(system_message.content.to_text());
+    }
+    None
+}
+
 pub async fn json_stream<S, F>(mut stream: S, mut handle: F) -> Result<()>
 where
     S: Stream<Item = Result<bytes::Bytes, reqwest::Error>> + Unpin,