refactor(vscode): share autocomplete FIM model selection

packages/kilo-vscode/src/services/autocomplete/AutocompleteServiceManager.ts

@@ -11,24 +11,16 @@ import { disposeLog } from "./next-edit/log"
 import { NextEditSuggestionManager } from "./next-edit/NextEditSuggestionManager"
 import { toAllowedMercuryRecentSnippets } from "./next-edit/recentSnippetsAdapter"
 import type { KiloConnectionService } from "../cli-backend"
-import { hasValidCredentials } from "./fim"
-import {
-  DEFAULT_AUTOCOMPLETE_MODEL,
-  getAutocompleteModel,
-  getAutocompleteModelById,
-} from "../../shared/autocomplete-models"
+import { hasValidCredentials, fimModel as notebookModel } from "./fim"
+import { DEFAULT_AUTOCOMPLETE_MODEL, getAutocompleteModel } from "../../shared/autocomplete-models"
 
 const CONFIG_SECTION = "kilo-code.new.autocomplete"
 
 export function selector(kind: "classic" | "next-edit"): vscode.DocumentSelector {
   return kind === "classic" ? [{ scheme: "file" }, { scheme: "vscode-notebook-cell" }] : [{ scheme: "file" }]
 }
 
-export function notebookModel(provider?: string, model?: string) {
-  const info = getAutocompleteModel(provider, model)
-  if (info.kind !== "edit") return info
-  return getAutocompleteModelById(info.fimModelID)
-}
+export { notebookModel }
 
 export interface AutocompleteServiceSettings {
   enableAutoTrigger?: boolean

packages/kilo-vscode/src/services/autocomplete/chat-autocomplete/ChatTextAreaAutocomplete.ts

@@ -6,8 +6,7 @@ import { postprocessAutocompleteSuggestion } from "../classic-auto-complete/usel
 import { VisibleCodeTracker } from "../context/VisibleCodeTracker"
 import { FileIgnoreController } from "../shims/FileIgnoreController"
 import type { KiloConnectionService } from "../../cli-backend"
-import { generateFim, hasValidCredentials } from "../fim"
-import { getAutocompleteModel, getAutocompleteModelById } from "../../../shared/autocomplete-models"
+import { generateFim, hasValidCredentials, fimModel as getChatAutocompleteModel } from "../fim"
 import { finalizeChatSuggestion, buildChatPrefix } from "./chat-autocomplete-utils"
 
 interface ChatCompletionRequestMessage {
@@ -20,11 +19,7 @@ interface ChatCompletionResponseSender {
   postMessage(message: { type: "chatCompletionResult"; text: string; requestId: string }): void
 }
 
-export function getChatAutocompleteModel(provider?: string, model?: string) {
-  const info = getAutocompleteModel(provider, model)
-  if (info.kind !== "edit") return info
-  return getAutocompleteModelById(info.fimModelID)
-}
+export { getChatAutocompleteModel }
 
 /**
  * Chat textarea autocomplete with cached per-request objects.

packages/kilo-vscode/src/services/autocomplete/fim.ts

@@ -1,9 +1,15 @@
 import { ResponseMetaData } from "./types"
 import type { KiloConnectionService } from "../cli-backend"
-import { getAutocompleteModelById } from "../../shared/autocomplete-models"
+import { getAutocompleteModel, getAutocompleteModelById } from "../../shared/autocomplete-models"
 
 const FIM_MAX_TOKENS = 256
 
+export function fimModel(provider?: string, model?: string) {
+  const info = getAutocompleteModel(provider, model)
+  if (info.kind !== "edit") return info
+  return getAutocompleteModelById(info.fimModelID)
+}
+
 /**
  * Generate a FIM (Fill-in-the-Middle) completion via the CLI backend.
  * Uses the SDK's kilo.fim() SSE endpoint which handles auth and streaming.