Refactor tool configuration and model provider management to support in-place editing, deletion, and positional insertion

Author: robertpiosik

README.md

@@ -83,12 +83,12 @@ Pair-programming using natural language instructions.
 **🛠️ Intelligent Update** \
 Integrate changes from malformed markdown code blocks.
 
-**🛠️ Code Completions** \
-Get accurate code at cursor from state-of-the-art models.
-
 **🛠️ Prune Context** \
 Remove irrelevant files from the current context selection.
 
+**🛠️ Code Completions** \
+Get accurate code at cursor from state-of-the-art models.
+
 **🛠️ Commit Messages** \
 Meaningful summaries of changes adhering to your style.
 

apps/editor/package.json

@@ -1,7 +1,7 @@
 {
   "name": "gemini-coder",
   "displayName": "Code Web Chat",
-  "description": "CWC is a blazing fast AI coding with chatbots/APIs",
+  "description": "CWC lets you connect VS Code with chatbots",
   "version": "1.770.0",
   "scripts": {
     "build": "npx vsce package --no-dependencies",

apps/editor/src/views/panel/backend/message-handlers/handle-create-preset-group-or-separator.ts

@@ -134,19 +134,51 @@ export const handle_create_preset_group_or_separator = async (
   let insertion_index: number | undefined
 
   if (message.reference_index !== undefined) {
-    const position_quick_pick = await vscode.window.showQuickPick(
-      ['Insert Above', 'Insert Below'],
-      {
-        placeHolder: 'Where to insert?'
+    const position_quick_pick = await new Promise<string | undefined>(
+      (resolve) => {
+        const quick_pick = vscode.window.createQuickPick()
+        quick_pick.items = [
+          { label: 'Insert above' },
+          { label: 'Insert below' }
+        ]
+        quick_pick.title = 'Placement'
+        quick_pick.placeholder = 'Where to insert?'
+        quick_pick.buttons = [
+          {
+            iconPath: new vscode.ThemeIcon('close'),
+            tooltip: 'Close'
+          }
+        ]
+
+        let accepted = false
+        const disposables: vscode.Disposable[] = []
+
+        disposables.push(
+          quick_pick.onDidTriggerButton(() => {
+            quick_pick.hide()
+          }),
+          quick_pick.onDidAccept(() => {
+            accepted = true
+            resolve(quick_pick.selectedItems[0]?.label)
+            quick_pick.hide()
+          }),
+          quick_pick.onDidHide(() => {
+            if (!accepted) resolve(undefined)
+            disposables.forEach((d) => d.dispose())
+            quick_pick.dispose()
+          })
+        )
+
+        quick_pick.show()
       }
     )
     if (!position_quick_pick) return
 
     insertion_index =
-      position_quick_pick === 'Insert Above'
+      position_quick_pick == 'Insert above'
         ? message.reference_index
         : message.reference_index + 1
-  } else if (message.placement === 'top') {
+  } else if (message.placement == 'top') {
     insertion_index = 0
   }
 

apps/editor/src/views/panel/backend/panel-provider.ts

@@ -114,6 +114,8 @@ import { CancelTokenSource } from 'axios'
 import { update_last_used_preset_or_group } from './message-handlers/update-last-used-preset-or-group'
 import { dictionary } from '@shared/constants/dictionary'
 import { DEFAULT_CONTEXT_SIZE_WARNING_THRESHOLD } from '@/constants/values'
+import { upsert_configuration } from '../../utils/upsert-configuration'
+import { delete_configuration } from '../../utils/delete-configuration'
 
 export class PanelProvider implements vscode.WebviewViewProvider {
   private _webview_view: vscode.WebviewView | undefined
@@ -769,6 +771,30 @@ export class PanelProvider implements vscode.WebviewViewProvider {
             message.command == 'SAVE_PRUNE_CONTEXT_INSTRUCTIONS_PREFIX'
           ) {
             await handle_save_prune_context_instructions_prefix(message.prefix)
+          } else if (message.command == 'UPSERT_CONFIGURATION') {
+            const tool_type_map: Record<string, string> = {
+              'edit-context': 'edit-context',
+              'code-at-cursor': 'code-completions',
+              'prune-context': 'prune-context'
+            }
+            await upsert_configuration({
+              context: this.context,
+              tool_type: tool_type_map[message.tool_type] || message.tool_type,
+              configuration_id: message.configuration_id,
+              create_on_top: message.create_on_top,
+              insertion_index: message.insertion_index
+            })
+          } else if (message.command == 'DELETE_CONFIGURATION') {
+            const tool_type_map: Record<string, any> = {
+              'edit-context': 'edit-context',
+              'code-at-cursor': 'code-completions',
+              'prune-context': 'prune-context'
+            }
+            await delete_configuration(
+              this.context,
+              message.configuration_id,
+              tool_type_map[message.api_prompt_type] || message.api_prompt_type
+            )
           } else if (message.command == 'OPEN_EXTERNAL_URL') {
             await handle_open_external_url(message)
           }

apps/editor/src/views/panel/frontend/Main/Main.tsx

@@ -398,10 +398,14 @@ export const Main: React.FC<Props> = (props) => {
     })
   }
 
-  const handle_manage_configurations = () => {
+  const handle_create_configuration = (params?: {
+    create_on_top?: boolean
+    insertion_index?: number
+  }) => {
     post_message(props.vscode, {
-      command: 'MANAGE_CONFIGURATIONS',
-      api_prompt_type: props.api_prompt_type
+      command: 'UPSERT_CONFIGURATION',
+      tool_type: props.api_prompt_type,
+      ...params
     })
   }
 
@@ -435,6 +439,24 @@ export const Main: React.FC<Props> = (props) => {
     }
   }
 
+  const handle_edit_configuration = (id: string) => {
+    if (props.api_prompt_type) {
+      post_message(props.vscode, {
+        command: 'UPSERT_CONFIGURATION',
+        tool_type: props.api_prompt_type,
+        configuration_id: id
+      })
+    }
+  }
+
+  const handle_delete_configuration = (id: string) => {
+    post_message(props.vscode, {
+      command: 'DELETE_CONFIGURATION',
+      api_prompt_type: props.api_prompt_type,
+      configuration_id: id
+    })
+  }
+
   const handle_toggle_pinned_configuration = (id: string) => {
     if (props.api_prompt_type) {
       post_message(props.vscode, {
@@ -738,7 +760,9 @@ export const Main: React.FC<Props> = (props) => {
       on_configuration_click={handle_configuration_click}
       on_configurations_reorder={handle_configurations_reorder}
       on_toggle_pinned_configuration={handle_toggle_pinned_configuration}
-      on_manage_configurations={handle_manage_configurations}
+      on_edit_configuration={handle_edit_configuration}
+      on_delete_configuration={handle_delete_configuration}
+      on_create_configuration={handle_create_configuration}
       on_search_click={handle_search_click}
       on_at_sign_click={handle_at_sign_click}
       on_hash_sign_click={handle_hash_sign_click}

apps/editor/src/views/panel/frontend/Main/MainView/MainView.tsx

@@ -45,7 +45,12 @@ type Props = {
     reordered_configurations: UiConfigurations.Configuration[]
   ) => void
   on_toggle_pinned_configuration: (id: string) => void
-  on_manage_configurations: () => void
+  on_edit_configuration: (id: string) => void
+  on_delete_configuration: (id: string) => void
+  on_create_configuration: (params?: {
+    create_on_top?: boolean
+    insertion_index?: number
+  }) => void
   currently_open_file_path?: string
   current_selection?: SelectionState | null
   chat_history: string[]
@@ -322,8 +327,10 @@ export const MainView: React.FC<Props> = (props) => {
               on_configuration_click={props.on_configuration_click}
               on_reorder={props.on_configurations_reorder}
               on_toggle_pinned={props.on_toggle_pinned_configuration}
+              on_edit={props.on_edit_configuration}
+              on_delete={props.on_delete_configuration}
               selected_configuration_id={props.selected_configuration_id}
-              on_manage_configurations={props.on_manage_configurations}
+              on_create={props.on_create_configuration}
               is_collapsed={props.configurations_collapsed}
               on_toggle_collapsed={props.on_configurations_collapsed_change}
             />

apps/editor/src/views/panel/types/messages.ts

@@ -493,6 +493,20 @@ export interface OpenExternalUrlMessage extends BaseMessage {
   url: string
 }
 
+export interface UpsertConfigurationMessage extends BaseMessage {
+  command: 'UPSERT_CONFIGURATION'
+  tool_type: string
+  configuration_id?: string
+  create_on_top?: boolean
+  insertion_index?: number
+}
+
+export interface DeleteConfigurationMessage extends BaseMessage {
+  command: 'DELETE_CONFIGURATION'
+  api_prompt_type: ApiPromptType
+  configuration_id: string
+}
+
 export type FrontendMessage =
   | GetInstructionsMessage
   | SaveInstructionsMessage
@@ -580,6 +594,8 @@ export type FrontendMessage =
   | GetPruneContextInstructionsPrefixMessage
   | SavePruneContextInstructionsPrefixMessage
   | OpenExternalUrlMessage
+  | UpsertConfigurationMessage
+  | DeleteConfigurationMessage
 
 // === FROM BACKEND TO FRONTEND ===
 export interface InstructionsMessage extends BaseMessage {

apps/editor/src/views/settings/backend/message-handlers/handle-delete-configuration.ts

@@ -1,11 +1,5 @@
-import * as vscode from 'vscode'
 import { SettingsProvider } from '@/views/settings/backend/settings-provider'
-import {
-  ModelProvidersManager,
-  get_tool_config_id,
-  ToolConfig
-} from '@/services/model-providers-manager'
-import { dictionary } from '@shared/constants/dictionary'
+import { delete_configuration } from '../../../utils/delete-configuration'
 
 export const handle_delete_configuration = async (
   provider: SettingsProvider,
@@ -17,92 +11,5 @@ export const handle_delete_configuration = async (
     | 'commit-messages'
     | 'prune-context'
 ): Promise<void> => {
-  const providers_manager = new ModelProvidersManager(provider.context)
-
-  let get_configs: () => Promise<ToolConfig[]>
-  let save_configs: (configs: ToolConfig[]) => Promise<void>
-  let get_default_config: (() => Promise<ToolConfig | undefined>) | undefined
-  let set_default_config:
-    | ((config: ToolConfig | null) => Promise<void>)
-    | undefined
-
-  switch (type) {
-    case 'code-completions':
-      get_configs = () => providers_manager.get_code_completions_tool_configs()
-      save_configs = (c) =>
-        providers_manager.save_code_completions_tool_configs(c)
-      get_default_config = () =>
-        providers_manager.get_default_code_completions_config()
-      set_default_config = (c) =>
-        providers_manager.set_default_code_completions_config(c)
-      break
-    case 'edit-context':
-      get_configs = () => providers_manager.get_edit_context_tool_configs()
-      save_configs = (c) => providers_manager.save_edit_context_tool_configs(c)
-      break
-    case 'intelligent-update':
-      get_configs = () =>
-        providers_manager.get_intelligent_update_tool_configs()
-      save_configs = (c) =>
-        providers_manager.save_intelligent_update_tool_configs(c)
-      get_default_config = () =>
-        providers_manager.get_default_intelligent_update_config()
-      set_default_config = (c) =>
-        providers_manager.set_default_intelligent_update_config(c)
-      break
-    case 'commit-messages':
-      get_configs = () => providers_manager.get_commit_messages_tool_configs()
-      save_configs = (c) =>
-        providers_manager.save_commit_messages_tool_configs(c)
-      get_default_config = () =>
-        providers_manager.get_default_commit_messages_config()
-      set_default_config = (c) =>
-        providers_manager.set_default_commit_messages_config(c)
-      break
-    case 'prune-context':
-      get_configs = () => providers_manager.get_prune_context_tool_configs()
-      save_configs = (c) => providers_manager.save_prune_context_tool_configs(c)
-      get_default_config = () =>
-        providers_manager.get_default_prune_context_config()
-      set_default_config = (c) =>
-        providers_manager.set_default_prune_context_config(c)
-      break
-  }
-
-  const original_configs = await get_configs()
-  const config_to_delete = original_configs.find(
-    (c) => get_tool_config_id(c) === configuration_id
-  )
-  if (!config_to_delete) return
-
-  const confirmation = await vscode.window.showWarningMessage(
-    dictionary.warning_message.PLEASE_CONFIRM,
-    {
-      modal: true,
-      detail: dictionary.warning_message.CONFIRM_DELETE_CONFIGURATION(
-        config_to_delete.model,
-        config_to_delete.provider_name
-      )
-    },
-    'Delete'
-  )
-
-  if (confirmation != 'Delete') {
-    return
-  }
-
-  const updated_configs = original_configs.filter(
-    (c) => get_tool_config_id(c) !== configuration_id
-  )
-  await save_configs(updated_configs)
-
-  if (get_default_config && set_default_config) {
-    const original_default_config = await get_default_config()
-    if (
-      original_default_config &&
-      get_tool_config_id(original_default_config) === configuration_id
-    ) {
-      await set_default_config(null)
-    }
-  }
+  await delete_configuration(provider.context, configuration_id, type)
 }