fix: Use tool key name without version when removing tool from database

dcSpark · Feb 13, 2025 · a1afa7e · a1afa7e
1 parent a784ceb
commit a1afa7e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shinkai-bin/shinkai-node/src/network/v2_api/api_v2_commands_tools.rs b/shinkai-bin/shinkai-node/src/network/v2_api/api_v2_commands_tools.rs
@@ -2271,14 +2271,14 @@ impl Node {
             return Ok(());
         }
         let tool_router_key = tool_router_key.unwrap();
-
+        let tool_key_name = tool_router_key.to_string_without_version();
         let version = tool_router_key.version;
 
         // Attempt to remove the playground tool first
         let _ = db_write.remove_tool_playground(&tool_key);
 
         // Remove the tool from the database
-        match db_write.remove_tool(&tool_key, version) {
+        match db_write.remove_tool(&tool_key_name, version) {
             Ok(_) => {
                 let response = json!({ "status": "success", "message": "Tool and associated playground (if any) removed successfully" });
                 let _ = res.send(Ok(response)).await;