try opt in @workspace context once if users are in treatment group

Author: Will-ShaoHua

packages/core/src/codewhispererChat/controllers/chat/controller.ts

@@ -49,7 +49,6 @@ import { CodeWhispererSettings } from '../../../codewhisperer/util/codewhisperer
 import { getSelectedCustomization } from '../../../codewhisperer/util/customizationUtil'
 import { getHttpStatusCode, AwsClientResponseError } from '../../../shared/errors'
 import { uiEventRecorder } from '../../../amazonq/util/eventRecorder'
-import { globals, waitUntil } from '../../../shared'
 import { telemetry } from '../../../shared/telemetry'
 import { isSsoConnection } from '../../../auth/connection'
 import { inspect } from '../../../shared/utilities/collectionUtils'

packages/core/src/shared/featureConfig.ts

@@ -35,6 +35,7 @@ export const Features = {
     customizationArnOverride: 'customizationArnOverride',
     dataCollectionFeature: 'IDEProjectContextDataCollection',
     projectContextFeature: 'ProjectContextV2',
+    workspaceContextFeature: 'WorkspaceContext',
     test: 'testFeature',
 } as const
 
@@ -83,6 +84,21 @@ export class FeatureConfigProvider {
         }
     }
 
+    getWorkspaceContextGroup(): 'control' | 'treatment' {
+        const variation = this.featureConfigs.get(Features.projectContextFeature)?.variation
+
+        switch (variation) {
+            case 'CONTROL':
+                return 'control'
+
+            case 'TREATMENT':
+                return 'treatment'
+
+            default:
+                return 'control'
+        }
+    }
+
     public async listFeatureEvaluations(): Promise<ListFeatureEvaluationsResponse> {
         const request: ListFeatureEvaluationsRequest = {
             userContext: {
@@ -157,7 +173,7 @@ export class FeatureConfigProvider {
                     await vscode.commands.executeCommand('aws.amazonq.refreshStatusBar')
                 }
             }
-            if (Auth.instance.isInternalAmazonUser()) {
+            if (this.getWorkspaceContextGroup() === 'treatment') {
                 // Enable local workspace index by default only once, for Amzn users.
                 const isSet = globals.globalState.get<boolean>('aws.amazonq.workspaceIndexToggleOn') || false
                 if (!isSet) {