remove Tool Selection Strategy preview (#6250)

crates/goose-server/src/openapi.rs

@@ -361,7 +361,6 @@ derive_utoipa!(Icon as IconSchema);
         super::routes::agent::agent_add_extension,
         super::routes::agent::agent_remove_extension,
         super::routes::agent::update_agent_provider,
-        super::routes::agent::update_router_tool_selector,
         super::routes::action_required::confirm_tool_action,
         super::routes::reply::reply,
         super::routes::session::list_sessions,
@@ -525,7 +524,6 @@ derive_utoipa!(Icon as IconSchema);
         super::routes::agent::ReadResourceResponse,
         super::routes::agent::CallToolRequest,
         super::routes::agent::CallToolResponse,
-        super::routes::agent::UpdateRouterToolSelectorRequest,
         super::routes::agent::StartAgentRequest,
         super::routes::agent::ResumeAgentRequest,
         super::routes::agent::UpdateFromSessionRequest,

crates/goose-server/src/routes/agent.rs

@@ -53,11 +53,6 @@ pub struct GetToolsQuery {
     session_id: String,
 }
 
-#[derive(Deserialize, utoipa::ToSchema)]
-pub struct UpdateRouterToolSelectorRequest {
-    session_id: String,
-}
-
 #[derive(Deserialize, utoipa::ToSchema)]
 pub struct StartAgentRequest {
     working_dir: String,
@@ -288,7 +283,7 @@ async fn resume_agent(
                     })?;
 
             agent
-                .load_provider(provider)
+                .update_provider(provider, &payload.session_id)
                 .await
                 .map_err(|e| ErrorResponse {
                     message: format!("Could not configure agent: {}", e),
@@ -501,35 +496,6 @@ async fn update_agent_provider(
     Ok(())
 }
 
-#[utoipa::path(
-    post,
-    path = "/agent/update_router_tool_selector",
-    request_body = UpdateRouterToolSelectorRequest,
-    responses(
-        (status = 200, description = "Tool selection strategy updated successfully", body = String),
-        (status = 401, description = "Unauthorized - invalid secret key"),
-        (status = 424, description = "Agent not initialized"),
-        (status = 500, description = "Internal server error")
-    )
-)]
-async fn update_router_tool_selector(
-    State(state): State<Arc<AppState>>,
-    Json(payload): Json<UpdateRouterToolSelectorRequest>,
-) -> Result<Json<String>, StatusCode> {
-    let agent = state.get_agent_for_route(payload.session_id).await?;
-    agent
-        .update_router_tool_selector(None, Some(true))
-        .await
-        .map_err(|e| {
-            error!("Failed to update tool selection strategy: {}", e);
-            StatusCode::INTERNAL_SERVER_ERROR
-        })?;
-
-    Ok(Json(
-        "Tool selection strategy updated successfully".to_string(),
-    ))
-}
-
 #[utoipa::path(
     post,
     path = "/agent/add_extension",
@@ -695,10 +661,6 @@ pub fn routes(state: Arc<AppState>) -> Router {
         .route("/agent/read_resource", post(read_resource))
         .route("/agent/call_tool", post(call_tool))
         .route("/agent/update_provider", post(update_agent_provider))
-        .route(
-            "/agent/update_router_tool_selector",
-            post(update_router_tool_selector),
-        )
         .route("/agent/update_from_session", post(update_from_session))
         .route("/agent/add_extension", post(agent_add_extension))
         .route("/agent/remove_extension", post(agent_remove_extension))