alexhancock/mcp-crate-cleanup (#4885)

2025-09-29 17:00:17 -04:00
parent 2bd18a757b
commit cb0eca9966
78 changed files with 1039 additions and 1090 deletions
@@ -108,8 +108,8 @@ impl Default for SecurityInspector {
 mod tests {
    use super::*;
    use crate::conversation::message::ToolRequest;
-    use mcp_core::ToolCall;
-    use serde_json::json;
+    use rmcp::model::CallToolRequestParam;
+    use rmcp::object;

    #[tokio::test]
    async fn test_security_inspector() {
@@ -118,9 +118,9 @@ mod tests {
        // Test with a potentially dangerous tool call
        let tool_requests = vec![ToolRequest {
            id: "test_req".to_string(),
-            tool_call: Ok(ToolCall {
-                name: "shell".to_string(),
-                arguments: json!({"command": "rm -rf /"}),
+            tool_call: Ok(CallToolRequestParam {
+                name: "shell".into(),
+                arguments: Some(object!({"command": "rm -rf /"})),
            }),
        }];