From e410a4c4faae2ca09203e16698603d4bcfc3d39a Mon Sep 17 00:00:00 2001 From: dal Date: Wed, 23 Apr 2025 19:02:38 -0600 Subject: [PATCH] push up o4-mini --- 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 | 4 ++-- api/libs/agents/src/agents/modes/mod.rs | 2 +- api/libs/agents/src/agents/modes/planning.rs | 2 +- .../src/tools/categories/file_tools/filter_dashboards.rs | 2 +- 8 files changed, 9 insertions(+), 9 deletions(-) diff --git a/api/libs/agents/src/agents/buster_multi_agent.rs b/api/libs/agents/src/agents/buster_multi_agent.rs index 533c19d51..c36df2fb7 100644 --- a/api/libs/agents/src/agents/buster_multi_agent.rs +++ b/api/libs/agents/src/agents/buster_multi_agent.rs @@ -115,7 +115,7 @@ impl BusterMultiAgent { // Create agent, passing the provider let agent = Arc::new(Agent::new( - "o3-mini".to_string(), // Initial model (can be overridden by first mode) + "o4-mini".to_string(), // Initial model (can be overridden by first mode) user_id, session_id, "buster_multi_agent".to_string(), diff --git a/api/libs/agents/src/agents/modes/analysis.rs b/api/libs/agents/src/agents/modes/analysis.rs index 99f3eccc6..7d58f9a15 100644 --- a/api/libs/agents/src/agents/modes/analysis.rs +++ b/api/libs/agents/src/agents/modes/analysis.rs @@ -30,7 +30,7 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration { // Note: This prompt doesn't use {DATASETS} // 2. Define the model for this mode (Using default based on original MODEL = None) - let model = "o3-mini".to_string(); + let model = "o4-mini".to_string(); // 3. Define the tool loader closure let tool_loader: Box< 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 554b14fde..f4d078444 100644 --- a/api/libs/agents/src/agents/modes/data_catalog_search.rs +++ b/api/libs/agents/src/agents/modes/data_catalog_search.rs @@ -32,7 +32,7 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration { // Note: This prompt doesn't use {TODAYS_DATE} // 2. Define the model for this mode - let model = "o3-mini".to_string(); // Use o3-mini as requested + let model = "o4-mini".to_string(); // Use o4-mini as requested // 3. Define the tool loader closure let tool_loader: Box) -> Pin> + Send>> + Send + Sync> = 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 56bb8778d..2e3197978 100644 --- a/api/libs/agents/src/agents/modes/follow_up_initialization.rs +++ b/api/libs/agents/src/agents/modes/follow_up_initialization.rs @@ -42,7 +42,7 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration { .replace("{TODAYS_DATE}", &agent_data.todays_date); // 2. Define the model for this mode (Using a default, adjust if needed) - let model = "o3-mini".to_string(); // Assuming default based on original MODEL = None + let model = "o4-mini".to_string(); // Assuming default based on original MODEL = None // 3. Define the tool loader closure let tool_loader: Box) -> Pin> + Send>> + Send + Sync> = diff --git a/api/libs/agents/src/agents/modes/initialization.rs b/api/libs/agents/src/agents/modes/initialization.rs index 74e67b67c..d278af901 100644 --- a/api/libs/agents/src/agents/modes/initialization.rs +++ b/api/libs/agents/src/agents/modes/initialization.rs @@ -26,8 +26,8 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration { // 2. Define the model for this mode (Using a default, adjust if needed) // Since the original MODEL was None, we might use the agent's default - // or specify a standard one like "o3-mini". Let's use "o3-mini". - let model = "o3-mini".to_string(); + // or specify a standard one like "o4-mini". Let's use "o4-mini". + let model = "o4-mini".to_string(); // 3. Define the tool loader closure let tool_loader: Box< diff --git a/api/libs/agents/src/agents/modes/mod.rs b/api/libs/agents/src/agents/modes/mod.rs index df883450a..582555737 100644 --- a/api/libs/agents/src/agents/modes/mod.rs +++ b/api/libs/agents/src/agents/modes/mod.rs @@ -31,7 +31,7 @@ pub struct ModeAgentData { pub struct ModeConfiguration { /// The system prompt to use for the LLM call in this mode. pub prompt: String, - /// The specific LLM model identifier (e.g., "o3-mini") to use for this mode. + /// The specific LLM model identifier (e.g., "o4-mini") to use for this mode. pub model: String, /// An async function/closure responsible for clearing existing tools /// and loading the specific tools required for this mode onto the agent. diff --git a/api/libs/agents/src/agents/modes/planning.rs b/api/libs/agents/src/agents/modes/planning.rs index 3637a9db5..7ca8e61a9 100644 --- a/api/libs/agents/src/agents/modes/planning.rs +++ b/api/libs/agents/src/agents/modes/planning.rs @@ -28,7 +28,7 @@ pub fn get_configuration(agent_data: &ModeAgentData) -> ModeConfiguration { .replace("{DATASETS}", &agent_data.dataset_names.join(", ")); // 2. Define the model for this mode (Using default based on original MODEL = None) - let model = "o3-mini".to_string(); + let model = "o4-mini".to_string(); // 3. Define the tool loader closure let tool_loader: Box< diff --git a/api/libs/agents/src/tools/categories/file_tools/filter_dashboards.rs b/api/libs/agents/src/tools/categories/file_tools/filter_dashboards.rs index fe4c5c07b..12530854a 100644 --- a/api/libs/agents/src/tools/categories/file_tools/filter_dashboards.rs +++ b/api/libs/agents/src/tools/categories/file_tools/filter_dashboards.rs @@ -673,7 +673,7 @@ mod tests { fn test_tool_parameter_validation() { let tool = FilterDashboardsTool { agent: Arc::new(Agent::new( - "o3-mini".to_string(), + "o4-mini".to_string(), HashMap::new(), Uuid::new_v4(), Uuid::new_v4(),