diff --git a/api/libs/agents/src/agents/buster_multi_agent.rs b/api/libs/agents/src/agents/buster_multi_agent.rs index 3ea9de629..47e4ae5a6 100644 --- a/api/libs/agents/src/agents/buster_multi_agent.rs +++ b/api/libs/agents/src/agents/buster_multi_agent.rs @@ -118,7 +118,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 751bf2b63..4b22c8be6 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/follow_up_initialization.rs b/api/libs/agents/src/agents/modes/follow_up_initialization.rs index d394eb307..b42e63935 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 71e7913df..b295f9b29 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 7001a5d9c..3b5590085 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(),