@@ -13,7 +13,6 @@ import software.amazon.awssdk.services.codewhispererruntime.model.ListAvailableC
1313import software.aws.toolkits.core.utils.debug
1414import software.aws.toolkits.core.utils.getLogger
1515import software.aws.toolkits.jetbrains.core.awsClient
16- import software.aws.toolkits.jetbrains.isDeveloperMode
1716import software.aws.toolkits.jetbrains.utils.isQExpired
1817
1918@Service
@@ -109,8 +108,7 @@ class CodeWhispererFeatureConfigService {
109108
110109 fun getNewAutoTriggerUX (): Boolean = getFeatureValueForKey(NEW_AUTO_TRIGGER_UX ).stringValue() == " TREATMENT"
111110
112- // TODO: remove dev mode flag
113- fun getInlineCompletion (): Boolean = if (isDeveloperMode()) true else getFeatureValueForKey(INLINE_COMPLETION ).boolValue()
111+ fun getInlineCompletion (): Boolean = getFeatureValueForKey(INLINE_COMPLETION ).stringValue() == " TREATMENT"
114112
115113 // Get the feature value for the given key.
116114 // In case of a misconfiguration, it will return a default feature value of Boolean false.
@@ -121,9 +119,7 @@ class CodeWhispererFeatureConfigService {
121119 companion object {
122120 fun getInstance (): CodeWhispererFeatureConfigService = service()
123121 private const val TEST_FEATURE_NAME = " testFeature"
124-
125- // TODO: update
126- private const val INLINE_COMPLETION = " inlineCompletion"
122+ private const val INLINE_COMPLETION = " ProjectContextV2"
127123 private const val DATA_COLLECTION_FEATURE = " IDEProjectContextDataCollection"
128124 const val CUSTOMIZATION_ARN_OVERRIDE_NAME = " customizationArnOverride"
129125 private const val NEW_AUTO_TRIGGER_UX = " newAutoTriggerUX"
@@ -148,6 +144,11 @@ class CodeWhispererFeatureConfigService {
148144 " CONTROL" ,
149145 FeatureValue .builder().stringValue(" CONTROL" ).build()
150146 ),
147+ INLINE_COMPLETION to FeatureContext (
148+ INLINE_COMPLETION ,
149+ " CONTROL" ,
150+ FeatureValue .builder().stringValue(" CONTROL" ).build()
151+ )
151152 )
152153 }
153154}
0 commit comments