mirror of
https://github.com/Nezumi-2711/9router.git
synced 2026-09-22 13:38:31 +00:00
feat(ui): add model support for custom providers and improve UX
- Support custom provider models in combos and model selection - Display custom provider names instead of technical IDs - Make model fields readonly (selected via modal) - Only show connected providers (remove fallback) Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -88,22 +88,6 @@ export default function CLIToolsPageClient({ machineId }) {
|
|||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
if (models.length === 0) {
|
|
||||||
Object.entries(PROVIDER_MODELS).forEach(([alias, providerModels]) => {
|
|
||||||
providerModels.forEach(m => {
|
|
||||||
const modelValue = `${alias}/${m.id}`;
|
|
||||||
models.push({
|
|
||||||
value: modelValue,
|
|
||||||
label: `${alias}/${m.id}`,
|
|
||||||
provider: alias,
|
|
||||||
alias: alias,
|
|
||||||
connectionName: alias,
|
|
||||||
modelId: m.id,
|
|
||||||
});
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return models;
|
return models;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
"use client";
|
"use client";
|
||||||
|
|
||||||
import { useState, useEffect } from "react";
|
import { useState, useEffect, useCallback } from "react";
|
||||||
import { Card, Button, Modal, Input, CardSkeleton, ModelSelectModal } from "@/shared/components";
|
import { Card, Button, Modal, Input, CardSkeleton, ModelSelectModal } from "@/shared/components";
|
||||||
import { useCopyToClipboard } from "@/shared/hooks/useCopyToClipboard";
|
import { useCopyToClipboard } from "@/shared/hooks/useCopyToClipboard";
|
||||||
|
import { isOpenAICompatibleProvider, isAnthropicCompatibleProvider } from "@/shared/constants/providers";
|
||||||
|
|
||||||
// Validate combo name: only a-z, A-Z, 0-9, -, _
|
// Validate combo name: only a-z, A-Z, 0-9, -, _
|
||||||
const VALID_NAME_REGEX = /^[a-zA-Z0-9_-]+$/;
|
const VALID_NAME_REGEX = /^[a-zA-Z0-9_-]+$/;
|
||||||
@@ -17,7 +18,7 @@ export default function CombosPage() {
|
|||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
fetchData();
|
fetchData();
|
||||||
}, []);
|
}, []); // eslint-disable-line react-hooks/exhaustive-deps
|
||||||
|
|
||||||
const fetchData = async () => {
|
const fetchData = async () => {
|
||||||
try {
|
try {
|
||||||
@@ -235,21 +236,32 @@ function ComboFormModal({ isOpen, combo, onClose, onSave, activeProviders }) {
|
|||||||
const [saving, setSaving] = useState(false);
|
const [saving, setSaving] = useState(false);
|
||||||
const [nameError, setNameError] = useState("");
|
const [nameError, setNameError] = useState("");
|
||||||
const [modelAliases, setModelAliases] = useState({});
|
const [modelAliases, setModelAliases] = useState({});
|
||||||
|
const [providerNodes, setProviderNodes] = useState([]);
|
||||||
|
|
||||||
// Fetch model aliases when modal opens
|
const fetchModalData = async () => {
|
||||||
useEffect(() => {
|
try {
|
||||||
if (isOpen) {
|
const [aliasesRes, nodesRes] = await Promise.all([
|
||||||
const fetchModelAliases = async () => {
|
fetch("/api/models/alias"),
|
||||||
try {
|
fetch("/api/provider-nodes"),
|
||||||
const res = await fetch("/api/models/alias");
|
]);
|
||||||
const data = await res.json();
|
|
||||||
if (res.ok) setModelAliases(data.aliases || {});
|
if (!aliasesRes.ok || !nodesRes.ok) {
|
||||||
} catch (error) {
|
throw new Error(`Failed to fetch data: aliases=${aliasesRes.status}, nodes=${nodesRes.status}`);
|
||||||
console.log("Error fetching model aliases:", error);
|
}
|
||||||
}
|
|
||||||
};
|
const [aliasesData, nodesData] = await Promise.all([
|
||||||
fetchModelAliases();
|
aliasesRes.json(),
|
||||||
|
nodesRes.json(),
|
||||||
|
]);
|
||||||
|
setModelAliases(aliasesData.aliases || {});
|
||||||
|
setProviderNodes(nodesData.nodes || []);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error fetching modal data:", error);
|
||||||
}
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (isOpen) fetchModalData();
|
||||||
}, [isOpen]);
|
}, [isOpen]);
|
||||||
|
|
||||||
const validateName = (value) => {
|
const validateName = (value) => {
|
||||||
@@ -282,11 +294,20 @@ function ComboFormModal({ isOpen, combo, onClose, onSave, activeProviders }) {
|
|||||||
setModels(models.filter((_, i) => i !== index));
|
setModels(models.filter((_, i) => i !== index));
|
||||||
};
|
};
|
||||||
|
|
||||||
const handleModelChange = (index, value) => {
|
// Format model display name with readable provider name
|
||||||
const newModels = [...models];
|
const formatModelDisplay = useCallback((modelValue) => {
|
||||||
newModels[index] = value;
|
const parts = modelValue.split('/');
|
||||||
setModels(newModels);
|
if (parts.length !== 2) return modelValue;
|
||||||
};
|
|
||||||
|
const [providerId, modelId] = parts;
|
||||||
|
const matchedNode = providerNodes.find(node => node.id === providerId);
|
||||||
|
|
||||||
|
if (matchedNode) {
|
||||||
|
return `${matchedNode.name}/${modelId}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
return modelValue;
|
||||||
|
}, [providerNodes]);
|
||||||
|
|
||||||
const handleMoveUp = (index) => {
|
const handleMoveUp = (index) => {
|
||||||
if (index === 0) return;
|
if (index === 0) return;
|
||||||
@@ -352,14 +373,10 @@ function ComboFormModal({ isOpen, combo, onClose, onSave, activeProviders }) {
|
|||||||
{/* Index badge */}
|
{/* Index badge */}
|
||||||
<span className="text-[10px] font-medium text-text-muted w-3 text-center shrink-0">{index + 1}</span>
|
<span className="text-[10px] font-medium text-text-muted w-3 text-center shrink-0">{index + 1}</span>
|
||||||
|
|
||||||
{/* Model Input */}
|
{/* Model display - show readable name only */}
|
||||||
<input
|
<div className="flex-1 min-w-0 px-1.5 py-0.5 text-xs text-text-main truncate">
|
||||||
type="text"
|
{formatModelDisplay(model)}
|
||||||
value={model}
|
</div>
|
||||||
onChange={(e) => handleModelChange(index, e.target.value)}
|
|
||||||
placeholder="provider/model"
|
|
||||||
className="flex-1 min-w-0 px-1.5 py-0.5 text-xs font-mono bg-transparent border-0 focus:outline-none text-text-main placeholder:text-text-muted/50"
|
|
||||||
/>
|
|
||||||
|
|
||||||
{/* Priority arrows - horizontal, always visible */}
|
{/* Priority arrows - horizontal, always visible */}
|
||||||
<div className="flex items-center gap-0.5">
|
<div className="flex items-center gap-0.5">
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ import { useState, useMemo, useEffect } from "react";
|
|||||||
import PropTypes from "prop-types";
|
import PropTypes from "prop-types";
|
||||||
import Modal from "./Modal";
|
import Modal from "./Modal";
|
||||||
import { getModelsByProviderId, PROVIDER_ID_TO_ALIAS } from "@/shared/constants/models";
|
import { getModelsByProviderId, PROVIDER_ID_TO_ALIAS } from "@/shared/constants/models";
|
||||||
import { OAUTH_PROVIDERS, APIKEY_PROVIDERS } from "@/shared/constants/providers";
|
import { OAUTH_PROVIDERS, APIKEY_PROVIDERS, isOpenAICompatibleProvider, isAnthropicCompatibleProvider } from "@/shared/constants/providers";
|
||||||
|
|
||||||
// Provider order: OAuth first, then API Key (matches dashboard/providers)
|
// Provider order: OAuth first, then API Key (matches dashboard/providers)
|
||||||
const PROVIDER_ORDER = [
|
const PROVIDER_ORDER = [
|
||||||
@@ -23,15 +23,38 @@ export default function ModelSelectModal({
|
|||||||
}) {
|
}) {
|
||||||
const [searchQuery, setSearchQuery] = useState("");
|
const [searchQuery, setSearchQuery] = useState("");
|
||||||
const [combos, setCombos] = useState([]);
|
const [combos, setCombos] = useState([]);
|
||||||
|
const [providerNodes, setProviderNodes] = useState([]);
|
||||||
|
|
||||||
// Fetch combos when modal opens
|
const fetchCombos = async () => {
|
||||||
useEffect(() => {
|
try {
|
||||||
if (isOpen) {
|
const res = await fetch("/api/combos");
|
||||||
fetch("/api/combos")
|
if (!res.ok) throw new Error(`Failed to fetch combos: ${res.status}`);
|
||||||
.then(res => res.json())
|
const data = await res.json();
|
||||||
.then(data => setCombos(data.combos || []))
|
setCombos(data.combos || []);
|
||||||
.catch(() => setCombos([]));
|
} catch (error) {
|
||||||
|
console.error("Error fetching combos:", error);
|
||||||
|
setCombos([]);
|
||||||
}
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (isOpen) fetchCombos();
|
||||||
|
}, [isOpen]);
|
||||||
|
|
||||||
|
const fetchProviderNodes = async () => {
|
||||||
|
try {
|
||||||
|
const res = await fetch("/api/provider-nodes");
|
||||||
|
if (!res.ok) throw new Error(`Failed to fetch provider nodes: ${res.status}`);
|
||||||
|
const data = await res.json();
|
||||||
|
setProviderNodes(data.nodes || []);
|
||||||
|
} catch (error) {
|
||||||
|
console.error("Error fetching provider nodes:", error);
|
||||||
|
setProviderNodes([]);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (isOpen) fetchProviderNodes();
|
||||||
}, [isOpen]);
|
}, [isOpen]);
|
||||||
|
|
||||||
const allProviders = useMemo(() => ({ ...OAUTH_PROVIDERS, ...APIKEY_PROVIDERS }), []);
|
const allProviders = useMemo(() => ({ ...OAUTH_PROVIDERS, ...APIKEY_PROVIDERS }), []);
|
||||||
@@ -40,13 +63,16 @@ export default function ModelSelectModal({
|
|||||||
const groupedModels = useMemo(() => {
|
const groupedModels = useMemo(() => {
|
||||||
const groups = {};
|
const groups = {};
|
||||||
|
|
||||||
// Get active provider IDs
|
// Get all active provider IDs from connections
|
||||||
const activeProviderIds = activeProviders.length > 0
|
const activeConnectionIds = activeProviders.map(p => p.provider);
|
||||||
? activeProviders.map(p => p.provider)
|
|
||||||
: PROVIDER_ORDER;
|
// Only show connected providers (including both standard and custom)
|
||||||
|
const providerIdsToShow = new Set([
|
||||||
|
...activeConnectionIds, // Only connected providers
|
||||||
|
]);
|
||||||
|
|
||||||
// Sort by PROVIDER_ORDER
|
// Sort by PROVIDER_ORDER
|
||||||
const sortedProviderIds = [...activeProviderIds].sort((a, b) => {
|
const sortedProviderIds = [...providerIdsToShow].sort((a, b) => {
|
||||||
const indexA = PROVIDER_ORDER.indexOf(a);
|
const indexA = PROVIDER_ORDER.indexOf(a);
|
||||||
const indexB = PROVIDER_ORDER.indexOf(b);
|
const indexB = PROVIDER_ORDER.indexOf(b);
|
||||||
return (indexA === -1 ? 999 : indexA) - (indexB === -1 ? 999 : indexB);
|
return (indexA === -1 ? 999 : indexA) - (indexB === -1 ? 999 : indexB);
|
||||||
@@ -55,8 +81,8 @@ export default function ModelSelectModal({
|
|||||||
sortedProviderIds.forEach((providerId) => {
|
sortedProviderIds.forEach((providerId) => {
|
||||||
const alias = PROVIDER_ID_TO_ALIAS[providerId] || providerId;
|
const alias = PROVIDER_ID_TO_ALIAS[providerId] || providerId;
|
||||||
const providerInfo = allProviders[providerId] || { name: providerId, color: "#666" };
|
const providerInfo = allProviders[providerId] || { name: providerId, color: "#666" };
|
||||||
|
const isCustomProvider = isOpenAICompatibleProvider(providerId) || isAnthropicCompatibleProvider(providerId);
|
||||||
|
|
||||||
// For passthrough providers, get models from aliases
|
|
||||||
if (providerInfo.passthroughModels) {
|
if (providerInfo.passthroughModels) {
|
||||||
const aliasModels = Object.entries(modelAliases)
|
const aliasModels = Object.entries(modelAliases)
|
||||||
.filter(([, fullModel]) => fullModel.startsWith(`${alias}/`))
|
.filter(([, fullModel]) => fullModel.startsWith(`${alias}/`))
|
||||||
@@ -67,13 +93,43 @@ export default function ModelSelectModal({
|
|||||||
}));
|
}));
|
||||||
|
|
||||||
if (aliasModels.length > 0) {
|
if (aliasModels.length > 0) {
|
||||||
|
// Check for custom name from providerNodes (for compatible providers)
|
||||||
|
const matchedNode = providerNodes.find(node => node.id === providerId);
|
||||||
|
const displayName = matchedNode?.name || providerInfo.name;
|
||||||
|
|
||||||
groups[providerId] = {
|
groups[providerId] = {
|
||||||
name: providerInfo.name,
|
name: displayName,
|
||||||
alias: alias,
|
alias: alias,
|
||||||
color: providerInfo.color,
|
color: providerInfo.color,
|
||||||
models: aliasModels,
|
models: aliasModels,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
} else if (isCustomProvider) {
|
||||||
|
// Match provider node to get custom name
|
||||||
|
const matchedNode = providerNodes.find(node => node.id === providerId);
|
||||||
|
const displayName = matchedNode?.name || providerInfo.name;
|
||||||
|
|
||||||
|
// Get models from modelAliases using providerId (not prefix)
|
||||||
|
// modelAliases format: { alias: "providerId/modelId" }
|
||||||
|
const nodeModels = Object.entries(modelAliases)
|
||||||
|
.filter(([, fullModel]) => fullModel.startsWith(`${providerId}/`))
|
||||||
|
.map(([aliasName, fullModel]) => ({
|
||||||
|
id: fullModel.replace(`${providerId}/`, ""),
|
||||||
|
name: aliasName,
|
||||||
|
value: fullModel,
|
||||||
|
}));
|
||||||
|
|
||||||
|
// Only add to groups if there are models (consistent with other provider types)
|
||||||
|
if (nodeModels.length > 0) {
|
||||||
|
groups[providerId] = {
|
||||||
|
name: displayName,
|
||||||
|
alias: matchedNode?.prefix || providerId,
|
||||||
|
color: providerInfo.color,
|
||||||
|
models: nodeModels,
|
||||||
|
isCustom: true,
|
||||||
|
hasModels: true,
|
||||||
|
};
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
const models = getModelsByProviderId(providerId);
|
const models = getModelsByProviderId(providerId);
|
||||||
if (models.length > 0) {
|
if (models.length > 0) {
|
||||||
@@ -92,7 +148,7 @@ export default function ModelSelectModal({
|
|||||||
});
|
});
|
||||||
|
|
||||||
return groups;
|
return groups;
|
||||||
}, [activeProviders, modelAliases, allProviders]);
|
}, [activeProviders, modelAliases, allProviders, providerNodes]);
|
||||||
|
|
||||||
// Filter combos by search query
|
// Filter combos by search query
|
||||||
const filteredCombos = useMemo(() => {
|
const filteredCombos = useMemo(() => {
|
||||||
@@ -112,11 +168,12 @@ export default function ModelSelectModal({
|
|||||||
const matchedModels = group.models.filter(
|
const matchedModels = group.models.filter(
|
||||||
(m) =>
|
(m) =>
|
||||||
m.name.toLowerCase().includes(query) ||
|
m.name.toLowerCase().includes(query) ||
|
||||||
m.id.toLowerCase().includes(query) ||
|
m.id.toLowerCase().includes(query)
|
||||||
group.name.toLowerCase().includes(query)
|
|
||||||
);
|
);
|
||||||
|
|
||||||
if (matchedModels.length > 0) {
|
const providerNameMatches = group.name.toLowerCase().includes(query);
|
||||||
|
|
||||||
|
if (matchedModels.length > 0 || providerNameMatches) {
|
||||||
filtered[providerId] = {
|
filtered[providerId] = {
|
||||||
...group,
|
...group,
|
||||||
models: matchedModels,
|
models: matchedModels,
|
||||||
@@ -210,7 +267,6 @@ export default function ModelSelectModal({
|
|||||||
</span>
|
</span>
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{/* Models as wrap chips - compact */}
|
|
||||||
<div className="flex flex-wrap gap-1.5">
|
<div className="flex flex-wrap gap-1.5">
|
||||||
{group.models.map((model) => {
|
{group.models.map((model) => {
|
||||||
const isSelected = selectedModel === model.value;
|
const isSelected = selectedModel === model.value;
|
||||||
|
|||||||
Reference in New Issue
Block a user