add to pre-endpoint check

mongodb-js · Sep 18, 2023 · 5be4dbb · 5be4dbb
1 parent 7d021f6
commit 5be4dbb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/atlas-service/src/main.ts b/packages/atlas-service/src/main.ts
@@ -93,7 +93,10 @@ export async function throwIfNotOk(
 }
 
 function throwIfAINotEnabled(atlasService: typeof AtlasService) {
-  if (!preferences.getPreferences().cloudFeatureRolloutAccess?.GEN_AI_COMPASS) {
+  if (
+    !preferences.getPreferences().cloudFeatureRolloutAccess?.GEN_AI_COMPASS &&
+    !preferences.getPreferences().enableAIWithoutRolloutAccess
+  ) {
     throw new Error(
       "Compass' AI functionality is not currently enabled. Please try again later."
     );