Internal MCP Crate Cleanup (#4800)
This commit is contained in:
@@ -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 /"})),
|
||||
}),
|
||||
}];
|
||||
|
||||
|
||||
Reference in New Issue
Block a user