From 4f8d895154c602fb51a7694911efd96d52129f82 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 24 Apr 2024 07:56:24 +0800
Subject: refactor: handling of system message (#432)

---
 src/client/claude.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'src/client/claude.rs')

diff --git a/src/client/claude.rs b/src/client/claude.rs
index 24bffbf..5c2a912 100644
--- a/src/client/claude.rs
+++ b/src/client/claude.rs
@@ -1,5 +1,5 @@
 use super::{
-    patch_system_message, ClaudeClient, Client, ExtraConfig, ImageUrl, MessageContent,
+    extract_sytem_message, ClaudeClient, Client, ExtraConfig, ImageUrl, MessageContent,
     MessageContentPart, Model, ModelConfig, PromptType, ReplyHandler, SendData,
 };
 
@@ -141,7 +141,7 @@ fn build_body(data: SendData, model: &Model) -> Result<Value> {
         stream,
     } = data;
 
-    patch_system_message(&mut messages);
+    let system_message = extract_sytem_message(&mut messages);
 
     let mut network_image_urls = vec![];
     let messages: Vec<Value> = messages
@@ -196,6 +196,10 @@ fn build_body(data: SendData, model: &Model) -> Result<Value> {
         "messages": messages,
     });
 
+    if let Some(system) = system_message {
+        body["system"] = system.into();
+    }
+
     if let Some(v) = temperature {
         body["temperature"] = v.into();
     }
-- 
cgit v1.2.3