Preview (#541)

* Strong typing (#533)

* refactor: :recycle: get rid of continuedev.src.continuedev structure

* refactor: :recycle: switching back to server folder

* feat: :sparkles: make config.py imports shorter

* feat: :bookmark: publish as pre-release vscode extension

* refactor: :recycle: refactor and add more completion params to ui

* build: :building_construction: download from preview S3

* fix: :bug: fix paths

* fix: :green_heart: package:pre-release

* ci: :green_heart: more time for tests

* fix: :green_heart: fix build scripts

* fix: :bug: fix import in run.py

* fix: :bookmark: update version to try again

* ci: 💚 Update package.json version [skip ci]

* refactor: :fire: don't check for old extensions version

* fix: :bug: small bug fixes

* fix: :bug: fix config.py import paths

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: platform-specific builds test #1

* feat: :green_heart: ship with binary

* fix: :green_heart: fix copy statement to include.exe for windows

* fix: :green_heart: cd extension before packaging

* chore: :loud_sound: count tokens generated

* fix: :green_heart: remove npm_config_arch

* fix: :green_heart: publish as pre-release!

* chore: :bookmark: update version

* perf: :green_heart: hardcode distro paths

* fix: :bug: fix yaml syntax error

* chore: :bookmark: update version

* fix: :green_heart: update permissions and version

* feat: :bug: kill old server if needed

* feat: :lipstick: update marketplace icon for pre-release

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: auto-reload for config.py

* feat: :wrench: update default config.py imports

* feat: :sparkles: codelens in config.py

* feat: :sparkles: select model param count from UI

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: more model options, ollama error handling

* perf: :zap: don't show server loading immediately

* fix: :bug: fixing small UI details

* ci: 💚 Update package.json version [skip ci]

* feat: :rocket: headers param on LLM class

* fix: :bug: fix headers for openai.;y

* feat: :sparkles: highlight code on cmd+shift+L

* ci: 💚 Update package.json version [skip ci]

* feat: :lipstick: sticky top bar in gui.tsx

* fix: :loud_sound: websocket logging and horizontal scrollbar

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: allow AzureOpenAI Service through GGML

* ci: 💚 Update package.json version [skip ci]

* fix: :bug: fix automigration

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: upload binaries in ci, download apple silicon

* chore: :fire: remove notes

* fix: :green_heart: use curl to download binary

* fix: :green_heart: set permissions on apple silicon binary

* fix: :green_heart: testing

* fix: :green_heart: cleanup file

* fix: :green_heart: fix preview.yaml

* fix: :green_heart: only upload once per binary

* fix: :green_heart: install rosetta

* ci: :green_heart: download binary after tests

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: prepare ci for merge to main

---------

Co-authored-by: GitHub Action <action@github.com>

1 files changed, 62 insertions, 0 deletions

diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index ec03f73e..6f36a6bd 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -4,6 +4,7 @@ import * as path from "path";
 import * as os from "os";
 import { DIFF_DIRECTORY, diffManager } from "../diffs";
 import { getMetaKeyLabel } from "../util/util";
+import { debugPanelWebview } from "../debugPanel";
 class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
   public provideCodeLenses(
     document: vscode.TextDocument,
@@ -82,8 +83,61 @@ class DiffViewerCodeLensProvider implements vscode.CodeLensProvider {
   }
 }
 
+class ConfigPyCodeLensProvider implements vscode.CodeLensProvider {
+  public provideCodeLenses(
+    document: vscode.TextDocument,
+    _: vscode.CancellationToken
+  ): vscode.CodeLens[] | Thenable<vscode.CodeLens[]> {
+    const codeLenses: vscode.CodeLens[] = [];
+
+    if (
+      !document.uri.fsPath.endsWith(".continue/config.py") &&
+      !document.uri.fsPath.endsWith(".continue\\config.py")
+    ) {
+      return codeLenses;
+    }
+
+    const lines = document.getText().split(os.EOL);
+    const lineOfModels = lines.findIndex((line) =>
+      line.includes("models=Models(")
+    );
+
+    if (lineOfModels >= 0) {
+      const range = new vscode.Range(lineOfModels, 0, lineOfModels + 1, 0);
+      codeLenses.push(
+        new vscode.CodeLens(range, {
+          title: `+ Add a Model`,
+          command: "continue.addModel",
+        })
+      );
+    }
+
+    const lineOfSystemMessage = lines.findIndex((line) =>
+      line.replace(" ", "").includes("config=ContinueConfig(")
+    );
+
+    if (lineOfSystemMessage >= 0) {
+      const range = new vscode.Range(
+        lineOfSystemMessage,
+        0,
+        lineOfSystemMessage + 1,
+        0
+      );
+      codeLenses.push(
+        new vscode.CodeLens(range, {
+          title: `✏️ Edit in UI`,
+          command: "continue.openSettingsUI",
+        })
+      );
+    }
+
+    return codeLenses;
+  }
+}
+
 let diffsCodeLensDisposable: vscode.Disposable | undefined = undefined;
 let suggestionsCodeLensDisposable: vscode.Disposable | undefined = undefined;
+let configPyCodeLensDisposable: vscode.Disposable | undefined = undefined;
 
 export function registerAllCodeLensProviders(context: vscode.ExtensionContext) {
   if (suggestionsCodeLensDisposable) {
@@ -92,6 +146,9 @@ export function registerAllCodeLensProviders(context: vscode.ExtensionContext) {
   if (diffsCodeLensDisposable) {
     diffsCodeLensDisposable.dispose();
   }
+  if (configPyCodeLensDisposable) {
+    configPyCodeLensDisposable.dispose();
+  }
   suggestionsCodeLensDisposable = vscode.languages.registerCodeLensProvider(
     "*",
     new SuggestionsCodeLensProvider()
@@ -100,6 +157,11 @@ export function registerAllCodeLensProviders(context: vscode.ExtensionContext) {
     "*",
     new DiffViewerCodeLensProvider()
   );
+  configPyCodeLensDisposable = vscode.languages.registerCodeLensProvider(
+    "*",
+    new ConfigPyCodeLensProvider()
+  );
   context.subscriptions.push(suggestionsCodeLensDisposable);
   context.subscriptions.push(diffsCodeLensDisposable);
+  context.subscriptions.push(configPyCodeLensDisposable);
 }