use serde::{Deserialize, Serialize};
use serde_json::Value;
#[derive(Debug, Deserialize)]
pub struct JsonRpcRequest {
#[allow(dead_code)]
pub jsonrpc: String,
pub id: Option<Value>,
pub method: String,
#[serde(default)]
pub params: Value,
}
#[derive(Debug, Serialize)]
pub struct JsonRpcResponse {
pub jsonrpc: &'static str,
#[serde(skip_serializing_if = "Option::is_none")]
pub id: Option<Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub result: Option<Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub error: Option<JsonRpcError>,
}
#[derive(Debug, Serialize)]
pub struct JsonRpcError {
pub code: i32,
pub message: String,
}
impl JsonRpcResponse {
pub fn success(id: Option<Value>, result: Value) -> Self {
Self {
jsonrpc: "2.0",
id,
result: Some(result),
error: None,
}
}
pub fn error(id: Option<Value>, code: i32, message: String) -> Self {
Self {
jsonrpc: "2.0",
id,
result: None,
error: Some(JsonRpcError { code, message }),
}
}
}
#[derive(Debug, Serialize)]
pub struct ToolDef {
pub name: String,
pub description: String,
#[serde(rename = "inputSchema")]
pub input_schema: Value,
}
#[derive(Debug, Serialize)]
pub struct ToolCallResult {
pub content: Vec<ToolContent>,
#[serde(rename = "isError", skip_serializing_if = "std::ops::Not::not")]
pub is_error: bool,
}
#[derive(Debug, Serialize)]
pub struct ToolContent {
#[serde(rename = "type")]
pub content_type: &'static str,
pub text: String,
}
impl ToolCallResult {
pub fn text(text: String) -> Self {
Self {
content: vec![ToolContent {
content_type: "text",
text,
}],
is_error: false,
}
}
pub fn error(message: String) -> Self {
Self {
content: vec![ToolContent {
content_type: "text",
text: message,
}],
is_error: true,
}
}
}