diff --git a/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/CodeWhispererFeatureConfigService.kt b/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/CodeWhispererFeatureConfigService.kt index 80cbbcf9f54..5193d4e704f 100644 --- a/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/CodeWhispererFeatureConfigService.kt +++ b/plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonq/CodeWhispererFeatureConfigService.kt @@ -108,6 +108,9 @@ class CodeWhispererFeatureConfigService { fun getNewAutoTriggerUX(): Boolean = getFeatureValueForKey(NEW_AUTO_TRIGGER_UX).stringValue() == "TREATMENT" + @Suppress("UNUSED") + fun getInlineCompletion(): Boolean = getFeatureValueForKey(INLINE_COMPLETION).stringValue() == "TREATMENT" + // Get the feature value for the given key. // In case of a misconfiguration, it will return a default feature value of Boolean false. private fun getFeatureValueForKey(name: String): FeatureValue = @@ -117,6 +120,7 @@ class CodeWhispererFeatureConfigService { companion object { fun getInstance(): CodeWhispererFeatureConfigService = service() private const val TEST_FEATURE_NAME = "testFeature" + private const val INLINE_COMPLETION = "ProjectContextV2" private const val DATA_COLLECTION_FEATURE = "IDEProjectContextDataCollection" const val CUSTOMIZATION_ARN_OVERRIDE_NAME = "customizationArnOverride" private const val NEW_AUTO_TRIGGER_UX = "newAutoTriggerUX" @@ -141,6 +145,11 @@ class CodeWhispererFeatureConfigService { "CONTROL", FeatureValue.builder().stringValue("CONTROL").build() ), + INLINE_COMPLETION to FeatureContext( + INLINE_COMPLETION, + "CONTROL", + FeatureValue.builder().stringValue("CONTROL").build() + ) ) } }