fix: tool deletion button (#606)

apps/shinkai-desktop/src/components/playground-tool/components/remove-tool-button.tsx

@@ -5,7 +5,13 @@ import { toast } from 'sonner';
 
 import { useAuth } from '../../../store/auth';
 
-export default function RemoveToolButton({ toolKey }: { toolKey: string }) {
+export default function RemoveToolButton({
+  isPlaygroundTool,
+  toolKey,
+}: {
+  isPlaygroundTool: boolean;
+  toolKey: string;
+}) {
   const auth = useAuth((state) => state.auth);
   const navigate = useNavigate();
 
@@ -32,6 +38,7 @@ export default function RemoveToolButton({ toolKey }: { toolKey: string }) {
           toolKey: toolKey ?? '',
           nodeAddress: auth?.node_address ?? '',
           token: auth?.api_v2_key ?? '',
+          isPlaygroundTool,
         });
       }}
       size="sm"

apps/shinkai-desktop/src/components/tools/components/tool-card.tsx

@@ -245,7 +245,10 @@ export default function ToolCard({
               Go Playground
             </Link>
           )}
-          <RemoveToolButton toolKey={toolKey as string} />
+          <RemoveToolButton
+            isPlaygroundTool={isPlaygroundTool}
+            toolKey={toolKey as string}
+          />
         </div>
       </div>
     </SubpageLayout>

libs/shinkai-message-ts/src/api/tools/index.ts

@@ -28,6 +28,7 @@ import {
   ImportToolResponse,
   PayInvoiceRequest,
   RemovePlaygroundToolRequest,
+  RemoveToolRequest,
   RemoveToolRequestRequest,
   SaveToolCodeRequest,
   SaveToolCodeResponse,
@@ -333,6 +334,22 @@ export const restoreToolConversation = async (
   return response.data as UndoToolImplementationResponse;
 };
 
+export const removeTool = async (
+  nodeAddress: string,
+  bearerToken: string,
+  payload: RemoveToolRequest,
+) => {
+  const response = await httpClient.delete(
+    urlJoin(nodeAddress, '/v2/remove_tool'),
+    {
+      params: { tool_key: payload.tool_key },
+      headers: { Authorization: `Bearer ${bearerToken}` },
+      responseType: 'json',
+    },
+  );
+  return response.data;
+};
+
 export const removePlaygroundTool = async (
   nodeAddress: string,
   bearerToken: string,

libs/shinkai-message-ts/src/api/tools/types.ts

@@ -305,6 +305,10 @@ export type RemovePlaygroundToolRequest = {
   tool_key: string;
 };
 
+export type RemoveToolRequest = {
+  tool_key: string;
+};
+
 export type ImportToolRequest = {
   url: string;
 };

libs/shinkai-node-state/src/v2/mutations/removeTool/index.ts

@@ -1,11 +1,28 @@
-import { removePlaygroundTool as removePlaygroundToolApi } from '@shinkai_network/shinkai-message-ts/api/tools/index';
+import {
+  removePlaygroundTool as removePlaygroundToolApi,
+  removeTool as removeToolApi,
+} from '@shinkai_network/shinkai-message-ts/api/tools/index';
 
 import { RemoveToolInput } from './types';
 
 export const removeTool = async ({
   nodeAddress,
   token,
   toolKey,
+  isPlaygroundTool,
+}: RemoveToolInput) => {
+  if (isPlaygroundTool) {
+    return await removePlaygroundTool({ nodeAddress, token, toolKey });
+  }
+  return await removeToolApi(nodeAddress, token, {
+    tool_key: toolKey,
+  });
+};
+
+export const removePlaygroundTool = async ({
+  nodeAddress,
+  token,
+  toolKey,
 }: RemoveToolInput) => {
   return await removePlaygroundToolApi(nodeAddress, token, {
     tool_key: toolKey,

libs/shinkai-node-state/src/v2/mutations/removeTool/types.ts

@@ -7,4 +7,5 @@ export type RemoveToolOutput = {
 export type RemoveToolInput = Token & {
   nodeAddress: string;
   toolKey: string;
+  isPlaygroundTool?: boolean;
 };

libs/shinkai-node-state/src/v2/mutations/removeTool/useRemoveTool.ts

@@ -23,6 +23,7 @@ export const useRemoveTool = (options?: Options) => {
           {
             nodeAddress: variables.nodeAddress,
             token: variables.token,
+            isPlaygroundTool: variables.isPlaygroundTool,
           },
         ],
       });