chore: use query

web/app/components/workflow/block-selector/tool-picker.tsx

@@ -51,7 +51,7 @@ const ToolPicker: FC<Props> = ({
 
   const { data: buildInTools } = useAllBuiltInTools()
   const { data: customTools } = useAllCustomTools()
-  const { invalidate: invalidateCustomTools } = useInvalidateAllCustomTools()
+  const invalidateCustomTools = useInvalidateAllCustomTools()
   const { data: workflowTools } = useAllWorkflowTools()
 
   const handleAddedCustomTool = invalidateCustomTools

web/service/use-tools.ts

@@ -30,13 +30,11 @@ export const useAllCustomTools = () => {
 
 export const useInvalidateAllCustomTools = () => {
   const queryClient = useQueryClient()
-  return {
+  return () => {
-    invalidate: () => {
+    queryClient.invalidateQueries(
-      queryClient.invalidateQueries(
+      {
-        {
+        queryKey: useAllCustomToolsKey,
-          queryKey: useAllCustomToolsKey,
+      })
-        })
-    },
   }
 }