Replace compaction notifications with system notifications (#5218)

crates/goose-server/src/openapi.rs

@@ -18,8 +18,9 @@ use goose::config::declarative_providers::{
     DeclarativeProviderConfig, LoadedProvider, ProviderEngine,
 };
 use goose::conversation::message::{
-    ConversationCompacted, FrontendToolRequest, Message, MessageContent, MessageMetadata,
-    RedactedThinkingContent, ThinkingContent, ToolConfirmationRequest, ToolRequest, ToolResponse,
+    FrontendToolRequest, Message, MessageContent, MessageMetadata, RedactedThinkingContent,
+    SystemNotificationContent, SystemNotificationType, ThinkingContent, ToolConfirmationRequest,
+    ToolRequest, ToolResponse,
 };
 
 use utoipa::openapi::schema::{
@@ -350,7 +351,6 @@ derive_utoipa!(Icon as IconSchema);
         super::routes::agent::update_router_tool_selector,
         super::routes::reply::confirm_permission,
         super::routes::reply::reply,
-        super::routes::context::manage_context,
         super::routes::session::list_sessions,
         super::routes::session::get_session,
         super::routes::session::get_session_insights,
@@ -393,8 +393,6 @@ derive_utoipa!(Icon as IconSchema);
         super::routes::config_management::UpdateCustomProviderRequest,
         super::routes::reply::PermissionConfirmationRequest,
         super::routes::reply::ChatRequest,
-        super::routes::context::ContextManageRequest,
-        super::routes::context::ContextManageResponse,
         super::routes::session::ImportSessionRequest,
         super::routes::session::SessionListResponse,
         super::routes::session::UpdateSessionDescriptionRequest,
@@ -420,7 +418,8 @@ derive_utoipa!(Icon as IconSchema);
         RedactedThinkingContent,
         FrontendToolRequest,
         ResourceContentsSchema,
-        ConversationCompacted,
+        SystemNotificationType,
+        SystemNotificationContent,
         JsonObjectSchema,
         RoleSchema,
         ProviderMetadata,

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

@@ -1,68 +0,0 @@
-use crate::state::AppState;
-use axum::{extract::State, http::StatusCode, routing::post, Json, Router};
-use goose::conversation::{message::Message, Conversation};
-use serde::{Deserialize, Serialize};
-use std::sync::Arc;
-use utoipa::ToSchema;
-
-/// Request payload for context management operations
-#[derive(Debug, Deserialize, ToSchema)]
-#[serde(rename_all = "camelCase")]
-pub struct ContextManageRequest {
-    /// Collection of messages to be managed
-    pub messages: Vec<Message>,
-    /// Optional session ID for session-specific agent
-    pub session_id: String,
-}
-
-/// Response from context management operations
-#[derive(Debug, Serialize, ToSchema)]
-#[serde(rename_all = "camelCase")]
-pub struct ContextManageResponse {
-    /// Processed messages after the operation
-    pub messages: Vec<Message>,
-    /// Token counts for each processed message
-    pub token_counts: Vec<usize>,
-}
-
-#[utoipa::path(
-    post,
-    path = "/context/manage",
-    request_body = ContextManageRequest,
-    responses(
-        (status = 200, description = "Context managed successfully", body = ContextManageResponse),
-        (status = 401, description = "Unauthorized - Invalid or missing API key"),
-        (status = 412, description = "Precondition failed - Agent not available"),
-        (status = 500, description = "Internal server error")
-    ),
-    security(
-        ("api_key" = [])
-    ),
-    tag = "Context Management"
-)]
-async fn manage_context(
-    State(state): State<Arc<AppState>>,
-    Json(request): Json<ContextManageRequest>,
-) -> Result<Json<ContextManageResponse>, StatusCode> {
-    let agent = state.get_agent_for_route(request.session_id).await?;
-
-    let conversation = Conversation::new_unvalidated(request.messages);
-
-    let (processed_messages, token_counts, _) =
-        goose::context_mgmt::compact_messages(&agent, &conversation, false)
-            .await
-            .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
-    // TODO(Douwe): store into db
-
-    Ok(Json(ContextManageResponse {
-        messages: processed_messages.messages().to_vec(),
-        token_counts,
-    }))
-}
-
-// Configure routes for this module
-pub fn routes(state: Arc<AppState>) -> Router {
-    Router::new()
-        .route("/context/manage", post(manage_context))
-        .with_state(state)
-}

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

@@ -1,7 +1,6 @@
 pub mod agent;
 pub mod audio;
 pub mod config_management;
-pub mod context;
 pub mod errors;
 pub mod extension;
 pub mod recipe;
@@ -23,7 +22,6 @@ pub fn configure(state: Arc<crate::state::AppState>) -> Router {
         .merge(reply::routes(state.clone()))
         .merge(agent::routes(state.clone()))
         .merge(audio::routes(state.clone()))
-        .merge(context::routes(state.clone()))
         .merge(extension::routes(state.clone()))
         .merge(config_management::routes(state.clone()))
         .merge(recipe::routes(state.clone()))