summaryrefslogtreecommitdiffstats
path: root/src/client/claude.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/claude.rs')
-rw-r--r--src/client/claude.rs8
1 files changed, 6 insertions, 2 deletions
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();
}