ctx-mgmt: ctx session management (dev mode only) (#2415)

crates/goose-server/src/openapi.rs

@@ -9,6 +9,8 @@ use goose::message::{
 };
 use goose::permission::permission_confirmation::PrincipalType;
 use goose::providers::base::{ConfigKey, ModelInfo, ProviderMetadata};
+use goose::session::info::SessionInfo;
+use goose::session::SessionMetadata;
 use mcp_core::content::{Annotations, Content, EmbeddedResource, ImageContent, TextContent};
 use mcp_core::handler::ToolResultSchema;
 use mcp_core::resource::ResourceContents;
@@ -33,7 +35,9 @@ use utoipa::OpenApi;
         super::routes::config_management::upsert_permissions,
         super::routes::agent::get_tools,
         super::routes::reply::confirm_permission,
-        super::routes::context::manage_context, // Added this path
+        super::routes::context::manage_context,
+        super::routes::session::list_sessions,
+        super::routes::session::get_session_history
     ),
     components(schemas(
         super::routes::config_management::UpsertConfigQuery,
@@ -48,6 +52,8 @@ use utoipa::OpenApi;
         super::routes::reply::PermissionConfirmationRequest,
         super::routes::context::ContextManageRequest,
         super::routes::context::ContextManageResponse,
+        super::routes::session::SessionListResponse,
+        super::routes::session::SessionHistoryResponse,
         Message,
         MessageContent,
         Content,
@@ -76,6 +82,8 @@ use utoipa::OpenApi;
         PermissionLevel,
         PrincipalType,
         ModelInfo,
+        SessionInfo,
+        SessionMetadata,
     ))
 )]
 pub struct ApiDoc;

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

@@ -11,20 +11,41 @@ use axum::{
 use goose::message::Message;
 use goose::session;
 use goose::session::info::{get_session_info, SessionInfo, SortOrder};
+use goose::session::SessionMetadata;
 use serde::Serialize;
+use utoipa::ToSchema;
 
-#[derive(Serialize)]
-struct SessionListResponse {
+#[derive(Serialize, ToSchema)]
+#[serde(rename_all = "camelCase")]
+pub struct SessionListResponse {
+    /// List of available session information objects
     sessions: Vec<SessionInfo>,
 }
 
-#[derive(Serialize)]
-struct SessionHistoryResponse {
+#[derive(Serialize, ToSchema)]
+#[serde(rename_all = "camelCase")]
+pub struct SessionHistoryResponse {
+    /// Unique identifier for the session
     session_id: String,
-    metadata: session::SessionMetadata,
+    /// Session metadata containing creation time and other details
+    metadata: SessionMetadata,
+    /// List of messages in the session conversation
     messages: Vec<Message>,
 }
 
+#[utoipa::path(
+    get,
+    path = "/sessions",
+    responses(
+        (status = 200, description = "List of available sessions retrieved successfully", body = SessionListResponse),
+        (status = 401, description = "Unauthorized - Invalid or missing API key"),
+        (status = 500, description = "Internal server error")
+    ),
+    security(
+        ("api_key" = [])
+    ),
+    tag = "Session Management"
+)]
 // List all available sessions
 async fn list_sessions(
     State(state): State<Arc<AppState>>,
@@ -38,6 +59,23 @@ async fn list_sessions(
     Ok(Json(SessionListResponse { sessions }))
 }
 
+#[utoipa::path(
+    get,
+    path = "/sessions/{session_id}",
+    params(
+        ("session_id" = String, Path, description = "Unique identifier for the session")
+    ),
+    responses(
+        (status = 200, description = "Session history retrieved successfully", body = SessionHistoryResponse),
+        (status = 401, description = "Unauthorized - Invalid or missing API key"),
+        (status = 404, description = "Session not found"),
+        (status = 500, description = "Internal server error")
+    ),
+    security(
+        ("api_key" = [])
+    ),
+    tag = "Session Management"
+)]
 // Get a specific session's history
 async fn get_session_history(
     State(state): State<Arc<AppState>>,