feat(knowledge-management): 添加知识管理搜索功能和统计接口

- 新增知识条目搜索查询和响应DTO
- 实现知识管理统计功能,包括总数、文件数和总大小
- 添加数据库查询方法支持文件搜索和统计计算
- 创建知识条目搜索控制器提供REST API
- 在前端添加知识管理搜索页面和相关组件
- 更新前端路由配置添加搜索页面入口
- 移除RAG索引服务中的重复统计功能
- 优化前端页面统计数据显示和刷新逻辑
This commit is contained in:
2026-01-31 09:30:37 +08:00
parent 97170a90fe
commit 790385bd80
24 changed files with 598 additions and 159 deletions

View File

@@ -1,48 +1,23 @@
import { useCallback, useEffect, useState } from "react";
import { Card, Button, Table, Tooltip, message, Statistic } from "antd";
import { useState } from "react";
import { Card, Button, Table, Tooltip, message } from "antd";
import { DeleteOutlined, EditOutlined } from "@ant-design/icons";
import { SearchControls } from "@/components/SearchControls";
import { useNavigate } from "react-router";
import CardView from "@/components/CardView";
import {
deleteKnowledgeBaseByIdUsingDelete,
getKnowledgeBaseStatisticsUsingGet,
queryKnowledgeBasesUsingPost,
} from "../knowledge-base.api";
import useFetchData from "@/hooks/useFetchData";
import { KnowledgeBaseItem, KnowledgeBaseStatistics } from "../knowledge-base.model";
import { KnowledgeBaseItem } from "../knowledge-base.model";
import CreateKnowledgeBase from "../components/CreateKnowledgeBase";
import { mapKnowledgeBase } from "../knowledge-base.const";
import { formatBytes } from "@/utils/unit";
type StatisticsItem = {
title: string;
value: number | string;
};
const DEFAULT_STATISTICS: StatisticsItem[] = [
{
title: "知识库总数",
value: 0,
},
{
title: "文件总数",
value: 0,
},
{
title: "总大小",
value: "0 B",
},
];
export default function KnowledgeBasePage() {
const navigate = useNavigate();
const [viewMode, setViewMode] = useState<"card" | "list">("card");
const [isEdit, setIsEdit] = useState(false);
const [currentKB, setCurrentKB] = useState<KnowledgeBaseItem | null>(null);
const [statisticsData, setStatisticsData] = useState<StatisticsItem[]>(
DEFAULT_STATISTICS
);
const {
loading,
tableData,
@@ -57,43 +32,11 @@ export default function KnowledgeBasePage() {
(kb) => mapKnowledgeBase(kb, false) // 在首页不显示索引模型和文本理解模型字段
);
const fetchStatistics = useCallback(async () => {
try {
const { data } = await getKnowledgeBaseStatisticsUsingGet();
const stats = data as KnowledgeBaseStatistics | undefined;
setStatisticsData([
{
title: "知识库总数",
value: stats?.totalKnowledgeBases ?? 0,
},
{
title: "文件总数",
value: stats?.totalFiles ?? 0,
},
{
title: "总大小",
value: formatBytes(stats?.totalSize ?? 0),
},
]);
} catch {
message.error("统计数据加载失败");
setStatisticsData(DEFAULT_STATISTICS);
}
}, []);
const refreshAll = useCallback(async () => {
await Promise.all([fetchData(), fetchStatistics()]);
}, [fetchData, fetchStatistics]);
useEffect(() => {
fetchStatistics();
}, [fetchStatistics]);
const handleDeleteKB = async (kb: KnowledgeBaseItem) => {
try {
await deleteKnowledgeBaseByIdUsingDelete(kb.id);
message.success("知识库删除成功");
await refreshAll();
fetchData();
} catch {
message.error("知识库删除失败");
}
@@ -197,7 +140,7 @@ export default function KnowledgeBasePage() {
isEdit={isEdit}
data={currentKB}
onUpdate={() => {
refreshAll();
fetchData();
}}
onClose={() => {
setIsEdit(false);
@@ -207,20 +150,6 @@ export default function KnowledgeBasePage() {
</div>
</div>
<div className="grid grid-cols-1 gap-4">
<Card>
<div className="grid grid-cols-3">
{statisticsData.map((item) => (
<Statistic
title={item.title}
key={item.title}
value={`${item.value}`}
/>
))}
</div>
</Card>
</div>
<SearchControls
searchTerm={searchParams.keyword}
onSearchChange={handleKeywordChange}
@@ -231,7 +160,7 @@ export default function KnowledgeBasePage() {
viewMode={viewMode}
onViewModeChange={setViewMode}
showViewToggle
onReload={refreshAll}
onReload={fetchData}
/>
{viewMode === "card" ? (
<CardView

View File

@@ -38,11 +38,6 @@ export function queryKnowledgeBaseFilesSearchUsingGet(params: RequestParams) {
return get("/api/knowledge-base/files/search", params);
}
// 获取知识库统计
export function getKnowledgeBaseStatisticsUsingGet() {
return get("/api/knowledge-base/statistics");
}
// 添加文件到知识库
export function addKnowledgeBaseFilesUsingPost(baseId: string, data: RequestPayload) {
return post(`/api/knowledge-base/${baseId}/files`, data);

View File

@@ -25,12 +25,6 @@ export interface KnowledgeBaseItem {
chat: never;
}
export interface KnowledgeBaseStatistics {
totalKnowledgeBases: number;
totalFiles: number;
totalSize: number;
}
export interface KBFile {
id: string;
fileName: string;