use std::collections::HashMap;
use serde::{Deserialize, Serialize};
use serde_json::Value;
pub type MessageId = String;
pub type SessionId = String;
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum MessageType {
Request,
Response,
Notification,
Error,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Message {
pub id: MessageId,
#[serde(rename = "type")]
pub message_type: MessageType,
#[serde(skip_serializing_if = "Option::is_none")]
pub session_id: Option<SessionId>,
#[serde(skip_serializing_if = "Option::is_none")]
pub metadata: Option<HashMap<String, Value>>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Request {
#[serde(flatten)]
pub message: Message,
pub tool: String,
pub params: Value,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Response {
#[serde(flatten)]
pub message: Message,
pub data: Value,
#[serde(skip_serializing_if = "Option::is_none")]
pub partial: Option<bool>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ErrorMessage {
#[serde(flatten)]
pub message: Message,
pub code: i32,
pub error: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub details: Option<Value>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Notification {
#[serde(flatten)]
pub message: Message,
pub name: String,
pub data: Value,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ToolMetadata {
pub name: String,
pub description: String,
pub parameters: Value,
#[serde(skip_serializing_if = "Option::is_none")]
pub return_schema: Option<Value>,
#[serde(default)]
pub streaming: bool,
#[serde(default)]
pub categories: Vec<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub version: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub author: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub documentation_url: Option<String>,
#[serde(default)]
pub deprecated: bool,
#[serde(skip_serializing_if = "Option::is_none")]
pub deprecation_message: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ListToolsResponse {
#[serde(flatten)]
pub message: Message,
pub tools: Vec<ToolMetadata>,
}