Jack Amadeo
102 lines
3.1 KiB
Rust
102 lines
3.1 KiB
Rust
/// The protocol messages exchanged between client and server
|
|
use rmcp::model::Tool;
|
|
use rmcp::model::{Content, Prompt, PromptMessage, Resource, ResourceContents};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
// Standard JSON-RPC error codes
|
|
pub const PARSE_ERROR: i32 = -32700;
|
|
pub const INVALID_REQUEST: i32 = -32600;
|
|
pub const METHOD_NOT_FOUND: i32 = -32601;
|
|
pub const INVALID_PARAMS: i32 = -32602;
|
|
pub const INTERNAL_ERROR: i32 = -32603;
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct InitializeResult {
|
|
pub protocol_version: String,
|
|
pub capabilities: ServerCapabilities,
|
|
pub server_info: Implementation,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub instructions: Option<String>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
pub struct Implementation {
|
|
pub name: String,
|
|
pub version: String,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
pub struct ServerCapabilities {
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub prompts: Option<PromptsCapability>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub resources: Option<ResourcesCapability>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub tools: Option<ToolsCapability>,
|
|
// Add other capabilities as needed
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct PromptsCapability {
|
|
pub list_changed: Option<bool>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct ResourcesCapability {
|
|
pub subscribe: Option<bool>,
|
|
pub list_changed: Option<bool>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct ToolsCapability {
|
|
pub list_changed: Option<bool>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct ListResourcesResult {
|
|
pub resources: Vec<Resource>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub next_cursor: Option<String>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
pub struct ReadResourceResult {
|
|
pub contents: Vec<ResourceContents>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct ListToolsResult {
|
|
pub tools: Vec<Tool>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub next_cursor: Option<String>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct CallToolResult {
|
|
pub content: Vec<Content>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub is_error: Option<bool>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
pub struct ListPromptsResult {
|
|
pub prompts: Vec<Prompt>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
|
|
pub struct GetPromptResult {
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
pub description: Option<String>,
|
|
pub messages: Vec<PromptMessage>,
|
|
}
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct EmptyResult {}
|