Merge pull request #1 from jgmartin/feature/resource-templates

add ResourceTemplate

Author: jgmartin

README.md

@@ -30,18 +30,33 @@ mcp_sdk_rs = "0.1.0"
 use mcp_sdk_rs::{Client, transport::WebSocketTransport};
 
 #[tokio::main]
-async fn main() -> Result<(), Box<dyn std::error::Error>> {
-    // Create a WebSocket transport
-    let transport = WebSocketTransport::new("ws://localhost:8080").await?;
-
-    // Create and connect the client
-    let client = Client::new(transport);
-    client.connect().await?;
-
-    // Make requests
-    let response = client.request("method_name", Some(params)).await?;
-
-    Ok(())
+async fn main() -> Result<(), Error> {
+    // Create a transport
+    let transport = WebSocketTransport::new(self.url.as_str())
+        .await
+        .map_err(|_| Error::Internal)?;
+
+    //Create mpsc channels for communication between the client and session
+    let (request_tx, request_rx): (UnboundedSender<Message>, UnboundedReceiver<Message>) =
+        tokio::sync::mpsc::unbounded_channel();
+    let (response_tx, response_rx): (UnboundedSender<Message>, UnboundedReceiver<Message>) =
+        tokio::sync::mpsc::unbounded_channel();
+
+    // Create and start the session
+    // Optionally pass an implementation of ClientHandler for custom handling of requests and notifications
+    let session = Session::new(Arc::new(transport), response_tx, request_rx, None);
+    session.start().await.map_err(|_| Error::Internal)?;
+
+    // Create the client and make requests, receive notifications etc
+    let client = Client::new(request_tx, response_rx);
+    let response = client.request(
+        "tools/call",
+        Some(json!({
+            "name": "methondName",
+            "arguments": json!({})
+        })),
+    )
+    .await?
 }
 ```
 

src/client/mod.rs

@@ -118,6 +118,7 @@ pub trait ClientHandler: Send + Sync {
 pub struct DefaultClientHandler;
 #[async_trait]
 impl ClientHandler for DefaultClientHandler {
+    /// Handle an incoming request
     async fn handle_request(&self, method: String, _params: Option<Value>) -> Result<Value, Error> {
         match method.as_str() {
             "sampling/createMessage" => {
@@ -128,6 +129,7 @@ impl ClientHandler for DefaultClientHandler {
         }
     }
 
+    /// Handle an incoming notification
     async fn handle_notification(
         &self,
         method: String,

src/types.rs

@@ -66,6 +66,23 @@ pub struct Resource {
     pub size: Option<String>,
 }
 
+/// A resource template
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct ResourceTemplate {
+    /// Unique identifier for the resource
+    #[serde(rename = "uriTemplate")]
+    pub uri_template: String,
+    /// Human-readable name
+    pub name: String,
+    /// Optional description
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub description: Option<String>,
+    /// The MIME type of this resource, if known
+    #[serde(skip_serializing_if = "Option::is_none")]
+    #[serde(rename = "mimeType")]
+    pub mime_type: Option<String>,
+}
+
 /// Model preferences for completion requests
 #[derive(Debug, Clone, Serialize, Deserialize)]
 pub struct ModelPreferences {