首页

video-gen-app/src/store/useAuthStore.ts

@@ -10,6 +10,7 @@ interface AuthState {
   checkAuth: () => Promise<void>;
   changePassword: (oldPwd: string, newPwd: string) => Promise<void>;
   refreshUser: () => Promise<void>;
+  setUserCredits: (credits: number) => void;
 }
 
 export const useAuthStore = create<AuthState>((set) => ({
@@ -52,4 +53,10 @@ export const useAuthStore = create<AuthState>((set) => ({
       console.error('Failed to refresh user:', error);
     }
   },
+
+  // 用通知接口返回的 credits.balance 直接覆盖当前 user.credits
+  // 避免再请求一次 getUser 接口
+  setUserCredits: (credits: number) => {
+    set((state) => (state.user ? { user: { ...state.user, credits } } : {}));
+  },
 }));