Adds capability to send new context commands to AB groups

Author: spfink

packages/amazonq/.changes/next-release/Feature-37eb706c-57a1-4751-888d-220b2f68ee4d.json

@@ -0,0 +1,4 @@
+{
+	"type": "Feature",
+	"description": "Adds capability to send new context commands to AB groups"
+}

packages/core/src/amazonq/webview/ui/tabs/generator.ts

@@ -3,12 +3,13 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { ChatItemType, MynahUIDataModel } from '@aws/mynah-ui'
+import { ChatItemType, MynahUIDataModel, QuickActionCommandGroup } from '@aws/mynah-ui'
 import { TabType } from '../storages/tabsStorage'
 import { FollowUpGenerator } from '../followUps/generator'
 import { QuickActionGenerator } from '../quickActions/generator'
 import { TabTypeDataMap } from './constants'
 import { agentWalkthroughDataModel } from '../walkthrough/agent'
+import { FeatureConfigProvider, Features } from '../../../../shared'
 
 export interface TabDataGeneratorProps {
     isFeatureDevEnabled: boolean
@@ -50,7 +51,7 @@ export class TabDataGenerator {
                 'Amazon Q Developer uses generative AI. You may need to verify responses. See the [AWS Responsible AI Policy](https://aws.amazon.com/machine-learning/responsible-ai/policy/).',
             quickActionCommands: this.quickActionsGenerator.generateForTab(tabType),
             promptInputPlaceholder: TabTypeDataMap[tabType].placeholder,
-            contextCommands: TabTypeDataMap[tabType].contextCommands,
+            contextCommands: this.getContextCommands(tabType),
             chatItems: needWelcomeMessages
                 ? [
                       {
@@ -66,4 +67,36 @@ export class TabDataGenerator {
         }
         return tabData
     }
+
+    private getContextCommands(tabType: TabType): QuickActionCommandGroup[] | undefined {
+        if (tabType === 'agentWalkthrough' || tabType === 'welcome') {
+            return
+        }
+
+        const commandHighlight = FeatureConfigProvider.getFeature(Features.highlightCommand)
+
+        const commandName = commandHighlight?.value.stringValue
+        const commandDescription = commandHighlight?.variation
+
+        if (commandName === undefined || commandName === '') {
+            return TabTypeDataMap[tabType].contextCommands
+        } else {
+            const highlightCommand: QuickActionCommandGroup = {
+                groupName: 'Additional Commands',
+                commands: [
+                    {
+                        command: commandName,
+                        description: commandDescription,
+                    },
+                ],
+            }
+
+            const contextCommands = TabTypeDataMap[tabType].contextCommands
+            if (contextCommands === undefined) {
+                return [highlightCommand]
+            } else {
+                return [...contextCommands, highlightCommand]
+            }
+        }
+    }
 }

packages/core/src/shared/featureConfig.ts

@@ -36,6 +36,7 @@ export const Features = {
     dataCollectionFeature: 'IDEProjectContextDataCollection',
     projectContextFeature: 'ProjectContextV2',
     test: 'testFeature',
+    highlightCommand: 'highlightCommand',
 } as const
 
 export type FeatureName = (typeof Features)[keyof typeof Features]