context_management: handle summarization in UI (#2377)

crates/mcp-core/src/content.rs

@@ -5,8 +5,9 @@ use super::role::Role;
 use crate::resource::ResourceContents;
 use chrono::{DateTime, Utc};
 use serde::{Deserialize, Serialize};
+use utoipa::ToSchema;
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(ToSchema, Debug, Clone, PartialEq, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct Annotations {
     #[serde(skip_serializing_if = "Option::is_none")]
@@ -14,6 +15,8 @@ pub struct Annotations {
     #[serde(skip_serializing_if = "Option::is_none")]
     pub priority: Option<f32>,
     #[serde(skip_serializing_if = "Option::is_none")]
+    #[schema(value_type = String, format = "date-time", example = "2023-01-01T00:00:00Z")]
+    // for openapi
     pub timestamp: Option<DateTime<Utc>>,
 }
 
@@ -33,7 +36,7 @@ impl Annotations {
     }
 }
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(ToSchema, Debug, Clone, PartialEq, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct TextContent {
     pub text: String,
@@ -41,7 +44,7 @@ pub struct TextContent {
     pub annotations: Option<Annotations>,
 }
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(ToSchema, Debug, Clone, PartialEq, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct ImageContent {
     pub data: String,
@@ -50,7 +53,7 @@ pub struct ImageContent {
     pub annotations: Option<Annotations>,
 }
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(ToSchema, Debug, Clone, PartialEq, Serialize, Deserialize)]
 #[serde(rename_all = "camelCase")]
 pub struct EmbeddedResource {
     pub resource: ResourceContents,
@@ -67,7 +70,7 @@ impl EmbeddedResource {
     }
 }
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(ToSchema, Debug, Clone, PartialEq, Serialize, Deserialize)]
 #[serde(tag = "type", rename_all = "camelCase")]
 pub enum Content {
     Text(TextContent),

crates/mcp-core/src/handler.rs

@@ -1,8 +1,11 @@
 use async_trait::async_trait;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
+#[allow(unused_imports)] // this is used in schema below
+use serde_json::json;
 use serde_json::Value;
 use thiserror::Error;
+use utoipa::ToSchema;
 
 #[non_exhaustive]
 #[derive(Error, Debug, Clone, Deserialize, Serialize, PartialEq)]
@@ -19,6 +22,18 @@ pub enum ToolError {
 
 pub type ToolResult<T> = std::result::Result<T, ToolError>;
 
+// Define schema manually without generics issues
+#[derive(ToSchema)]
+#[schema(example = json!({"success": true, "data": {}}))]
+pub struct ToolResultSchema {
+    #[schema(example = "Operation completed successfully")]
+    pub message: Option<String>,
+    #[schema(example = true)]
+    pub success: bool,
+    #[schema(value_type = Object)]
+    pub data: Option<serde_json::Value>,
+}
+
 #[derive(Error, Debug)]
 pub enum ResourceError {
     #[error("Execution failed: {0}")]

crates/mcp-core/src/resource.rs

@@ -1,10 +1,10 @@
+use crate::content::Annotations;
 /// Resources that servers provide to clients
 use anyhow::{anyhow, Result};
 use chrono::{DateTime, Utc};
 use serde::{Deserialize, Serialize};
 use url::Url;
-
-use crate::content::Annotations;
+use utoipa::ToSchema;
 
 const EPSILON: f32 = 1e-6; // Tolerance for floating point comparison
 
@@ -28,6 +28,7 @@ pub struct Resource {
 
 #[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
 #[serde(rename_all = "camelCase", untagged)]
+#[derive(ToSchema)]
 pub enum ResourceContents {
     TextResourceContents {
         uri: String,

crates/mcp-core/src/role.rs

@@ -1,7 +1,8 @@
 /// Roles to describe the origin/ownership of content
 use serde::{Deserialize, Serialize};
+use utoipa::ToSchema;
 
-#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
+#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, ToSchema)]
 #[serde(rename_all = "lowercase")]
 pub enum Role {
     User,