mirror of
https://github.com/outbackdingo/optimclaw.git
synced 2026-08-26 15:40:18 +00:00
Artem
9be29b2c22
- Fix parse_custom_headers to preserve commas in values by splitting
only on commas followed by a header-name:colon pattern (manual scan
instead of simple split(','))
- Use matches! macro for backend exclusion check in app.rs
- Merge SSE metadata extraction into single pass (was iterating twice)
- Replace fragile substring-based context_length with explicit match
on known Gemini model IDs via gemini_context_length()
- Add missing models to regression test (8 models, not 5)
100 lines
3.1 KiB
Rust
100 lines
3.1 KiB
Rust
use ironclaw::llm::ChatMessage;
|
|
use ironclaw::llm::gemini_oauth::GeminiOauthProvider;
|
|
|
|
/// Regression: Cloud Code API routing for Gemini 2.0+ models.
|
|
/// Gemini 1.x → legacy generativelanguage.googleapis.com
|
|
/// Gemini 2.0+ → Cloud Code API (cloudcode-pa.googleapis.com)
|
|
#[test]
|
|
fn test_regression_cloud_code_api_routing() {
|
|
// Legacy models (1.x) → false
|
|
assert!(!GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-1.5-pro"
|
|
));
|
|
assert!(!GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-1.5-flash"
|
|
));
|
|
|
|
// 2.0+ models → true
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-2.0-flash"
|
|
));
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-2.5-pro"
|
|
));
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-2.5-flash"
|
|
));
|
|
|
|
// Preview models with hyphen → true
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-3.1-pro-preview"
|
|
));
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-3-flash-preview"
|
|
));
|
|
|
|
// Gemini 3 family → true
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"gemini-3-pro"
|
|
));
|
|
}
|
|
|
|
/// Regression: "preview" false-positive fix.
|
|
/// `model.contains("-preview")` (with hyphen) prevents models whose name
|
|
/// happens to include "preview" without a hyphen prefix from being
|
|
/// mis-routed to Cloud Code API.
|
|
#[test]
|
|
fn test_regression_preview_false_positive_fix() {
|
|
// "my-preview-custom" still matches (contains "-preview")
|
|
assert!(GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"my-preview-custom"
|
|
));
|
|
|
|
// "mypreviewcustom" does NOT match (no hyphen before "preview")
|
|
assert!(!GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"mypreviewcustom"
|
|
));
|
|
|
|
// Non-Gemini models without "-preview" → false
|
|
assert!(!GeminiOauthProvider::model_uses_cloud_code_api(
|
|
"not-a-gemini-model"
|
|
));
|
|
}
|
|
|
|
/// Regression: model list consistency.
|
|
/// Wizard, list_models(), and LLM_PROVIDERS.md all return the same 8 models.
|
|
#[test]
|
|
fn test_regression_standardized_model_list() {
|
|
let expected_models = [
|
|
"gemini-3.1-pro-preview",
|
|
"gemini-3.1-pro-preview-customtools",
|
|
"gemini-3-pro-preview",
|
|
"gemini-3-flash-preview",
|
|
"gemini-3.1-flash-lite-preview",
|
|
"gemini-2.5-pro",
|
|
"gemini-2.5-flash",
|
|
"gemini-2.5-flash-lite",
|
|
];
|
|
|
|
// All standardized models must route to Cloud Code API (all are >= 2.0)
|
|
for model in &expected_models {
|
|
assert!(
|
|
GeminiOauthProvider::model_uses_cloud_code_api(model),
|
|
"Standardized model '{}' should route to Cloud Code API",
|
|
model
|
|
);
|
|
}
|
|
}
|
|
|
|
/// Regression: ChatMessage helper constructors.
|
|
#[test]
|
|
fn test_regression_chat_message_helpers() {
|
|
let user_msg = ChatMessage::user("hello");
|
|
assert_eq!(user_msg.role, ironclaw::llm::Role::User);
|
|
assert_eq!(user_msg.content, "hello");
|
|
|
|
let system_msg = ChatMessage::system("you are helpful");
|
|
assert_eq!(system_msg.role, ironclaw::llm::Role::System);
|
|
assert_eq!(system_msg.content, "you are helpful");
|
|
}
|