diff options
| author | sigoden <sigoden@gmail.com> | 2024-06-05 21:16:06 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-05 21:16:06 +0800 |
| commit | ce8d6d750d2f9522a0206d74e996fb45c5e521b6 (patch) | |
| tree | fe85359336e8d2d0e759165b0f6a25d8c7a2e782 /src | |
| parent | 4b27378443bc74c31cca26435b99a7550a40d2f7 (diff) | |
| download | aichat-ce8d6d750d2f9522a0206d74e996fb45c5e521b6.tar.gz | |
revert: feat: use systemInstruction for gemini-1.5-* models (#564) (#567)
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/vertexai.rs | 11 |
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(); } |
