front connectors permissions error handling

dust-tt · Jan 2, 2025 · 044cc1d · 044cc1d
1 parent 0313b5b
commit 044cc1d
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 2 deletions.
diff --git a/connectors/src/api/get_connector_permissions.ts b/connectors/src/api/get_connector_permissions.ts
@@ -108,7 +108,7 @@ const _getConnectorPermissions = async (
         return apiError(req, res, {
           status_code: 401,
           api_error: {
-            type: "connector_oauth_error",
+            type: "connector_authorization_error",
             message: pRes.error.message,
           },
         });

diff --git a/front/lib/api/data_source_view.ts b/front/lib/api/data_source_view.ts
@@ -109,7 +109,12 @@ async function getContentNodesForManagedDataSourceView(
     });
 
     if (connectorsRes.isErr()) {
-      if (connectorsRes.error.type === "connector_rate_limit_error") {
+      if (
+        [
+          "connector_rate_limit_error",
+          "connector_authorization_error",
+        ].includes(connectorsRes.error.type)
+      ) {
         return new Err(new Error(connectorsRes.error.message));
       }
       return new Err(

diff --git a/front/pages/api/w/[wId]/data_sources/[dsId]/managed/permissions/index.ts b/front/pages/api/w/[wId]/data_sources/[dsId]/managed/permissions/index.ts
@@ -282,6 +282,18 @@ export async function getManagedDataSourcePermissionsHandler(
         },
       });
     }
+    if (permissionsRes.error.type === "connector_authorization_error") {
+      return apiError(req, res, {
+        status_code: 401,
+        api_error: {
+          type: "data_source_auth_error",
+          message:
+            "Authorization error while retrieving the data source permissions.",
+        },
+      });
+    }
+    // Other error codes are invalid requests from front to connectors or 500s which should be
+    // treated as 500.
     return apiError(req, res, {
       status_code: 500,
       api_error: {

diff --git a/types/src/connectors/api.ts b/types/src/connectors/api.ts
@@ -13,6 +13,7 @@ export type ConnectorsAPIErrorType =
   | "connector_update_unauthorized"
   | "connector_oauth_target_mismatch"
   | "connector_oauth_error"
+  | "connector_authorization_error"
   | "slack_channel_not_found"
   | "connector_rate_limit_error"
   | "slack_configuration_not_found"