Merge pull request #6581 from continuedev/jacob/feature/static

feat: Test static contextualization

Author: sestinj

core/autocomplete/CompletionProvider.ts

@@ -10,7 +10,7 @@ import { BracketMatchingService } from "./filtering/BracketMatchingService.js";
 import { CompletionStreamer } from "./generation/CompletionStreamer.js";
 import { postprocessCompletion } from "./postprocessing/index.js";
 import { shouldPrefilter } from "./prefiltering/index.js";
-import { getAllSnippets } from "./snippets/index.js";
+import { getAllSnippetsWithoutRace } from "./snippets/index.js";
 import { renderPrompt } from "./templating/index.js";
 import { GetLspDefinitionsFunction } from "./types.js";
 import { AutocompleteDebouncer } from "./util/AutocompleteDebouncer.js";
@@ -121,6 +121,12 @@ export class CompletionProvider {
       ...config?.tabAutocompleteOptions,
       ...llm.autocompleteOptions,
     };
+
+    // Enable static contextualization if defined.
+    if (config?.experimental?.enableStaticContextualization) {
+      options.experimental_enableStaticContextualization = true;
+    }
+
     return options;
   }
 
@@ -171,7 +177,7 @@ export class CompletionProvider {
       }
 
       const [snippetPayload, workspaceDirs] = await Promise.all([
-        getAllSnippets({
+        getAllSnippetsWithoutRace({
           helper,
           ide: this.ide,
           getDefinitionsFromLsp: this.getDefinitionsFromLsp,
@@ -258,6 +264,10 @@ export class CompletionProvider {
         ...helper.options,
       };
 
+      if (options.experimental_enableStaticContextualization) {
+        outcome.enabledStaticContextualization = true;
+      }
+
       //////////
 
       // Save to cache

core/autocomplete/context/ContextRetrievalService.ts

@@ -2,23 +2,27 @@ import { IDE } from "../..";
 import {
   AutocompleteCodeSnippet,
   AutocompleteSnippetType,
+  AutocompleteStaticSnippet,
 } from "../snippets/types";
 import { HelperVars } from "../util/HelperVars";
 
 import { ImportDefinitionsService } from "./ImportDefinitionsService";
 import { getSymbolsForSnippet } from "./ranking";
 import { RootPathContextService } from "./root-path-context/RootPathContextService";
+import { StaticContextService } from "./static-context/StaticContextService";
 
 export class ContextRetrievalService {
   private importDefinitionsService: ImportDefinitionsService;
   private rootPathContextService: RootPathContextService;
+  private staticContextService: StaticContextService;
 
   constructor(private readonly ide: IDE) {
     this.importDefinitionsService = new ImportDefinitionsService(this.ide);
     this.rootPathContextService = new RootPathContextService(
       this.importDefinitionsService,
       this.ide,
     );
+    this.staticContextService = new StaticContextService(this.ide);
   }
 
   public async getSnippetsFromImportDefinitions(
@@ -71,6 +75,12 @@ export class ContextRetrievalService {
     );
   }
 
+  public async getStaticContextSnippets(
+    helper: HelperVars,
+  ): Promise<AutocompleteStaticSnippet[]> {
+    return this.staticContextService.getContext(helper);
+  }
+
   /**
    * Initialize the import definitions cache for a file.
    * This is normally done automatically when the active text editor changes,