summaryrefslogtreecommitdiffstats
path: root/src/config
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2025-02-10 14:57:08 +0800
committerGitHub <noreply@github.com>2025-02-10 14:57:08 +0800
commit69974365ee786af5b643c335e2a12dd7817b84be (patch)
treecdb43b4911907fc918b0719379ffdad4bb880f8d /src/config
parentbaf0c1ed07fc185fd237fbdf51aecaa8ed926ba6 (diff)
downloadaichat-69974365ee786af5b643c335e2a12dd7817b84be.tar.gz
feat: new model field `system_prompt_prefix` (#1163)
Diffstat (limited to 'src/config')
-rw-r--r--src/config/input.rs8
1 files changed, 3 insertions, 5 deletions
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<ChatCompletionsData> {
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();