summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-06-05 21:16:06 +0800
committerGitHub <noreply@github.com>2024-06-05 21:16:06 +0800
commitce8d6d750d2f9522a0206d74e996fb45c5e521b6 (patch)
treefe85359336e8d2d0e759165b0f6a25d8c7a2e782 /src
parent4b27378443bc74c31cca26435b99a7550a40d2f7 (diff)
downloadaichat-ce8d6d750d2f9522a0206d74e996fb45c5e521b6.tar.gz
revert: feat: use systemInstruction for gemini-1.5-* models (#564) (#567)
Diffstat (limited to 'src')
-rw-r--r--src/client/vertexai.rs11
1 files changed, 1 insertions, 10 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index 5211438..dc75c9f 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -262,12 +262,7 @@ pub fn gemini_build_chat_completions_body(
stream: _,
} = data;
- let mut system_message = None;
- if model.name().starts_with("gemini-1.5") {
- system_message = extract_system_message(&mut messages);
- } else {
- patch_system_message(&mut messages);
- };
+ patch_system_message(&mut messages);
let mut network_image_urls = vec![];
let contents: Vec<Value> = messages
@@ -338,10 +333,6 @@ pub fn gemini_build_chat_completions_body(
let mut body = json!({ "contents": contents, "generationConfig": {} });
- if let Some(v) = system_message {
- body["systemInstruction"] = json!({ "parts": [{ "text": v }] });
- }
-
if let Some(v) = model.max_tokens_param() {
body["generationConfig"]["maxOutputTokens"] = v.into();
}