[VS Code] Temp add Omnibox FF

lib/shared/src/experimentation/FeatureFlagProvider.ts

@@ -123,6 +123,8 @@ export enum FeatureFlag {
      * Whether the user will see the CTA about upgrading to Sourcegraph Teams
      */
     SourcegraphTeamsUpgradeCTA = 'teams-upgrade-available-cta',
+
+    TempCodyExperimentalOnebox = 'cody-experimental-one-box',
 }
 
 const ONE_HOUR = 60 * 60 * 1000

vscode/src/chat/chat-view/ChatController.ts

@@ -14,6 +14,7 @@ import {
     DOTCOM_URL,
     type DefaultChatCommands,
     type EventSource,
+    FeatureFlag,
     type Guardrails,
     ModelUsage,
     type NLSSearchDynamicFilter,
@@ -58,6 +59,7 @@ import {
     skip,
     skipPendingOperation,
     startWith,
+    storeLastValue,
     subscriptionDisposable,
     telemetryRecorder,
     tracer,
@@ -193,6 +195,7 @@ export class ChatController implements vscode.Disposable, vscode.WebviewViewProv
 
     public dispose(): void {
         vscode.Disposable.from(...this.disposables).dispose()
+        this.featureCodyExperimentalOneBox.subscription.unsubscribe()
         this.disposables = []
     }
 
@@ -693,10 +696,15 @@ export class ChatController implements vscode.Disposable, vscode.WebviewViewProv
         })
     }
 
+    // TODO(naman/tom): Remove this FF check before the Cody release on 29th.
+    private featureCodyExperimentalOneBox = storeLastValue(
+        featureFlagProvider.evaluatedFeatureFlag(FeatureFlag.TempCodyExperimentalOnebox)
+    )
+
     private async isOmniBoxEnabled(): Promise<boolean> {
         const config = await ClientConfigSingleton.getInstance().getConfig()
 
-        return !!config?.omniBoxEnabled
+        return !!config?.omniBoxEnabled && !!this.featureCodyExperimentalOneBox.value.last
     }
 
     private async getIntentAndScores({

vscode/webviews/utils/useOmniBox.tsx

@@ -4,8 +4,10 @@ import { useFeatureFlag } from './useFeatureFlags'
 
 export const useOmniBox = (): boolean => {
     const config = useClientConfig()
+    // TODO(naman/tom): Remove this FF check before the Cody release on 29th.
+    const tempFFCheck = useFeatureFlag(FeatureFlag.TempCodyExperimentalOnebox)
 
-    return !!config?.omniBoxEnabled
+    return !!config?.omniBoxEnabled && !!tempFFCheck
 }
 
 export const useOmniBoxDebug = (): boolean | undefined => {