fix query tool

dojoengine · Mar 4, 2025 · d85f68f · d85f68f
1 parent c85f931
commit d85f68f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/torii/server/src/handlers/mcp.rs b/crates/torii/server/src/handlers/mcp.rs
@@ -540,7 +540,7 @@ impl McpHandler {
 
     async fn handle_query_tool(&self, request: JsonRpcRequest) -> JsonRpcResponse {
         if let Some(params) = request.params {
-            if let Some(query) = params.get("arguments").and_then(Value::as_str) {
+            if let Some(query) = params.get("query").and_then(Value::as_str) {
                 match sqlx::query(query).fetch_all(&*self.pool).await {
                     Ok(rows) => {
                         // Convert rows to JSON using shared mapping function