mirror of
https://github.com/harivansh-afk/sandbox-agent.git
synced 2026-04-15 07:04:48 +00:00
feat: model list (#96)
This commit is contained in:
parent
6a3345b954
commit
b74539172b
24 changed files with 1174 additions and 52 deletions
|
|
@ -3,6 +3,7 @@ import {
|
|||
SandboxAgentError,
|
||||
SandboxAgent,
|
||||
type AgentInfo,
|
||||
type AgentModelInfo,
|
||||
type AgentModeInfo,
|
||||
type PermissionEventData,
|
||||
type QuestionEventData,
|
||||
|
|
@ -89,6 +90,8 @@ export default function App() {
|
|||
|
||||
const [agents, setAgents] = useState<AgentInfo[]>([]);
|
||||
const [modesByAgent, setModesByAgent] = useState<Record<string, AgentModeInfo[]>>({});
|
||||
const [modelsByAgent, setModelsByAgent] = useState<Record<string, AgentModelInfo[]>>({});
|
||||
const [defaultModelByAgent, setDefaultModelByAgent] = useState<Record<string, string>>({});
|
||||
const [sessions, setSessions] = useState<SessionInfo[]>([]);
|
||||
const [agentsLoading, setAgentsLoading] = useState(false);
|
||||
const [agentsError, setAgentsError] = useState<string | null>(null);
|
||||
|
|
@ -96,6 +99,8 @@ export default function App() {
|
|||
const [sessionsError, setSessionsError] = useState<string | null>(null);
|
||||
const [modesLoadingByAgent, setModesLoadingByAgent] = useState<Record<string, boolean>>({});
|
||||
const [modesErrorByAgent, setModesErrorByAgent] = useState<Record<string, string | null>>({});
|
||||
const [modelsLoadingByAgent, setModelsLoadingByAgent] = useState<Record<string, boolean>>({});
|
||||
const [modelsErrorByAgent, setModelsErrorByAgent] = useState<Record<string, string | null>>({});
|
||||
|
||||
const [agentId, setAgentId] = useState("claude");
|
||||
const [agentMode, setAgentMode] = useState("");
|
||||
|
|
@ -252,10 +257,14 @@ export default function App() {
|
|||
stopTurnStream();
|
||||
setAgents([]);
|
||||
setSessions([]);
|
||||
setModelsByAgent({});
|
||||
setDefaultModelByAgent({});
|
||||
setAgentsLoading(false);
|
||||
setSessionsLoading(false);
|
||||
setAgentsError(null);
|
||||
setSessionsError(null);
|
||||
setModelsLoadingByAgent({});
|
||||
setModelsErrorByAgent({});
|
||||
};
|
||||
|
||||
const refreshAgents = async () => {
|
||||
|
|
@ -268,6 +277,7 @@ export default function App() {
|
|||
for (const agent of agentList) {
|
||||
if (agent.installed) {
|
||||
loadModes(agent.id);
|
||||
loadModels(agent.id);
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
|
|
@ -314,6 +324,29 @@ export default function App() {
|
|||
}
|
||||
};
|
||||
|
||||
const loadModels = async (targetId: string) => {
|
||||
setModelsLoadingByAgent((prev) => ({ ...prev, [targetId]: true }));
|
||||
setModelsErrorByAgent((prev) => ({ ...prev, [targetId]: null }));
|
||||
try {
|
||||
const data = await getClient().getAgentModels(targetId);
|
||||
const models = data.models ?? [];
|
||||
setModelsByAgent((prev) => ({ ...prev, [targetId]: models }));
|
||||
if (data.defaultModel) {
|
||||
setDefaultModelByAgent((prev) => ({ ...prev, [targetId]: data.defaultModel! }));
|
||||
} else {
|
||||
setDefaultModelByAgent((prev) => {
|
||||
const next = { ...prev };
|
||||
delete next[targetId];
|
||||
return next;
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
setModelsErrorByAgent((prev) => ({ ...prev, [targetId]: "Unable to load models." }));
|
||||
} finally {
|
||||
setModelsLoadingByAgent((prev) => ({ ...prev, [targetId]: false }));
|
||||
}
|
||||
};
|
||||
|
||||
const sendMessage = async () => {
|
||||
const prompt = message.trim();
|
||||
if (!prompt || !sessionId || turnStreaming) return;
|
||||
|
|
@ -825,6 +858,12 @@ export default function App() {
|
|||
}
|
||||
}, [connected, agentId]);
|
||||
|
||||
useEffect(() => {
|
||||
if (connected && agentId && !modelsByAgent[agentId]) {
|
||||
loadModels(agentId);
|
||||
}
|
||||
}, [connected, agentId]);
|
||||
|
||||
useEffect(() => {
|
||||
const modes = modesByAgent[agentId];
|
||||
if (modes && modes.length > 0 && !agentMode) {
|
||||
|
|
@ -836,6 +875,15 @@ export default function App() {
|
|||
const activeModes = modesByAgent[agentId] ?? [];
|
||||
const modesLoading = modesLoadingByAgent[agentId] ?? false;
|
||||
const modesError = modesErrorByAgent[agentId] ?? null;
|
||||
const modelOptions = modelsByAgent[agentId] ?? [];
|
||||
const modelsLoading = modelsLoadingByAgent[agentId] ?? false;
|
||||
const modelsError = modelsErrorByAgent[agentId] ?? null;
|
||||
const defaultModel = defaultModelByAgent[agentId] ?? "";
|
||||
const selectedModelId = model || defaultModel;
|
||||
const selectedModel = modelOptions.find((entry) => entry.id === selectedModelId);
|
||||
const variantOptions = selectedModel?.variants ?? [];
|
||||
const defaultVariant = selectedModel?.defaultVariant ?? "";
|
||||
const supportsVariants = Boolean(currentAgent?.capabilities?.variants);
|
||||
const agentDisplayNames: Record<string, string> = {
|
||||
claude: "Claude Code",
|
||||
codex: "Codex",
|
||||
|
|
@ -936,6 +984,13 @@ export default function App() {
|
|||
permissionMode={permissionMode}
|
||||
model={model}
|
||||
variant={variant}
|
||||
modelOptions={modelOptions}
|
||||
defaultModel={defaultModel}
|
||||
modelsLoading={modelsLoading}
|
||||
modelsError={modelsError}
|
||||
variantOptions={variantOptions}
|
||||
defaultVariant={defaultVariant}
|
||||
supportsVariants={supportsVariants}
|
||||
streamMode={streamMode}
|
||||
activeModes={activeModes}
|
||||
currentAgentVersion={currentAgent?.version ?? null}
|
||||
|
|
|
|||
|
|
@ -10,6 +10,7 @@ import {
|
|||
GitBranch,
|
||||
HelpCircle,
|
||||
Image,
|
||||
Layers,
|
||||
MessageSquare,
|
||||
Paperclip,
|
||||
PlayCircle,
|
||||
|
|
@ -37,7 +38,8 @@ const badges = [
|
|||
{ key: "fileChanges", label: "File Changes", icon: FileDiff },
|
||||
{ key: "mcpTools", label: "MCP", icon: Plug },
|
||||
{ key: "streamingDeltas", label: "Deltas", icon: Activity },
|
||||
{ key: "itemStarted", label: "Item Start", icon: CircleDot }
|
||||
{ key: "itemStarted", label: "Item Start", icon: CircleDot },
|
||||
{ key: "variants", label: "Variants", icon: Layers }
|
||||
] as const;
|
||||
|
||||
type BadgeItem = (typeof badges)[number];
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
import { MessageSquare, PauseCircle, PlayCircle, Plus, Square, Terminal } from "lucide-react";
|
||||
import { useEffect, useRef, useState } from "react";
|
||||
import type { AgentInfo, AgentModeInfo, PermissionEventData, QuestionEventData } from "sandbox-agent";
|
||||
import type { AgentInfo, AgentModelInfo, AgentModeInfo, PermissionEventData, QuestionEventData } from "sandbox-agent";
|
||||
import ApprovalsTab from "../debug/ApprovalsTab";
|
||||
import ChatInput from "./ChatInput";
|
||||
import ChatMessages from "./ChatMessages";
|
||||
|
|
@ -28,6 +28,13 @@ const ChatPanel = ({
|
|||
permissionMode,
|
||||
model,
|
||||
variant,
|
||||
modelOptions,
|
||||
defaultModel,
|
||||
modelsLoading,
|
||||
modelsError,
|
||||
variantOptions,
|
||||
defaultVariant,
|
||||
supportsVariants,
|
||||
streamMode,
|
||||
activeModes,
|
||||
currentAgentVersion,
|
||||
|
|
@ -70,6 +77,13 @@ const ChatPanel = ({
|
|||
permissionMode: string;
|
||||
model: string;
|
||||
variant: string;
|
||||
modelOptions: AgentModelInfo[];
|
||||
defaultModel: string;
|
||||
modelsLoading: boolean;
|
||||
modelsError: string | null;
|
||||
variantOptions: string[];
|
||||
defaultVariant: string;
|
||||
supportsVariants: boolean;
|
||||
streamMode: "poll" | "sse" | "turn";
|
||||
activeModes: AgentModeInfo[];
|
||||
currentAgentVersion?: string | null;
|
||||
|
|
@ -277,6 +291,13 @@ const ChatPanel = ({
|
|||
permissionMode={permissionMode}
|
||||
model={model}
|
||||
variant={variant}
|
||||
modelOptions={modelOptions}
|
||||
defaultModel={defaultModel}
|
||||
modelsLoading={modelsLoading}
|
||||
modelsError={modelsError}
|
||||
variantOptions={variantOptions}
|
||||
defaultVariant={defaultVariant}
|
||||
supportsVariants={supportsVariants}
|
||||
activeModes={activeModes}
|
||||
modesLoading={modesLoading}
|
||||
modesError={modesError}
|
||||
|
|
|
|||
|
|
@ -1,10 +1,17 @@
|
|||
import type { AgentModeInfo } from "sandbox-agent";
|
||||
import type { AgentModelInfo, AgentModeInfo } from "sandbox-agent";
|
||||
|
||||
const ChatSetup = ({
|
||||
agentMode,
|
||||
permissionMode,
|
||||
model,
|
||||
variant,
|
||||
modelOptions,
|
||||
defaultModel,
|
||||
modelsLoading,
|
||||
modelsError,
|
||||
variantOptions,
|
||||
defaultVariant,
|
||||
supportsVariants,
|
||||
activeModes,
|
||||
hasSession,
|
||||
modesLoading,
|
||||
|
|
@ -18,6 +25,13 @@ const ChatSetup = ({
|
|||
permissionMode: string;
|
||||
model: string;
|
||||
variant: string;
|
||||
modelOptions: AgentModelInfo[];
|
||||
defaultModel: string;
|
||||
modelsLoading: boolean;
|
||||
modelsError: string | null;
|
||||
variantOptions: string[];
|
||||
defaultVariant: string;
|
||||
supportsVariants: boolean;
|
||||
activeModes: AgentModeInfo[];
|
||||
hasSession: boolean;
|
||||
modesLoading: boolean;
|
||||
|
|
@ -27,6 +41,16 @@ const ChatSetup = ({
|
|||
onModelChange: (value: string) => void;
|
||||
onVariantChange: (value: string) => void;
|
||||
}) => {
|
||||
const showModelSelect = modelsLoading || Boolean(modelsError) || modelOptions.length > 0;
|
||||
const hasModelOptions = modelOptions.length > 0;
|
||||
const showVariantSelect =
|
||||
supportsVariants && (modelsLoading || Boolean(modelsError) || variantOptions.length > 0);
|
||||
const hasVariantOptions = variantOptions.length > 0;
|
||||
const modelCustom =
|
||||
model && hasModelOptions && !modelOptions.some((entry) => entry.id === model);
|
||||
const variantCustom =
|
||||
variant && hasVariantOptions && !variantOptions.includes(variant);
|
||||
|
||||
return (
|
||||
<div className="setup-row">
|
||||
<div className="setup-field">
|
||||
|
|
@ -71,26 +95,82 @@ const ChatSetup = ({
|
|||
|
||||
<div className="setup-field">
|
||||
<span className="setup-label">Model</span>
|
||||
<input
|
||||
className="setup-input"
|
||||
value={model}
|
||||
onChange={(e) => onModelChange(e.target.value)}
|
||||
placeholder="Model"
|
||||
title="Model"
|
||||
disabled={!hasSession}
|
||||
/>
|
||||
{showModelSelect ? (
|
||||
<select
|
||||
className="setup-select"
|
||||
value={model}
|
||||
onChange={(e) => onModelChange(e.target.value)}
|
||||
title="Model"
|
||||
disabled={!hasSession || modelsLoading || Boolean(modelsError)}
|
||||
>
|
||||
{modelsLoading ? (
|
||||
<option value="">Loading models...</option>
|
||||
) : modelsError ? (
|
||||
<option value="">{modelsError}</option>
|
||||
) : (
|
||||
<>
|
||||
<option value="">
|
||||
{defaultModel ? `Default (${defaultModel})` : "Default"}
|
||||
</option>
|
||||
{modelCustom && <option value={model}>{model} (custom)</option>}
|
||||
{modelOptions.map((entry) => (
|
||||
<option key={entry.id} value={entry.id}>
|
||||
{entry.name ?? entry.id}
|
||||
</option>
|
||||
))}
|
||||
</>
|
||||
)}
|
||||
</select>
|
||||
) : (
|
||||
<input
|
||||
className="setup-input"
|
||||
value={model}
|
||||
onChange={(e) => onModelChange(e.target.value)}
|
||||
placeholder="Model"
|
||||
title="Model"
|
||||
disabled={!hasSession}
|
||||
/>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="setup-field">
|
||||
<span className="setup-label">Variant</span>
|
||||
<input
|
||||
className="setup-input"
|
||||
value={variant}
|
||||
onChange={(e) => onVariantChange(e.target.value)}
|
||||
placeholder="Variant"
|
||||
title="Variant"
|
||||
disabled={!hasSession}
|
||||
/>
|
||||
{showVariantSelect ? (
|
||||
<select
|
||||
className="setup-select"
|
||||
value={variant}
|
||||
onChange={(e) => onVariantChange(e.target.value)}
|
||||
title="Variant"
|
||||
disabled={!hasSession || !supportsVariants || modelsLoading || Boolean(modelsError)}
|
||||
>
|
||||
{modelsLoading ? (
|
||||
<option value="">Loading variants...</option>
|
||||
) : modelsError ? (
|
||||
<option value="">{modelsError}</option>
|
||||
) : (
|
||||
<>
|
||||
<option value="">
|
||||
{defaultVariant ? `Default (${defaultVariant})` : "Default"}
|
||||
</option>
|
||||
{variantCustom && <option value={variant}>{variant} (custom)</option>}
|
||||
{variantOptions.map((entry) => (
|
||||
<option key={entry} value={entry}>
|
||||
{entry}
|
||||
</option>
|
||||
))}
|
||||
</>
|
||||
)}
|
||||
</select>
|
||||
) : (
|
||||
<input
|
||||
className="setup-input"
|
||||
value={variant}
|
||||
onChange={(e) => onVariantChange(e.target.value)}
|
||||
placeholder={supportsVariants ? "Variant" : "Variants unsupported"}
|
||||
title="Variant"
|
||||
disabled={!hasSession || !supportsVariants}
|
||||
/>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
|
|
|
|||
|
|
@ -14,6 +14,7 @@ export type FeatureCoverageView = AgentCapabilities & {
|
|||
mcpTools?: boolean;
|
||||
streamingDeltas?: boolean;
|
||||
itemStarted?: boolean;
|
||||
variants?: boolean;
|
||||
};
|
||||
|
||||
export const emptyFeatureCoverage: FeatureCoverageView = {
|
||||
|
|
@ -34,5 +35,6 @@ export const emptyFeatureCoverage: FeatureCoverageView = {
|
|||
mcpTools: false,
|
||||
streamingDeltas: false,
|
||||
itemStarted: false,
|
||||
variants: false,
|
||||
sharedProcess: false
|
||||
};
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue