From 0ff54ceec356952b06c5be610df6ddac247ae1b9 Mon Sep 17 00:00:00 2001
From: Bradley Axen <baxen@squareup.com>
Date: Sun, 23 Nov 2025 16:51:00 -0800
Subject: [PATCH] [fix] generic check for gemini compat (#5842)

---
 crates/goose/src/providers/formats/databricks.rs | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/crates/goose/src/providers/formats/databricks.rs b/crates/goose/src/providers/formats/databricks.rs
index 54668e9dfe..cbc047e6ee 100644
--- a/crates/goose/src/providers/formats/databricks.rs
+++ b/crates/goose/src/providers/formats/databricks.rs
@@ -263,7 +263,7 @@ pub fn format_tools(tools: &[Tool], model_name: &str) -> anyhow::Result<Vec<Valu
     let mut tool_names = std::collections::HashSet::new();
     let mut result = Vec::new();
 
-    let is_gemini = model_name.starts_with("gemini");
+    let is_gemini = model_name.contains("gemini");
 
     for tool in tools {
         if !tool_names.insert(&tool.name) {
@@ -697,7 +697,11 @@ mod tests {
             "http://json-schema.org/draft-07/schema#"
         );
 
-        let spec = format_tools(&[tool], "gemini-2-5-flash")?;
+        let spec = format_tools(&[tool.clone()], "gemini-2-5-flash")?;
+        assert!(spec[0]["function"]["parameters"].get("$schema").is_none());
+        assert_eq!(spec[0]["function"]["parameters"]["type"], "object");
+
+        let spec = format_tools(&[tool], "databricks-gemini-3-pro")?;
         assert!(spec[0]["function"]["parameters"].get("$schema").is_none());
         assert_eq!(spec[0]["function"]["parameters"]["type"], "object");