feat(pkb): Complete PKB module frontend migration with V3 design
Summary: - Implement PKB Dashboard and Workspace pages based on V3 prototype - Add single-layer header with integrated Tab navigation - Implement 3 work modes: Full Text, Deep Read, Batch Processing - Integrate Ant Design X Chat component for AI conversations - Create BatchModeComplete with template selection and document processing - Add compact work mode selector with dropdown design Backend: - Migrate PKB controllers and services to /modules/pkb structure - Register v2 API routes at /api/v2/pkb/knowledge - Maintain dual API routes for backward compatibility Technical details: - Use Zustand for state management - Handle SSE streaming responses for AI chat - Support document selection for Deep Read mode - Implement batch processing with progress tracking Known issues: - Batch processing API integration pending - Knowledge assets page navigation needs optimization Status: Frontend functional, pending refinement
This commit is contained in:
@@ -0,0 +1,510 @@
|
||||
/**
|
||||
* 完整的批处理模式组件
|
||||
* 精细化设计，支持模板选择、文档选择、执行和结果展示
|
||||
*/
|
||||
|
||||
import React, { useState, useEffect } from 'react';
|
||||
import { Button, Table, Progress, Alert, message, Card, Steps, Checkbox, Radio } from 'antd';
|
||||
import { Play, Download, RotateCw, FileText, CheckCircle2, Zap } from 'lucide-react';
|
||||
import type { KnowledgeBase, Document } from '../../api/knowledgeBaseApi';
|
||||
|
||||
interface BatchModeCompleteProps {
|
||||
kbId: string;
|
||||
kbInfo: KnowledgeBase;
|
||||
documents: Document[];
|
||||
template: string;
|
||||
}
|
||||
|
||||
interface BatchTemplate {
|
||||
id: string;
|
||||
name: string;
|
||||
description: string;
|
||||
fields: { key: string; label: string }[];
|
||||
}
|
||||
|
||||
interface BatchResult {
|
||||
documentId: string;
|
||||
documentName: string;
|
||||
status: 'pending' | 'processing' | 'completed' | 'error';
|
||||
progress: number;
|
||||
result?: Record<string, string>;
|
||||
error?: string;
|
||||
}
|
||||
|
||||
const TEMPLATES: BatchTemplate[] = [
|
||||
{
|
||||
id: 'clinicalResearch',
|
||||
name: '临床研究信息提取',
|
||||
description: '提取研究目的、方法、样本量、结论等核心信息',
|
||||
fields: [
|
||||
{ key: 'title', label: '研究标题' },
|
||||
{ key: 'purpose', label: '研究目的' },
|
||||
{ key: 'method', label: '研究方法' },
|
||||
{ key: 'sampleSize', label: '样本量' },
|
||||
{ key: 'intervention', label: '干预措施' },
|
||||
{ key: 'outcome', label: '主要结局' },
|
||||
{ key: 'conclusion', label: '研究结论' },
|
||||
{ key: 'limitation', label: '研究局限' },
|
||||
],
|
||||
},
|
||||
{
|
||||
id: 'drug_safety',
|
||||
name: '药物安全性分析',
|
||||
description: '提取药物不良反应、禁忌症、注意事项等',
|
||||
fields: [
|
||||
{ key: 'drugName', label: '药物名称' },
|
||||
{ key: 'adverseReactions', label: '不良反应' },
|
||||
{ key: 'contraindications', label: '禁忌症' },
|
||||
{ key: 'warnings', label: '警告事项' },
|
||||
],
|
||||
},
|
||||
{
|
||||
id: 'patient_baseline',
|
||||
name: '患者基线特征',
|
||||
description: '提取患者年龄、性别、诊断、既往史等基线信息',
|
||||
fields: [
|
||||
{ key: 'age', label: '年龄' },
|
||||
{ key: 'gender', label: '性别' },
|
||||
{ key: 'diagnosis', label: '主要诊断' },
|
||||
{ key: 'comorbidities', label: '合并症' },
|
||||
],
|
||||
},
|
||||
];
|
||||
|
||||
export const BatchModeComplete: React.FC<BatchModeCompleteProps> = ({
|
||||
kbId,
|
||||
documents,
|
||||
template: initialTemplate,
|
||||
}) => {
|
||||
const [step, setStep] = useState(0); // 0: 配置, 1: 执行中, 2: 结果
|
||||
const [selectedTemplate, setSelectedTemplate] = useState<BatchTemplate | null>(TEMPLATES[0]);
|
||||
const [selectedDocs, setSelectedDocs] = useState<string[]>([]);
|
||||
const [results, setResults] = useState<BatchResult[]>([]);
|
||||
const [, setIsExecuting] = useState(false);
|
||||
|
||||
const completedDocs = documents.filter(doc => doc.status === 'completed');
|
||||
|
||||
useEffect(() => {
|
||||
// 初始化模板（如果有传入则使用，否则默认第一个）
|
||||
if (initialTemplate) {
|
||||
const template = TEMPLATES.find(t => t.id === initialTemplate);
|
||||
if (template) setSelectedTemplate(template);
|
||||
}
|
||||
}, [initialTemplate]);
|
||||
|
||||
// 处理文档选择
|
||||
const handleDocSelect = (docId: string, checked: boolean) => {
|
||||
if (checked) {
|
||||
if (selectedDocs.length >= 50) {
|
||||
message.warning('最多选择50篇文档');
|
||||
return;
|
||||
}
|
||||
setSelectedDocs(prev => [...prev, docId]);
|
||||
} else {
|
||||
setSelectedDocs(prev => prev.filter(id => id !== docId));
|
||||
}
|
||||
};
|
||||
|
||||
// 全选/取消全选
|
||||
const handleSelectAll = (checked: boolean) => {
|
||||
if (checked) {
|
||||
const allIds = completedDocs.slice(0, 50).map(d => d.id);
|
||||
setSelectedDocs(allIds);
|
||||
} else {
|
||||
setSelectedDocs([]);
|
||||
}
|
||||
};
|
||||
|
||||
// 执行批处理
|
||||
const handleExecute = async () => {
|
||||
if (selectedDocs.length < 3) {
|
||||
message.warning('请至少选择3篇文档');
|
||||
return;
|
||||
}
|
||||
|
||||
if (!selectedTemplate) {
|
||||
message.error('请选择批处理模板');
|
||||
return;
|
||||
}
|
||||
|
||||
setStep(1);
|
||||
setIsExecuting(true);
|
||||
|
||||
// 初始化结果
|
||||
const initialResults: BatchResult[] = selectedDocs.map(docId => {
|
||||
const doc = documents.find(d => d.id === docId);
|
||||
return {
|
||||
documentId: docId,
|
||||
documentName: doc?.filename || '未知文档',
|
||||
status: 'pending',
|
||||
progress: 0,
|
||||
};
|
||||
});
|
||||
setResults(initialResults);
|
||||
|
||||
try {
|
||||
// 调用批处理API
|
||||
const response = await fetch('/api/v2/pkb/batch-tasks/batch/execute', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
kb_id: kbId,
|
||||
template_id: selectedTemplate.id,
|
||||
document_ids: selectedDocs,
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error('批处理执行失败');
|
||||
}
|
||||
|
||||
const data = await response.json();
|
||||
const taskId = data.task_id;
|
||||
|
||||
// 轮询任务状态
|
||||
const pollInterval = setInterval(async () => {
|
||||
try {
|
||||
const statusRes = await fetch(`/api/v2/pkb/batch-tasks/${taskId}`);
|
||||
const statusData = await statusRes.json();
|
||||
|
||||
// 更新进度
|
||||
setResults(prev => prev.map((r, idx) => {
|
||||
const docResult = statusData.results?.[idx];
|
||||
if (docResult) {
|
||||
return {
|
||||
...r,
|
||||
status: docResult.status,
|
||||
progress: docResult.progress || 0,
|
||||
result: docResult.result,
|
||||
error: docResult.error,
|
||||
};
|
||||
}
|
||||
return r;
|
||||
}));
|
||||
|
||||
// 检查是否全部完成
|
||||
if (statusData.status === 'completed' || statusData.status === 'failed') {
|
||||
clearInterval(pollInterval);
|
||||
setIsExecuting(false);
|
||||
setStep(2);
|
||||
|
||||
if (statusData.status === 'completed') {
|
||||
message.success('批处理完成！');
|
||||
} else {
|
||||
message.error('批处理失败');
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('轮询任务状态失败:', error);
|
||||
}
|
||||
}, 2000);
|
||||
|
||||
} catch (error: unknown) {
|
||||
const errorMessage = error instanceof Error ? error.message : '执行失败';
|
||||
message.error(errorMessage);
|
||||
setIsExecuting(false);
|
||||
}
|
||||
};
|
||||
|
||||
// 导出Excel
|
||||
const handleExport = () => {
|
||||
if (!selectedTemplate) return;
|
||||
|
||||
// 构建CSV数据
|
||||
const headers = ['文档名称', ...selectedTemplate.fields.map(f => f.label)];
|
||||
const rows = results
|
||||
.filter(r => r.status === 'completed' && r.result)
|
||||
.map(r => [
|
||||
r.documentName,
|
||||
...selectedTemplate.fields.map(f => r.result?.[f.key] || '-'),
|
||||
]);
|
||||
|
||||
const csvContent = [
|
||||
headers.join(','),
|
||||
...rows.map(row => row.map(cell => `"${cell}"`).join(',')),
|
||||
].join('\n');
|
||||
|
||||
// 下载
|
||||
const blob = new Blob(['\ufeff' + csvContent], { type: 'text/csv;charset=utf-8;' });
|
||||
const link = document.createElement('a');
|
||||
link.href = URL.createObjectURL(blob);
|
||||
link.download = `批处理结果_${selectedTemplate.name}_${new Date().toISOString().split('T')[0]}.csv`;
|
||||
link.click();
|
||||
};
|
||||
|
||||
// 重置
|
||||
const handleReset = () => {
|
||||
setStep(0);
|
||||
setSelectedDocs([]);
|
||||
setResults([]);
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="h-full flex flex-col overflow-hidden bg-white">
|
||||
{/* 步骤指示器 */}
|
||||
<div className="flex-shrink-0 px-8 py-5 border-b border-gray-100 bg-gray-50/50">
|
||||
<Steps
|
||||
current={step}
|
||||
size="small"
|
||||
items={[
|
||||
{ title: '配置任务', description: '选择模板和文档' },
|
||||
{ title: '执行中', description: '正在处理文档' },
|
||||
{ title: '查看结果', description: '导出数据' },
|
||||
]}
|
||||
/>
|
||||
</div>
|
||||
|
||||
{/* 步骤1: 配置 */}
|
||||
{step === 0 && (
|
||||
<div className="flex-1 overflow-y-auto p-6">
|
||||
<div className="max-w-4xl mx-auto space-y-6">
|
||||
{/* 模板选择 */}
|
||||
<Card
|
||||
title={
|
||||
<div className="flex items-center">
|
||||
<Zap className="w-4 h-4 mr-2 text-orange-500" />
|
||||
<span>选择批处理模板</span>
|
||||
</div>
|
||||
}
|
||||
size="small"
|
||||
className="shadow-sm"
|
||||
>
|
||||
<Radio.Group
|
||||
value={selectedTemplate?.id}
|
||||
onChange={(e) => {
|
||||
const template = TEMPLATES.find(t => t.id === e.target.value);
|
||||
setSelectedTemplate(template || null);
|
||||
}}
|
||||
className="w-full"
|
||||
>
|
||||
<div className="space-y-3">
|
||||
{TEMPLATES.map(t => (
|
||||
<div
|
||||
key={t.id}
|
||||
className={`p-4 border rounded-lg cursor-pointer transition-all ${
|
||||
selectedTemplate?.id === t.id
|
||||
? 'border-blue-400 bg-blue-50/50'
|
||||
: 'border-gray-200 hover:border-blue-200 hover:bg-gray-50'
|
||||
}`}
|
||||
onClick={() => setSelectedTemplate(t)}
|
||||
>
|
||||
<Radio value={t.id} className="w-full">
|
||||
<div className="ml-2">
|
||||
<div className="font-medium text-slate-800">{t.name}</div>
|
||||
<div className="text-xs text-slate-500 mt-1">{t.description}</div>
|
||||
<div className="flex flex-wrap gap-1.5 mt-2">
|
||||
{t.fields.slice(0, 5).map(f => (
|
||||
<span key={f.key} className="text-xs bg-gray-100 text-slate-600 px-2 py-0.5 rounded">
|
||||
{f.label}
|
||||
</span>
|
||||
))}
|
||||
{t.fields.length > 5 && (
|
||||
<span className="text-xs text-slate-400">+{t.fields.length - 5}项</span>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
</Radio>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
</Radio.Group>
|
||||
</Card>
|
||||
|
||||
{/* 文档选择 */}
|
||||
<Card
|
||||
title={
|
||||
<div className="flex items-center justify-between w-full">
|
||||
<div className="flex items-center">
|
||||
<FileText className="w-4 h-4 mr-2 text-blue-500" />
|
||||
<span>选择文档</span>
|
||||
<span className="ml-2 text-xs text-slate-400 font-normal">
|
||||
({completedDocs.length} 篇可用)
|
||||
</span>
|
||||
</div>
|
||||
<Checkbox
|
||||
checked={selectedDocs.length === completedDocs.length && completedDocs.length > 0}
|
||||
indeterminate={selectedDocs.length > 0 && selectedDocs.length < completedDocs.length}
|
||||
onChange={(e) => handleSelectAll(e.target.checked)}
|
||||
>
|
||||
全选
|
||||
</Checkbox>
|
||||
</div>
|
||||
}
|
||||
size="small"
|
||||
className="shadow-sm"
|
||||
>
|
||||
<div className="space-y-1 max-h-[280px] overflow-y-auto">
|
||||
{completedDocs.length === 0 ? (
|
||||
<div className="text-center py-8 text-slate-400">
|
||||
<FileText className="w-8 h-8 mx-auto mb-2 opacity-50" />
|
||||
<p>暂无已完成解析的文档</p>
|
||||
</div>
|
||||
) : (
|
||||
completedDocs.map(doc => (
|
||||
<div
|
||||
key={doc.id}
|
||||
className={`flex items-center p-3 rounded-lg cursor-pointer transition-all ${
|
||||
selectedDocs.includes(doc.id)
|
||||
? 'bg-blue-50 border border-blue-200'
|
||||
: 'hover:bg-gray-50 border border-transparent'
|
||||
}`}
|
||||
onClick={() => handleDocSelect(doc.id, !selectedDocs.includes(doc.id))}
|
||||
>
|
||||
<Checkbox
|
||||
checked={selectedDocs.includes(doc.id)}
|
||||
onChange={(e) => {
|
||||
e.stopPropagation();
|
||||
handleDocSelect(doc.id, e.target.checked);
|
||||
}}
|
||||
/>
|
||||
<div className="w-8 h-8 bg-red-50 text-red-500 rounded flex items-center justify-center ml-3 mr-3 flex-shrink-0">
|
||||
<FileText className="w-4 h-4" />
|
||||
</div>
|
||||
<div className="flex-1 min-w-0">
|
||||
<div className="text-sm font-medium text-slate-700 truncate">{doc.filename}</div>
|
||||
<div className="text-xs text-slate-400">
|
||||
{doc.tokensCount ? `${(doc.tokensCount / 1000).toFixed(0)}k tokens` : ''}
|
||||
</div>
|
||||
</div>
|
||||
{selectedDocs.includes(doc.id) && (
|
||||
<CheckCircle2 className="w-4 h-4 text-blue-500 flex-shrink-0" />
|
||||
)}
|
||||
</div>
|
||||
))
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="mt-4 pt-4 border-t border-gray-100">
|
||||
<Alert
|
||||
message={
|
||||
<span className="text-sm">
|
||||
已选择 <strong className="text-blue-600">{selectedDocs.length}</strong> 篇文档
|
||||
{selectedDocs.length < 3 && <span className="text-orange-600 ml-2">（至少需要3篇）</span>}
|
||||
</span>
|
||||
}
|
||||
type={selectedDocs.length >= 3 ? 'success' : 'warning'}
|
||||
showIcon
|
||||
/>
|
||||
</div>
|
||||
</Card>
|
||||
|
||||
{/* 开始按钮 */}
|
||||
<Button
|
||||
type="primary"
|
||||
size="large"
|
||||
icon={<Play className="w-4 h-4" />}
|
||||
onClick={handleExecute}
|
||||
disabled={selectedDocs.length < 3 || !selectedTemplate}
|
||||
className="w-full h-12 text-base font-medium shadow-lg"
|
||||
>
|
||||
开始批处理 ({selectedDocs.length} 篇文档)
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* 步骤2: 执行中 */}
|
||||
{step === 1 && (
|
||||
<div className="flex-1 overflow-y-auto p-6">
|
||||
<div className="max-w-3xl mx-auto">
|
||||
<Card
|
||||
title={
|
||||
<div className="flex items-center">
|
||||
<Zap className="w-4 h-4 mr-2 text-orange-500 animate-pulse" />
|
||||
<span>批处理执行中...</span>
|
||||
</div>
|
||||
}
|
||||
size="small"
|
||||
className="shadow-sm"
|
||||
>
|
||||
<div className="space-y-4">
|
||||
{results.map((result) => (
|
||||
<div key={result.documentId} className="border-b border-gray-100 pb-4 last:border-b-0 last:pb-0">
|
||||
<div className="flex justify-between items-center mb-2">
|
||||
<div className="flex items-center min-w-0 flex-1">
|
||||
<FileText className="w-4 h-4 text-red-400 mr-2 flex-shrink-0" />
|
||||
<span className="font-medium text-sm text-slate-700 truncate">
|
||||
{result.documentName}
|
||||
</span>
|
||||
</div>
|
||||
<span className="text-xs ml-2 flex-shrink-0">
|
||||
{result.status === 'completed' && <span className="text-green-600">✅ 完成</span>}
|
||||
{result.status === 'processing' && <span className="text-blue-600">⏳ 处理中</span>}
|
||||
{result.status === 'error' && <span className="text-red-600">❌ 失败</span>}
|
||||
{result.status === 'pending' && <span className="text-slate-400">⏸️ 等待</span>}
|
||||
</span>
|
||||
</div>
|
||||
<Progress
|
||||
percent={result.progress}
|
||||
status={result.status === 'error' ? 'exception' : result.status === 'completed' ? 'success' : 'active'}
|
||||
size="small"
|
||||
strokeColor={result.status === 'completed' ? '#22c55e' : '#3b82f6'}
|
||||
/>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
</Card>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* 步骤3: 结果 */}
|
||||
{step === 2 && selectedTemplate && (
|
||||
<div className="flex-1 overflow-hidden flex flex-col p-6">
|
||||
<div className="flex-shrink-0 mb-4 flex justify-between items-center">
|
||||
<div className="flex items-center text-sm text-slate-600">
|
||||
<CheckCircle2 className="w-4 h-4 text-green-500 mr-2" />
|
||||
已完成 {results.filter(r => r.status === 'completed').length} / {results.length} 篇文档处理
|
||||
</div>
|
||||
<div className="flex space-x-3">
|
||||
<Button icon={<RotateCw className="w-4 h-4" />} onClick={handleReset}>
|
||||
重新开始
|
||||
</Button>
|
||||
<Button
|
||||
type="primary"
|
||||
icon={<Download className="w-4 h-4" />}
|
||||
onClick={handleExport}
|
||||
>
|
||||
导出Excel
|
||||
</Button>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div className="flex-1 overflow-auto bg-white border border-gray-200 rounded-lg shadow-sm">
|
||||
<Table
|
||||
dataSource={results.filter(r => r.status === 'completed')}
|
||||
rowKey="documentId"
|
||||
pagination={false}
|
||||
scroll={{ x: 'max-content', y: '100%' }}
|
||||
size="small"
|
||||
columns={[
|
||||
{
|
||||
title: '文档名称',
|
||||
dataIndex: 'documentName',
|
||||
key: 'documentName',
|
||||
fixed: 'left',
|
||||
width: 200,
|
||||
ellipsis: true,
|
||||
render: (text) => (
|
||||
<div className="flex items-center">
|
||||
<FileText className="w-4 h-4 text-red-400 mr-2 flex-shrink-0" />
|
||||
<span className="truncate">{text}</span>
|
||||
</div>
|
||||
),
|
||||
},
|
||||
...selectedTemplate.fields.map(field => ({
|
||||
title: field.label,
|
||||
dataIndex: ['result', field.key],
|
||||
key: field.key,
|
||||
width: 180,
|
||||
ellipsis: true,
|
||||
render: (text: string) => text || '-',
|
||||
})),
|
||||
]}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
};
|
||||
Reference in New Issue
Block a user