Merge pull request #200 from rishab11250/feature/hf-token-frontend

frontend/src/lib/api.ts

@@ -201,6 +201,29 @@ class ApiClient {
     return res.json();
   }
 
+  async put<T>(path: string, body?: unknown, options?: FetchOptions): Promise<T> {
+    const res = await this.fetchWithConnectionError(`${this.baseUrl}${path}`, {
+      method: "PUT",
+      headers: this.getHeaders(options?.token),
+      body: body ? JSON.stringify(body) : undefined,
+      ...options,
+    });
+
+    // Auto-refresh on 401
+    if (res.status === 401 && !options?._skipRefresh) {
+      const newToken = await this.tryRefreshToken();
+      if (newToken) {
+        return this.put<T>(path, body, { ...options, token: newToken, _skipRefresh: true });
+      }
+    }
+
+    if (!res.ok) {
+      throw new Error(await this.getErrorMessage(res, res.statusText || "Request failed"));
+    }
+
+    return res.json();
+  }
+
   async postForm<T>(path: string, formData: FormData, options?: FetchOptions): Promise<T> {
     const token = options?.token || this.getToken();
     const headers: HeadersInit = {};

frontend/src/store/auth-store.ts

@@ -8,6 +8,7 @@ export interface AuthUser {
   username: string;
   email: string;
   is_admin: boolean;
+  hf_token?: string;
   created_at: string;
 }
 
@@ -23,6 +24,7 @@ interface AuthStore {
   initializeAuth: () => Promise<void>;
   syncTokensRefreshed: (detail?: { accessToken?: string; user?: AuthUser | null }) => void;
   syncLoggedOut: () => void;
+  setHfToken: (hfToken: string) => Promise<void>;
 }
 
 const getStoredToken = () =>
@@ -138,4 +140,9 @@ export const useAuthStore = create<AuthStore>((set, get) => ({
       initialized: true,
     });
   },
+
+  async setHfToken(hfToken: string) {
+    const response = await api.put<AuthUser>("/api/v1/auth/hf-token", { hf_token: hfToken });
+    set({ user: response });
+  },
 }));