Separate recommended models from rest to make it more clear

.changeset/model-picker-sections.md

@@ -0,0 +1,5 @@
+---
+"kilo-code": patch
+---
+
+Add section headers to model selection dropdowns for "Recommended models" and "All models"

webview-ui/src/components/kilocode/chat/ModelSelector.tsx

@@ -1,13 +1,13 @@
 import { useMemo } from "react"
-import { SelectDropdown, DropdownOptionType } from "@/components/ui"
+import { SelectDropdown, DropdownOptionType, type DropdownOption } from "@/components/ui"
 import { OPENROUTER_DEFAULT_PROVIDER_NAME, type ProviderSettings } from "@roo-code/types"
 import { vscode } from "@src/utils/vscode"
 import { useAppTranslation } from "@src/i18n/TranslationContext"
 import { cn } from "@src/lib/utils"
 import { prettyModelName } from "../../../utils/prettyModelName"
 import { useProviderModels } from "../hooks/useProviderModels"
 import { getModelIdKey, getSelectedModelId } from "../hooks/useSelectedModel"
-import { usePreferredModels } from "@/components/ui/hooks/kilocode/usePreferredModels"
+import { useGroupedModelIds } from "@/components/ui/hooks/kilocode/usePreferredModels"
 
 interface ModelSelectorProps {
 	currentApiConfigName?: string
@@ -32,15 +32,66 @@ export const ModelSelector = ({
 	const modelIdKey = getModelIdKey({ provider })
 	const isAutocomplete = apiConfiguration.profileType === "autocomplete"
 
-	const modelsIds = usePreferredModels(providerModels)
+	const { preferredModelIds, restModelIds } = useGroupedModelIds(providerModels)
 	const options = useMemo(() => {
-		const missingModelIds = modelsIds.indexOf(selectedModelId) >= 0 ? [] : [selectedModelId]
-		return missingModelIds.concat(modelsIds).map((modelId) => ({
-			value: modelId,
-			label: providerModels[modelId]?.displayName ?? prettyModelName(modelId),
-			type: DropdownOptionType.ITEM,
-		}))
-	}, [modelsIds, providerModels, selectedModelId])
+		const result: DropdownOption[] = []
+
+		// Check if selected model is missing from the lists
+		const allModelIds = [...preferredModelIds, ...restModelIds]
+		const isMissingSelectedModel = selectedModelId && !allModelIds.includes(selectedModelId)
+
+		// Add "Recommended models" section if there are preferred models
+		if (preferredModelIds.length > 0) {
+			result.push({
+				value: "__label_recommended__",
+				label: t("settings:modelPicker.recommendedModels"),
+				type: DropdownOptionType.LABEL,
+			})
+
+			preferredModelIds.forEach((modelId) => {
+				result.push({
+					value: modelId,
+					label: providerModels[modelId]?.displayName ?? prettyModelName(modelId),
+					type: DropdownOptionType.ITEM,
+				})
+			})
+		}
+
+		// Add "All models" section
+		if (restModelIds.length > 0) {
+			result.push({
+				value: "__label_all__",
+				label: t("settings:modelPicker.allModels"),
+				type: DropdownOptionType.LABEL,
+			})
+
+			// Add missing selected model at the top of "All models" if not in any list
+			if (isMissingSelectedModel) {
+				result.push({
+					value: selectedModelId,
+					label: providerModels[selectedModelId]?.displayName ?? prettyModelName(selectedModelId),
+					type: DropdownOptionType.ITEM,
+				})
+			}
+
+			restModelIds.forEach((modelId) => {
+				result.push({
+					value: modelId,
+					label: providerModels[modelId]?.displayName ?? prettyModelName(modelId),
+					type: DropdownOptionType.ITEM,
+				})
+			})
+		} else if (isMissingSelectedModel) {
+			// If there are no rest models but we have a missing selected model, add it
+			result.push({
+				value: selectedModelId,
+				label: providerModels[selectedModelId]?.displayName ?? prettyModelName(selectedModelId),
+				type: DropdownOptionType.ITEM,
+			})
+		}
+
+		return result
+	}, [preferredModelIds, restModelIds, providerModels, selectedModelId, t])
 
 	const disabled = isLoading || isError || isAutocomplete
 

webview-ui/src/components/kilocode/chat/__tests__/ModelSelector.spec.tsx

@@ -14,8 +14,11 @@ vi.mock("@/i18n/TranslationContext", () => ({
 	}),
 }))
 
+// Create a mock function that can be controlled per test
+const mockUseGroupedModelIds = vi.fn()
+
 vi.mock("@/components/ui/hooks/kilocode/usePreferredModels", () => ({
-	usePreferredModels: () => ["model-1", "model-2"],
+	useGroupedModelIds: () => mockUseGroupedModelIds(),
 }))
 
 // Create a mock function that can be controlled per test
@@ -37,9 +40,17 @@ describe("ModelSelector", () => {
 	}
 
 	beforeEach(() => {
-		// Reset mock before each test
+		// Reset mocks before each test
 		mockUseProviderModels.mockReset()
-		// Default mock implementation
+		mockUseGroupedModelIds.mockReset()
+
+		// Default mock implementation for useGroupedModelIds (no preferred models)
+		mockUseGroupedModelIds.mockReturnValue({
+			preferredModelIds: [],
+			restModelIds: ["model-1", "model-2"],
+		})
+
+		// Default mock implementation for useProviderModels
 		mockUseProviderModels.mockReturnValue({
 			provider: "openai",
 			providerModels: {
@@ -188,4 +199,98 @@ describe("ModelSelector", () => {
 		const dropdownTrigger = screen.queryByTestId("dropdown-trigger")
 		expect(dropdownTrigger).not.toBeInTheDocument()
 	})
+
+	describe("preferred models sections", () => {
+		test("builds options with section headers when preferred models exist", () => {
+			// Setup mock to return preferred models
+			mockUseGroupedModelIds.mockReturnValue({
+				preferredModelIds: ["preferred-1", "preferred-2"],
+				restModelIds: ["model-1", "model-2"],
+			})
+
+			mockUseProviderModels.mockReturnValue({
+				provider: "openai",
+				providerModels: {
+					"preferred-1": { displayName: "Preferred Model 1", preferredIndex: 0 },
+					"preferred-2": { displayName: "Preferred Model 2", preferredIndex: 1 },
+					"model-1": { displayName: "Model 1" },
+					"model-2": { displayName: "Model 2" },
+				},
+				providerDefaultModel: "model-1",
+				isLoading: false,
+				isError: false,
+			})
+
+			render(
+				<ModelSelector
+					currentApiConfigName="test-profile"
+					apiConfiguration={{
+						apiProvider: "openai",
+						apiModelId: "model-1",
+					}}
+					fallbackText="Select a model"
+				/>,
+			)
+
+			// Should render the dropdown
+			const dropdownTrigger = screen.getByTestId("dropdown-trigger")
+			expect(dropdownTrigger).toBeInTheDocument()
+		})
+
+		test("does not add section headers when no preferred models exist", () => {
+			// Setup mock with no preferred models
+			mockUseGroupedModelIds.mockReturnValue({
+				preferredModelIds: [],
+				restModelIds: ["model-1", "model-2"],
+			})
+
+			render(
+				<ModelSelector
+					currentApiConfigName="test-profile"
+					apiConfiguration={{
+						apiProvider: "openai",
+						apiModelId: "model-1",
+					}}
+					fallbackText="Select a model"
+				/>,
+			)
+
+			// Should render the dropdown
+			const dropdownTrigger = screen.getByTestId("dropdown-trigger")
+			expect(dropdownTrigger).toBeInTheDocument()
+		})
+
+		test("handles only preferred models without rest models", () => {
+			// Setup mock with only preferred models (edge case)
+			mockUseGroupedModelIds.mockReturnValue({
+				preferredModelIds: ["preferred-1"],
+				restModelIds: [],
+			})
+
+			mockUseProviderModels.mockReturnValue({
+				provider: "openai",
+				providerModels: {
+					"preferred-1": { displayName: "Preferred Model 1", preferredIndex: 0 },
+				},
+				providerDefaultModel: "preferred-1",
+				isLoading: false,
+				isError: false,
+			})
+
+			render(
+				<ModelSelector
+					currentApiConfigName="test-profile"
+					apiConfiguration={{
+						apiProvider: "openai",
+						apiModelId: "preferred-1",
+					}}
+					fallbackText="Select a model"
+				/>,
+			)
+
+			// Should render the dropdown with only preferred models section
+			const dropdownTrigger = screen.getByTestId("dropdown-trigger")
+			expect(dropdownTrigger).toBeInTheDocument()
+		})
+	})
 })

webview-ui/src/components/settings/ModelPicker.tsx

@@ -1,4 +1,4 @@
-import { useState, useCallback, useEffect, useRef, Fragment } from "react" // kilocode_change Fragment
+import { useState, useCallback, useEffect, useRef, useMemo, Fragment } from "react" // kilocode_change Fragment, useMemo
 import { VSCodeLink } from "@vscode/webview-ui-toolkit/react"
 import { Trans } from "react-i18next"
 import { ChevronsUpDown, Check, X, Info } from "lucide-react"
@@ -7,7 +7,7 @@ import type { ProviderSettings, ModelInfo, OrganizationAllowList } from "@roo-co
 
 import { useAppTranslation } from "@src/i18n/TranslationContext"
 import { useSelectedModel } from "@/components/ui/hooks/useSelectedModel"
-import { usePreferredModels } from "@/components/ui/hooks/kilocode/usePreferredModels" // kilocode_change
+import { useGroupedModelIds } from "@/components/ui/hooks/kilocode/usePreferredModels" // kilocode_change
 // import { filterModels } from "./utils/organizationFilters" // kilocode_change: not doing this
 import { cn } from "@src/lib/utils"
 import {
@@ -21,7 +21,6 @@ import {
 	PopoverContent,
 	PopoverTrigger,
 	Button,
-	SelectSeparator, // kilocode_change
 } from "@src/components/ui"
 import { useEscapeKey } from "@src/hooks/useEscapeKey"
 
@@ -88,8 +87,9 @@ export const ModelPicker = ({
 	const selectTimeoutRef = useRef<NodeJS.Timeout | null>(null)
 	const closeTimeoutRef = useRef<NodeJS.Timeout | null>(null)
 
-	// kilocode_change start
-	const modelIds = usePreferredModels(models)
+	// kilocode_change start: Use grouped model IDs for section headers
+	const { preferredModelIds, restModelIds } = useGroupedModelIds(models)
+	const modelIds = useMemo(() => [...preferredModelIds, ...restModelIds], [preferredModelIds, restModelIds])
 	const [isPricingExpanded, setIsPricingExpanded] = useState(false)
 	// kilocode_change end
 
@@ -205,36 +205,51 @@ export const ModelPicker = ({
 										</div>
 									)}
 								</CommandEmpty>
-								<CommandGroup>
-									{/* kilocode_change start */}
-									{modelIds.map((model, i) => {
-										const isPreferred = Number.isInteger(models?.[model]?.preferredIndex)
-										const previousModelWasPreferred = Number.isInteger(
-											models?.[modelIds[i - 1]]?.preferredIndex,
-										)
-										return (
-											<Fragment key={model}>
-												{!isPreferred && previousModelWasPreferred ? <SelectSeparator /> : null}
-												<CommandItem
-													value={model}
-													onSelect={onSelect}
-													data-testid={`model-option-${model}`}
-													className={cn(isPreferred ? "font-semibold" : "")}>
-													<span className="truncate" title={model}>
-														{model}
-													</span>
-													<Check
-														className={cn(
-															"size-4 p-0.5 ml-auto",
-															model === selectedModelId ? "opacity-100" : "opacity-0",
-														)}
-													/>
-												</CommandItem>
-											</Fragment>
-										)
-									})}
-									{/* kilocode_change end */}
-								</CommandGroup>
+								{/* kilocode_change start: Section headers for recommended and all models */}
+								{preferredModelIds.length > 0 && (
+									<CommandGroup heading={t("settings:modelPicker.recommendedModels")}>
+										{preferredModelIds.map((model) => (
+											<CommandItem
+												key={model}
+												value={model}
+												onSelect={onSelect}
+												data-testid={`model-option-${model}`}
+												className="font-semibold">
+												<span className="truncate" title={model}>
+													{model}
+												</span>
+												<Check
+													className={cn(
+														"size-4 p-0.5 ml-auto",
+														model === selectedModelId ? "opacity-100" : "opacity-0",
+													)}
+												/>
+											</CommandItem>
+										))}
+									</CommandGroup>
+								)}
+								{restModelIds.length > 0 && (
+									<CommandGroup heading={t("settings:modelPicker.allModels")}>
+										{restModelIds.map((model) => (
+											<CommandItem
+												key={model}
+												value={model}
+												onSelect={onSelect}
+												data-testid={`model-option-${model}`}>
+												<span className="truncate" title={model}>
+													{model}
+												</span>
+												<Check
+													className={cn(
+														"size-4 p-0.5 ml-auto",
+														model === selectedModelId ? "opacity-100" : "opacity-0",
+													)}
+												/>
+											</CommandItem>
+										))}
+									</CommandGroup>
+								)}
+								{/* kilocode_change end */}
 							</CommandList>
 							{searchValue && !modelIds.includes(searchValue) && (
 								<div className="p-1 border-t border-vscode-input-border">