fix: handle 401 errors (#6611)

upstream https://github.com/toeverything/blocksuite/pull/6809

packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts

@@ -9,7 +9,11 @@ import {
   type QueryOptions,
   type RequestOptions,
 } from '@affine/graphql';
-import { GeneralNetworkError, PaymentRequiredError } from '@blocksuite/blocks';
+import {
+  GeneralNetworkError,
+  PaymentRequiredError,
+  UnauthorizedError,
+} from '@blocksuite/blocks';
 
 type OptionsField<T extends GraphQLQuery> =
   RequestOptions<T>['variables'] extends { options: infer U } ? U : never;
@@ -21,10 +25,16 @@ const fetcher = async <Query extends GraphQLQuery>(
     return await defaultFetcher<Query>(options);
   } catch (_err) {
     const error = Array.isArray(_err) ? _err.at(0) : _err;
-    if (error.extensions?.code === 402) {
+    const code = error.extensions?.code;
-      throw new PaymentRequiredError();
+
+    switch (code) {
+      case 401:
+        throw new UnauthorizedError();
+      case 402:
+        throw new PaymentRequiredError();
+      default:
+        throw new GeneralNetworkError();
     }
-    throw new GeneralNetworkError();
   }
 };
 

packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/provider.ts

@@ -1,4 +1,4 @@
-import { openSettingModalAtom } from '@affine/core/atoms';
+import { authAtom, openSettingModalAtom } from '@affine/core/atoms';
 import { getBaseUrl } from '@affine/graphql';
 import { assertExists } from '@blocksuite/global/utils';
 import { AIProvider } from '@blocksuite/presets';
@@ -295,4 +295,11 @@ export function setupAIProvider() {
       open: true,
     });
   });
+
+  AIProvider.slots.requestLogin.on(() => {
+    getCurrentStore().set(authAtom, s => ({
+      ...s,
+      openModal: true,
+    }));
+  });
 }