Revert "feat: fix the problem in the Operator Market frontend pages"

This commit is contained in:
Kecheng Sha
2025-12-29 12:00:37 +08:00
committed by GitHub
parent 8f30f71a68
commit 0df7a872e4
213 changed files with 45537 additions and 45547 deletions

View File

@@ -1,332 +1,332 @@
import type React from "react";
import { useEffect, useState } from "react";
import { Table, Badge, Button, Breadcrumb, Tooltip, App, Card, Input, Empty, Spin } from "antd";
import {
DeleteOutlined,
EditOutlined,
ReloadOutlined,
} from "@ant-design/icons";
import { useNavigate, useParams } from "react-router";
import DetailHeader from "@/components/DetailHeader";
import { SearchControls } from "@/components/SearchControls";
import { KBFile, KnowledgeBaseItem } from "../knowledge-base.model";
import { mapFileData, mapKnowledgeBase } from "../knowledge-base.const";
import {
deleteKnowledgeBaseByIdUsingDelete,
deleteKnowledgeBaseFileByIdUsingDelete,
queryKnowledgeBaseByIdUsingGet,
queryKnowledgeBaseFilesUsingGet,
retrieveKnowledgeBaseContent,
} from "../knowledge-base.api";
import useFetchData from "@/hooks/useFetchData";
import AddDataDialog from "../components/AddDataDialog";
import CreateKnowledgeBase from "../components/CreateKnowledgeBase";
interface StatisticItem {
icon?: React.ReactNode;
label: string;
value: string | number;
}
interface RagChunk {
id: string;
text: string;
metadata: string;
}
interface RecallResult {
score: number;
entity: RagChunk;
id?: string | object;
primaryKey?: string;
}
const KnowledgeBaseDetailPage: React.FC = () => {
const navigate = useNavigate();
const { message } = App.useApp();
const { id } = useParams<{ id: string }>();
const [knowledgeBase, setKnowledgeBase] = useState<KnowledgeBaseItem | undefined>(undefined);
const [showEdit, setShowEdit] = useState(false);
const [activeTab, setActiveTab] = useState<'fileList' | 'recallTest'>('fileList');
const [recallLoading, setRecallLoading] = useState(false);
const [recallResults, setRecallResults] = useState<RecallResult[]>([]);
const [recallQuery, setRecallQuery] = useState("");
const fetchKnowledgeBaseDetails = async (id: string) => {
const { data } = await queryKnowledgeBaseByIdUsingGet(id);
setKnowledgeBase(mapKnowledgeBase(data));
};
useEffect(() => {
if (id) {
fetchKnowledgeBaseDetails(id);
}
}, [id]);
const {
loading,
tableData: files,
searchParams,
pagination,
fetchData: fetchFiles,
setSearchParams,
handleFiltersChange,
handleKeywordChange,
} = useFetchData<KBFile>(
(params) => id ? queryKnowledgeBaseFilesUsingGet(id, params) : Promise.resolve({ data: [] }),
mapFileData
);
// File table logic
const handleDeleteFile = async (file: KBFile) => {
try {
await deleteKnowledgeBaseFileByIdUsingDelete(knowledgeBase!.id, {
ids: [file.id]
});
message.success("文件已删除");
fetchFiles();
} catch {
message.error("文件删除失败");
}
};
const handleDeleteKB = async (kb: KnowledgeBaseItem) => {
await deleteKnowledgeBaseByIdUsingDelete(kb.id);
message.success("知识库已删除");
navigate("/data/knowledge-base");
};
const handleRefreshPage = () => {
if (knowledgeBase) {
fetchKnowledgeBaseDetails(knowledgeBase.id);
}
fetchFiles();
setShowEdit(false);
};
const handleRecallTest = async () => {
if (!recallQuery || !knowledgeBase?.id) return;
setRecallLoading(true);
try {
const result = await retrieveKnowledgeBaseContent({
query: recallQuery,
topK: 10,
threshold: 0.2,
knowledgeBaseIds: [knowledgeBase.id],
});
setRecallResults(result?.data || []);
} catch {
setRecallResults([]);
}
setRecallLoading(false);
};
const operations = [
{
key: "edit",
label: "编辑知识库",
icon: <EditOutlined className="w-4 h-4" />,
onClick: () => {
setShowEdit(true);
},
},
{
key: "refresh",
label: "刷新知识库",
icon: <ReloadOutlined className="w-4 h-4" />,
onClick: () => {
handleRefreshPage();
},
},
{
key: "delete",
label: "删除知识库",
danger: true,
confirm: {
title: "确认删除该知识库吗?",
description: "删除后将无法恢复,请谨慎操作。",
cancelText: "取消",
okText: "删除",
okType: "danger",
onConfirm: () => knowledgeBase && handleDeleteKB(knowledgeBase),
},
icon: <DeleteOutlined className="w-4 h-4" />,
},
];
const fileOps = [
{
key: "delete",
label: "删除文件",
icon: <DeleteOutlined className="w-4 h-4" />,
danger: true,
onClick: handleDeleteFile,
},
];
const fileColumns = [
{
title: "文件名",
dataIndex: "name",
key: "name",
width: 200,
ellipsis: true,
fixed: "left" as const,
},
{
title: "状态",
dataIndex: "status",
key: "vectorizationStatus",
width: 120,
render: (status: unknown) => {
if (typeof status === 'object' && status !== null) {
const s = status as { color?: string; label?: string };
return <Badge color={s.color} text={s.label} />;
}
return <Badge color="default" text={String(status)} />;
},
},
{
title: "分块数",
dataIndex: "chunkCount",
key: "chunkCount",
width: 100,
ellipsis: true,
},
{
title: "创建时间",
dataIndex: "createdAt",
key: "createdAt",
ellipsis: true,
width: 180,
},
{
title: "更新时间",
dataIndex: "updatedAt",
key: "updatedAt",
ellipsis: true,
width: 180,
},
{
title: "操作",
key: "actions",
align: "right" as const,
width: 100,
render: (_: unknown, file: KBFile) => (
<div>
{fileOps.map((op) => (
<Tooltip key={op.key} title={op.label}>
<Button
type="text"
icon={op.icon}
danger={op?.danger}
onClick={() => op.onClick(file)}
/>
</Tooltip>
))}
</div>
),
},
];
return (
<div className="h-full flex flex-col">
<div className="mb-4">
<Breadcrumb>
<Breadcrumb.Item>
<a onClick={() => navigate("/data/knowledge-base")}></a>
</Breadcrumb.Item>
<Breadcrumb.Item>{knowledgeBase?.name}</Breadcrumb.Item>
</Breadcrumb>
</div>
<DetailHeader
data={knowledgeBase}
statistics={knowledgeBase && Array.isArray((knowledgeBase as { statistics?: StatisticItem[] }).statistics)
? ((knowledgeBase as { statistics?: StatisticItem[] }).statistics ?? [])
: []}
operations={operations}
/>
<CreateKnowledgeBase
showBtn={false}
isEdit={showEdit}
data={knowledgeBase}
onUpdate={handleRefreshPage}
onClose={() => setShowEdit(false)}
/>
<div className="flex-1 border-card p-6 mt-4">
<div className="flex items-center justify-between mb-4 gap-3">
<div className="flex items-center gap-2">
<Button type={activeTab === 'fileList' ? 'primary' : 'default'} onClick={() => setActiveTab('fileList')}>
</Button>
<Button type={activeTab === 'recallTest' ? 'primary' : 'default'} onClick={() => setActiveTab('recallTest')}>
</Button>
</div>
{activeTab === 'fileList' && (
<>
<div className="flex-1">
<SearchControls
searchTerm={searchParams.keyword}
onSearchChange={handleKeywordChange}
searchPlaceholder="搜索文件名..."
filters={[]}
onFiltersChange={handleFiltersChange}
onClearFilters={() => setSearchParams({ ...searchParams, filter: { type: [], status: [], tags: [] } })}
showViewToggle={false}
showReload={false}
/>
</div>
<AddDataDialog knowledgeBase={knowledgeBase} onDataAdded={handleRefreshPage} />
</>
)}
</div>
{activeTab === 'fileList' ? (
<Table
loading={loading}
columns={fileColumns}
dataSource={files}
rowKey="id"
pagination={pagination}
scroll={{ y: "calc(100vh - 30rem)" }}
/>
) : (
<div className="p-2">
<div style={{ fontSize: 14, fontWeight: 300, marginBottom: 8 }}></div>
<div className="flex items-center mb-4">
<Input.Search
value={recallQuery}
onChange={e => setRecallQuery(e.target.value)}
onSearch={handleRecallTest}
placeholder="请输入召回测试问题"
enterButton="检索"
loading={recallLoading}
style={{ width: "100%", fontSize: 18, height: 48 }}
/>
</div>
{recallLoading ? (
<Spin className="mt-8" />
) : recallResults.length === 0 ? (
<Empty description="暂无召回结果" />
) : (
<div className="grid grid-cols-1 md:grid-cols-2 gap-4">
{recallResults.map((item, idx) => (
<Card key={idx} title={`得分:${item.score?.toFixed(4) ?? "-"}`}
extra={<span style={{ fontSize: 12 }}>ID: {item.entity?.id ?? "-"}</span>}
style={{ wordBreak: "break-all" }}
>
<div style={{ marginBottom: 8, fontWeight: 500 }}>{item.entity?.text ?? ""}</div>
<div style={{ fontSize: 12, color: '#888' }}>
metadata: <pre style={{ whiteSpace: 'pre-wrap', wordBreak: 'break-all', margin: 0 }}>{item.entity?.metadata}</pre>
</div>
</Card>
))}
</div>
)}
</div>
)}
</div>
</div>
);
};
export default KnowledgeBaseDetailPage;
import type React from "react";
import { useEffect, useState } from "react";
import { Table, Badge, Button, Breadcrumb, Tooltip, App, Card, Input, Empty, Spin } from "antd";
import {
DeleteOutlined,
EditOutlined,
ReloadOutlined,
} from "@ant-design/icons";
import { useNavigate, useParams } from "react-router";
import DetailHeader from "@/components/DetailHeader";
import { SearchControls } from "@/components/SearchControls";
import { KBFile, KnowledgeBaseItem } from "../knowledge-base.model";
import { mapFileData, mapKnowledgeBase } from "../knowledge-base.const";
import {
deleteKnowledgeBaseByIdUsingDelete,
deleteKnowledgeBaseFileByIdUsingDelete,
queryKnowledgeBaseByIdUsingGet,
queryKnowledgeBaseFilesUsingGet,
retrieveKnowledgeBaseContent,
} from "../knowledge-base.api";
import useFetchData from "@/hooks/useFetchData";
import AddDataDialog from "../components/AddDataDialog";
import CreateKnowledgeBase from "../components/CreateKnowledgeBase";
interface StatisticItem {
icon?: React.ReactNode;
label: string;
value: string | number;
}
interface RagChunk {
id: string;
text: string;
metadata: string;
}
interface RecallResult {
score: number;
entity: RagChunk;
id?: string | object;
primaryKey?: string;
}
const KnowledgeBaseDetailPage: React.FC = () => {
const navigate = useNavigate();
const { message } = App.useApp();
const { id } = useParams<{ id: string }>();
const [knowledgeBase, setKnowledgeBase] = useState<KnowledgeBaseItem | undefined>(undefined);
const [showEdit, setShowEdit] = useState(false);
const [activeTab, setActiveTab] = useState<'fileList' | 'recallTest'>('fileList');
const [recallLoading, setRecallLoading] = useState(false);
const [recallResults, setRecallResults] = useState<RecallResult[]>([]);
const [recallQuery, setRecallQuery] = useState("");
const fetchKnowledgeBaseDetails = async (id: string) => {
const { data } = await queryKnowledgeBaseByIdUsingGet(id);
setKnowledgeBase(mapKnowledgeBase(data));
};
useEffect(() => {
if (id) {
fetchKnowledgeBaseDetails(id);
}
}, [id]);
const {
loading,
tableData: files,
searchParams,
pagination,
fetchData: fetchFiles,
setSearchParams,
handleFiltersChange,
handleKeywordChange,
} = useFetchData<KBFile>(
(params) => id ? queryKnowledgeBaseFilesUsingGet(id, params) : Promise.resolve({ data: [] }),
mapFileData
);
// File table logic
const handleDeleteFile = async (file: KBFile) => {
try {
await deleteKnowledgeBaseFileByIdUsingDelete(knowledgeBase!.id, {
ids: [file.id]
});
message.success("文件已删除");
fetchFiles();
} catch {
message.error("文件删除失败");
}
};
const handleDeleteKB = async (kb: KnowledgeBaseItem) => {
await deleteKnowledgeBaseByIdUsingDelete(kb.id);
message.success("知识库已删除");
navigate("/data/knowledge-base");
};
const handleRefreshPage = () => {
if (knowledgeBase) {
fetchKnowledgeBaseDetails(knowledgeBase.id);
}
fetchFiles();
setShowEdit(false);
};
const handleRecallTest = async () => {
if (!recallQuery || !knowledgeBase?.id) return;
setRecallLoading(true);
try {
const result = await retrieveKnowledgeBaseContent({
query: recallQuery,
topK: 10,
threshold: 0.2,
knowledgeBaseIds: [knowledgeBase.id],
});
setRecallResults(result?.data || []);
} catch {
setRecallResults([]);
}
setRecallLoading(false);
};
const operations = [
{
key: "edit",
label: "编辑知识库",
icon: <EditOutlined className="w-4 h-4" />,
onClick: () => {
setShowEdit(true);
},
},
{
key: "refresh",
label: "刷新知识库",
icon: <ReloadOutlined className="w-4 h-4" />,
onClick: () => {
handleRefreshPage();
},
},
{
key: "delete",
label: "删除知识库",
danger: true,
confirm: {
title: "确认删除该知识库吗?",
description: "删除后将无法恢复,请谨慎操作。",
cancelText: "取消",
okText: "删除",
okType: "danger",
onConfirm: () => knowledgeBase && handleDeleteKB(knowledgeBase),
},
icon: <DeleteOutlined className="w-4 h-4" />,
},
];
const fileOps = [
{
key: "delete",
label: "删除文件",
icon: <DeleteOutlined className="w-4 h-4" />,
danger: true,
onClick: handleDeleteFile,
},
];
const fileColumns = [
{
title: "文件名",
dataIndex: "name",
key: "name",
width: 200,
ellipsis: true,
fixed: "left" as const,
},
{
title: "状态",
dataIndex: "status",
key: "vectorizationStatus",
width: 120,
render: (status: unknown) => {
if (typeof status === 'object' && status !== null) {
const s = status as { color?: string; label?: string };
return <Badge color={s.color} text={s.label} />;
}
return <Badge color="default" text={String(status)} />;
},
},
{
title: "分块数",
dataIndex: "chunkCount",
key: "chunkCount",
width: 100,
ellipsis: true,
},
{
title: "创建时间",
dataIndex: "createdAt",
key: "createdAt",
ellipsis: true,
width: 180,
},
{
title: "更新时间",
dataIndex: "updatedAt",
key: "updatedAt",
ellipsis: true,
width: 180,
},
{
title: "操作",
key: "actions",
align: "right" as const,
width: 100,
render: (_: unknown, file: KBFile) => (
<div>
{fileOps.map((op) => (
<Tooltip key={op.key} title={op.label}>
<Button
type="text"
icon={op.icon}
danger={op?.danger}
onClick={() => op.onClick(file)}
/>
</Tooltip>
))}
</div>
),
},
];
return (
<div className="h-full flex flex-col">
<div className="mb-4">
<Breadcrumb>
<Breadcrumb.Item>
<a onClick={() => navigate("/data/knowledge-base")}></a>
</Breadcrumb.Item>
<Breadcrumb.Item>{knowledgeBase?.name}</Breadcrumb.Item>
</Breadcrumb>
</div>
<DetailHeader
data={knowledgeBase}
statistics={knowledgeBase && Array.isArray((knowledgeBase as { statistics?: StatisticItem[] }).statistics)
? ((knowledgeBase as { statistics?: StatisticItem[] }).statistics ?? [])
: []}
operations={operations}
/>
<CreateKnowledgeBase
showBtn={false}
isEdit={showEdit}
data={knowledgeBase}
onUpdate={handleRefreshPage}
onClose={() => setShowEdit(false)}
/>
<div className="flex-1 border-card p-6 mt-4">
<div className="flex items-center justify-between mb-4 gap-3">
<div className="flex items-center gap-2">
<Button type={activeTab === 'fileList' ? 'primary' : 'default'} onClick={() => setActiveTab('fileList')}>
</Button>
<Button type={activeTab === 'recallTest' ? 'primary' : 'default'} onClick={() => setActiveTab('recallTest')}>
</Button>
</div>
{activeTab === 'fileList' && (
<>
<div className="flex-1">
<SearchControls
searchTerm={searchParams.keyword}
onSearchChange={handleKeywordChange}
searchPlaceholder="搜索文件名..."
filters={[]}
onFiltersChange={handleFiltersChange}
onClearFilters={() => setSearchParams({ ...searchParams, filter: { type: [], status: [], tags: [] } })}
showViewToggle={false}
showReload={false}
/>
</div>
<AddDataDialog knowledgeBase={knowledgeBase} onDataAdded={handleRefreshPage} />
</>
)}
</div>
{activeTab === 'fileList' ? (
<Table
loading={loading}
columns={fileColumns}
dataSource={files}
rowKey="id"
pagination={pagination}
scroll={{ y: "calc(100vh - 30rem)" }}
/>
) : (
<div className="p-2">
<div style={{ fontSize: 14, fontWeight: 300, marginBottom: 8 }}></div>
<div className="flex items-center mb-4">
<Input.Search
value={recallQuery}
onChange={e => setRecallQuery(e.target.value)}
onSearch={handleRecallTest}
placeholder="请输入召回测试问题"
enterButton="检索"
loading={recallLoading}
style={{ width: "100%", fontSize: 18, height: 48 }}
/>
</div>
{recallLoading ? (
<Spin className="mt-8" />
) : recallResults.length === 0 ? (
<Empty description="暂无召回结果" />
) : (
<div className="grid grid-cols-1 md:grid-cols-2 gap-4">
{recallResults.map((item, idx) => (
<Card key={idx} title={`得分:${item.score?.toFixed(4) ?? "-"}`}
extra={<span style={{ fontSize: 12 }}>ID: {item.entity?.id ?? "-"}</span>}
style={{ wordBreak: "break-all" }}
>
<div style={{ marginBottom: 8, fontWeight: 500 }}>{item.entity?.text ?? ""}</div>
<div style={{ fontSize: 12, color: '#888' }}>
metadata: <pre style={{ whiteSpace: 'pre-wrap', wordBreak: 'break-all', margin: 0 }}>{item.entity?.metadata}</pre>
</div>
</Card>
))}
</div>
)}
</div>
)}
</div>
</div>
);
};
export default KnowledgeBaseDetailPage;