diff --git a/tests/gemini_oauth_regression.rs b/tests/gemini_oauth_regression.rs index c6a6cbb1..b7551554 100644 --- a/tests/gemini_oauth_regression.rs +++ b/tests/gemini_oauth_regression.rs @@ -5,10 +5,18 @@ fn test_regression_gemini_oauth_fields() { // This test ensures that the CompletionResponse and ToolCompletionResponse // include the newly added caching fields, which was a critical compilation fix. // Since we are using the public API, if it compiles and runs, the fields are present. - + // Test model metadata logic (which we updated) - assert!(!ironclaw::llm::gemini_oauth::GeminiOauthProvider::model_uses_cloud_code_api("gemini-1.5-pro")); - assert!(ironclaw::llm::gemini_oauth::GeminiOauthProvider::model_uses_cloud_code_api("gemini-2.0-flash")); + assert!( + !ironclaw::llm::gemini_oauth::GeminiOauthProvider::model_uses_cloud_code_api( + "gemini-1.5-pro" + ) + ); + assert!( + ironclaw::llm::gemini_oauth::GeminiOauthProvider::model_uses_cloud_code_api( + "gemini-2.0-flash" + ) + ); } #[tokio::test]