From 8f1d8dec5fceda13574ca7e6fbc4093d6ee10f8e Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 14 Nov 2023 10:34:42 +0800 Subject: refactor: palm client system message (#245) --- src/client/palm.rs | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) (limited to 'src/client') diff --git a/src/client/palm.rs b/src/client/palm.rs index dd063da..a2aec8a 100644 --- a/src/client/palm.rs +++ b/src/client/palm.rs @@ -113,19 +113,16 @@ fn build_body(data: SendData, _model: String) -> Value { .. } = data; - let mut context = None; if messages[0].role.is_system() { - let message = messages.remove(0); - context = 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 messages: Vec = messages.into_iter().map(|v| json!({ "content": v.content })).collect(); - let mut prompt = json!({ "messages": messages }); - - if let Some(context) = context { - prompt["context"] = context.into(); - }; + let prompt = json!({ "messages": messages }); let mut body = json!({ "prompt": prompt, -- cgit v1.2.3