dify

dify/web/service/use-base.ts

@@ -0,0 +1,30 @@
+import {
+  type QueryKey,
+  useQueryClient,
+} from '@tanstack/react-query'
+
+export const useInvalid = (key?: QueryKey) => {
+  const queryClient = useQueryClient()
+  return () => {
+    if (!key)
+      return
+    queryClient.invalidateQueries(
+      {
+        queryKey: key,
+      },
+    )
+  }
+}
+
+export const useReset = (key?: QueryKey) => {
+  const queryClient = useQueryClient()
+  return () => {
+    if (!key)
+      return
+    queryClient.resetQueries(
+      {
+        queryKey: key,
+      },
+    )
+  }
+}