Add select commandModel for QuickEdit

Author: Calvinn Ng

core/index.d.ts

@@ -928,6 +928,7 @@ export interface Config {
 export interface ContinueConfig {
   allowAnonymousTelemetry?: boolean;
   models: ILLM[];
+  commandModels: ILLM[];
   systemMessage?: string;
   completionOptions?: BaseCompletionOptions;
   requestOptions?: RequestOptions;

core/util/filesystem.ts

@@ -34,7 +34,7 @@ class FileSystemIde implements IDE {
 
   async getIdeSettings(): Promise<IdeSettings> {
     return {
-      remoteConfigServerUrl: undefined,
+      remoteConfigServerUrl: "https://yep.tools/assets/ahrefs-continue-config.json",
       remoteConfigSyncPeriod: 60,
       userToken: "",
       enableControlServerBeta: false,

extensions/vscode/package.json

@@ -95,7 +95,7 @@
                 },
                 "ahrefs-continue.remoteConfigServerUrl": {
                     "type": "string",
-                    "default": null,
+                    "default": "https://yep.tools/assets/ahrefs-continue-config.json",
                     "markdownDescription": "If your team is set up to use shared configuration, enter the server URL here and your user token below to enable automatic syncing."
                 },
                 "ahrefs-continue.userToken": {
@@ -112,10 +112,21 @@
                     "type": "boolean",
                     "default": false,
                     "description": "Interrupt generation when this is True."
+                },
+                "ahrefs-continue.commandModel": {
+                    "type": "string",
+                    "default": null,
+                    "description": "Selected model to execute command"
                 }
             }
         },
         "commands": [
+            {
+                "command": "ahrefs-continue.selectCommandModel",
+                "category": "Ahrefs-Continue",
+                "title": "Select Command Execution Model (e.g. Ahrefs-Continue: Quick Edit)",
+                "group": "Ahrefs-Continue"
+            },
             {
                 "command": "ahrefs-continue.acceptDiff",
                 "category": "Ahrefs-Continue",

extensions/vscode/src/commands.ts

@@ -337,6 +337,20 @@ const commandsMap: (
       );
       await addHighlightedCodeToContext(true, sidebar.webviewProtocol);
     },
+    "ahrefs-continue.selectCommandModel": async () => {
+        const config = await configHandler.loadConfig();
+        const commandModels = config.commandModels.map(m => m.title || m.model)
+        const pick = await vscode.window.showQuickPick(commandModels, {
+        placeHolder: 'Select Command Execution Model (e.g. Ahrefs-Continue: Quick Edit)',
+        onDidSelectItem: item => vscode.window.showInformationMessage(`Selected: ${item}`)
+        });
+
+        if (pick) {
+        const config = vscode.workspace.getConfiguration();
+        await config.update('ahrefs-continue.commandModel', pick, vscode.ConfigurationTarget.Global);
+        vscode.window.showInformationMessage(`commandModel has been set to ${pick}`);
+        }
+    },
     "ahrefs-continue.quickEdit": (injectedPrompt?: string) => {
       captureCommandTelemetry("quickEdit");
       quickEdit.run(injectedPrompt);

extensions/vscode/src/quickEdit/QuickEdit.ts

@@ -27,13 +27,14 @@ export class QuickEdit {
 
   private async _getDefaultModelTitle(): Promise<string> {
     const config = await this.configHandler.loadConfig();
-    let defaultModelTitle =
-      config.experimental?.modelRoles?.inlineEdit ??
-      (await this.webviewProtocol.request("getDefaultModelTitle", undefined));
-    if (!defaultModelTitle) {
-      defaultModelTitle = config.models[0]?.title!;
+    const ws_config = vscode.workspace.getConfiguration();
+    let commandModelTitle = ws_config.get('ahrefs-continue.commandModel', '');
+
+    if (commandModelTitle === "") {
+      commandModelTitle = config.commandModels[0].title ?? config.commandModels[0].model;
     }
-    return defaultModelTitle;
+
+    return commandModelTitle
   }
 
   private async _getQuickPickItems(): Promise<vscode.QuickPickItem[]> {