Greptile review fixes

frontend/src/hooks/api/identities/mutations.tsx

@@ -305,7 +305,7 @@ export const useClearIdentityUniversalAuthLockouts = () => {
     },
     onSuccess: (_, { identityId }) => {
       queryClient.invalidateQueries({
-        queryKey: identitiesKeys.clearIdentityUniversalAuthLockouts(identityId)
+        queryKey: identitiesKeys.getIdentityUniversalAuth(identityId)
       });
     }
   });

frontend/src/hooks/api/identities/queries.tsx

@@ -49,9 +49,7 @@ export const identitiesKeys = {
   getIdentityTokensTokenAuth: (identityId: string) =>
     [{ identityId }, "identity-tokens-token-auth"] as const,
   getIdentityProjectMemberships: (identityId: string) =>
-    [{ identityId }, "identity-project-memberships"] as const,
-  clearIdentityUniversalAuthLockouts: (identityId: string) =>
-    [{ identityId }, "clear-identity-universal-auth-lockouts"] as const
+    [{ identityId }, "identity-project-memberships"] as const
 };
 
 export const useGetIdentityById = (identityId: string) => {

frontend/src/pages/organization/IdentityDetailsByIDPage/components/ViewIdentityAuthModal/ViewIdentityUniversalAuthContent.tsx

@@ -40,14 +40,20 @@ export const ViewIdentityUniversalAuthContent = ({
   });
 
   async function clearLockouts() {
-    const deleted = await clearLockoutsFn({ identityId });
-
-    createNotification({
-      text: `Successfully cleared ${deleted} lockout${deleted === 1 ? "" : "s"}`,
-      type: "success"
-    });
-
-    setLockedOutState(false);
+    try {
+      const deleted = await clearLockoutsFn({ identityId });
+      createNotification({
+        text: `Successfully cleared ${deleted} lockout${deleted === 1 ? "" : "s"}`,
+        type: "success"
+      });
+      setLockedOutState(false);
+    } catch (error) {
+      console.error(error);
+      createNotification({
+        text: "Failed to clear lockouts. Please try again.",
+        type: "error"
+      });
+    }
   }
 
   if (isPending || clientSecretsPending) {