logging changes for nes

Author: hitesh-1997

Diff for: lib/shared/src/experimentation/FeatureFlagProvider.ts

@@ -32,6 +32,9 @@ export enum FeatureFlag {
     // Enable the deepseek-v2 as the default model via Fireworks
     CodyAutocompleteDeepseekV2LiteBase = 'cody-autocomplete-deepseek-v2-lite-base',
 
+    // Data collection variants used for completions and next edit completions
+    CodyAutocompleteDataCollectionFlag = 'cody-autocomplete-data-collection-flag',
+
     // Enable various feature flags to experiment with FIM trained fine-tuned models via Fireworks
     CodyAutocompleteFIMModelExperimentBaseFeatureFlag = 'cody-autocomplete-fim-model-experiment-flag-v2',
     CodyAutocompleteFIMModelExperimentControl = 'cody-autocomplete-fim-model-experiment-control-v2',

Diff for: vscode/src/completions/completion-provider-config.ts

@@ -27,6 +27,7 @@ class CompletionProviderConfig {
             FeatureFlag.CodyAutocompletePreloadingExperimentVariant2,
             FeatureFlag.CodyAutocompletePreloadingExperimentVariant3,
             FeatureFlag.CodyAutocompleteDisableLowPerfLangDelay,
+            FeatureFlag.CodyAutocompleteDataCollectionFlag,
         ]
         await Promise.all(featureFlagsUsed.map(flag => featureFlagProvider.evaluateFeatureFlag(flag)))
     }
@@ -193,6 +194,12 @@ class CompletionProviderConfig {
             .evaluatedFeatureFlag(FeatureFlag.CodyAutocompleteDisableLowPerfLangDelay)
             .pipe(distinctUntilChanged())
     }
+
+    public get completionDataCollectionFlag(): Observable<boolean> {
+        return featureFlagProvider
+            .evaluatedFeatureFlag(FeatureFlag.CodyAutocompleteDataCollectionFlag)
+            .pipe(distinctUntilChanged())
+    }
 }
 
 /**

Diff for: vscode/src/completions/context/context-mixer.ts

@@ -4,9 +4,12 @@ import {
     type AutocompleteContextSnippet,
     type DocumentContext,
     contextFiltersProvider,
+    subscriptionDisposable,
     wrapInActiveSpan,
 } from '@sourcegraph/cody-shared'
 
+import {GitHubDotComRepoMetadata} from '../../repository/repo-metadata-from-git-api';
+import { completionProviderConfig } from '../completion-provider-config'
 import type { LastInlineCompletionCandidate } from '../get-inline-completions'
 import {
     DefaultCompletionsContextRanker,
@@ -21,6 +24,8 @@ interface GetContextOptions {
     abortSignal?: AbortSignal
     maxChars: number
     lastCandidate?: LastInlineCompletionCandidate
+    gitUrl?: string
+    isDotComUser?: boolean
 }
 
 export interface ContextSummary {
@@ -72,8 +77,19 @@ export interface GetContextResult {
  * ranged for the top ranked document from all retrieval sources before we move on to the second
  * document).
  */
-export class ContextMixer {
-    constructor(private strategyFactory: ContextStrategyFactory) {}
+export class ContextMixer implements vscode.Disposable {
+    private disposables: vscode.Disposable[] = []
+    private dataCollectionFlag = false
+
+    constructor(private strategyFactory: ContextStrategyFactory) {
+        this.disposables.push(
+            subscriptionDisposable(
+                completionProviderConfig.completionDataCollectionFlag.subscribe(dataCollectionFlag => {
+                    this.dataCollectionFlag = dataCollectionFlag
+                })
+            )
+        )
+    }
 
     public async getContext(options: GetContextOptions): Promise<GetContextResult> {
         const start = performance.now()
@@ -176,6 +192,22 @@ export class ContextMixer {
             rankedContextCandidates: Array.from(fusedResults),
         }
     }
+
+    private shouldLogContext(gitUrl: string, isDotComUser: boolean): boolean {
+        if (!isDotComUser || !this.dataCollectionFlag) {
+            return false
+        }
+        const instance = GitHubDotComRepoMetadata.getInstance()
+        const gitRepoMetadata = instance.getRepoMetadataIfCached(gitUrl)
+        return gitRepoMetadata?.isPublic ?? false
+    }
+
+    public dispose(): void {
+        for (const disposable of this.disposables) {
+            disposable.dispose()
+        }
+        this.disposables = []
+    }
 }
 
 async function filter(snippets: AutocompleteContextSnippet[]): Promise<AutocompleteContextSnippet[]> {

Diff for: vscode/src/completions/get-inline-completions.ts

@@ -447,6 +447,8 @@ async function doGetInlineCompletions(
                 abortSignal,
                 maxChars: provider.contextSizeHints.totalChars,
                 lastCandidate,
+                gitUrl: gitIdentifiersForFile?.gitUrl,
+                isDotComUser: isDotComUser,
             })
         ),
         remainingInterval > 0

Diff for: vscode/src/completions/inline-completion-item-provider.ts

@@ -211,6 +211,7 @@ export class InlineCompletionItemProvider
         this.disposables.push(strategyFactory)
 
         this.contextMixer = new ContextMixer(strategyFactory)
+        this.disposables.push(this.contextMixer)
 
         this.smartThrottleService = new SmartThrottleService()
         this.disposables.push(this.smartThrottleService)