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,26 @@
|
||||
/**
|
||||
* 批处理模式组件
|
||||
*/
|
||||
|
||||
import React from 'react';
|
||||
import { BatchModeComplete } from './BatchModeComplete';
|
||||
import type { KnowledgeBase, Document } from '../../api/knowledgeBaseApi';
|
||||
|
||||
interface BatchModeProps {
|
||||
kbId: string;
|
||||
kbInfo?: KnowledgeBase;
|
||||
documents?: Document[];
|
||||
template?: string;
|
||||
}
|
||||
|
||||
export const BatchMode: React.FC<BatchModeProps> = ({ kbId, kbInfo, documents, template }) => {
|
||||
// 直接渲染完整的批处理组件
|
||||
return (
|
||||
<BatchModeComplete
|
||||
kbId={kbId}
|
||||
kbInfo={kbInfo!}
|
||||
documents={documents || []}
|
||||
template={template || 'clinicalResearch'}
|
||||
/>
|
||||
);
|
||||
};
|
||||
@@ -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>
|
||||
);
|
||||
};
|
||||
@@ -0,0 +1,116 @@
|
||||
/**
|
||||
* 逐篇精读模式组件 - ChatGPT风格全屏聊天
|
||||
*/
|
||||
|
||||
import React from 'react';
|
||||
import { Empty } from 'antd';
|
||||
import { FileText } from 'lucide-react';
|
||||
import { ChatContainer } from '@/shared/components/Chat';
|
||||
import type { KnowledgeBase, Document } from '../../api/knowledgeBaseApi';
|
||||
|
||||
interface DeepReadModeProps {
|
||||
kbId: string;
|
||||
kbInfo: KnowledgeBase;
|
||||
selectedDocuments: Document[];
|
||||
}
|
||||
|
||||
export const DeepReadMode: React.FC<DeepReadModeProps> = ({
|
||||
kbId,
|
||||
selectedDocuments
|
||||
}) => {
|
||||
if (!selectedDocuments || selectedDocuments.length === 0) {
|
||||
return (
|
||||
<div className="h-full flex items-center justify-center bg-slate-50">
|
||||
<div className="text-center">
|
||||
<div className="w-16 h-16 mx-auto bg-purple-50 rounded-full flex items-center justify-center mb-4">
|
||||
<FileText className="w-8 h-8 text-purple-400" />
|
||||
</div>
|
||||
<h3 className="text-lg font-semibold text-slate-700 mb-2">选择文档开始精读</h3>
|
||||
<p className="text-sm text-slate-500 max-w-xs">
|
||||
请在上方"逐篇精读"下拉框中选择一篇文档进行深度分析
|
||||
</p>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
const selectedDoc = selectedDocuments[0];
|
||||
const selectedDocIds = selectedDocuments.map(d => d.id);
|
||||
|
||||
return (
|
||||
<div className="h-full flex flex-col bg-white">
|
||||
{/* Chat组件 - 全屏展开 */}
|
||||
<div className="flex-1 overflow-hidden">
|
||||
<ChatContainer
|
||||
conversationType="pkb"
|
||||
conversationKey={`kb-deepread-${kbId}-${selectedDoc.id}`}
|
||||
defaultMessages={[{
|
||||
id: 'welcome',
|
||||
role: 'assistant',
|
||||
content: `您好！我已准备好深度解读文档《${selectedDoc.filename}》。\n\n我可以帮您：\n- 📖 逐段解读文献内容\n- 🎯 提炼核心观点和结论\n- 💡 分析研究方法和局限性\n\n请告诉我您想深入了解哪方面？`,
|
||||
status: 'success',
|
||||
timestamp: Date.now(),
|
||||
}]}
|
||||
providerConfig={{
|
||||
apiEndpoint: '/api/v1/chat/stream',
|
||||
requestFn: async (message: string) => {
|
||||
const response = await fetch('/api/v1/chat/stream', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Accept': 'text/event-stream',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
content: message,
|
||||
modelType: 'qwen-long',
|
||||
knowledgeBaseIds: [kbId],
|
||||
documentIds: selectedDocIds, // 🌟 关键参数：限定文档范围
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`API请求失败: ${response.status}`);
|
||||
}
|
||||
|
||||
// 处理流式响应
|
||||
const reader = response.body?.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let fullContent = '';
|
||||
|
||||
if (reader) {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
const chunk = decoder.decode(value);
|
||||
const lines = chunk.split('\n');
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('data: ')) {
|
||||
const data = line.slice(6);
|
||||
if (data === '[DONE]') break;
|
||||
|
||||
try {
|
||||
const json = JSON.parse(data);
|
||||
if (json.content) {
|
||||
fullContent += json.content;
|
||||
}
|
||||
} catch (e) {
|
||||
// 忽略解析错误
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
content: fullContent,
|
||||
messageId: Date.now().toString(),
|
||||
};
|
||||
},
|
||||
}}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
@@ -0,0 +1,117 @@
|
||||
/**
|
||||
* 全文阅读模式组件 - ChatGPT风格全屏聊天
|
||||
*/
|
||||
|
||||
import React from 'react';
|
||||
import { ChatContainer } from '@/shared/components/Chat';
|
||||
import type { KnowledgeBase, Document } from '../../api/knowledgeBaseApi';
|
||||
|
||||
interface FullTextModeProps {
|
||||
kbId: string;
|
||||
kbInfo: KnowledgeBase;
|
||||
documents: Document[];
|
||||
}
|
||||
|
||||
export const FullTextMode: React.FC<FullTextModeProps> = ({ kbId, documents }) => {
|
||||
// 准备全文文档ID
|
||||
const fullTextDocumentIds = documents
|
||||
.filter(doc => doc.status === 'completed')
|
||||
.map(doc => doc.id);
|
||||
|
||||
const totalTokens = documents.reduce((sum, d) => sum + (d.tokensCount || 0), 0);
|
||||
|
||||
return (
|
||||
<div className="h-full flex flex-col bg-white">
|
||||
{/* Chat组件 - 全屏展开，ChatGPT风格 */}
|
||||
<div className="flex-1 overflow-hidden">
|
||||
<ChatContainer
|
||||
conversationType="pkb"
|
||||
conversationKey={`kb-fulltext-${kbId}`}
|
||||
defaultMessages={[{
|
||||
id: 'welcome',
|
||||
role: 'assistant',
|
||||
content: `您好！我已加载全部 ${documents.length} 篇文档（共${totalTokens.toLocaleString()} tokens）。\n\n我可以帮您：\n- 📚 综合多篇文献进行对比分析\n- 🔍 查找特定主题的共识与争议\n- 📊 总结研究趋势和发展脉络\n\n请告诉我您想了解什么？`,
|
||||
status: 'success',
|
||||
timestamp: Date.now(),
|
||||
}]}
|
||||
customMessageRenderer={(msgInfo) => {
|
||||
const msg = msgInfo.message;
|
||||
if (msg.role === 'assistant' && msg.content) {
|
||||
// 处理参考文献格式
|
||||
const processedContent = msg.content.replace(
|
||||
/\*\*参考文献\*\*/g,
|
||||
'\n\n**📚 参考文献**\n'
|
||||
);
|
||||
return <div className="whitespace-pre-wrap">{processedContent}</div>;
|
||||
}
|
||||
return <div className="whitespace-pre-wrap">{msg.content}</div>;
|
||||
}}
|
||||
providerConfig={{
|
||||
apiEndpoint: '/api/v1/chat/stream',
|
||||
requestFn: async (message: string) => {
|
||||
const response = await fetch('/api/v1/chat/stream', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Accept': 'text/event-stream',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
content: message,
|
||||
modelType: 'qwen-long',
|
||||
knowledgeBaseIds: [kbId],
|
||||
fullTextDocumentIds, // 🌟 关键参数：全文阅读
|
||||
}),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`API请求失败: ${response.status}`);
|
||||
}
|
||||
|
||||
// 处理流式响应
|
||||
const reader = response.body?.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let fullContent = '';
|
||||
|
||||
if (reader) {
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
|
||||
const chunk = decoder.decode(value);
|
||||
const lines = chunk.split('\n');
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('data: ')) {
|
||||
const data = line.slice(6);
|
||||
if (data === '[DONE]') break;
|
||||
|
||||
try {
|
||||
const json = JSON.parse(data);
|
||||
if (json.content) {
|
||||
fullContent += json.content;
|
||||
}
|
||||
} catch (e) {
|
||||
// 忽略解析错误
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return {
|
||||
content: fullContent,
|
||||
messageId: Date.now().toString(),
|
||||
};
|
||||
},
|
||||
}}
|
||||
onMessageReceived={(msg) => {
|
||||
console.log('[全文阅读模式] 收到AI回复:', msg);
|
||||
}}
|
||||
onError={(error) => {
|
||||
console.error('[全文阅读模式] 错误:', error);
|
||||
}}
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
@@ -0,0 +1,165 @@
|
||||
/**
|
||||
* 工作模式选择器组件
|
||||
* 支持：全文阅读、逐篇精读、批处理
|
||||
*/
|
||||
|
||||
import React, { useState } from 'react';
|
||||
import { Collapse, Radio, Progress, Select, Alert } from 'antd';
|
||||
import { Globe, Search, Package } from 'lucide-react';
|
||||
import type { WorkMode } from '../../types/workspace';
|
||||
import type { KnowledgeBase, Document } from '../../api/knowledgeBaseApi';
|
||||
|
||||
interface WorkModeSelectorProps {
|
||||
currentMode: WorkMode;
|
||||
kbInfo: KnowledgeBase;
|
||||
documents: Document[];
|
||||
onChange: (mode: WorkMode) => void;
|
||||
onDocumentSelect?: (docs: string[]) => void;
|
||||
onTemplateSelect?: (template: string) => void;
|
||||
}
|
||||
|
||||
export const WorkModeSelector: React.FC<WorkModeSelectorProps> = ({
|
||||
currentMode,
|
||||
kbInfo,
|
||||
documents,
|
||||
onChange,
|
||||
onDocumentSelect,
|
||||
onTemplateSelect,
|
||||
}) => {
|
||||
const [selectedDocs, setSelectedDocs] = useState<string[]>([]);
|
||||
const [selectedTemplate, setSelectedTemplate] = useState<string>('');
|
||||
|
||||
// 计算Token使用率
|
||||
const calculateTokenUsage = () => {
|
||||
const totalTokens = documents.reduce((sum, doc) => sum + (doc.tokensCount || 0), 0);
|
||||
const maxTokens = 200000; // 假设最大200k tokens
|
||||
return Math.min(Math.round((totalTokens / maxTokens) * 100), 100);
|
||||
};
|
||||
|
||||
const handleDocumentChange = (values: string[]) => {
|
||||
setSelectedDocs(values);
|
||||
onDocumentSelect?.(values);
|
||||
};
|
||||
|
||||
const handleTemplateChange = (value: string) => {
|
||||
setSelectedTemplate(value);
|
||||
onTemplateSelect?.(value);
|
||||
};
|
||||
|
||||
return (
|
||||
<Collapse
|
||||
defaultActiveKey={['modes']}
|
||||
className="bg-white border border-gray-100 rounded-lg shadow-sm"
|
||||
bordered={false}
|
||||
>
|
||||
<Collapse.Panel
|
||||
header={
|
||||
<div className="font-semibold text-slate-800 text-sm flex items-center">
|
||||
<span className="text-base mr-2">📚</span>
|
||||
工作模式
|
||||
</div>
|
||||
}
|
||||
key="modes"
|
||||
>
|
||||
<Radio.Group
|
||||
value={currentMode}
|
||||
onChange={(e) => onChange(e.target.value)}
|
||||
className="w-full space-y-4"
|
||||
>
|
||||
{/* 全文阅读模式 - 精细优化 */}
|
||||
<Radio value="full_text" className="w-full">
|
||||
<div className="flex justify-between items-center w-full">
|
||||
<div className="flex-1">
|
||||
<div className="flex items-center mb-1.5">
|
||||
<Globe className="w-4 h-4 mr-2 text-blue-600" />
|
||||
<span className="font-semibold text-slate-800 text-sm">全文阅读模式</span>
|
||||
</div>
|
||||
<div className="text-xs text-slate-500 ml-6 leading-relaxed">
|
||||
加载全部 {kbInfo.fileCount} 篇文档，AI具备全知视角，适合文献综述
|
||||
</div>
|
||||
</div>
|
||||
<Progress
|
||||
type="circle"
|
||||
percent={calculateTokenUsage()}
|
||||
width={42}
|
||||
strokeColor="#3b82f6"
|
||||
trailColor="#e5e7eb"
|
||||
strokeWidth={8}
|
||||
/>
|
||||
</div>
|
||||
</Radio>
|
||||
|
||||
{/* 逐篇精读模式 - 精细优化 */}
|
||||
<Radio value="deep_read" className="w-full">
|
||||
<div className="flex-1">
|
||||
<div className="flex items-center mb-1.5">
|
||||
<Search className="w-4 h-4 mr-2 text-purple-600" />
|
||||
<span className="font-semibold text-slate-800 text-sm">逐篇精读模式</span>
|
||||
</div>
|
||||
<div className="text-xs text-slate-500 ml-6 leading-relaxed">
|
||||
选择1-5篇文档进行深度解读，适合单篇文献精读
|
||||
</div>
|
||||
{currentMode === 'deep_read' && (
|
||||
<div className="ml-6 mt-3" onClick={(e) => e.stopPropagation()}>
|
||||
<Select
|
||||
mode="multiple"
|
||||
placeholder="请选择文档（最多5篇）"
|
||||
className="w-full"
|
||||
maxCount={5}
|
||||
value={selectedDocs}
|
||||
onChange={handleDocumentChange}
|
||||
getPopupContainer={(trigger) => trigger.parentElement || document.body}
|
||||
options={documents
|
||||
.filter(doc => doc.status === 'completed')
|
||||
.map(doc => ({
|
||||
label: doc.filename,
|
||||
value: doc.id,
|
||||
}))}
|
||||
/>
|
||||
{selectedDocs.length > 0 && (
|
||||
<Alert
|
||||
message={`已选择 ${selectedDocs.length} 篇文档`}
|
||||
type="info"
|
||||
showIcon
|
||||
className="mt-2"
|
||||
/>
|
||||
)}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</Radio>
|
||||
|
||||
{/* 批处理模式 - 精细优化 */}
|
||||
<Radio value="batch" className="w-full">
|
||||
<div className="flex-1">
|
||||
<div className="flex items-center mb-1.5">
|
||||
<Package className="w-4 h-4 mr-2 text-orange-600" />
|
||||
<span className="font-semibold text-slate-800 text-sm">批处理模式</span>
|
||||
</div>
|
||||
<div className="text-xs text-slate-500 ml-6 leading-relaxed">
|
||||
批量提取信息，生成结构化表格，适合数据提取
|
||||
</div>
|
||||
{currentMode === 'batch' && (
|
||||
<div className="ml-6 mt-3" onClick={(e) => e.stopPropagation()}>
|
||||
<Select
|
||||
placeholder="请选择批处理模板"
|
||||
className="w-full"
|
||||
value={selectedTemplate}
|
||||
onChange={handleTemplateChange}
|
||||
getPopupContainer={(trigger) => trigger.parentElement || document.body}
|
||||
options={[
|
||||
{ label: '临床研究信息提取', value: 'clinicalResearch' },
|
||||
{ label: '药物安全性分析', value: 'drug_safety' },
|
||||
{ label: '患者基线特征', value: 'patient_baseline' },
|
||||
]}
|
||||
/>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</Radio>
|
||||
</Radio.Group>
|
||||
</Collapse.Panel>
|
||||
</Collapse>
|
||||
);
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user