From ce8d6d750d2f9522a0206d74e996fb45c5e521b6 Mon Sep 17 00:00:00 2001 From: sigoden Date: Wed, 5 Jun 2024 21:16:06 +0800 Subject: revert: feat: use systemInstruction for gemini-1.5-* models (#564) (#567) --- src/client/vertexai.rs | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) (limited to 'src/client') 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 = 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(); } -- cgit v1.2.3