feat(core): add error boundary for ai-usage

toeverything · Apr 11, 2024 · d20ed7a · d20ed7a
1 parent ef8c3d3
commit d20ed7a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/packages/frontend/core/src/components/affine/setting-modal/account-setting/index.tsx b/packages/frontend/core/src/components/affine/setting-modal/account-setting/index.tsx
@@ -5,6 +5,7 @@ import {
 } from '@affine/component/setting-components';
 import { Avatar } from '@affine/component/ui/avatar';
 import { Button } from '@affine/component/ui/button';
+import { SWRErrorBoundary } from '@affine/core/components/pure/swr-error-bundary';
 import { useAsyncCallback } from '@affine/core/hooks/affine-async-hooks';
 import {
   removeAvatarMutation,
@@ -257,9 +258,11 @@ export const AccountSetting: FC = () => {
       <Suspense>
         <StoragePanel />
       </Suspense>
-      <Suspense>
-        <AIUsagePanel />
-      </Suspense>
+      <SWRErrorBoundary fallback={<div />}>
+        <Suspense>
+          <AIUsagePanel />
+        </Suspense>
+      </SWRErrorBoundary>
       <SettingRow
         name={t[`Sign out`]()}
         desc={t['com.affine.setting.sign.out.message']()}