feat(amazonq): enable pre flare experiment

Author: andrewyuq

packages/amazonq/src/lsp/client.ts

@@ -21,6 +21,7 @@ import {
 import {
     AuthUtil,
     CodeWhispererSettings,
+    FeatureConfigProvider,
     getSelectedCustomization,
     TelemetryHelper,
     vsCodeState,
@@ -339,7 +340,7 @@ async function onLanguageServerReady(
     // tutorial for inline chat
     const inlineChatTutorialAnnotation = new InlineChatTutorialAnnotation(inlineTutorialAnnotation)
 
-    const enableInlineRollback = true
+    const enableInlineRollback = FeatureConfigProvider.instance.getPreFlareRollbackGroup() === 'treatment'
     if (enableInlineRollback) {
         // use VSC inline
         await activateInline(client)

packages/core/src/shared/featureConfig.ts

@@ -106,6 +106,20 @@ export class FeatureConfigProvider {
         }
     }
 
+    getPreFlareRollbackGroup(): 'control' | 'treatment' | 'default' {
+        const variation = this.featureConfigs.get(Features.projectContextFeature)?.variation
+        switch (variation) {
+            case 'CONTROL':
+                return 'control'
+            case 'TREATMENT':
+                return 'treatment'
+            case 'DEFAULT':
+                return 'default'
+            default:
+                return 'default'
+        }
+    }
+
     public async listFeatureEvaluations(): Promise<ListFeatureEvaluationsResponse> {
         const profile = AuthUtil.instance.regionProfileManager.activeRegionProfile
         const request: ListFeatureEvaluationsRequest = {