✨ disableInFiles option for autocomplete

pzaback · Apr 22, 2024 · f05df63 · f05df63
1 parent 1cb0455
commit f05df63
Show file tree

Hide file tree

Showing 7 changed files with 55 additions and 9 deletions.
diff --git a/core/autocomplete/completionProvider.ts b/core/autocomplete/completionProvider.ts
@@ -1,4 +1,6 @@
 import Handlebars from "handlebars";
+import ignore from "ignore";
+import path from "path";
 import { v4 as uuidv4 } from "uuid";
 import { IDE, ILLM, Position, TabAutocompleteOptions } from "..";
 import { RangeInFileWithContents } from "../commands/util";
@@ -367,13 +369,6 @@ export class CompletionProvider {
     input: AutocompleteInput,
     token: AbortSignal | undefined,
   ): Promise<AutocompleteOutcome | undefined> {
-    // Create abort signal if not given
-    if (!token) {
-      const controller = new AbortController();
-      token = controller.signal;
-      this._abortControllers.set(input.completionId, controller);
-    }
-
     try {
       // Debounce
       const uuid = uuidv4();
@@ -385,6 +380,36 @@ export class CompletionProvider {
         ...config.tabAutocompleteOptions,
       };
 
+      // Check whether autocomplete is disabled for this file
+      if (options.disableInFiles) {
+        // Relative path needed for `ignore`
+        const workspaceDirs = await this.ide.getWorkspaceDirs();
+        let filepath = input.filepath;
+        for (const workspaceDir of workspaceDirs) {
+          if (filepath.startsWith(workspaceDir)) {
+            filepath = path.relative(workspaceDir, filepath);
+            break;
+          }
+        }
+
+        // Worst case we can check filetype glob patterns
+        if (filepath === input.filepath) {
+          filepath = getBasename(filepath);
+        }
+
+        const pattern = ignore().add(options.disableInFiles);
+        if (pattern.ignores(filepath)) {
+          return undefined;
+        }
+      }
+
+      // Create abort signal if not given
+      if (!token) {
+        const controller = new AbortController();
+        token = controller.signal;
+        this._abortControllers.set(input.completionId, controller);
+      }
+
       // Allow disabling autocomplete from config.json
       if (options.disable) {
         return undefined;

diff --git a/core/index.d.ts b/core/index.d.ts
@@ -550,8 +550,8 @@ export type ModelName =
   | "claude-3-haiku-20240307"
   | "claude-2.1"
   // Cohere
-  |"command-r"
-  |"command-r-plus"
+  | "command-r"
+  | "command-r-plus"
   // Gemini
   | "gemini-pro"
   | "gemini-1.5-pro-latest"
@@ -677,6 +677,7 @@ export interface TabAutocompleteOptions {
   useCache: boolean;
   onlyMyCode: boolean;
   useOtherFiles: boolean;
+  disableInFiles?: string[];
 }
 
 export interface ContinueUIConfig {

diff --git a/core/util/parameters.ts b/core/util/parameters.ts
@@ -16,6 +16,7 @@ export const DEFAULT_AUTOCOMPLETE_OPTS: TabAutocompleteOptions = {
   useCache: true,
   onlyMyCode: true,
   useOtherFiles: false,
+  disableInFiles: undefined,
 };
 
 export const RETRIEVAL_PARAMS = {

diff --git a/docs/static/schemas/config.json b/docs/static/schemas/config.json
@@ -1821,6 +1821,10 @@
             "useOtherFiles": {
               "type": "boolean",
               "description": "Defaults to true. If set to false, Continue will not attempt to include snippets from other files."
+            },
+            "disableInFiles": {
+              "type": "boolean",
+              "description": "A list of files / glob patterns in which to disable tab autocomplete. For example, *.csv if you'd like to disable autocomplete in .csv files."
             }
           },
           "required": []

diff --git a/extensions/intellij/src/main/resources/config_schema.json b/extensions/intellij/src/main/resources/config_schema.json
@@ -1821,6 +1821,10 @@
             "useOtherFiles": {
               "type": "boolean",
               "description": "Defaults to true. If set to false, Continue will not attempt to include snippets from other files."
+            },
+            "disableInFiles": {
+              "type": "boolean",
+              "description": "A list of files / glob patterns in which to disable tab autocomplete. For example, *.csv if you'd like to disable autocomplete in .csv files."
             }
           },
           "required": []

diff --git a/extensions/vscode/config_schema.json b/extensions/vscode/config_schema.json
@@ -1821,6 +1821,13 @@
             "useOtherFiles": {
               "type": "boolean",
               "description": "Defaults to true. If set to false, Continue will not attempt to include snippets from other files."
+            },
+            "disableInFiles": {
+              "description": "A list of files / glob patterns in which to disable tab autocomplete. For example, *.csv if you'd like to disable autocomplete in .csv files.",
+              "type": "array",
+              "items": {
+                "type": "string"
+              }
             }
           },
           "required": []

diff --git a/extensions/vscode/continue_rc_schema.json b/extensions/vscode/continue_rc_schema.json
@@ -2055,6 +2055,10 @@
             "useOtherFiles": {
               "type": "boolean",
               "description": "Defaults to true. If set to false, Continue will not attempt to include snippets from other files."
+            },
+            "disableInFiles": {
+              "type": "boolean",
+              "description": "A list of files / glob patterns in which to disable tab autocomplete. For example, *.csv if you'd like to disable autocomplete in .csv files."
             }
           },
           "required": []