revert: feat: use systemInstruction for gemini-1.5-* models (#564) (#567)

pull/568/head
sigoden 4 months ago committed by GitHub
parent 4b27378443
commit ce8d6d750d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -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();
}

Loading…
Cancel
Save