From 69974365ee786af5b643c335e2a12dd7817b84be Mon Sep 17 00:00:00 2001 From: sigoden Date: Mon, 10 Feb 2025 14:57:08 +0800 Subject: feat: new model field `system_prompt_prefix` (#1163) --- src/config/input.rs | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'src/config') diff --git a/src/config/input.rs b/src/config/input.rs index c16e1e2..1067b2c 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -1,8 +1,8 @@ use super::*; use crate::client::{ - init_client, patch_system_message, ChatCompletionsData, Client, ImageUrl, Message, - MessageContent, MessageContentPart, MessageContentToolCalls, MessageRole, Model, + init_client, patch_messages, ChatCompletionsData, Client, ImageUrl, Message, MessageContent, + MessageContentPart, MessageContentToolCalls, MessageRole, Model, }; use crate::function::ToolResult; use crate::utils::{base64_encode, is_loader_protocol, sha256, AbortSignal}; @@ -238,9 +238,7 @@ impl Input { stream: bool, ) -> Result { let mut messages = self.build_messages()?; - if model.no_system_message() { - patch_system_message(&mut messages); - } + patch_messages(&mut messages, model); model.guard_max_input_tokens(&messages)?; let temperature = self.role().temperature(); let top_p = self.role().top_p(); -- cgit v1.2.3