diff options
| -rw-r--r-- | models.yaml | 2 | ||||
| -rw-r--r-- | src/client/vertexai.rs | 7 |
2 files changed, 3 insertions, 6 deletions
diff --git a/models.yaml b/models.yaml index a573d11..5edc9b3 100644 --- a/models.yaml +++ b/models.yaml @@ -108,6 +108,7 @@ input_price: 0 output_price: 0 supports_function_calling: true + no_system_message: true - name: text-embedding-004 type: embedding input_price: 0 @@ -415,6 +416,7 @@ input_price: 0.125 output_price: 0.375 supports_function_calling: true + no_system_message: true - name: claude-3-5-sonnet-v2@20241022 max_input_tokens: 200000 max_output_tokens: 8192 diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 3ce73e0..7ba205e 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -309,12 +309,7 @@ pub fn gemini_build_chat_completions_body( stream: _, } = data; - let system_message = if model.name().starts_with("gemini-1.5-") { - extract_system_message(&mut messages) - } else { - patch_system_message(&mut messages); - None - }; + let system_message = extract_system_message(&mut messages); let mut network_image_urls = vec![]; let contents: Vec<Value> = messages |