Test KB index button move
This commit is contained in:
Xin Wang
@@ -473,6 +473,7 @@ const KnowledgeBaseDetail: React.FC<{
|
|||||||
const [debugLoading, setDebugLoading] = useState(false);
|
const [debugLoading, setDebugLoading] = useState(false);
|
||||||
const [debugResults, setDebugResults] = useState<KnowledgeSearchResultItem[]>([]);
|
const [debugResults, setDebugResults] = useState<KnowledgeSearchResultItem[]>([]);
|
||||||
const [debugError, setDebugError] = useState('');
|
const [debugError, setDebugError] = useState('');
|
||||||
|
const [debugDialogOpen, setDebugDialogOpen] = useState(false);
|
||||||
const filteredDocs = kb.documents.filter((d) => d.name.toLowerCase().includes(docSearch.toLowerCase()));
|
const filteredDocs = kb.documents.filter((d) => d.name.toLowerCase().includes(docSearch.toLowerCase()));
|
||||||
const docNameById = new Map(kb.documents.map((doc) => [doc.id, doc.name]));
|
const docNameById = new Map(kb.documents.map((doc) => [doc.id, doc.name]));
|
||||||
|
|
||||||
@@ -535,6 +536,7 @@ const KnowledgeBaseDetail: React.FC<{
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="flex items-center gap-2">
|
<div className="flex items-center gap-2">
|
||||||
|
<Button variant="outline" onClick={() => setDebugDialogOpen(true)}>调试检索</Button>
|
||||||
<Button variant="outline" onClick={onEdit}><Settings2 className="mr-2 h-4 w-4" /> 编辑配置</Button>
|
<Button variant="outline" onClick={onEdit}><Settings2 className="mr-2 h-4 w-4" /> 编辑配置</Button>
|
||||||
<Button onClick={onImport}><Upload className="mr-2 h-4 w-4" /> 新增知识 (导入)</Button>
|
<Button onClick={onImport}><Upload className="mr-2 h-4 w-4" /> 新增知识 (导入)</Button>
|
||||||
<Button variant="ghost" className="text-destructive" onClick={onDelete}><Trash2 className="mr-2 h-4 w-4" /> 删除</Button>
|
<Button variant="ghost" className="text-destructive" onClick={onDelete}><Trash2 className="mr-2 h-4 w-4" /> 删除</Button>
|
||||||
@@ -609,71 +611,6 @@ const KnowledgeBaseDetail: React.FC<{
|
|||||||
</table>
|
</table>
|
||||||
</Card>
|
</Card>
|
||||||
|
|
||||||
<Card className="overflow-hidden border-white/5">
|
|
||||||
<div className="p-4 border-b border-white/5 bg-white/5">
|
|
||||||
<h3 className="font-medium text-white">调试检索</h3>
|
|
||||||
<p className="text-xs text-muted-foreground mt-1">
|
|
||||||
输入测试文本，使用当前知识库的 embedding 配置检索文档片段并查看匹配度。
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
<div className="p-4 space-y-4">
|
|
||||||
<div className="grid grid-cols-1 md:grid-cols-[1fr_120px_120px] gap-3">
|
|
||||||
<Input
|
|
||||||
placeholder="输入要测试的检索文本..."
|
|
||||||
value={debugQuery}
|
|
||||||
onChange={(e) => setDebugQuery(e.target.value)}
|
|
||||||
onKeyDown={(e) => {
|
|
||||||
if (e.key === 'Enter' && !debugLoading) void runDebugSearch();
|
|
||||||
}}
|
|
||||||
className="bg-black/20 border-transparent focus:bg-black/40"
|
|
||||||
/>
|
|
||||||
<Input
|
|
||||||
type="number"
|
|
||||||
min={1}
|
|
||||||
max={20}
|
|
||||||
value={debugTopK}
|
|
||||||
onChange={(e) => setDebugTopK(parseInt(e.target.value || '1', 10))}
|
|
||||||
className="bg-black/20 border-transparent focus:bg-black/40"
|
|
||||||
/>
|
|
||||||
<Button onClick={() => void runDebugSearch()} disabled={debugLoading}>
|
|
||||||
{debugLoading ? '检索中...' : '开始检索'}
|
|
||||||
</Button>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
{!!debugError && (
|
|
||||||
<div className="text-sm text-destructive">{debugError}</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{!debugLoading && debugResults.length === 0 && !debugError && (
|
|
||||||
<div className="text-sm text-muted-foreground">暂无结果，输入文本后点击“开始检索”。</div>
|
|
||||||
)}
|
|
||||||
|
|
||||||
{debugResults.length > 0 && (
|
|
||||||
<div className="space-y-3">
|
|
||||||
{debugResults.map((item, idx) => {
|
|
||||||
const docId = String(item.metadata?.document_id || '');
|
|
||||||
const chunkIndex = item.metadata?.chunk_index;
|
|
||||||
const docName = docNameById.get(docId);
|
|
||||||
return (
|
|
||||||
<div key={`${docId}-${chunkIndex}-${idx}`} className="rounded-lg border border-white/10 bg-white/5 p-3 space-y-2">
|
|
||||||
<div className="flex flex-wrap items-center gap-2 text-xs">
|
|
||||||
<Badge variant="outline">#{idx + 1}</Badge>
|
|
||||||
<Badge variant="outline">{docName || docId || 'unknown_doc'}</Badge>
|
|
||||||
<Badge variant="outline">chunk {chunkIndex ?? '-'}</Badge>
|
|
||||||
<Badge variant="outline">distance: {typeof item.distance === 'number' ? item.distance.toFixed(6) : '-'}</Badge>
|
|
||||||
<Badge variant="outline">匹配度: {renderMatchPercent(item.distance)}</Badge>
|
|
||||||
</div>
|
|
||||||
<p className="text-sm text-white/90 whitespace-pre-wrap leading-relaxed">
|
|
||||||
{item.content || '(空内容)'}
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
})}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</Card>
|
|
||||||
|
|
||||||
<Dialog
|
<Dialog
|
||||||
isOpen={!!indexingDoc}
|
isOpen={!!indexingDoc}
|
||||||
onClose={() => {
|
onClose={() => {
|
||||||
@@ -712,6 +649,77 @@ const KnowledgeBaseDetail: React.FC<{
|
|||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
</Dialog>
|
</Dialog>
|
||||||
|
|
||||||
|
<Dialog
|
||||||
|
isOpen={debugDialogOpen}
|
||||||
|
onClose={() => setDebugDialogOpen(false)}
|
||||||
|
title="调试检索"
|
||||||
|
footer={
|
||||||
|
<>
|
||||||
|
<Button variant="ghost" onClick={() => setDebugDialogOpen(false)}>关闭</Button>
|
||||||
|
<Button onClick={() => void runDebugSearch()} disabled={debugLoading}>
|
||||||
|
{debugLoading ? '检索中...' : '开始检索'}
|
||||||
|
</Button>
|
||||||
|
</>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<div className="space-y-4">
|
||||||
|
<p className="text-xs text-muted-foreground">
|
||||||
|
输入测试文本，使用当前知识库的 embedding 配置检索文档片段并查看匹配度。
|
||||||
|
</p>
|
||||||
|
<div className="grid grid-cols-1 md:grid-cols-[1fr_120px] gap-3">
|
||||||
|
<Input
|
||||||
|
placeholder="输入要测试的检索文本..."
|
||||||
|
value={debugQuery}
|
||||||
|
onChange={(e) => setDebugQuery(e.target.value)}
|
||||||
|
onKeyDown={(e) => {
|
||||||
|
if (e.key === 'Enter' && !debugLoading) void runDebugSearch();
|
||||||
|
}}
|
||||||
|
className="bg-black/20 border-transparent focus:bg-black/40"
|
||||||
|
/>
|
||||||
|
<Input
|
||||||
|
type="number"
|
||||||
|
min={1}
|
||||||
|
max={20}
|
||||||
|
value={debugTopK}
|
||||||
|
onChange={(e) => setDebugTopK(parseInt(e.target.value || '1', 10))}
|
||||||
|
className="bg-black/20 border-transparent focus:bg-black/40"
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{!!debugError && (
|
||||||
|
<div className="text-sm text-destructive">{debugError}</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{!debugLoading && debugResults.length === 0 && !debugError && (
|
||||||
|
<div className="text-sm text-muted-foreground">暂无结果，输入文本后点击“开始检索”。</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{debugResults.length > 0 && (
|
||||||
|
<div className="space-y-3 max-h-[52vh] overflow-y-auto pr-1 custom-scrollbar">
|
||||||
|
{debugResults.map((item, idx) => {
|
||||||
|
const docId = String(item.metadata?.document_id || '');
|
||||||
|
const chunkIndex = item.metadata?.chunk_index;
|
||||||
|
const docName = docNameById.get(docId);
|
||||||
|
return (
|
||||||
|
<div key={`${docId}-${chunkIndex}-${idx}`} className="rounded-lg border border-white/10 bg-white/5 p-3 space-y-2">
|
||||||
|
<div className="flex flex-wrap items-center gap-2 text-xs">
|
||||||
|
<Badge variant="outline">#{idx + 1}</Badge>
|
||||||
|
<Badge variant="outline">{docName || docId || 'unknown_doc'}</Badge>
|
||||||
|
<Badge variant="outline">chunk {chunkIndex ?? '-'}</Badge>
|
||||||
|
<Badge variant="outline">distance: {typeof item.distance === 'number' ? item.distance.toFixed(6) : '-'}</Badge>
|
||||||
|
<Badge variant="outline">匹配度: {renderMatchPercent(item.distance)}</Badge>
|
||||||
|
</div>
|
||||||
|
<p className="text-sm text-white/90 whitespace-pre-wrap leading-relaxed">
|
||||||
|
{item.content || '(空内容)'}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</Dialog>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|||||||
Reference in New Issue
Block a user