mlogclub
345 lines
11 KiB
TypeScript
345 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: "工作流状态",
|
|
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">
|
|
未发布工作流，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 mcpTools = item.mcpTools ?? [];
|
|
const mcpServerCodes = Array.from(
|
|
new Set(mcpTools.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">{mcpTools.length} MCP Tools</Badge>
|
|
</div>
|
|
<div className="flex flex-wrap gap-1">
|
|
{mcpServerCodes.length === 0 ? (
|
|
<span className="text-sm text-muted-foreground">
|
|
{t("aiAgent.noMcpServer")}
|
|
</span>
|
|
) : (
|
|
mcpServerCodes.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"
|
|
allowFullscreen
|
|
contentClassName="h-[800px] max-h-[calc(100vh-2rem)]"
|
|
bodyScrollable={false}
|
|
headerClassName="sr-only"
|
|
closeOnEsc={false}
|
|
>
|
|
{configOpen ? (
|
|
<AIAgentConfigWorkbench
|
|
agentId={configAgentId}
|
|
onAgentCreated={(agent) => setConfigAgentId(agent.id)}
|
|
onAgentSaved={() => crudActions?.onRefresh()}
|
|
onCancel={() => setConfigOpen(false)}
|
|
/>
|
|
) : null}
|
|
</ProjectDialog>
|
|
</>
|
|
);
|
|
}
|