mlogclub
343 lines
11 KiB
TypeScript
343 lines
11 KiB
TypeScript
"use client";
|
||
|
||
import { BotMessageSquareIcon, GitBranchIcon, PowerIcon } from "lucide-react";
|
||
import { useMemo, useState } from "react";
|
||
|
||
import {
|
||
DashboardCrudPage,
|
||
type DashboardCrudActionState,
|
||
createDashboardStatusColumn,
|
||
createDashboardStatusToggleAction,
|
||
type DashboardCrudColumn,
|
||
type DashboardCrudFilter,
|
||
} from "@/components/dashboard/crud";
|
||
import { ProjectDialog } from "@/components/project-dialog";
|
||
import { Badge } from "@/components/ui/badge";
|
||
import {
|
||
createAIAgent,
|
||
deleteAIAgent,
|
||
fetchAIAgents,
|
||
updateAIAgent,
|
||
updateAIAgentSort,
|
||
updateAIAgentStatus,
|
||
type AIAgent,
|
||
type CreateAIAgentPayload,
|
||
} from "@/lib/api/admin";
|
||
import { IMConversationServiceMode, Status } from "@/lib/generated/enums";
|
||
import { useI18n } from "@/i18n/provider";
|
||
import { AIAgentConfigWorkbench } from "./_components/config-workbench";
|
||
|
||
type TFunction = (key: string, values?: Record<string, string | number>) => string;
|
||
|
||
function getStatusOptions(t: TFunction) {
|
||
return [
|
||
{ value: "all", label: t("aiAgent.allStatuses") },
|
||
{ value: String(Status.Ok), label: t("aiAgent.enabled") },
|
||
{ value: String(Status.Disabled), label: t("aiAgent.disabled") },
|
||
{ value: String(Status.Deleted), label: t("status.deleted") },
|
||
];
|
||
}
|
||
|
||
function getStatusLabel(value: string, t: TFunction) {
|
||
return (
|
||
getStatusOptions(t).find((item) => item.value === value)?.label ??
|
||
t("aiAgent.allStatuses")
|
||
);
|
||
}
|
||
|
||
function getServiceModeLabel(mode: number, t: TFunction) {
|
||
switch (mode) {
|
||
case IMConversationServiceMode.AIOnly:
|
||
return t("aiAgent.serviceAiOnly");
|
||
case IMConversationServiceMode.HumanOnly:
|
||
return t("aiAgent.serviceHumanOnly");
|
||
case IMConversationServiceMode.AIFirst:
|
||
return t("aiAgent.serviceAiFirst");
|
||
default:
|
||
return "-";
|
||
}
|
||
}
|
||
|
||
function getNextStatus(item: AIAgent) {
|
||
return item.status === Status.Ok ? Status.Disabled : Status.Ok;
|
||
}
|
||
|
||
function isWorkflowPublished(item: AIAgent) {
|
||
return Boolean(item.workflowPublished ?? item.workflowVersionId > 0);
|
||
}
|
||
|
||
export default function DashboardAIAgentsPage() {
|
||
const t = useI18n();
|
||
const statusOptions = useMemo(() => getStatusOptions(t), [t]);
|
||
const [configAgentId, setConfigAgentId] = useState<number | null>(null);
|
||
const [configOpen, setConfigOpen] = useState(false);
|
||
const [crudActions, setCrudActions] = useState<DashboardCrudActionState | null>(null);
|
||
|
||
const filters = useMemo<DashboardCrudFilter[]>(
|
||
() => [
|
||
{
|
||
name: "name",
|
||
label: t("aiAgent.filterName"),
|
||
placeholder: t("aiAgent.filterName"),
|
||
defaultValue: "",
|
||
trim: true,
|
||
className: "w-full sm:w-56",
|
||
},
|
||
{
|
||
name: "status",
|
||
label: t("aiAgent.allStatuses"),
|
||
type: "select",
|
||
defaultValue: "all",
|
||
allValue: "all",
|
||
options: statusOptions,
|
||
className: "w-full sm:w-52",
|
||
},
|
||
],
|
||
[statusOptions, t],
|
||
);
|
||
|
||
const columns = useMemo<DashboardCrudColumn<AIAgent>[]>(
|
||
() => [
|
||
{
|
||
key: "agent",
|
||
label: "Agent",
|
||
render: (item) => (
|
||
<div className="flex items-center gap-3">
|
||
<div className="flex size-10 items-center justify-center rounded-md bg-muted text-muted-foreground">
|
||
<BotMessageSquareIcon className="size-4" />
|
||
</div>
|
||
<div>
|
||
<div className="font-medium">{item.name}</div>
|
||
<div className="font-mono text-xs text-muted-foreground">
|
||
Agent ID: {item.id}
|
||
</div>
|
||
</div>
|
||
</div>
|
||
),
|
||
},
|
||
{
|
||
key: "aiConfig",
|
||
label: t("aiAgent.columnAiConfig"),
|
||
render: (item) => item.aiConfigName || "-",
|
||
},
|
||
{
|
||
key: "serviceMode",
|
||
label: t("aiAgent.columnServiceMode"),
|
||
render: (item) => getServiceModeLabel(item.serviceMode, t),
|
||
},
|
||
{
|
||
key: "workflow",
|
||
label: "Playbook 状态",
|
||
render: (item) => {
|
||
const published = isWorkflowPublished(item);
|
||
return (
|
||
<div className="space-y-1">
|
||
<div className="flex flex-wrap items-center gap-1.5">
|
||
<Badge variant={published ? "default" : "outline"}>
|
||
{item.workflowStateText || (published ? "已发布" : "未发布")}
|
||
</Badge>
|
||
{published ? (
|
||
<span className="font-mono text-xs text-muted-foreground">
|
||
#{item.workflowVersionId}
|
||
</span>
|
||
) : null}
|
||
</div>
|
||
{!published ? (
|
||
<div className="text-xs text-muted-foreground">
|
||
未发布 Playbook，AI 不会自动回复
|
||
</div>
|
||
) : (
|
||
<div className="text-xs text-muted-foreground">
|
||
当前生效版本 #{item.workflowVersionId}
|
||
</div>
|
||
)}
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
key: "skills",
|
||
label: t("aiAgent.columnSkills"),
|
||
render: (item) => {
|
||
const skills = item.skills ?? [];
|
||
return (
|
||
<div className="flex flex-wrap gap-1">
|
||
{skills.length === 0 ? (
|
||
<span className="text-sm text-muted-foreground">
|
||
{t("aiAgent.ragOnly")}
|
||
</span>
|
||
) : (
|
||
skills.map((skill) => (
|
||
<Badge key={skill.id} variant="outline">
|
||
{skill.name}
|
||
</Badge>
|
||
))
|
||
)}
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
{
|
||
key: "capabilities",
|
||
label: t("aiAgent.columnCapabilities"),
|
||
render: (item) => {
|
||
const skills = item.skills ?? [];
|
||
const directTools = item.directTools ?? [];
|
||
const directToolServerCodes = Array.from(
|
||
new Set(directTools.map((tool) => tool.serverCode).filter(Boolean)),
|
||
);
|
||
return (
|
||
<div className="space-y-2">
|
||
<div className="flex flex-wrap gap-1">
|
||
<Badge variant="secondary">{skills.length} Skills</Badge>
|
||
<Badge variant="secondary">{directTools.length} Tools</Badge>
|
||
</div>
|
||
<div className="flex flex-wrap gap-1">
|
||
{directToolServerCodes.length === 0 ? (
|
||
<span className="text-sm text-muted-foreground">
|
||
{t("aiAgent.noMcpServer")}
|
||
</span>
|
||
) : (
|
||
directToolServerCodes.map((serverCode) => (
|
||
<Badge key={serverCode} variant="outline">
|
||
{serverCode}
|
||
</Badge>
|
||
))
|
||
)}
|
||
</div>
|
||
</div>
|
||
);
|
||
},
|
||
},
|
||
createDashboardStatusColumn<AIAgent, number>({
|
||
label: t("aiAgent.columnStatus"),
|
||
getStatus: (item) => item.status,
|
||
getLabel: (status) => getStatusLabel(String(status), t),
|
||
getBadgeVariant: (status) =>
|
||
status === Status.Ok ? "default" : "secondary",
|
||
isEnabled: (status) => status === Status.Ok,
|
||
toggle: {
|
||
getNextStatus,
|
||
updateStatus: (item, nextStatus) =>
|
||
updateAIAgentStatus(item.id, nextStatus),
|
||
successMessage: (item, nextStatus) =>
|
||
t("aiAgent.statusChanged", {
|
||
name: item.name,
|
||
status:
|
||
nextStatus === Status.Ok
|
||
? t("aiAgent.enabled")
|
||
: t("aiAgent.stop"),
|
||
}),
|
||
errorMessage: t("aiAgent.statusUpdateFailed"),
|
||
ariaLabel: (item) => t("aiAgent.toggleStatus", { name: item.name }),
|
||
},
|
||
}),
|
||
],
|
||
[t],
|
||
);
|
||
|
||
return (
|
||
<>
|
||
<DashboardCrudPage<AIAgent, CreateAIAgentPayload>
|
||
filters={filters}
|
||
columns={columns}
|
||
fetchList={(query) =>
|
||
fetchAIAgents({
|
||
name: typeof query.name === "string" ? query.name : undefined,
|
||
status: typeof query.status === "string" ? query.status : undefined,
|
||
page: Number(query.page),
|
||
limit: Number(query.limit),
|
||
})
|
||
}
|
||
getItemId={(item) => item.id}
|
||
createItem={createAIAgent}
|
||
updateItem={(item, payload) => updateAIAgent({ id: item.id, ...payload })}
|
||
onCreateItem={() => {
|
||
setConfigAgentId(null);
|
||
setConfigOpen(true);
|
||
}}
|
||
onEditItem={(item) => {
|
||
setConfigAgentId(item.id);
|
||
setConfigOpen(true);
|
||
}}
|
||
deleteItem={(item) => deleteAIAgent(item.id)}
|
||
rowActions={[
|
||
{
|
||
key: "workflow",
|
||
icon: <GitBranchIcon />,
|
||
label: t("aiAgent.configure"),
|
||
run: ({ item }) => {
|
||
setConfigAgentId(item.id);
|
||
setConfigOpen(true);
|
||
},
|
||
},
|
||
createDashboardStatusToggleAction<AIAgent, number>({
|
||
icon: <PowerIcon />,
|
||
label: (item) =>
|
||
item.status === Status.Ok ? t("aiAgent.stop") : t("aiAgent.enabled"),
|
||
getNextStatus,
|
||
updateStatus: (item, nextStatus) =>
|
||
updateAIAgentStatus(item.id, nextStatus),
|
||
successMessage: (item, nextStatus) =>
|
||
t("aiAgent.statusChanged", {
|
||
name: item.name,
|
||
status:
|
||
nextStatus === Status.Ok
|
||
? t("aiAgent.enabled")
|
||
: t("aiAgent.stop"),
|
||
}),
|
||
errorMessage: t("aiAgent.statusUpdateFailed"),
|
||
}),
|
||
]}
|
||
sort={{
|
||
enabled: true,
|
||
onReorder: (items) => updateAIAgentSort(items.map((item) => item.id)),
|
||
successMessage: t("aiAgent.sortUpdated"),
|
||
errorMessage: t("aiAgent.sortUpdateFailed"),
|
||
handleLabel: t("aiAgent.dragSort", { name: "" }),
|
||
}}
|
||
onActionStateChange={setCrudActions}
|
||
labels={{
|
||
refresh: t("aiAgent.refresh"),
|
||
create: t("aiAgent.new"),
|
||
query: t("aiAgent.query"),
|
||
loading: t("aiAgent.loadingRows"),
|
||
empty: t("aiAgent.emptyRows"),
|
||
actions: t("aiAgent.columnActions"),
|
||
edit: t("aiAgent.configure"),
|
||
delete: t("aiAgent.delete"),
|
||
processing: t("aiAgent.processing"),
|
||
moreActions: (item) => t("aiAgent.moreActions", { name: item.name }),
|
||
loadFailed: t("aiAgent.loadFailed"),
|
||
saveFailed: t("aiAgent.saveFailed"),
|
||
deleteFailed: t("aiAgent.deleteFailed"),
|
||
created: (payload) => t("aiAgent.created", { name: payload.name }),
|
||
updated: (_item, payload) => t("aiAgent.updated", { name: payload.name }),
|
||
deleted: (item) => t("aiAgent.deleted", { name: item.name }),
|
||
}}
|
||
/>
|
||
<ProjectDialog
|
||
open={configOpen}
|
||
onOpenChange={(open) => {
|
||
setConfigOpen(open);
|
||
if (!open) setConfigAgentId(null);
|
||
}}
|
||
title={t("aiAgent.configure")}
|
||
size="xxl"
|
||
defaultFullscreen
|
||
bodyScrollable={false}
|
||
headerClassName="sr-only"
|
||
closeOnEsc={false}
|
||
>
|
||
{configOpen ? (
|
||
<AIAgentConfigWorkbench
|
||
agentId={configAgentId}
|
||
onAgentCreated={(agent) => setConfigAgentId(agent.id)}
|
||
onAgentSaved={() => crudActions?.onRefresh()}
|
||
/>
|
||
) : null}
|
||
</ProjectDialog>
|
||
</>
|
||
);
|
||
}
|