"use client"; import { zodResolver } from "@hookform/resolvers/zod"; import { ArrowDownIcon, ArrowUpIcon, InfoIcon, PlusIcon, Trash2Icon, } from "lucide-react"; import { useEffect, useMemo, useState, type ReactNode } from "react"; import { Controller, Resolver, useForm } from "react-hook-form"; import { toast } from "sonner"; import { z } from "zod/v4"; import { OptionCombobox } from "@/components/option-combobox"; import { ProjectDialog } from "@/components/project-dialog"; import { Badge } from "@/components/ui/badge"; import { Button } from "@/components/ui/button"; import { Field, FieldContent, FieldError, FieldLabel, } from "@/components/ui/field"; import { Input } from "@/components/ui/input"; import { Popover, PopoverContent, PopoverDescription, PopoverTrigger, } from "@/components/ui/popover"; import { Textarea } from "@/components/ui/textarea"; import { type AIAgent, type AIConfig, type AdminAgentTeam, type CreateAIAgentPayload, type KnowledgeBase, type MCPToolCatalogItem, type SkillDefinition, fetchAIAgent, fetchAIConfigsAll, fetchAgentTeamsAll, fetchMCPCatalog, fetchKnowledgeBasesAll, fetchSkillDefinitionsAll, } from "@/lib/api/admin"; import { AIAgentHandoffMode, AIAgentHandoffModeLabels, AIModelType, IMConversationServiceMode, IMConversationServiceModeLabels, Status, } from "@/lib/generated/enums"; import { getEnumOptions } from "@/lib/enums"; import { FieldDescription } from "@base-ui/react"; type DirectToolItem = CreateAIAgentPayload["directTools"][number]; type DirectToolOption = { value: string; label: string; meta: DirectToolItem; sourceType: string; autoInjected: boolean; groupLabel: string; }; type EditDialogProps = { open: boolean; saving: boolean; itemId: number | null; onOpenChange: (open: boolean) => void; onSubmit: (payload: CreateAIAgentPayload) => Promise; }; const schema = z.object({ name: z.string().trim().min(1, "名称不能为空"), description: z.string().trim(), aiConfigId: z.string().trim().regex(/^\d+$/, "请选择 AI 配置"), serviceMode: z.string().trim().min(1, "请选择服务模式"), systemPrompt: z.string().trim(), welcomeMessage: z.string().trim(), replyTimeoutSeconds: z .number() .min(0, "回复超时秒数必须是大于等于 0 的整数"), handoffMode: z.string().trim().min(1, "请选择转人工模式"), fallbackMessage: z.string().trim(), remark: z.string().trim(), }); type EditForm = z.infer; const resolver = zodResolver(schema as never) as Resolver< z.input, undefined, z.output >; const serviceModeOptions = getEnumOptions(IMConversationServiceModeLabels).map( (option) => ({ value: String(option.value), label: option.label, }), ); const handoffModeOptions = getEnumOptions(AIAgentHandoffModeLabels).map( (option) => ({ value: String(option.value), label: option.label, }), ); function buildForm(item: AIAgent | null): EditForm { if (!item) { return { name: "", description: "", aiConfigId: "", serviceMode: String(IMConversationServiceMode.AIFirst), systemPrompt: "", welcomeMessage: "", replyTimeoutSeconds: 180, handoffMode: String(AIAgentHandoffMode.WaitPool), fallbackMessage: "我暂时没有找到足够准确的信息。你可以补充订单号、产品名或更具体的问题,我再继续帮你查。", remark: "", }; } return { name: item.name, description: item.description || "", aiConfigId: item.aiConfigId > 0 ? String(item.aiConfigId) : "", serviceMode: String(item.serviceMode), systemPrompt: item.systemPrompt || "", welcomeMessage: item.welcomeMessage || "", replyTimeoutSeconds: item.replyTimeoutSeconds ?? 180, handoffMode: String(item.handoffMode), fallbackMessage: item.fallbackMessage || "", remark: item.remark || "", }; } function buildPayload( form: EditForm, knowledgeIds: number[], teamIds: number[], skillIds: number[], directTools: CreateAIAgentPayload["directTools"], ): CreateAIAgentPayload { return { name: form.name.trim(), description: form.description.trim(), aiConfigId: Number(form.aiConfigId), serviceMode: Number(form.serviceMode), systemPrompt: form.systemPrompt.trim(), welcomeMessage: form.welcomeMessage.trim(), replyTimeoutSeconds: Number(form.replyTimeoutSeconds), teamIds, handoffMode: Number(form.handoffMode), fallbackMessage: form.fallbackMessage.trim(), knowledgeIds, skillIds, directTools, remark: form.remark.trim(), }; } export function EditDialog({ open, saving, itemId, onOpenChange, onSubmit, }: EditDialogProps) { if (!open) { return null; } return ( ); } function EditDialogBody({ open, saving, itemId, onOpenChange, onSubmit, }: EditDialogProps) { const formId = "ai-agent-edit-form"; const [loading, setLoading] = useState(false); const form = useForm< z.input, undefined, z.output >({ resolver, defaultValues: buildForm(null), }); const { control, handleSubmit, register, reset, watch, formState: { errors }, } = form; const [selectedKnowledgeIds, setSelectedKnowledgeIds] = useState( [], ); const [selectedTeamIds, setSelectedTeamIds] = useState([]); const [selectedSkillIds, setSelectedSkillIds] = useState([]); const [knowledgeToAdd, setKnowledgeToAdd] = useState(""); const [teamToAdd, setTeamToAdd] = useState(""); const [skillToAdd, setSkillToAdd] = useState(""); const [directToolGroupToAdd, setDirectToolGroupToAdd] = useState(""); const [directToolToAdd, setDirectToolToAdd] = useState(""); const [aiConfigs, setAIConfigs] = useState([]); const [knowledgeBases, setKnowledgeBases] = useState([]); const [agentTeams, setAgentTeams] = useState([]); const [skills, setSkills] = useState([]); const [directTools, setDirectTools] = useState([]); const [directToolOptions, setDirectToolOptions] = useState( [], ); const [toolCatalog, setToolCatalog] = useState([]); useEffect(() => { async function loadDetail() { if (!itemId) { reset(buildForm(null)); setSelectedKnowledgeIds([]); setSelectedTeamIds([]); setSelectedSkillIds([]); setDirectTools([]); setKnowledgeToAdd(""); setTeamToAdd(""); setSkillToAdd(""); setDirectToolGroupToAdd(""); setDirectToolToAdd(""); return; } setLoading(true); try { const data = await fetchAIAgent(itemId); reset(buildForm(data)); setSelectedKnowledgeIds(data.knowledgeIds ?? []); setSelectedTeamIds((data.teams ?? []).map((team) => team.id)); setSelectedSkillIds(data.skillIds ?? []); setDirectTools(data.directTools ?? []); setKnowledgeToAdd(""); setTeamToAdd(""); setSkillToAdd(""); setDirectToolGroupToAdd(""); setDirectToolToAdd(""); } catch (error) { toast.error( error instanceof Error ? error.message : "加载 AI Agent 详情失败", ); } finally { setLoading(false); } } void loadDetail(); }, [itemId, reset]); useEffect(() => { async function loadAIConfigs() { try { const data = await fetchAIConfigsAll({ modelType: AIModelType.LLM, }); setAIConfigs(data); } catch (error) { toast.error( error instanceof Error ? error.message : "加载 AI 配置失败", ); } } void loadAIConfigs(); }, []); useEffect(() => { async function loadAgentTeams() { try { const data = await fetchAgentTeamsAll(); setAgentTeams(data); } catch (error) { toast.error(error instanceof Error ? error.message : "加载客服组失败"); } } void loadAgentTeams(); }, []); useEffect(() => { async function loadKnowledgeBases() { try { const data = await fetchKnowledgeBasesAll({ status: Status.Ok, }); setKnowledgeBases(data); } catch (error) { toast.error(error instanceof Error ? error.message : "加载知识库失败"); } } void loadKnowledgeBases(); }, []); useEffect(() => { async function loadSkills() { try { const data = await fetchSkillDefinitionsAll({ status: Status.Ok, }); setSkills(data); } catch (error) { toast.error(error instanceof Error ? error.message : "加载 Skills 失败"); } } void loadSkills(); }, []); useEffect(() => { async function loadDirectToolOptions() { try { const catalog = await fetchMCPCatalog(); setToolCatalog(catalog); setDirectToolOptions( catalog .filter((tool) => !tool.autoInjected) .map((tool) => ({ value: tool.toolCode, label: `${tool.title || tool.toolName} · ${tool.toolCode}`, sourceType: tool.sourceType, autoInjected: tool.autoInjected, groupLabel: tool.sourceType === "builtin" ? "内置工具" : tool.serverCode, meta: { toolCode: tool.toolCode, serverCode: tool.serverCode, toolName: tool.toolName, title: tool.title || tool.toolName, description: tool.description || "", arguments: undefined, }, })), ); } catch (error) { toast.error( error instanceof Error ? error.message : "加载 Direct Tools 失败", ); } } void loadDirectToolOptions(); }, []); const aiConfigOptions = useMemo( () => aiConfigs.map((item) => ({ value: String(item.id), label: `${item.name} · ${item.modelName}`, })), [aiConfigs], ); const teamOptions = useMemo( () => agentTeams.map((item: AdminAgentTeam) => ({ value: String(item.id), label: item.name, })), [agentTeams], ); const knowledgeOptions = useMemo( () => knowledgeBases.map((item) => ({ value: String(item.id), label: item.name, })), [knowledgeBases], ); const skillOptions = useMemo( () => skills.map((item) => ({ value: String(item.id), label: item.name, })), [skills], ); const addableKnowledgeOptions = useMemo( () => knowledgeOptions.filter( (option) => !selectedKnowledgeIds.includes(Number(option.value)), ), [knowledgeOptions, selectedKnowledgeIds], ); const selectedKnowledgeOptions = useMemo( () => selectedKnowledgeIds .map((id) => knowledgeOptions.find((option) => Number(option.value) === id), ) .filter( (option): option is { value: string; label: string } => !!option, ), [knowledgeOptions, selectedKnowledgeIds], ); const addableSkillOptions = useMemo( () => skillOptions.filter( (option) => !selectedSkillIds.includes(Number(option.value)), ), [selectedSkillIds, skillOptions], ); const selectedSkillOptions = useMemo( () => selectedSkillIds .map((id) => skillOptions.find((option) => Number(option.value) === id)) .filter( (option): option is { value: string; label: string } => !!option, ), [selectedSkillIds, skillOptions], ); const addableDirectToolOptions = useMemo( () => directToolOptions.filter( (option) => option.groupLabel === directToolGroupToAdd && !directTools.some((tool) => tool.toolCode === option.value), ), [directToolOptions, directToolGroupToAdd, directTools], ); const directToolGroupOptions = useMemo( () => Array.from( new Map( directToolOptions.map((option) => [ option.groupLabel, { value: option.groupLabel, label: option.groupLabel, }, ]), ).values(), ), [directToolOptions], ); const directToolsGrouped = useMemo(() => { const groups = new Map(); for (const tool of directTools) { const groupLabel = tool.serverCode === "builtin" || tool.toolCode.startsWith("builtin/") ? "内置工具" : tool.serverCode || "未分组"; const current = groups.get(groupLabel) ?? []; current.push(tool); groups.set(groupLabel, current); } return Array.from(groups.entries()); }, [directTools]); const addableTeamOptions = useMemo( () => teamOptions.filter( (option) => !selectedTeamIds.includes(Number(option.value)), ), [teamOptions, selectedTeamIds], ); const selectedTeamOptions = useMemo( () => selectedTeamIds .map((id) => teamOptions.find((option) => Number(option.value) === id)) .filter( (option): option is { value: string; label: string } => !!option, ), [selectedTeamIds, teamOptions], ); const handoffMode = watch("handoffMode"); const selectedHandoffModeLabel = handoffModeOptions.find((item) => item.value === handoffMode)?.label ?? "未选择"; async function onFormSubmit(values: EditForm) { await onSubmit( buildPayload( values, selectedKnowledgeIds, selectedTeamIds, selectedSkillIds, directTools, ), ); } function handleAddTeam(value: string) { const id = Number(value); if (!Number.isFinite(id) || id <= 0 || selectedTeamIds.includes(id)) { return; } setSelectedTeamIds((prev) => [...prev, id]); setTeamToAdd(""); } function handleRemoveTeam(id: number) { setSelectedTeamIds((prev) => prev.filter((item) => item !== id)); } function handleAddKnowledge(value: string) { const id = Number(value); if (!Number.isFinite(id) || id <= 0 || selectedKnowledgeIds.includes(id)) { return; } setSelectedKnowledgeIds((prev) => [...prev, id]); setKnowledgeToAdd(""); } function handleMoveKnowledge(index: number, direction: -1 | 1) { const targetIndex = index + direction; if (targetIndex < 0 || targetIndex >= selectedKnowledgeIds.length) { return; } setSelectedKnowledgeIds((prev) => { const next = [...prev]; const current = next[index]; next[index] = next[targetIndex]; next[targetIndex] = current; return next; }); } function handleRemoveKnowledge(id: number) { setSelectedKnowledgeIds((prev) => prev.filter((item) => item !== id)); } function handleAddSkill(value: string) { const id = Number(value); if (!Number.isFinite(id) || id <= 0 || selectedSkillIds.includes(id)) { return; } setSelectedSkillIds((prev) => [...prev, id]); setSkillToAdd(""); } function handleRemoveSkill(id: number) { setSelectedSkillIds((prev) => prev.filter((item) => item !== id)); } function handleAddDirectTool(value: string) { const option = directToolOptions.find((item) => item.value === value); if (!option) { return; } setDirectTools((prev) => { if ( prev.some( (item) => item.toolCode === option.meta.toolCode, ) ) { return prev; } return [...prev, option.meta]; }); setDirectToolGroupToAdd(option.groupLabel); setDirectToolToAdd(""); } function handleRemoveDirectTool(value: string) { setDirectTools((prev) => prev.filter((item) => item.toolCode !== value)); } return ( } > {loading ? (
加载中...
) : (
名称 AI 配置 ( )} /> 服务模式 ( )} />
描述