Handle 401 errors by redirecting to the sign-in page in usePrompts an…

…d apiClient
astriaai · Nov 15, 2024 · c91359d · c91359d
1 parent 622f504
commit c91359d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/src/hooks/usePrompts.ts b/src/hooks/usePrompts.ts
@@ -2,6 +2,7 @@ import { useEffect, useCallback, useRef } from "react";
 import { useStore } from "@/store/promptStore";
 import { fetchGalleryPrompts, fetchUserPrompts } from "@/api/prompts";
 import type { Prompt } from "../types";
+import { AxiosError } from "axios";
 
 
 export type PromptType = "user" | "gallery" | null;
@@ -42,8 +43,11 @@ const usePrompts = (promptId?: number): UsePromptsResult => {
         addUserPrompts(prompts);
       }
       return true;
-    } catch (error) {
+    } catch (error: unknown) {
       console.error(`Error fetching ${isGallery ? "gallery" : "user"} prompts:`, error);
+      if ((error as AxiosError).response?.status === 401) {
+        window.location.href = "https://www.astria.ai/users/sign_in";
+      }
       return false;
     }
   }, [galleryOffset, userOffset, limit, addGalleryPrompts, addUserPrompts]);

diff --git a/src/services/apiClient.ts b/src/services/apiClient.ts
@@ -58,14 +58,8 @@ export const post = async <T>(url: string, data: object): Promise<ApiResponse<T>
 
 const handleApiError = (error: AxiosError): void => {
   if (error.response?.status === 401) {
-    window.location.href = 'https://astria.ai/login';
+    window.location.href = 'https://www.astria.ai/users/sign_in';
     return;
-  } else if (error.response) {
-    console.error('API Error:', error.response.status, error.response.data);
-  } else if (error.request) {
-    console.error('No response received:', error.request);
-  } else {
-    console.error('Error:', error.message);
   }
 };