Don't require Atlas login, use the unauthenticated NLQ endpoints.

Author: lerouxb

packages/compass-generative-ai/src/atlas-ai-service.spec.ts

@@ -323,20 +323,6 @@ describe('AtlasAiService', function () {
       });
 
       describe('optIntoGenAIFeatures', function () {
-        beforeEach(async function () {
-          // Reset preferences
-          await preferences.savePreferences({
-            optInGenAIFeatures: false,
-            enableUnauthenticatedGenAI: true,
-          });
-        });
-
-        afterEach(async function () {
-          await preferences.savePreferences({
-            enableUnauthenticatedGenAI: false,
-          });
-        });
-
         it('should save preference when cloud preset', async function () {
           const fetchStub = sandbox.stub().resolves(makeResponse({}));
           global.fetch = fetchStub;

packages/compass-generative-ai/src/atlas-ai-service.ts

@@ -197,9 +197,11 @@ const aiURLConfig = {
   // There are two different sets of endpoints we use for our requests.
   // Down the line we'd like to only use the admin api, however,
   // we cannot currently call that from the Atlas UI. Pending CLOUDP-251201
+  // NOTE: The unauthenticated endpoints are also rate limited by IP address
+  // rather than by logged in user.
   'admin-api': {
-    aggregation: 'ai/api/v1/mql-aggregation',
-    query: 'ai/api/v1/mql-query',
+    aggregation: 'unauth/ai/api/v1/mql-aggregation',
+    query: 'unauth/ai/api/v1/mql-query',
   },
   cloud: {
     aggregation: (groupId: string) => `ai/v1/groups/${groupId}/mql-aggregation`,
@@ -277,17 +279,7 @@ export class AtlasAiService {
   }
 
   async ensureAiFeatureAccess({ signal }: { signal?: AbortSignal } = {}) {
-    if (this.preferences.getPreferences().enableUnauthenticatedGenAI) {
-      return getStore().dispatch(optIntoGenAIWithModalPrompt({ signal }));
-    }
-
-    // When the ai feature is attempted to be opened we make sure
-    // the user is signed into Atlas and opted in.
-
-    if (this.apiURLPreset === 'cloud') {
-      return getStore().dispatch(optIntoGenAIWithModalPrompt({ signal }));
-    }
-    return getStore().dispatch(signIntoAtlasWithModalPrompt({ signal }));
+    return getStore().dispatch(optIntoGenAIWithModalPrompt({ signal }));
   }
 
   private getQueryOrAggregationFromUserInput = async <T>(

packages/compass-preferences-model/src/feature-flags.ts

@@ -28,7 +28,6 @@ export type FeatureFlags = {
   showIndexesGuidanceVariant: boolean;
   enableContextMenus: boolean;
   enableSearchActivationProgramP1: boolean;
-  enableUnauthenticatedGenAI: boolean;
 };
 
 export const featureFlags: Required<{
@@ -151,13 +150,6 @@ export const featureFlags: Required<{
     },
   },
 
-  enableUnauthenticatedGenAI: {
-    stage: 'development',
-    description: {
-      short: 'Enable GenAI for unauthenticated users',
-    },
-  },
-
   /**
    * Feature flag for CLOUDP-308952.
    */