602534a1b7
- Add agent generation service with OpenAI/Anthropic support, streaming, and error handling - Add POST /agent/generate API endpoint with SSE streaming and cost tracking - Add API client functions for agent generation with AbortController support - Create AgentGenerationModal component with live preview, streaming, and import flow - Integrate AI Generate button into NewAgentDialog with configuration options - Add comprehensive unit tests for service and component - Add changeset for @gsxdsm/fusion minor bump
405 lines
16 KiB
TypeScript
405 lines
16 KiB
TypeScript
import { useState, useEffect, useCallback } from "react";
|
||
import type { AgentCapability, ModelInfo, AgentGenerationSpec } from "../api";
|
||
import { createAgent, fetchModels } from "../api";
|
||
import { CustomModelDropdown } from "./CustomModelDropdown";
|
||
import { ProviderIcon } from "./ProviderIcon";
|
||
import { AgentGenerationModal } from "./AgentGenerationModal";
|
||
|
||
export interface NewAgentDialogProps {
|
||
isOpen: boolean;
|
||
onClose: () => void;
|
||
onCreated: () => void;
|
||
projectId?: string;
|
||
}
|
||
|
||
const AGENT_ROLES: { value: AgentCapability; label: string; icon: string }[] = [
|
||
{ value: "triage", label: "Triage", icon: "🔍" },
|
||
{ value: "executor", label: "Executor", icon: "⚡" },
|
||
{ value: "reviewer", label: "Reviewer", icon: "👁" },
|
||
{ value: "merger", label: "Merger", icon: "🔀" },
|
||
{ value: "scheduler", label: "Scheduler", icon: "⏰" },
|
||
{ value: "engineer", label: "Engineer", icon: "🛠" },
|
||
{ value: "custom", label: "Custom", icon: "🔧" },
|
||
];
|
||
|
||
type ThinkingLevel = "off" | "minimal" | "low" | "medium" | "high";
|
||
|
||
/** Set of valid AgentCapability values for mapping generated roles */
|
||
const VALID_CAPABILITIES = new Set<string>(["triage", "executor", "reviewer", "merger", "scheduler", "engineer", "custom"]);
|
||
|
||
interface RuntimeConfig {
|
||
model: string;
|
||
thinkingLevel: ThinkingLevel;
|
||
maxTurns: number;
|
||
}
|
||
|
||
export function NewAgentDialog({ isOpen, onClose, onCreated, projectId }: NewAgentDialogProps) {
|
||
const [step, setStep] = useState(0);
|
||
const [name, setName] = useState("");
|
||
const [title, setTitle] = useState("");
|
||
const [icon, setIcon] = useState("");
|
||
const [role, setRole] = useState<AgentCapability>("custom");
|
||
const [runtimeConfig, setRuntimeConfig] = useState<RuntimeConfig>({
|
||
model: "",
|
||
thinkingLevel: "off",
|
||
maxTurns: 10,
|
||
});
|
||
const [isSubmitting, setIsSubmitting] = useState(false);
|
||
const [error, setError] = useState<string | null>(null);
|
||
const [isGenerationModalOpen, setIsGenerationModalOpen] = useState(false);
|
||
|
||
// Model dropdown state
|
||
const [availableModels, setAvailableModels] = useState<ModelInfo[]>([]);
|
||
const [modelsLoading, setModelsLoading] = useState(false);
|
||
const [favoriteProviders, setFavoriteProviders] = useState<string[]>([]);
|
||
const [favoriteModels, setFavoriteModels] = useState<string[]>([]);
|
||
|
||
// Load models on mount (global data, not per-agent)
|
||
useEffect(() => {
|
||
setModelsLoading(true);
|
||
fetchModels()
|
||
.then((response) => {
|
||
setAvailableModels(response.models);
|
||
setFavoriteProviders(response.favoriteProviders);
|
||
setFavoriteModels(response.favoriteModels);
|
||
})
|
||
.catch(() => {
|
||
// Gracefully handle — dropdown will show empty list
|
||
})
|
||
.finally(() => setModelsLoading(false));
|
||
}, []);
|
||
|
||
// Selected model in "provider/modelId" format, or "" for default
|
||
const selectedModel = runtimeConfig.model.includes("/")
|
||
? runtimeConfig.model
|
||
: "";
|
||
|
||
const handleGenerated = useCallback((spec: AgentGenerationSpec) => {
|
||
// Map generated role to AgentCapability, default to "custom" if unrecognized
|
||
const mappedRole = VALID_CAPABILITIES.has(spec.role)
|
||
? (spec.role as AgentCapability)
|
||
: "custom";
|
||
|
||
setName(spec.title);
|
||
setTitle(spec.description);
|
||
setIcon(spec.icon);
|
||
setRole(mappedRole);
|
||
setRuntimeConfig(c => ({
|
||
...c,
|
||
thinkingLevel: spec.thinkingLevel,
|
||
maxTurns: spec.maxTurns,
|
||
}));
|
||
setIsGenerationModalOpen(false);
|
||
// Advance to Step 1 so user can review model selection
|
||
setStep(1);
|
||
}, []);
|
||
|
||
const handleModelChange = useCallback((value: string) => {
|
||
// value is "provider/modelId" or "" for default
|
||
setRuntimeConfig(c => ({ ...c, model: value }));
|
||
}, []);
|
||
|
||
const handleToggleFavorite = useCallback(async (provider: string) => {
|
||
const currentFavorites = favoriteProviders;
|
||
const isFavorite = currentFavorites.includes(provider);
|
||
const newFavorites = isFavorite
|
||
? currentFavorites.filter(p => p !== provider)
|
||
: [provider, ...currentFavorites];
|
||
setFavoriteProviders(newFavorites);
|
||
}, [favoriteProviders]);
|
||
|
||
const handleToggleModelFavorite = useCallback(async (modelId: string) => {
|
||
const currentFavorites = favoriteModels;
|
||
const isFavorite = currentFavorites.includes(modelId);
|
||
const newFavorites = isFavorite
|
||
? currentFavorites.filter(m => m !== modelId)
|
||
: [modelId, ...currentFavorites];
|
||
setFavoriteModels(newFavorites);
|
||
}, [favoriteModels]);
|
||
|
||
if (!isOpen) return null;
|
||
|
||
const handleClose = () => {
|
||
setStep(0);
|
||
setName("");
|
||
setTitle("");
|
||
setIcon("");
|
||
setRole("custom");
|
||
setRuntimeConfig({ model: "", thinkingLevel: "off", maxTurns: 10 });
|
||
setError(null);
|
||
setIsGenerationModalOpen(false);
|
||
onClose();
|
||
};
|
||
|
||
const handleCreate = async () => {
|
||
if (!name.trim()) return;
|
||
setIsSubmitting(true);
|
||
setError(null);
|
||
try {
|
||
const runtimeCfg: Record<string, unknown> = {};
|
||
if (runtimeConfig.model.trim()) runtimeCfg.model = runtimeConfig.model.trim();
|
||
if (runtimeConfig.thinkingLevel !== "off") runtimeCfg.thinkingLevel = runtimeConfig.thinkingLevel;
|
||
if (runtimeConfig.maxTurns !== 10) runtimeCfg.maxTurns = runtimeConfig.maxTurns;
|
||
await createAgent({
|
||
name: name.trim(),
|
||
role,
|
||
...(title.trim() ? { title: title.trim() } : {}),
|
||
...(icon.trim() ? { icon: icon.trim() } : {}),
|
||
...(Object.keys(runtimeCfg).length > 0 ? { runtimeConfig: runtimeCfg } : {}),
|
||
}, projectId);
|
||
handleClose();
|
||
onCreated();
|
||
} catch (err: unknown) {
|
||
setError(err instanceof Error ? err.message : "Failed to create agent");
|
||
} finally {
|
||
setIsSubmitting(false);
|
||
}
|
||
};
|
||
|
||
const selectedRole = AGENT_ROLES.find(r => r.value === role);
|
||
|
||
return (
|
||
<>
|
||
<div className="agent-dialog-overlay" onClick={(e) => { if (e.target === e.currentTarget) handleClose(); }}>
|
||
<div className="agent-dialog" role="dialog" aria-modal="true" aria-label="Create new agent">
|
||
{/* Header */}
|
||
<div className="agent-dialog-header">
|
||
<span style={{ fontWeight: 600, fontSize: 15 }}>New Agent</span>
|
||
<button
|
||
className="btn-icon"
|
||
onClick={handleClose}
|
||
aria-label="Close"
|
||
style={{ background: "none", border: "none", cursor: "pointer", color: "var(--text-muted)", fontSize: 18, lineHeight: 1 }}
|
||
>
|
||
×
|
||
</button>
|
||
</div>
|
||
|
||
{/* Step indicator */}
|
||
<div className="agent-dialog-steps">
|
||
{[0, 1, 2].map(i => (
|
||
<div
|
||
key={i}
|
||
className={`agent-dialog-step${i === step ? " active" : i < step ? " completed" : ""}`}
|
||
aria-label={`Step ${i + 1}`}
|
||
/>
|
||
))}
|
||
</div>
|
||
|
||
{/* Body */}
|
||
<div className="agent-dialog-body">
|
||
{step === 0 && (
|
||
<div>
|
||
<div className="agent-dialog-field">
|
||
<label htmlFor="agent-name">Name <span style={{ color: "var(--state-error-text, #f85149)" }}>*</span></label>
|
||
<input
|
||
id="agent-name"
|
||
type="text"
|
||
className="input"
|
||
placeholder="e.g. Frontend Reviewer"
|
||
value={name}
|
||
onChange={e => setName(e.target.value)}
|
||
autoFocus
|
||
style={{ width: "100%", boxSizing: "border-box" }}
|
||
/>
|
||
</div>
|
||
<div className="agent-dialog-field">
|
||
<label htmlFor="agent-title">Title <span style={{ color: "var(--text-muted)", fontWeight: 400 }}>(optional)</span></label>
|
||
<input
|
||
id="agent-title"
|
||
type="text"
|
||
className="input"
|
||
placeholder="e.g. Senior Code Reviewer"
|
||
value={title}
|
||
onChange={e => setTitle(e.target.value)}
|
||
style={{ width: "100%", boxSizing: "border-box" }}
|
||
/>
|
||
</div>
|
||
<div className="agent-dialog-field">
|
||
<label>Role</label>
|
||
<div className="agent-role-grid">
|
||
{AGENT_ROLES.map(r => (
|
||
<button
|
||
key={r.value}
|
||
type="button"
|
||
className={`agent-role-option${role === r.value ? " selected" : ""}`}
|
||
onClick={() => setRole(r.value)}
|
||
>
|
||
<span className="agent-role-option-icon">{r.icon}</span>
|
||
<span style={{ fontSize: 12, marginTop: 4 }}>{r.label}</span>
|
||
</button>
|
||
))}
|
||
</div>
|
||
</div>
|
||
{/* AI-assisted generation */}
|
||
<div style={{ marginTop: 8, borderTop: "1px solid var(--border)", paddingTop: 12 }}>
|
||
<button
|
||
type="button"
|
||
className="btn"
|
||
onClick={() => setIsGenerationModalOpen(true)}
|
||
style={{ width: "100%", display: "flex", alignItems: "center", justifyContent: "center", gap: 6 }}
|
||
>
|
||
<span>✨</span>
|
||
Generate with AI
|
||
</button>
|
||
<p style={{ color: "var(--text-muted)", fontSize: 11, textAlign: "center", margin: "6px 0 0" }}>
|
||
Describe your agent's role and let AI generate a specification
|
||
</p>
|
||
</div>
|
||
</div>
|
||
)}
|
||
|
||
{step === 1 && (
|
||
<div>
|
||
<div className="agent-dialog-field">
|
||
<label>Model</label>
|
||
{modelsLoading ? (
|
||
<div style={{ color: "var(--text-muted)", fontSize: 13, padding: "8px 0" }}>Loading models…</div>
|
||
) : (
|
||
<CustomModelDropdown
|
||
id="agent-model"
|
||
label="Model"
|
||
value={selectedModel}
|
||
onChange={handleModelChange}
|
||
models={availableModels}
|
||
placeholder="Select a model…"
|
||
favoriteProviders={favoriteProviders}
|
||
onToggleFavorite={handleToggleFavorite}
|
||
favoriteModels={favoriteModels}
|
||
onToggleModelFavorite={handleToggleModelFavorite}
|
||
/>
|
||
)}
|
||
</div>
|
||
<div className="agent-dialog-field">
|
||
<label htmlFor="agent-thinking">Thinking Level</label>
|
||
<select
|
||
id="agent-thinking"
|
||
className="select"
|
||
value={runtimeConfig.thinkingLevel}
|
||
onChange={e => setRuntimeConfig(c => ({ ...c, thinkingLevel: e.target.value as ThinkingLevel }))}
|
||
style={{ width: "100%" }}
|
||
>
|
||
<option value="off">Off</option>
|
||
<option value="minimal">Minimal</option>
|
||
<option value="low">Low</option>
|
||
<option value="medium">Medium</option>
|
||
<option value="high">High</option>
|
||
</select>
|
||
</div>
|
||
<div className="agent-dialog-field">
|
||
<label htmlFor="agent-max-turns">Max Turns</label>
|
||
<input
|
||
id="agent-max-turns"
|
||
type="number"
|
||
className="input"
|
||
min={1}
|
||
max={500}
|
||
value={runtimeConfig.maxTurns}
|
||
onChange={e => setRuntimeConfig(c => ({ ...c, maxTurns: Math.max(1, parseInt(e.target.value, 10) || 1) }))}
|
||
style={{ width: "100%", boxSizing: "border-box" }}
|
||
/>
|
||
</div>
|
||
</div>
|
||
)}
|
||
|
||
{step === 2 && (
|
||
<div>
|
||
<p style={{ color: "var(--text-muted)", fontSize: 13, marginTop: 0, marginBottom: 12 }}>
|
||
Review your agent configuration before creating.
|
||
</p>
|
||
<div className="agent-dialog-summary">
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Name</span>
|
||
<span style={{ fontWeight: 600 }}>
|
||
{icon && <span style={{ marginRight: 6 }}>{icon}</span>}
|
||
{name}
|
||
</span>
|
||
</div>
|
||
{title && (
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Title</span>
|
||
<span>{title}</span>
|
||
</div>
|
||
)}
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Role</span>
|
||
<span>{selectedRole?.icon} {selectedRole?.label}</span>
|
||
</div>
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Model</span>
|
||
<span style={{ fontSize: 13 }}>
|
||
{selectedModel ? (
|
||
<>
|
||
<ProviderIcon provider={selectedModel.split("/")[0]} size="sm" />
|
||
{" "}
|
||
{(() => {
|
||
const slashIdx = selectedModel.indexOf("/");
|
||
const provider = selectedModel.slice(0, slashIdx);
|
||
const modelId = selectedModel.slice(slashIdx + 1);
|
||
const model = availableModels.find(m => m.provider === provider && m.id === modelId);
|
||
return model?.name || selectedModel;
|
||
})()}
|
||
</>
|
||
) : (
|
||
<em style={{ color: "var(--text-muted)" }}>default</em>
|
||
)}
|
||
</span>
|
||
</div>
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Thinking</span>
|
||
<span style={{ textTransform: "capitalize" }}>{runtimeConfig.thinkingLevel}</span>
|
||
</div>
|
||
<div className="agent-dialog-summary-row">
|
||
<span style={{ color: "var(--text-muted)", fontSize: 13 }}>Max Turns</span>
|
||
<span>{runtimeConfig.maxTurns}</span>
|
||
</div>
|
||
</div>
|
||
{error && (
|
||
<p style={{ color: "var(--state-error-text, #f85149)", fontSize: 13, marginTop: 12 }}>{error}</p>
|
||
)}
|
||
</div>
|
||
)}
|
||
</div>
|
||
|
||
{/* Footer */}
|
||
<div className="agent-dialog-footer">
|
||
{step > 0 && (
|
||
<button className="btn" onClick={() => setStep(s => s - 1)} disabled={isSubmitting}>
|
||
Back
|
||
</button>
|
||
)}
|
||
<button className="btn" onClick={handleClose} disabled={isSubmitting}>
|
||
Cancel
|
||
</button>
|
||
{step < 2 ? (
|
||
<button
|
||
className="btn btn--primary"
|
||
onClick={() => setStep(s => s + 1)}
|
||
disabled={step === 0 && !name.trim()}
|
||
>
|
||
Next
|
||
</button>
|
||
) : (
|
||
<button
|
||
className="btn btn--primary"
|
||
onClick={() => void handleCreate()}
|
||
disabled={isSubmitting || !name.trim()}
|
||
>
|
||
{isSubmitting ? "Creating..." : "Create"}
|
||
</button>
|
||
)}
|
||
</div>
|
||
</div>
|
||
</div>
|
||
|
||
{/* AI-assisted agent generation modal */}
|
||
<AgentGenerationModal
|
||
isOpen={isGenerationModalOpen}
|
||
onClose={() => setIsGenerationModalOpen(false)}
|
||
onGenerated={handleGenerated}
|
||
projectId={projectId}
|
||
/>
|
||
</>
|
||
);
|
||
}
|