feat: hide Kilo Gateway models that may train on prompts

.changeset/hide-prompt-training-models.md

@@ -0,0 +1,6 @@
+---
+"kilo-code": patch
+"@kilocode/cli": patch
+---
+
+Add a shared model setting to hide Kilo Gateway models that may train on your prompts across Kilo clients.

packages/kilo-vscode/src/KiloProvider.ts

@@ -2397,7 +2397,8 @@ export class KiloProvider implements vscode.WebviewViewProvider, TelemetryProper
     const refreshProviders =
       partial.provider !== undefined ||
       partial.disabled_providers !== undefined ||
-      partial.enabled_providers !== undefined
+      partial.enabled_providers !== undefined ||
+      partial.hide_prompt_training_models !== undefined
     const refreshAgents =
       partial.default_agent !== undefined ||
       partial.agent !== undefined ||

packages/kilo-vscode/tests/unit/kilo-provider-indexing-refresh.test.ts

@@ -109,6 +109,21 @@ describe("KiloProvider indexing refresh", () => {
     expect(indexing).toBe(0)
   })
 
+  it("refreshes providers when prompt-training model visibility changes", async () => {
+    const conn = createConnection()
+    const provider = new KiloProvider({} as never, conn.service as never)
+    const internal = provider as unknown as Internals
+    let calls = 0
+    internal.connectionState = "connected"
+    internal.fetchAndSendProviders = async () => {
+      calls += 1
+    }
+
+    await internal.handleUpdateConfig({ hide_prompt_training_models: true })
+
+    expect(calls).toBe(1)
+  })
+
   it("passes scoped unset paths to the config overlay endpoint", async () => {
     const conn = createConnection()
     const provider = new KiloProvider({} as never, conn.service as never)

packages/kilo-vscode/webview-ui/src/components/settings/ModelsTab.tsx

@@ -1,5 +1,6 @@
 import { Component, For, Show, createMemo } from "solid-js"
 import { Card } from "@kilocode/kilo-ui/card"
+import { Switch } from "@kilocode/kilo-ui/switch"
 import { useConfig } from "../../context/config"
 import { useLanguage } from "../../context/language"
 import { useProvider } from "../../context/provider"
@@ -154,7 +155,6 @@ const ModelsTab: Component = () => {
         <SettingsRow
           title={language.t("settings.autocomplete.model.title")}
           description={language.t("settings.autocomplete.model.description")}
-          last
         >
           <ModelSelectorBase
             value={getAutocompleteSelection(autocompleteProvider(), autocompleteModel())}
@@ -168,6 +168,19 @@ const ModelsTab: Component = () => {
             description={language.t("settings.autocomplete.model.description")}
           />
         </SettingsRow>
+        <SettingsRow
+          title={language.t("settings.models.hidePromptTraining.title")}
+          description={language.t("settings.models.hidePromptTraining.description")}
+          last
+        >
+          <Switch
+            checked={config().hide_prompt_training_models === true}
+            onChange={(checked: boolean) => updateConfig({ hide_prompt_training_models: checked })}
+            hideLabel
+          >
+            {language.t("settings.models.hidePromptTraining.title")}
+          </Switch>
+        </SettingsRow>
       </Card>
 
       <h4 style={{ "margin-top": "24px", "margin-bottom": "8px" }}>{language.t("settings.providers.modeModels")}</h4>

packages/kilo-vscode/webview-ui/src/components/settings/settings-io.ts

@@ -37,6 +37,7 @@ export const KNOWN_KEYS: ReadonlyArray<string> = [
   "layout",
   "auto_collapse_reasoning",
   "terminal_command_display",
+  "hide_prompt_training_models",
   "indexing",
   "experimental",
 ]

packages/kilo-vscode/webview-ui/src/i18n/en.ts

@@ -1504,6 +1504,9 @@ export const dict = {
   "settings.providers.subagentModel.title": "Subagent Model",
   "settings.providers.subagentModel.description":
     "Default model and reasoning effort for task-tool subagents. Leave unset to inherit the calling agent's model.",
+  "settings.models.hidePromptTraining.title": "Hide Prompt-Training Models",
+  "settings.models.hidePromptTraining.description":
+    "Hide Kilo Gateway models whose providers may use your prompts for training.",
   "settings.providers.modeModels": "Model per Mode",
   "settings.providers.modeModels.description":
     "Override the default model for specific modes. If not set, the global default model is used.",