From c08c72134bad0a488d1725853999036e20815f1b Mon Sep 17 00:00:00 2001
From: dal <dallin@buster.so>
Date: Thu, 8 May 2025 14:49:01 -0600
Subject: [PATCH] refactor: update reasoning effort and model names across
 agent configurations

---
 api/libs/agents/src/agent.rs                                 | 2 +-
 api/libs/agents/src/agents/buster_multi_agent.rs             | 2 +-
 api/libs/agents/src/agents/modes/analysis.rs                 | 2 +-
 api/libs/agents/src/agents/modes/data_catalog_search.rs      | 2 +-
 api/libs/agents/src/agents/modes/follow_up_initialization.rs | 2 +-
 api/libs/agents/src/agents/modes/initialization.rs           | 2 +-
 api/libs/agents/src/agents/modes/planning.rs                 | 2 +-
 api/libs/agents/src/agents/modes/review.rs                   | 2 +-
 8 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/api/libs/agents/src/agent.rs b/api/libs/agents/src/agent.rs
index eb0235a46..3a1387898 100644
--- a/api/libs/agents/src/agent.rs
+++ b/api/libs/agents/src/agent.rs
@@ -652,7 +652,7 @@ impl Agent {
                 session_id: thread_ref.id.to_string(),
                 trace_id: Uuid::new_v4().to_string(),
             }),
-            reasoning_effort: Some("high".to_string()),
+            reasoning_effort: Some("medium".to_string()),
             ..Default::default()
         };
 
diff --git a/api/libs/agents/src/agents/buster_multi_agent.rs b/api/libs/agents/src/agents/buster_multi_agent.rs
index 16d380d16..3bc2e0000 100644
--- a/api/libs/agents/src/agents/buster_multi_agent.rs
+++ b/api/libs/agents/src/agents/buster_multi_agent.rs
@@ -178,7 +178,7 @@ impl BusterMultiAgent {
         let model = if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
             "o4-mini".to_string()
         } else {
-            "gemini-2.5-pro-exp-03-25".to_string()
+            "o4-mini".to_string()
         };
 
         // Create agent, passing the provider
diff --git a/api/libs/agents/src/agents/modes/analysis.rs b/api/libs/agents/src/agents/modes/analysis.rs
index eba8e59e5..33c1e585c 100644
--- a/api/libs/agents/src/agents/modes/analysis.rs
+++ b/api/libs/agents/src/agents/modes/analysis.rs
@@ -49,7 +49,7 @@ pub fn get_configuration(agent_data: &ModeAgentData, data_source_syntax: Option<
     let model = if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
         "o4-mini".to_string()
     } else {
-        "gemini-2.5-pro-exp-03-25".to_string()
+        "o4-mini".to_string()
     };
 
     // 3. Define the tool loader closure
diff --git a/api/libs/agents/src/agents/modes/data_catalog_search.rs b/api/libs/agents/src/agents/modes/data_catalog_search.rs
index 22728aa1b..6100cdb64 100644
--- a/api/libs/agents/src/agents/modes/data_catalog_search.rs
+++ b/api/libs/agents/src/agents/modes/data_catalog_search.rs
@@ -36,7 +36,7 @@ pub fn get_configuration(agent_data: &ModeAgentData, _data_source_syntax: Option
     let model = if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
         "o4-mini".to_string()
     } else {
-        "gemini-2.5-pro-exp-03-25".to_string()
+        "o4-mini".to_string()
     };
 
     // 3. Define the tool loader closure
diff --git a/api/libs/agents/src/agents/modes/follow_up_initialization.rs b/api/libs/agents/src/agents/modes/follow_up_initialization.rs
index b6d51d7ff..73f4fae55 100644
--- a/api/libs/agents/src/agents/modes/follow_up_initialization.rs
+++ b/api/libs/agents/src/agents/modes/follow_up_initialization.rs
@@ -41,7 +41,7 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration {
         if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
             "o4-mini".to_string()
         } else {
-            "gemini-2.5-pro-exp-03-25".to_string()
+            "o4-mini".to_string()
         };
 
     // 3. Define the tool loader closure
diff --git a/api/libs/agents/src/agents/modes/initialization.rs b/api/libs/agents/src/agents/modes/initialization.rs
index b86595b0a..40aa005fa 100644
--- a/api/libs/agents/src/agents/modes/initialization.rs
+++ b/api/libs/agents/src/agents/modes/initialization.rs
@@ -39,7 +39,7 @@ pub fn get_configuration(
         if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
             "o4-mini".to_string()
         } else {
-            "gemini-2.5-pro-exp-03-25".to_string()
+            "o4-mini".to_string()
         };
 
     // 3. Define the tool loader closure
diff --git a/api/libs/agents/src/agents/modes/planning.rs b/api/libs/agents/src/agents/modes/planning.rs
index b227fe60a..a894accfb 100644
--- a/api/libs/agents/src/agents/modes/planning.rs
+++ b/api/libs/agents/src/agents/modes/planning.rs
@@ -39,7 +39,7 @@ pub fn get_configuration(
         if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
             "o4-mini".to_string()
         } else {
-            "gemini-2.5-pro-exp-03-25".to_string()
+            "o4-mini".to_string()
         };
 
     // 3. Define the tool loader closure
diff --git a/api/libs/agents/src/agents/modes/review.rs b/api/libs/agents/src/agents/modes/review.rs
index 834b38ff7..f95022a70 100644
--- a/api/libs/agents/src/agents/modes/review.rs
+++ b/api/libs/agents/src/agents/modes/review.rs
@@ -31,7 +31,7 @@ pub fn get_configuration(
         if env::var("ENVIRONMENT").unwrap_or_else(|_| "development".to_string()) == "local" {
             "o4-mini".to_string()
         } else {
-            "gemini-2.5-flash-preview-04-17".to_string()
+            "gemini-2.0-flash-001".to_string()
         };
 
     // 3. Define the tool loader closure