[CallToolArgs] Set agent id inside when tools need to call other agents (#22)

Author: kageiit

Cargo.lock

@@ -21,7 +21,7 @@ jsonrpc-core = "18"
 warp = "0.3"
 jsonrpc-derive = "18"
 futures = "0.3"
-uuid = { version = "1", features = ["v4"] }
+uuid = { version = "1", features = ["v4", "v7", "serde"] }
 url = "2"
 base64 = "0.22"
 async-stream = "0.3"

Cargo.toml

@@ -6,6 +6,7 @@ use mcp_rs::{
     transport::{sse::SseTransport, stdio::StdioTransport, ws::WebSocketTransport},
 };
 use serde_json::json;
+use uuid::Uuid;
 
 #[derive(Parser, Debug)]
 #[command(name = "mcp-client", version, about = "MCP Client CLI")]
@@ -255,6 +256,7 @@ async fn main() -> Result<(), McpError> {
                         CallToolArgs::builder()
                             .session_id(session_id)
                             .tool_id(tool_id)
+                            .agent_id(Uuid::new_v4())
                             .build(),
                     ),
                 )

bin/client.rs

@@ -5,6 +5,7 @@ use std::{collections::HashMap, sync::Arc};
 use test_tool::{PingTool, TestTool};
 use tokio::sync::{mpsc, RwLock};
 use typed_builder::TypedBuilder;
+use uuid::Uuid;
 
 pub mod calculator;
 pub mod file_system;
@@ -117,6 +118,8 @@ pub struct CallToolArgs {
     #[serde(skip_serializing_if = "Option::is_none")]
     #[builder(default, setter(into))]
     pub session_id: Option<String>,
+
+    pub agent_id: Uuid,
 }
 
 // Tool Provider trait