Allow ACP providers without model lists

This commit is contained in:
Mohamed Boudra
2026-05-19 20:01:56 +07:00
parent f238cbc20c
commit bc7798af28
8 changed files with 353 additions and 80 deletions
@@ -2,6 +2,7 @@ import { describe, expect, it } from "vitest";
import type { AgentModelDefinition } from "@server/server/agent/agent-sdk-types";
import {
buildModelRows,
buildProviderGroups,
buildSelectedTriggerLabel,
filterAndRankModelRows,
matchesSearch,
@@ -24,6 +25,13 @@ describe("combined model selector helpers", () => {
defaultModeId: "auto",
modes: [],
},
{
id: "deepseek-tui",
label: "DeepSeek TUI",
description: "DeepSeek TUI provider",
defaultModeId: "default",
modes: [],
},
];
const claudeModels: AgentModelDefinition[] = [
@@ -112,6 +120,53 @@ describe("combined model selector helpers", () => {
expect(filterAndRankModelRows(rows, "gpt54").map((row) => row.modelId)).toEqual(["gpt-5.4"]);
});
it("includes providers that expose no models", () => {
const rows = buildModelRows(
providerDefinitions,
new Map([
["claude", claudeModels],
["deepseek-tui", []],
]),
);
const groups = buildProviderGroups(
providerDefinitions,
new Map([
["claude", claudeModels],
["deepseek-tui", []],
]),
rows,
"",
);
expect(groups).toEqual([
expect.objectContaining({
providerId: "claude",
hasNoModels: false,
}),
expect.objectContaining({
providerId: "deepseek-tui",
providerLabel: "DeepSeek TUI",
rows: [],
hasNoModels: true,
}),
]);
});
it("matches model-less providers by provider name", () => {
const groups = buildProviderGroups(
providerDefinitions,
new Map([
["claude", claudeModels],
["deepseek-tui", []],
]),
[],
"deepseek",
);
expect(groups.map((group) => group.providerId)).toEqual(["deepseek-tui"]);
});
it("keeps the selected trigger label model-only", () => {
expect(resolveProviderLabel(providerDefinitions, "codex")).toBe("Codex");
expect(buildSelectedTriggerLabel("GPT-5.4")).toBe("GPT-5.4");
@@ -47,9 +47,11 @@ function drillDownRowStyle({
import { getProviderIcon } from "@/components/provider-icons";
import {
buildModelRows,
buildProviderGroups,
buildSelectedTriggerLabel,
filterAndRankModelRows,
resolveProviderLabel,
type SelectorProviderGroup,
type SelectorModelRow,
} from "./combined-model-selector.utils";
@@ -122,31 +124,6 @@ function sortFavoritesFirst(
return [...favorites, ...rest];
}
function groupRowsByProvider(
rows: SelectorModelRow[],
): Array<{ providerId: string; providerLabel: string; rows: SelectorModelRow[] }> {
const grouped = new Map<
string,
{ providerId: string; providerLabel: string; rows: SelectorModelRow[] }
>();
for (const row of rows) {
const existing = grouped.get(row.provider);
if (existing) {
existing.rows.push(row);
continue;
}
grouped.set(row.provider, {
providerId: row.provider,
providerLabel: row.providerLabel,
rows: [row],
});
}
return Array.from(grouped.values());
}
function ModelRow({
row,
isSelected,
@@ -317,29 +294,41 @@ interface GroupProviderButtonProps {
providerId: string;
providerLabel: string;
rowCount: number;
hasNoModels: boolean;
disabled?: boolean;
onDrillDown: (providerId: string, providerLabel: string) => void;
onSelectDefault: (providerId: string) => void;
}
function GroupProviderButton({
providerId,
providerLabel,
rowCount,
hasNoModels,
disabled,
onDrillDown,
onSelectDefault,
}: GroupProviderButtonProps) {
const { theme } = useUnistyles();
const ProvIcon = getProviderIcon(providerId);
const handlePress = useCallback(() => {
if (hasNoModels) {
onSelectDefault(providerId);
return;
}
onDrillDown(providerId, providerLabel);
}, [onDrillDown, providerId, providerLabel]);
}, [hasNoModels, onDrillDown, onSelectDefault, providerId, providerLabel]);
return (
<Pressable onPress={handlePress} style={drillDownRowStyle}>
<Pressable disabled={disabled} onPress={handlePress} style={drillDownRowStyle}>
<ProvIcon size={theme.iconSize.sm} color={theme.colors.foregroundMuted} />
<Text style={styles.drillDownText}>{providerLabel}</Text>
<View style={styles.drillDownTrailing}>
<Text style={styles.drillDownCount}>
{rowCount} {rowCount === 1 ? "model" : "models"}
{hasNoModels ? "Default" : `${rowCount} ${rowCount === 1 ? "model" : "models"}`}
</Text>
<ChevronRight size={theme.iconSize.sm} color={theme.colors.foregroundMuted} />
{hasNoModels ? null : (
<ChevronRight size={theme.iconSize.sm} color={theme.colors.foregroundMuted} />
)}
</View>
</Pressable>
);
@@ -348,9 +337,13 @@ function GroupProviderButton({
function GroupedProviderRows({
groupedRows,
onDrillDown,
onSelectDefault,
canSelectProvider,
}: {
groupedRows: Array<{ providerId: string; providerLabel: string; rows: SelectorModelRow[] }>;
groupedRows: SelectorProviderGroup[];
onDrillDown: (providerId: string, providerLabel: string) => void;
onSelectDefault: (providerId: string) => void;
canSelectProvider: (provider: string) => boolean;
}) {
return (
<View>
@@ -362,7 +355,10 @@ function GroupedProviderRows({
providerId={group.providerId}
providerLabel={group.providerLabel}
rowCount={group.rows.length}
hasNoModels={group.hasNoModels}
disabled={group.hasNoModels && !canSelectProvider(group.providerId)}
onDrillDown={onDrillDown}
onSelectDefault={onSelectDefault}
/>
</View>
);
@@ -371,6 +367,38 @@ function GroupedProviderRows({
);
}
function DefaultProviderRow({
providerId,
isSelected,
disabled,
onSelect,
}: {
providerId: string;
isSelected: boolean;
disabled?: boolean;
onSelect: (provider: string, modelId: string) => void;
}) {
const { theme } = useUnistyles();
const ProviderIcon = getProviderIcon(providerId);
const handlePress = useCallback(() => {
onSelect(providerId, "");
}, [onSelect, providerId]);
const leadingSlot = useMemo(
() => <ProviderIcon size={theme.iconSize.sm} color={theme.colors.foregroundMuted} />,
[ProviderIcon, theme.iconSize.sm, theme.colors.foregroundMuted],
);
return (
<ComboboxItem
label="Default"
selected={isSelected}
disabled={disabled}
onPress={handlePress}
leadingSlot={leadingSlot}
/>
);
}
function ProviderModelRows({
rows,
selectedProvider,
@@ -472,7 +500,16 @@ function SelectorContent({
[favoriteKeys, visibleRows],
);
const allGroupedRows = useMemo(() => groupRowsByProvider(visibleRows), [visibleRows]);
const allGroupedRows = useMemo(
() => buildProviderGroups(providerDefinitions, allProviderModels, visibleRows, normalizedQuery),
[allProviderModels, normalizedQuery, providerDefinitions, visibleRows],
);
const handleSelectDefaultProvider = useCallback(
(providerId: string) => {
onSelect(providerId, "");
},
[onSelect],
);
const hasResults = favoriteRows.length > 0 || allGroupedRows.length > 0;
const emptyState = (
<View style={styles.emptyState}>
@@ -482,6 +519,18 @@ function SelectorContent({
);
if (view.kind === "provider") {
const providerModels = allProviderModels.get(view.providerId);
if (providerModels && providerModels.length === 0 && !normalizedQuery) {
return (
<DefaultProviderRow
providerId={view.providerId}
isSelected={view.providerId === selectedProvider && !selectedModel}
disabled={!canSelectProvider(view.providerId)}
onSelect={onSelect}
/>
);
}
if (visibleRows.length === 0) {
return emptyState;
}
@@ -513,7 +562,12 @@ function SelectorContent({
/>
{allGroupedRows.length > 0 ? (
<GroupedProviderRows groupedRows={allGroupedRows} onDrillDown={onDrillDown} />
<GroupedProviderRows
groupedRows={allGroupedRows}
onDrillDown={onDrillDown}
onSelectDefault={handleSelectDefaultProvider}
canSelectProvider={canSelectProvider}
/>
) : null}
{!hasResults ? emptyState : null}
@@ -598,6 +652,10 @@ export function CombinedModelSelector({
if (!hasSelectedProvider) {
return "Select model";
}
const models = allProviderModels.get(selectedProvider);
if (models && models.length === 0) {
return "Default";
}
return isLoading ? "Loading..." : "Select model";
}
const models = allProviderModels.get(selectedProvider);
@@ -5,6 +5,13 @@ import { compareMatchScores, scoreTextFields } from "@/utils/score-match";
export type SelectorModelRow = FavoriteModelRow;
export interface SelectorProviderGroup {
providerId: string;
providerLabel: string;
rows: SelectorModelRow[];
hasNoModels: boolean;
}
export function resolveProviderLabel(
providerDefinitions: AgentProviderDefinition[],
providerId: string,
@@ -75,3 +82,55 @@ export function filterAndRankModelRows(
return scored.map((entry) => entry.row);
}
export function buildProviderGroups(
providerDefinitions: AgentProviderDefinition[],
allProviderModels: Map<string, AgentModelDefinition[]>,
rows: SelectorModelRow[],
normalizedQuery: string,
): SelectorProviderGroup[] {
const rowsByProvider = new Map<string, SelectorModelRow[]>();
for (const row of rows) {
const providerRows = rowsByProvider.get(row.provider);
if (providerRows) {
providerRows.push(row);
} else {
rowsByProvider.set(row.provider, [row]);
}
}
const groups: SelectorProviderGroup[] = [];
for (const definition of providerDefinitions) {
const providerRows = rowsByProvider.get(definition.id) ?? [];
if (providerRows.length > 0) {
groups.push({
providerId: definition.id,
providerLabel: definition.label,
rows: providerRows,
hasNoModels: false,
});
continue;
}
const models = allProviderModels.get(definition.id);
if (!models || models.length > 0) {
continue;
}
const providerMatches =
!normalizedQuery ||
scoreTextFields(normalizedQuery, [definition.label, definition.id]) !== null;
if (!providerMatches) {
continue;
}
groups.push({
providerId: definition.id,
providerLabel: definition.label,
rows: [],
hasNoModels: true,
});
}
return groups;
}