From 6286251d320a7f512e49534c30b42452d30d93cc Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 19 Dec 2023 20:37:53 +0800 Subject: feat: support gemini (#273) --- src/client/palm.rs | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) (limited to 'src/client/palm.rs') diff --git a/src/client/palm.rs b/src/client/palm.rs index 37ed0ae..a519768 100644 --- a/src/client/palm.rs +++ b/src/client/palm.rs @@ -1,4 +1,4 @@ -use super::{PaLMClient, Client, ExtraConfig, Model, PromptType, SendData, TokensCountFactors, send_message_as_streaming, MessageContent}; +use super::{PaLMClient, Client, ExtraConfig, Model, PromptType, SendData, TokensCountFactors, send_message_as_streaming, patch_system_message}; use crate::{config::GlobalConfig, render::ReplyHandler, utils::PromptKind}; @@ -113,14 +113,7 @@ fn build_body(data: SendData, _model: String) -> Value { .. } = data; - if messages[0].role.is_system() { - let system_message = messages.remove(0); - if let (Some(message), MessageContent::Text(system_text)) = (messages.get_mut(0), system_message.content) { - if let MessageContent::Text(text) = message.content.clone() { - message.content = MessageContent::Text(format!("{}\n\n{}", system_text, text)) - } - } - } + patch_system_message(&mut messages); let messages: Vec = messages.into_iter().map(|v| json!({ "content": v.content })).collect(); -- cgit v1.2.3