Fix 法规对话
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
import React, { useEffect, useRef, useState } from 'react';
|
||||
import React, { useCallback, useEffect, useRef, useState } from 'react';
|
||||
import { useTheme } from '../../contexts';
|
||||
import type { ChatMessage, RetrievalData } from '../../types';
|
||||
import { getQuickQuestions, ragChat } from '../../api/rag';
|
||||
@@ -19,6 +19,7 @@ export const RagChatPage: React.FC = () => {
|
||||
const { theme } = useTheme();
|
||||
const nextMessageIdRef = useRef(1);
|
||||
const [messages, setMessages] = useState<ChatMessage[]>([]);
|
||||
// retrievals: right-panel shows sources of the most recent assistant reply
|
||||
const [retrievals, setRetrievals] = useState<RetrievalData[]>([]);
|
||||
const [input, setInput] = useState<string>('');
|
||||
const [loading, setLoading] = useState<boolean>(false);
|
||||
@@ -27,62 +28,60 @@ export const RagChatPage: React.FC = () => {
|
||||
const [quickQuestions, setQuickQuestions] = useState<string[]>(ragQuickQuestionsDefault);
|
||||
const [filterRegulationType, setFilterRegulationType] = useState<string>('');
|
||||
const [highlightedSourceIdx, setHighlightedSourceIdx] = useState<number | null>(null);
|
||||
const [sessionId, setSessionId] = useState<string | undefined>();
|
||||
|
||||
// Auto-scroll ref
|
||||
const messagesEndRef = useRef<HTMLDivElement>(null);
|
||||
// AbortController for cancelling in-flight requests
|
||||
const abortRef = useRef<AbortController | null>(null);
|
||||
|
||||
function nextMessageId() {
|
||||
const currentId = nextMessageIdRef.current;
|
||||
const id = nextMessageIdRef.current;
|
||||
nextMessageIdRef.current += 1;
|
||||
return currentId;
|
||||
return id;
|
||||
}
|
||||
|
||||
// Scroll to bottom whenever messages change
|
||||
useEffect(() => {
|
||||
messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
|
||||
}, [messages]);
|
||||
|
||||
async function loadQuickQuestions() {
|
||||
try {
|
||||
const response = await getQuickQuestions();
|
||||
setQuickQuestions(response.questions.map(q => q.question));
|
||||
} catch (error) {
|
||||
console.error('Failed to load quick questions:', error);
|
||||
} catch {
|
||||
// keep defaults
|
||||
}
|
||||
}
|
||||
|
||||
useEffect(() => {
|
||||
const timerId = window.setTimeout(() => {
|
||||
void loadQuickQuestions();
|
||||
}, 0);
|
||||
const timerId = window.setTimeout(() => { void loadQuickQuestions(); }, 0);
|
||||
return () => window.clearTimeout(timerId);
|
||||
}, []);
|
||||
|
||||
const sendMessage = (text: string) => {
|
||||
if (!text.trim()) return;
|
||||
/**
|
||||
* Core query executor — shared by sendMessage and regenerateLastAnswer.
|
||||
* Manages session_id, AbortController, SSE parsing, and state updates.
|
||||
*/
|
||||
const executeQuery = useCallback((text: string) => {
|
||||
// Cancel any in-flight request
|
||||
abortRef.current?.abort();
|
||||
abortRef.current = new AbortController();
|
||||
|
||||
const userMsg = { id: nextMessageId(), role: 'user' as const, content: text };
|
||||
setMessages((prev) => [...prev, userMsg]);
|
||||
setInput('');
|
||||
setLoading(true);
|
||||
setRetrievals([]);
|
||||
setHighlightedSourceIdx(null);
|
||||
|
||||
let currentResponse = '';
|
||||
const activeFilters = filterRegulationType.trim() || undefined;
|
||||
let currentResponse = '';
|
||||
// Capture the assistant message id so we can attach sources later
|
||||
let assistantMsgId: number | null = null;
|
||||
|
||||
void ragChat(
|
||||
text,
|
||||
5,
|
||||
(data: unknown) => {
|
||||
const sseData = data as {
|
||||
type: string;
|
||||
text?: string;
|
||||
docs?: Array<{
|
||||
id: string;
|
||||
score: number;
|
||||
preview: string;
|
||||
doc_name: string;
|
||||
clause: string;
|
||||
doc_id?: string;
|
||||
download_url?: string;
|
||||
}>;
|
||||
};
|
||||
|
||||
if (sseData.type === 'retrieved' && sseData.docs) {
|
||||
const retrievedDocs: RetrievalData[] = sseData.docs.map(d => ({
|
||||
(data) => {
|
||||
if (data.type === 'session' && data.session_id) {
|
||||
setSessionId(data.session_id);
|
||||
} else if (data.type === 'retrieved' && data.docs) {
|
||||
const docs: RetrievalData[] = data.docs.map(d => ({
|
||||
id: parseInt(d.id.replace('chunk-', ''), 10) || 1,
|
||||
file: d.doc_name,
|
||||
clause: d.clause,
|
||||
@@ -91,125 +90,98 @@ export const RagChatPage: React.FC = () => {
|
||||
docId: d.doc_id,
|
||||
downloadUrl: d.download_url,
|
||||
}));
|
||||
setRetrievals(retrievedDocs);
|
||||
} else if (sseData.type === 'chunk' && sseData.text) {
|
||||
currentResponse += sseData.text;
|
||||
setMessages((prev) => {
|
||||
const lastMsg = prev[prev.length - 1];
|
||||
if (lastMsg?.role === 'assistant') {
|
||||
return [...prev.slice(0, -1), { ...lastMsg, content: currentResponse }];
|
||||
setRetrievals(docs);
|
||||
// Attach sources to the assistant message once we know its id
|
||||
if (assistantMsgId !== null) {
|
||||
setMessages(prev => prev.map(m =>
|
||||
m.id === assistantMsgId ? { ...m, sources: docs } : m
|
||||
));
|
||||
}
|
||||
} else if (data.type === 'chunk' && data.text) {
|
||||
currentResponse += data.text;
|
||||
setMessages(prev => {
|
||||
const last = prev[prev.length - 1];
|
||||
if (last?.role === 'assistant' && last.id === assistantMsgId) {
|
||||
return [...prev.slice(0, -1), { ...last, content: currentResponse }];
|
||||
}
|
||||
return [...prev, { id: nextMessageId(), role: 'assistant' as const, content: currentResponse }];
|
||||
// First chunk: create assistant message
|
||||
const newId = nextMessageId();
|
||||
assistantMsgId = newId;
|
||||
return [...prev, { id: newId, role: 'assistant' as const, content: currentResponse }];
|
||||
});
|
||||
} else if (sseData.type === 'done') {
|
||||
} else if (data.type === 'done') {
|
||||
if (data.session_id) setSessionId(data.session_id);
|
||||
setLoading(false);
|
||||
} else if (sseData.type === 'error') {
|
||||
} else if (data.type === 'error') {
|
||||
setLoading(false);
|
||||
setMessages(prev => [
|
||||
...prev,
|
||||
{ id: nextMessageId(), role: 'assistant' as const, content: '抱歉，生成回答时出错，请稍后再试。' },
|
||||
]);
|
||||
}
|
||||
},
|
||||
(error: Error) => {
|
||||
(error) => {
|
||||
console.error('RAG chat error:', error);
|
||||
setLoading(false);
|
||||
setMessages((prev) => [
|
||||
setMessages(prev => [
|
||||
...prev,
|
||||
{ id: nextMessageId(), role: 'assistant' as const, content: '抱歉，连接服务器时出错，请稍后再试。' }
|
||||
{ id: nextMessageId(), role: 'assistant' as const, content: '抱歉，连接服务器时出错，请稍后再试。' },
|
||||
]);
|
||||
},
|
||||
() => {
|
||||
setLoading(false);
|
||||
},
|
||||
activeFilters
|
||||
() => { setLoading(false); },
|
||||
activeFilters,
|
||||
sessionId,
|
||||
abortRef.current.signal,
|
||||
);
|
||||
};
|
||||
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||||
}, [filterRegulationType, sessionId]);
|
||||
|
||||
const clearMessages = () => {
|
||||
setMessages([]);
|
||||
const sendMessage = (text: string) => {
|
||||
if (!text.trim() || loading) return;
|
||||
setMessages(prev => [...prev, { id: nextMessageId(), role: 'user' as const, content: text }]);
|
||||
setInput('');
|
||||
setLoading(true);
|
||||
setRetrievals([]);
|
||||
setShowClearConfirm(false);
|
||||
setHighlightedSourceIdx(null);
|
||||
executeQuery(text);
|
||||
};
|
||||
|
||||
const regenerateLastAnswer = () => {
|
||||
if (messages.length < 2) return;
|
||||
const lastUserMsg = messages.filter((m) => m.role === 'user').pop();
|
||||
if (loading) return;
|
||||
const lastUserMsg = [...messages].reverse().find(m => m.role === 'user');
|
||||
if (!lastUserMsg) return;
|
||||
|
||||
// Remove the last assistant message
|
||||
setMessages(prev => {
|
||||
const lastAssistantIdx = [...prev].reverse().findIndex(m => m.role === 'assistant');
|
||||
if (lastAssistantIdx === -1) return prev;
|
||||
const idx = prev.length - 1 - lastAssistantIdx;
|
||||
return [...prev.slice(0, idx)];
|
||||
});
|
||||
setLoading(true);
|
||||
setMessages((prev) => [...prev.slice(0, -1)]);
|
||||
setRetrievals([]);
|
||||
setHighlightedSourceIdx(null);
|
||||
executeQuery(lastUserMsg.content);
|
||||
};
|
||||
|
||||
let currentResponse = '';
|
||||
const activeFilters = filterRegulationType.trim() || undefined;
|
||||
|
||||
void ragChat(
|
||||
lastUserMsg.content,
|
||||
5,
|
||||
(data: unknown) => {
|
||||
const sseData = data as {
|
||||
type: string;
|
||||
text?: string;
|
||||
docs?: Array<{
|
||||
id: string;
|
||||
score: number;
|
||||
preview: string;
|
||||
doc_name: string;
|
||||
clause: string;
|
||||
doc_id?: string;
|
||||
download_url?: string;
|
||||
}>;
|
||||
};
|
||||
|
||||
if (sseData.type === 'retrieved' && sseData.docs) {
|
||||
const retrievedDocs: RetrievalData[] = sseData.docs.map(d => ({
|
||||
id: parseInt(d.id.replace('chunk-', ''), 10) || 1,
|
||||
file: d.doc_name,
|
||||
clause: d.clause,
|
||||
score: d.score,
|
||||
content: d.preview,
|
||||
docId: d.doc_id,
|
||||
downloadUrl: d.download_url,
|
||||
}));
|
||||
setRetrievals(retrievedDocs);
|
||||
} else if (sseData.type === 'chunk' && sseData.text) {
|
||||
currentResponse += sseData.text;
|
||||
setMessages((prev) => {
|
||||
const lastMsg = prev[prev.length - 1];
|
||||
if (lastMsg?.role === 'assistant') {
|
||||
return [...prev.slice(0, -1), { ...lastMsg, content: currentResponse }];
|
||||
}
|
||||
return [...prev, { id: nextMessageId(), role: 'assistant' as const, content: currentResponse }];
|
||||
});
|
||||
} else if (sseData.type === 'done') {
|
||||
setLoading(false);
|
||||
}
|
||||
},
|
||||
(error: Error) => {
|
||||
console.error('RAG chat error:', error);
|
||||
setLoading(false);
|
||||
setMessages((prev) => [
|
||||
...prev,
|
||||
{ id: nextMessageId(), role: 'assistant' as const, content: '抱歉，连接服务器时出错，请稍后再试。' }
|
||||
]);
|
||||
},
|
||||
() => {
|
||||
setLoading(false);
|
||||
},
|
||||
activeFilters
|
||||
);
|
||||
const clearMessages = () => {
|
||||
abortRef.current?.abort();
|
||||
setMessages([]);
|
||||
setRetrievals([]);
|
||||
setSessionId(undefined);
|
||||
setShowClearConfirm(false);
|
||||
setLoading(false);
|
||||
};
|
||||
|
||||
return (
|
||||
<div style={{
|
||||
flex: 1,
|
||||
display: 'flex',
|
||||
height: 'calc(100vh - 128px)',
|
||||
}}>
|
||||
<div style={{ flex: 1, display: 'flex', height: 'calc(100vh - 128px)' }}>
|
||||
{/* ── Left: chat panel ─────────────────────────────────── */}
|
||||
<div style={{
|
||||
flex: '0 0 60%',
|
||||
display: 'flex',
|
||||
flexDirection: 'column',
|
||||
borderRight: `1px solid ${theme.border}`,
|
||||
}}>
|
||||
{/* Message list */}
|
||||
<div style={{
|
||||
flex: 1,
|
||||
overflowY: 'auto',
|
||||
@@ -219,20 +191,11 @@ export const RagChatPage: React.FC = () => {
|
||||
gap: 20,
|
||||
}}>
|
||||
{messages.length === 0 ? (
|
||||
<div style={{
|
||||
textAlign: 'center',
|
||||
padding: 60,
|
||||
color: theme.text3,
|
||||
}}>
|
||||
<div style={{ textAlign: 'center', padding: 60, color: theme.text3 }}>
|
||||
<div style={{
|
||||
width: 72,
|
||||
height: 72,
|
||||
borderRadius: 16,
|
||||
background: theme.bgCard,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
margin: '0 auto 20px',
|
||||
width: 72, height: 72, borderRadius: 16,
|
||||
background: theme.bgCard, display: 'flex', alignItems: 'center',
|
||||
justifyContent: 'center', margin: '0 auto 20px',
|
||||
border: `1px solid ${theme.border}`,
|
||||
}}>
|
||||
<svg width="28" height="28" viewBox="0 0 24 24" fill="none">
|
||||
@@ -245,20 +208,14 @@ export const RagChatPage: React.FC = () => {
|
||||
) : (
|
||||
messages.map(msg => (
|
||||
<div key={msg.id} style={{
|
||||
display: 'flex',
|
||||
gap: 12,
|
||||
display: 'flex', gap: 12,
|
||||
flexDirection: msg.role === 'user' ? 'row-reverse' : 'row',
|
||||
}}>
|
||||
{msg.role === 'assistant' && (
|
||||
<div style={{
|
||||
width: 32,
|
||||
height: 32,
|
||||
borderRadius: 8,
|
||||
background: theme.gradientAccent,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
flexShrink: 0,
|
||||
width: 32, height: 32, borderRadius: 8,
|
||||
background: theme.gradientAccent, display: 'flex',
|
||||
alignItems: 'center', justifyContent: 'center', flexShrink: 0,
|
||||
}}>
|
||||
<svg width="16" height="16" viewBox="0 0 24 24" fill="none">
|
||||
<circle cx="12" cy="12" r="6" fill="#fff"/>
|
||||
@@ -271,16 +228,16 @@ export const RagChatPage: React.FC = () => {
|
||||
background: msg.role === 'user' ? theme.gradientAccent : theme.bgCard,
|
||||
borderRadius: 12,
|
||||
color: msg.role === 'user' ? '#fff' : theme.text,
|
||||
fontSize: 14,
|
||||
lineHeight: 1.6,
|
||||
whiteSpace: 'pre-wrap',
|
||||
fontSize: 14, lineHeight: 1.6, whiteSpace: 'pre-wrap',
|
||||
border: msg.role === 'assistant' ? `1px solid ${theme.border}` : 'none',
|
||||
}}>
|
||||
{msg.role === 'assistant' ? (
|
||||
<CitedAnswer
|
||||
text={msg.content}
|
||||
sources={retrievals}
|
||||
sources={msg.sources ?? retrievals}
|
||||
onCiteClick={(idx) => {
|
||||
const msgSources = msg.sources ?? retrievals;
|
||||
setRetrievals(msgSources);
|
||||
setHighlightedSourceIdx(idx);
|
||||
const el = document.getElementById(`source-${idx}`);
|
||||
if (el) el.scrollIntoView({ behavior: 'smooth', block: 'center' });
|
||||
@@ -289,12 +246,9 @@ export const RagChatPage: React.FC = () => {
|
||||
) : msg.content}
|
||||
{msg.role === 'assistant' && msg.retrievalIds && msg.retrievalIds.length > 0 && (
|
||||
<div style={{
|
||||
marginTop: 10,
|
||||
paddingTop: 10,
|
||||
marginTop: 10, paddingTop: 10,
|
||||
borderTop: `1px solid ${theme.border}`,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
gap: 6,
|
||||
display: 'flex', alignItems: 'center', gap: 6,
|
||||
}}>
|
||||
<svg width="12" height="12" viewBox="0 0 24 24" fill="none">
|
||||
<path d="M14 2H6C5 2 4 3 4 4V20C4 21 5 22 6 22H18C19 22 20 21 20 20V8L14 2Z" stroke={theme.accent} strokeWidth="1.5"/>
|
||||
@@ -311,150 +265,115 @@ export const RagChatPage: React.FC = () => {
|
||||
{loading && (
|
||||
<div style={{ display: 'flex', gap: 12 }}>
|
||||
<div style={{
|
||||
width: 32,
|
||||
height: 32,
|
||||
borderRadius: 8,
|
||||
background: theme.gradientAccent,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
width: 32, height: 32, borderRadius: 8,
|
||||
background: theme.gradientAccent, display: 'flex',
|
||||
alignItems: 'center', justifyContent: 'center',
|
||||
}}>
|
||||
<svg width="16" height="16" viewBox="0 0 24 24" fill="none">
|
||||
<circle cx="12" cy="12" r="6" fill="#fff"/>
|
||||
</svg>
|
||||
</div>
|
||||
<div style={{
|
||||
padding: '14px 18px',
|
||||
background: theme.bgCard,
|
||||
borderRadius: 12,
|
||||
border: `1px solid ${theme.border}`,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
gap: 8,
|
||||
padding: '14px 18px', background: theme.bgCard,
|
||||
borderRadius: 12, border: `1px solid ${theme.border}`,
|
||||
display: 'flex', alignItems: 'center', gap: 8,
|
||||
}}>
|
||||
<div style={{
|
||||
width: 6,
|
||||
height: 6,
|
||||
borderRadius: '50%',
|
||||
background: theme.accent,
|
||||
animation: 'pulse 1s infinite',
|
||||
width: 6, height: 6, borderRadius: '50%',
|
||||
background: theme.accent, animation: 'pulse 1s infinite',
|
||||
}} />
|
||||
<span style={{ fontSize: 13, color: theme.text2 }}>检索中...</span>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
{/* Scroll anchor */}
|
||||
<div ref={messagesEndRef} />
|
||||
</div>
|
||||
|
||||
<div style={{
|
||||
padding: '16px 32px 20px',
|
||||
background: theme.bg,
|
||||
borderTop: `1px solid ${theme.border}`,
|
||||
}}>
|
||||
<div style={{
|
||||
display: 'flex',
|
||||
gap: 8,
|
||||
marginBottom: 10,
|
||||
alignItems: 'center',
|
||||
}}>
|
||||
{/* Input area */}
|
||||
<div style={{ padding: '16px 32px 20px', background: theme.bg, borderTop: `1px solid ${theme.border}` }}>
|
||||
{/* Filter row */}
|
||||
<div style={{ display: 'flex', gap: 8, marginBottom: 10, alignItems: 'center' }}>
|
||||
<span className="mono" style={{ fontSize: 11, color: theme.text3, whiteSpace: 'nowrap' }}>法规类型</span>
|
||||
<input
|
||||
value={filterRegulationType}
|
||||
onChange={(e) => setFilterRegulationType(e.target.value)}
|
||||
onChange={e => setFilterRegulationType(e.target.value)}
|
||||
placeholder="如: GB / UN-ECE / IATF（留空不过滤）"
|
||||
style={{
|
||||
flex: 1,
|
||||
maxWidth: 280,
|
||||
padding: '5px 10px',
|
||||
fontSize: 12,
|
||||
background: theme.bgHover,
|
||||
border: `1px solid ${theme.border}`,
|
||||
borderRadius: 6,
|
||||
color: theme.text,
|
||||
outline: 'none',
|
||||
flex: 1, maxWidth: 280, padding: '5px 10px', fontSize: 12,
|
||||
background: theme.bgHover, border: `1px solid ${theme.border}`,
|
||||
borderRadius: 6, color: theme.text, outline: 'none',
|
||||
}}
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div style={{
|
||||
display: 'flex',
|
||||
gap: 8,
|
||||
marginBottom: 12,
|
||||
flexWrap: 'wrap',
|
||||
}}>
|
||||
{/* Quick questions */}
|
||||
<div style={{ display: 'flex', gap: 8, marginBottom: 12, flexWrap: 'wrap' }}>
|
||||
{quickQuestions.map(q => (
|
||||
<button
|
||||
key={q}
|
||||
onClick={() => sendMessage(q)}
|
||||
disabled={loading}
|
||||
style={{
|
||||
padding: '6px 14px',
|
||||
fontSize: 12,
|
||||
background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`,
|
||||
borderRadius: 6,
|
||||
color: theme.text2,
|
||||
cursor: 'pointer',
|
||||
padding: '6px 14px', fontSize: 12, background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`, borderRadius: 6,
|
||||
color: theme.text2, cursor: loading ? 'not-allowed' : 'pointer',
|
||||
}}
|
||||
>{q}</button>
|
||||
))}
|
||||
</div>
|
||||
|
||||
{/* Send row */}
|
||||
<div style={{ display: 'flex', gap: 10 }}>
|
||||
<input
|
||||
value={input}
|
||||
onChange={(e) => setInput(e.target.value)}
|
||||
onKeyDown={(e) => e.key === 'Enter' && sendMessage(input)}
|
||||
onChange={e => setInput(e.target.value)}
|
||||
onKeyDown={e => e.key === 'Enter' && !e.shiftKey && sendMessage(input)}
|
||||
placeholder="输入法规问题..."
|
||||
style={{
|
||||
flex: 1,
|
||||
padding: 12,
|
||||
fontSize: 14,
|
||||
background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`,
|
||||
borderRadius: 8,
|
||||
color: theme.text,
|
||||
outline: 'none',
|
||||
flex: 1, padding: 12, fontSize: 14,
|
||||
background: theme.bgCard, border: `1px solid ${theme.border}`,
|
||||
borderRadius: 8, color: theme.text, outline: 'none',
|
||||
}}
|
||||
/>
|
||||
<button
|
||||
onClick={() => sendMessage(input)}
|
||||
disabled={loading || !input.trim()}
|
||||
style={{
|
||||
padding: '12px 24px',
|
||||
fontSize: 14,
|
||||
fontWeight: 600,
|
||||
padding: '12px 24px', fontSize: 14, fontWeight: 600,
|
||||
background: loading || !input.trim() ? theme.bgHover : theme.gradientAccent,
|
||||
color: loading || !input.trim() ? theme.text3 : '#fff',
|
||||
border: 'none',
|
||||
borderRadius: 8,
|
||||
border: 'none', borderRadius: 8,
|
||||
cursor: loading || !input.trim() ? 'not-allowed' : 'pointer',
|
||||
}}
|
||||
>发送</button>
|
||||
{messages.length > 0 && (
|
||||
{loading && (
|
||||
<button
|
||||
onClick={() => { abortRef.current?.abort(); setLoading(false); }}
|
||||
style={{
|
||||
padding: '12px 16px', fontSize: 13, background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`, borderRadius: 8,
|
||||
color: theme.text2, cursor: 'pointer',
|
||||
}}
|
||||
>停止</button>
|
||||
)}
|
||||
{!loading && messages.length > 0 && (
|
||||
<button
|
||||
onClick={() => setShowClearConfirm(true)}
|
||||
style={{
|
||||
padding: '12px 16px',
|
||||
fontSize: 13,
|
||||
background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`,
|
||||
borderRadius: 8,
|
||||
color: theme.text2,
|
||||
cursor: 'pointer',
|
||||
padding: '12px 16px', fontSize: 13, background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`, borderRadius: 8,
|
||||
color: theme.text2, cursor: 'pointer',
|
||||
}}
|
||||
>清空</button>
|
||||
)}
|
||||
{messages.filter(m => m.role === 'assistant').length > 0 && (
|
||||
{!loading && messages.filter(m => m.role === 'assistant').length > 0 && (
|
||||
<button
|
||||
onClick={regenerateLastAnswer}
|
||||
disabled={loading}
|
||||
style={{
|
||||
padding: '12px 16px',
|
||||
fontSize: 13,
|
||||
background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`,
|
||||
borderRadius: 8,
|
||||
color: theme.text2,
|
||||
cursor: loading ? 'not-allowed' : 'pointer',
|
||||
padding: '12px 16px', fontSize: 13, background: theme.bgCard,
|
||||
border: `1px solid ${theme.border}`, borderRadius: 8,
|
||||
color: theme.text2, cursor: 'pointer',
|
||||
}}
|
||||
>重生成</button>
|
||||
)}
|
||||
@@ -462,27 +381,16 @@ export const RagChatPage: React.FC = () => {
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div style={{
|
||||
flex: '0 0 40%',
|
||||
display: 'flex',
|
||||
flexDirection: 'column',
|
||||
background: theme.bgCard,
|
||||
}}>
|
||||
{/* ── Right: retrieved sources panel ───────────────────── */}
|
||||
<div style={{ flex: '0 0 40%', display: 'flex', flexDirection: 'column', background: theme.bgCard }}>
|
||||
<div style={{
|
||||
padding: '20px 24px',
|
||||
borderBottom: `1px solid ${theme.border}`,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
gap: 10,
|
||||
padding: '20px 24px', borderBottom: `1px solid ${theme.border}`,
|
||||
display: 'flex', alignItems: 'center', gap: 10,
|
||||
}}>
|
||||
<div style={{
|
||||
width: 28,
|
||||
height: 28,
|
||||
borderRadius: 6,
|
||||
background: theme.gradientAccent,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
width: 28, height: 28, borderRadius: 6,
|
||||
background: theme.gradientAccent, display: 'flex',
|
||||
alignItems: 'center', justifyContent: 'center',
|
||||
}}>
|
||||
<svg width="14" height="14" viewBox="0 0 24 24" fill="none">
|
||||
<path d="M14 2H6C5 2 4 3 4 4V20C4 21 5 22 6 22H18C19 22 20 21 20 20V8L14 2Z" stroke="#fff" strokeWidth="1.5"/>
|
||||
@@ -493,20 +401,13 @@ export const RagChatPage: React.FC = () => {
|
||||
</span>
|
||||
{retrievals.length > 0 && (
|
||||
<span className="mono" style={{
|
||||
fontSize: 11,
|
||||
padding: '4px 10px',
|
||||
background: theme.bgHover,
|
||||
borderRadius: 4,
|
||||
color: theme.text3,
|
||||
fontSize: 11, padding: '4px 10px',
|
||||
background: theme.bgHover, borderRadius: 4, color: theme.text3,
|
||||
}}>{retrievals.length}</span>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div style={{
|
||||
flex: 1,
|
||||
overflowY: 'auto',
|
||||
padding: '16px 24px',
|
||||
}}>
|
||||
<div style={{ flex: 1, overflowY: 'auto', padding: '16px 24px' }}>
|
||||
{retrievals.length > 0 ? (
|
||||
<div style={{ display: 'flex', flexDirection: 'column', gap: 12 }}>
|
||||
{retrievals.map((r, i) => (
|
||||
@@ -515,30 +416,20 @@ export const RagChatPage: React.FC = () => {
|
||||
id={`source-${i + 1}`}
|
||||
onClick={() => setSelectedRetrieval(r)}
|
||||
style={{
|
||||
padding: 16,
|
||||
background: highlightedSourceIdx === i + 1 ? theme.bgElevated : theme.bgHover,
|
||||
borderRadius: 10,
|
||||
border: `1px solid ${highlightedSourceIdx === i + 1 ? theme.accent : theme.border}`,
|
||||
cursor: 'pointer',
|
||||
position: 'relative',
|
||||
padding: 16, background: highlightedSourceIdx === i + 1 ? theme.bgElevated : theme.bgHover,
|
||||
borderRadius: 10, border: `1px solid ${highlightedSourceIdx === i + 1 ? theme.accent : theme.border}`,
|
||||
cursor: 'pointer', position: 'relative',
|
||||
transition: 'border-color 0.2s, background 0.2s',
|
||||
}}
|
||||
>
|
||||
<div style={{
|
||||
position: 'absolute',
|
||||
left: 0,
|
||||
top: 16,
|
||||
bottom: 16,
|
||||
width: 3,
|
||||
background: theme.gradientAccent,
|
||||
borderRadius: 2,
|
||||
position: 'absolute', left: 0, top: 16, bottom: 16,
|
||||
width: 3, background: theme.gradientAccent, borderRadius: 2,
|
||||
}} />
|
||||
<div style={{ paddingLeft: 8 }}>
|
||||
<div style={{
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'space-between',
|
||||
marginBottom: 8,
|
||||
display: 'flex', alignItems: 'center',
|
||||
justifyContent: 'space-between', marginBottom: 8,
|
||||
}}>
|
||||
<span className="mono" style={{ fontSize: 11, fontWeight: 700, color: theme.accent }}>#{i + 1}</span>
|
||||
<div style={{ display: 'flex', alignItems: 'center', gap: 10 }}>
|
||||
@@ -547,17 +438,13 @@ export const RagChatPage: React.FC = () => {
|
||||
href={r.downloadUrl}
|
||||
target="_blank"
|
||||
rel="noreferrer"
|
||||
onClick={(event) => event.stopPropagation()}
|
||||
onClick={e => e.stopPropagation()}
|
||||
style={{ fontSize: 11, color: theme.accent, textDecoration: 'none' }}
|
||||
>
|
||||
下载文档
|
||||
</a>
|
||||
>下载文档</a>
|
||||
)}
|
||||
<span className="mono" style={{
|
||||
fontSize: 11,
|
||||
fontWeight: 600,
|
||||
color: theme.accent,
|
||||
}}>{(r.score * 100).toFixed(0)}%</span>
|
||||
<span className="mono" style={{ fontSize: 11, fontWeight: 600, color: theme.accent }}>
|
||||
{(r.score * 100).toFixed(0)}%
|
||||
</span>
|
||||
</div>
|
||||
</div>
|
||||
<div style={{ fontSize: 13, fontWeight: 500, marginBottom: 4, color: theme.text }}>{r.file}</div>
|
||||
@@ -565,32 +452,20 @@ export const RagChatPage: React.FC = () => {
|
||||
{r.clause}{r.docId ? ` · ${r.docId}` : ''}
|
||||
</div>
|
||||
<div style={{
|
||||
fontSize: 12,
|
||||
color: theme.text2,
|
||||
lineHeight: 1.5,
|
||||
overflow: 'hidden',
|
||||
textOverflow: 'ellipsis',
|
||||
whiteSpace: 'nowrap',
|
||||
fontSize: 12, color: theme.text2, lineHeight: 1.5,
|
||||
display: '-webkit-box', WebkitLineClamp: 3,
|
||||
WebkitBoxOrient: 'vertical', overflow: 'hidden',
|
||||
}}>{r.content}</div>
|
||||
</div>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
) : (
|
||||
<div style={{
|
||||
textAlign: 'center',
|
||||
padding: 40,
|
||||
color: theme.text3,
|
||||
}}>
|
||||
<div style={{ textAlign: 'center', padding: 40, color: theme.text3 }}>
|
||||
<div style={{
|
||||
width: 48,
|
||||
height: 48,
|
||||
borderRadius: 10,
|
||||
background: theme.bgHover,
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
margin: '0 auto 16px',
|
||||
width: 48, height: 48, borderRadius: 10,
|
||||
background: theme.bgHover, display: 'flex', alignItems: 'center',
|
||||
justifyContent: 'center', margin: '0 auto 16px',
|
||||
}}>
|
||||
<svg width="20" height="20" viewBox="0 0 24 24" fill="none">
|
||||
<path d="M14 2H6C5 2 4 3 4 4V20C4 21 5 22 6 22H18C19 22 20 21 20 20V8L14 2Z" stroke={theme.text3} strokeWidth="1.5"/>
|
||||
@@ -602,52 +477,33 @@ export const RagChatPage: React.FC = () => {
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* ── Clear confirm modal ───────────────────────────────── */}
|
||||
{showClearConfirm && (
|
||||
<div style={{
|
||||
position: 'fixed',
|
||||
top: 0,
|
||||
left: 0,
|
||||
right: 0,
|
||||
bottom: 0,
|
||||
background: 'rgba(0,0,0,0.6)',
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
zIndex: 1000,
|
||||
position: 'fixed', top: 0, left: 0, right: 0, bottom: 0,
|
||||
background: 'rgba(0,0,0,0.6)', display: 'flex',
|
||||
alignItems: 'center', justifyContent: 'center', zIndex: 1000,
|
||||
}}>
|
||||
<div style={{
|
||||
padding: 24,
|
||||
background: theme.bgCard,
|
||||
borderRadius: 16,
|
||||
maxWidth: 400,
|
||||
border: `1px solid ${theme.border}`,
|
||||
padding: 24, background: theme.bgCard, borderRadius: 16,
|
||||
maxWidth: 400, border: `1px solid ${theme.border}`,
|
||||
}}>
|
||||
<div style={{ fontSize: 15, fontWeight: 600, marginBottom: 12, color: theme.text }}>确定清空对话？</div>
|
||||
<div style={{ fontSize: 13, color: theme.text2, marginBottom: 20 }}>此操作不可恢复</div>
|
||||
<div style={{ fontSize: 13, color: theme.text2, marginBottom: 20 }}>此操作不可恢复，会话历史将被重置</div>
|
||||
<div style={{ display: 'flex', gap: 10, justifyContent: 'flex-end' }}>
|
||||
<button
|
||||
onClick={() => setShowClearConfirm(false)}
|
||||
style={{
|
||||
padding: '10px 18px',
|
||||
fontSize: 13,
|
||||
background: theme.bgHover,
|
||||
border: 'none',
|
||||
borderRadius: 8,
|
||||
color: theme.text2,
|
||||
cursor: 'pointer',
|
||||
padding: '10px 18px', fontSize: 13, background: theme.bgHover,
|
||||
border: 'none', borderRadius: 8, color: theme.text2, cursor: 'pointer',
|
||||
}}
|
||||
>取消</button>
|
||||
<button
|
||||
onClick={clearMessages}
|
||||
style={{
|
||||
padding: '10px 18px',
|
||||
fontSize: 13,
|
||||
fontWeight: 600,
|
||||
background: theme.accent,
|
||||
border: 'none',
|
||||
borderRadius: 8,
|
||||
color: '#fff',
|
||||
cursor: 'pointer',
|
||||
padding: '10px 18px', fontSize: 13, fontWeight: 600,
|
||||
background: theme.accent, border: 'none', borderRadius: 8,
|
||||
color: '#fff', cursor: 'pointer',
|
||||
}}
|
||||
>确认</button>
|
||||
</div>
|
||||
@@ -655,47 +511,31 @@ export const RagChatPage: React.FC = () => {
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* ── Source detail modal ───────────────────────────────── */}
|
||||
{selectedRetrieval && (
|
||||
<div
|
||||
onClick={() => setSelectedRetrieval(null)}
|
||||
style={{
|
||||
position: 'fixed',
|
||||
top: 0,
|
||||
left: 0,
|
||||
right: 0,
|
||||
bottom: 0,
|
||||
background: 'rgba(0,0,0,0.6)',
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
zIndex: 1000,
|
||||
position: 'fixed', top: 0, left: 0, right: 0, bottom: 0,
|
||||
background: 'rgba(0,0,0,0.6)', display: 'flex',
|
||||
alignItems: 'center', justifyContent: 'center', zIndex: 1000,
|
||||
}}
|
||||
>
|
||||
<div
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
onClick={e => e.stopPropagation()}
|
||||
style={{
|
||||
width: 520,
|
||||
maxWidth: '90%',
|
||||
maxHeight: '80%',
|
||||
padding: 24,
|
||||
background: theme.bgCard,
|
||||
borderRadius: 16,
|
||||
border: `1px solid ${theme.accent}`,
|
||||
width: 520, maxWidth: '90%', maxHeight: '80%',
|
||||
overflowY: 'auto', padding: 24, background: theme.bgCard,
|
||||
borderRadius: 16, border: `1px solid ${theme.accent}`,
|
||||
boxShadow: '0 8px 32px rgba(0,0,0,0.3)',
|
||||
}}
|
||||
>
|
||||
<div style={{
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'space-between',
|
||||
marginBottom: 16,
|
||||
display: 'flex', alignItems: 'center',
|
||||
justifyContent: 'space-between', marginBottom: 16,
|
||||
}}>
|
||||
<div style={{ display: 'flex', alignItems: 'center', gap: 10, flexWrap: 'wrap' }}>
|
||||
<div style={{
|
||||
padding: '4px 10px',
|
||||
background: theme.gradientAccent,
|
||||
borderRadius: 6,
|
||||
}}>
|
||||
<div style={{ padding: '4px 10px', background: theme.gradientAccent, borderRadius: 6 }}>
|
||||
<span className="mono" style={{ fontSize: 11, fontWeight: 600, color: '#fff' }}>
|
||||
{(selectedRetrieval.score * 100).toFixed(0)}%
|
||||
</span>
|
||||
@@ -707,23 +547,15 @@ export const RagChatPage: React.FC = () => {
|
||||
target="_blank"
|
||||
rel="noreferrer"
|
||||
style={{ fontSize: 12, color: theme.accent, textDecoration: 'none' }}
|
||||
>
|
||||
下载关联文档
|
||||
</a>
|
||||
>下载关联文档</a>
|
||||
)}
|
||||
</div>
|
||||
<button
|
||||
onClick={() => setSelectedRetrieval(null)}
|
||||
style={{
|
||||
width: 28,
|
||||
height: 28,
|
||||
background: theme.bgHover,
|
||||
border: 'none',
|
||||
borderRadius: 6,
|
||||
cursor: 'pointer',
|
||||
display: 'flex',
|
||||
alignItems: 'center',
|
||||
justifyContent: 'center',
|
||||
width: 28, height: 28, background: theme.bgHover,
|
||||
border: 'none', borderRadius: 6, cursor: 'pointer',
|
||||
display: 'flex', alignItems: 'center', justifyContent: 'center',
|
||||
}}
|
||||
>
|
||||
<svg width="14" height="14" viewBox="0 0 24 24" fill="none">
|
||||
@@ -731,23 +563,15 @@ export const RagChatPage: React.FC = () => {
|
||||
</svg>
|
||||
</button>
|
||||
</div>
|
||||
<div style={{
|
||||
padding: '10px 14px',
|
||||
background: theme.bgHover,
|
||||
borderRadius: 8,
|
||||
marginBottom: 16,
|
||||
}}>
|
||||
<div style={{ padding: '10px 14px', background: theme.bgHover, borderRadius: 8, marginBottom: 16 }}>
|
||||
<span className="mono" style={{ fontSize: 12, color: theme.accent }}>{selectedRetrieval.clause}</span>
|
||||
{selectedRetrieval.docId && (
|
||||
<span className="mono" style={{ fontSize: 11, color: theme.text3, marginLeft: 8 }}>{selectedRetrieval.docId}</span>
|
||||
)}
|
||||
</div>
|
||||
<div style={{
|
||||
fontSize: 14,
|
||||
lineHeight: 1.7,
|
||||
color: theme.text2,
|
||||
whiteSpace: 'pre-wrap',
|
||||
}}>{selectedRetrieval.content}</div>
|
||||
<div style={{ fontSize: 14, lineHeight: 1.7, color: theme.text2, whiteSpace: 'pre-wrap' }}>
|
||||
{selectedRetrieval.content}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
Reference in New Issue
Block a user