use serde::{Deserialize, Serialize};
#[derive(Deserialize)]
pub struct JsonRpcRequest {
#[allow(dead_code)]
pub jsonrpc: String,
pub id: Option<serde_json::Value>,
pub method: String,
pub params: Option<serde_json::Value>,
}
#[derive(Serialize)]
pub struct JsonRpcResponse {
pub jsonrpc: String,
pub id: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub result: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub error: Option<JsonRpcError>,
}
impl JsonRpcResponse {
pub fn ok(id: Option<serde_json::Value>, result: serde_json::Value) -> Self {
JsonRpcResponse {
jsonrpc: "2.0".into(),
id,
result: Some(result),
error: None,
}
}
pub fn err(id: Option<serde_json::Value>, code: i64, message: impl Into<String>) -> Self {
JsonRpcResponse {
jsonrpc: "2.0".into(),
id,
result: None,
error: Some(JsonRpcError {
code,
message: message.into(),
}),
}
}
}
#[derive(Serialize)]
pub struct JsonRpcError {
pub code: i64,
pub message: String,
}
#[derive(Serialize, Clone)]
pub struct Tool {
pub name: String,
pub description: String,
#[serde(rename = "inputSchema")]
pub input_schema: serde_json::Value,
}
#[derive(Serialize)]
pub struct ToolResult {
pub content: Vec<ToolContent>,
#[serde(rename = "isError", skip_serializing_if = "Option::is_none")]
pub is_error: Option<bool>,
}
impl ToolResult {
pub fn text(text: impl Into<String>) -> Self {
ToolResult {
content: vec![ToolContent {
content_type: "text".into(),
text: text.into(),
}],
is_error: None,
}
}
pub fn error(text: impl Into<String>) -> Self {
ToolResult {
content: vec![ToolContent {
content_type: "text".into(),
text: text.into(),
}],
is_error: Some(true),
}
}
}
#[derive(Serialize)]
pub struct ToolContent {
#[serde(rename = "type")]
pub content_type: String,
pub text: String,
}