fix: propagate real message IDs via SSE and fix regenerate model parameter

- Add message_id and user_message_id to SSE done events so frontend
  can replace temp IDs with real DB IDs, fixing "消息不存在" on regenerate
- Replace Body(default=None) with RegenerateRequest Pydantic model for
  proper JSON body parsing, fixing 422 Unprocessable Content
- Frontend always sends model in regenerate request body
- Pass selected model through message-item → regenerateMessage chain
- Various UI refinements to chat sidebar, quick questions, and layout

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
2026-05-27 09:00:31 +08:00
parent 53a9380d7a
commit 0251a5e3ff
10 changed files with 549 additions and 858 deletions
+39 -12
View File
@@ -282,8 +282,9 @@ async def stream_message(
) )
db.add(assistant_message) db.add(assistant_message)
db.commit() db.commit()
db.refresh(assistant_message)
yield f"data: {json.dumps({'type': 'done', 'session_id': session.id}, ensure_ascii=False)}\n\n" yield f"data: {json.dumps({'type': 'done', 'session_id': session.id, 'message_id': assistant_message.id, 'user_message_id': user_message.id}, ensure_ascii=False)}\n\n"
else: else:
# 普通模式:使用LangChain 1.0对话链 # 普通模式:使用LangChain 1.0对话链
@@ -321,8 +322,9 @@ async def stream_message(
) )
db.add(assistant_message) db.add(assistant_message)
db.commit() db.commit()
db.refresh(assistant_message)
yield f"data: {json.dumps({'type': 'done', 'session_id': session.id}, ensure_ascii=False)}\n\n" yield f"data: {json.dumps({'type': 'done', 'session_id': session.id, 'message_id': assistant_message.id, 'user_message_id': user_message.id}, ensure_ascii=False)}\n\n"
except Exception as e: except Exception as e:
yield f"data: {json.dumps({'error': str(e)})}\n\n" yield f"data: {json.dumps({'error': str(e)})}\n\n"
@@ -555,6 +557,11 @@ class MessageFeedbackRequest(BaseModel):
feedback: str # like, dislike feedback: str # like, dislike
class RegenerateRequest(BaseModel):
"""重新生成请求模型"""
model: Optional[str] = None
@router.put("/messages/{message_id}") @router.put("/messages/{message_id}")
async def edit_message( async def edit_message(
message_id: int, message_id: int,
@@ -588,22 +595,41 @@ async def edit_message(
async def regenerate_message( async def regenerate_message(
message_id: int, message_id: int,
current_user: str = Depends(get_current_user), current_user: str = Depends(get_current_user),
db: Session = Depends(get_db) db: Session = Depends(get_db),
request: RegenerateRequest = None,
): ):
"""重新生成AI回复""" """重新生成AI回复 — 支持 user 和 assistant 消息 ID"""
# 获取原始消息 model = request.model if request else None
original_message = db.query(ChatMessage).join(ChatSession).filter( # 查找目标消息
target_message = db.query(ChatMessage).join(ChatSession).filter(
ChatMessage.id == message_id, ChatMessage.id == message_id,
ChatMessage.role == "user",
ChatSession.user_id == get_user_id_by_username(db, current_user) ChatSession.user_id == get_user_id_by_username(db, current_user)
).first() ).first()
if not original_message: if not target_message:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, status_code=status.HTTP_404_NOT_FOUND,
detail="消息不存在" detail="消息不存在"
) )
# 如果传入的是 assistant 消息,找到同 session 中前一条 user 消息
if target_message.role == "assistant":
original_message = db.query(ChatMessage).filter(
ChatMessage.session_id == target_message.session_id,
ChatMessage.role == "user",
ChatMessage.created_at < target_message.created_at
).order_by(ChatMessage.created_at.desc()).first()
if not original_message:
raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND,
detail="找不到对应的用户消息"
)
# 删除该 assistant 消息本身
db.delete(target_message)
else:
original_message = target_message
# 删除该消息之后的所有消息 # 删除该消息之后的所有消息
later_messages = db.query(ChatMessage).filter( later_messages = db.query(ChatMessage).filter(
ChatMessage.session_id == original_message.session_id, ChatMessage.session_id == original_message.session_id,
@@ -620,7 +646,8 @@ async def regenerate_message(
original_message.content, original_message.content,
original_message.session_id, original_message.session_id,
db, db,
None # No knowledge base filtering for regeneration None, # No knowledge base filtering for regeneration
model
) )
# 创建新的AI回复 # 创建新的AI回复
@@ -679,7 +706,7 @@ async def feedback_message(
return {"message": "反馈提交成功", "feedback": request.feedback} return {"message": "反馈提交成功", "feedback": request.feedback}
async def run_rag_workflow_with_context(question: str, session_id: int, db: Session, knowledge_base_ids: Optional[List[int]] = None) -> Dict[str, Any]: async def run_rag_workflow_with_context(question: str, session_id: int, db: Session, knowledge_base_ids: Optional[List[int]] = None, model: Optional[str] = None) -> Dict[str, Any]:
"""运行带上下文的RAG工作流""" """运行带上下文的RAG工作流"""
try: try:
# 获取会话历史消息作为上下文 # 获取会话历史消息作为上下文
@@ -696,7 +723,7 @@ async def run_rag_workflow_with_context(question: str, session_id: int, db: Sess
context_messages.append({"role": "assistant", "content": msg.content}) context_messages.append({"role": "assistant", "content": msg.content})
# 运行RAG工作流 # 运行RAG工作流
rag_chain = create_rag_chain(knowledge_base_ids=knowledge_base_ids) rag_chain = create_rag_chain(knowledge_base_ids=knowledge_base_ids, model=model)
result = rag_chain.invoke(question) result = rag_chain.invoke(question)
# 如果有上下文,增强回答 # 如果有上下文,增强回答
@@ -710,5 +737,5 @@ async def run_rag_workflow_with_context(question: str, session_id: int, db: Sess
except Exception as e: except Exception as e:
print(f"RAG工作流执行失败: {str(e)}") print(f"RAG工作流执行失败: {str(e)}")
# 降级到基础问答 # 降级到基础问答
rag_chain = create_rag_chain() rag_chain = create_rag_chain(model=model)
return rag_chain.invoke(question) return rag_chain.invoke(question)
+37 -8
View File
@@ -8,12 +8,15 @@ import ChatInterface from "@/components/chat/chat-interface";
import Sidebar from "@/components/chat/sidebar"; import Sidebar from "@/components/chat/sidebar";
import MobileNav from "@/components/layout/mobile-nav"; import MobileNav from "@/components/layout/mobile-nav";
import LoadingSpinner from "@/components/ui/loading-spinner"; import LoadingSpinner from "@/components/ui/loading-spinner";
import { Button } from "@/components/ui/button";
import { PanelLeftClose, PanelLeftOpen } from "lucide-react";
export default function ChatPage() { export default function ChatPage() {
const router = useRouter(); const router = useRouter();
const { isAuthenticated, user, isLoading: authLoading } = useAuthStore(); const { isAuthenticated, user, isLoading: authLoading } = useAuthStore();
const { loadSessions, isLoading: chatLoading } = useChatStore(); const { loadSessions, isLoading: chatLoading } = useChatStore();
const [isInitialized, setIsInitialized] = useState(false); const [isInitialized, setIsInitialized] = useState(false);
const [sidebarOpen, setSidebarOpen] = useState(true);
useEffect(() => { useEffect(() => {
if (!authLoading && !isAuthenticated) { if (!authLoading && !isAuthenticated) {
@@ -22,7 +25,6 @@ export default function ChatPage() {
} }
if (isAuthenticated && !isInitialized) { if (isAuthenticated && !isInitialized) {
// 加载聊天会话
loadSessions(); loadSessions();
setIsInitialized(true); setIsInitialized(true);
} }
@@ -41,18 +43,45 @@ export default function ChatPage() {
} }
return ( return (
<div className="h-[calc(100vh-4rem)] bg-background flex"> <div className="h-[calc(100vh-4rem)] bg-background">
{/* 侧边栏 */} <div className="h-full max-w-7xl mx-auto px-4 sm:px-6 lg:px-8 flex gap-4 py-4">
<div className="w-80 border-r bg-card/50 flex-shrink-0"> {/* Mobile overlay */}
<Sidebar /> {sidebarOpen && (
<div
className="fixed inset-0 bg-black/30 z-30 md:hidden"
onClick={() => setSidebarOpen(false)}
/>
)}
{/* Sidebar — persistent on desktop, overlay on mobile */}
<div className={`
shrink-0 z-40
w-[260px] rounded-xl border border-border/40 bg-card shadow-sm
transition-all duration-200 ease-in-out overflow-hidden
fixed md:relative md:top-auto md:left-auto md:bottom-auto
${sidebarOpen ? 'translate-x-0' : '-translate-x-full md:hidden'}
`}>
<Sidebar onClose={() => setSidebarOpen(false)} />
</div> </div>
{/* 主聊天区域 */} {/* Main chat area */}
<div className="flex-1 flex flex-col min-w-0"> <div className="flex-1 flex flex-col min-w-0 relative rounded-xl border border-border/40 bg-card shadow-sm overflow-hidden">
{/* Toggle button */}
{!sidebarOpen && (
<Button
variant="ghost"
size="icon"
onClick={() => setSidebarOpen(true)}
className="absolute top-3 left-3 z-10 h-9 w-9 text-muted-foreground hover:text-foreground hover:bg-muted/50"
>
<PanelLeftOpen className="w-5 h-5" />
</Button>
)}
<ChatInterface /> <ChatInterface />
</div> </div>
</div>
{/* 移动端导航 */}
<MobileNav /> <MobileNav />
</div> </div>
); );
@@ -69,13 +69,6 @@ export default function CourseContentPage() {
setIsLoading(true); setIsLoading(true);
setError(null); setError(null);
const structure = await courseContentAPI.getCourseContent(); const structure = await courseContentAPI.getCourseContent();
console.log("加载的书籍结构:", structure);
console.log("章节数量:", structure?.chapters?.length || 0);
if (structure?.chapters) {
structure.chapters.forEach((chapter, index) => {
console.log(`章节 ${index + 1}:`, chapter.title, "节数:", chapter.sections?.length || 0);
});
}
setBookStructure(structure); setBookStructure(structure);
} catch (err) { } catch (err) {
console.error("加载课程内容失败:", err); console.error("加载课程内容失败:", err);
+112 -287
View File
@@ -4,7 +4,7 @@ import { useState, useRef, useEffect } from "react";
import { useChatStore } from "@/store/chat"; import { useChatStore } from "@/store/chat";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input"; import { Input } from "@/components/ui/input";
import { Send, Loader2, Bot, User, StopCircle, ArrowDown, Paperclip } from "lucide-react"; import { Send, Square, Bot, ChevronDown, Settings2 } from "lucide-react";
import MessageList from "./message-list"; import MessageList from "./message-list";
import QuickQuestions from "./quick-questions"; import QuickQuestions from "./quick-questions";
import ModeSelector, { ChatMode } from "./mode-selector"; import ModeSelector, { ChatMode } from "./mode-selector";
@@ -21,8 +21,11 @@ export default function ChatInterface() {
const [selectedKnowledgeBases, setSelectedKnowledgeBases] = useState<string[]>([]); const [selectedKnowledgeBases, setSelectedKnowledgeBases] = useState<string[]>([]);
const [systemKnowledgeBases, setSystemKnowledgeBases] = useState<KnowledgeBase[]>([]); const [systemKnowledgeBases, setSystemKnowledgeBases] = useState<KnowledgeBase[]>([]);
const [userKnowledgeBases, setUserKnowledgeBases] = useState<KnowledgeBase[]>([]); const [userKnowledgeBases, setUserKnowledgeBases] = useState<KnowledgeBase[]>([]);
const [showScrollDown, setShowScrollDown] = useState(false);
const [showSettings, setShowSettings] = useState(false);
const inputRef = useRef<HTMLInputElement>(null); const inputRef = useRef<HTMLInputElement>(null);
const messagesEndRef = useRef<HTMLDivElement>(null); const messagesEndRef = useRef<HTMLDivElement>(null);
const scrollContainerRef = useRef<HTMLDivElement>(null);
const { const {
currentSession, currentSession,
@@ -36,91 +39,44 @@ export default function ChatInterface() {
selectSession, selectSession,
} = useChatStore(); } = useChatStore();
// 调试日志
useEffect(() => {
console.log("[DEBUG-CHAT-INTERFACE] 组件状态:", {
currentSession: currentSession?.id,
messagesCount: messages.length,
isLoading,
isStreaming,
messages: messages.map(m => ({ id: m.id, role: m.role, contentLength: m.content.length }))
});
}, [currentSession, messages, isLoading, isStreaming]);
// 自动滚动到底部
useEffect(() => {
messagesEndRef.current?.scrollIntoView({
behavior: isStreaming ? "auto" : "smooth"
});
}, [messages, isStreaming]);
// 加载知识库(区分系统知识库和用户知识库)
useEffect(() => { useEffect(() => {
const loadKnowledgeBases = async () => { const loadKnowledgeBases = async () => {
try { try {
const bases = await knowledgeBaseAPI.getKnowledgeBases(); const bases = await knowledgeBaseAPI.getKnowledgeBases();
console.log("[DEBUG] 加载的知识库:", bases.map(kb => ({
id: kb.id,
name: kb.name,
is_system: kb.is_system
})));
// 转换为KnowledgeSelector需要的格式
const formattedBases: KnowledgeBase[] = bases.map(kb => ({ const formattedBases: KnowledgeBase[] = bases.map(kb => ({
id: kb.id.toString(), id: kb.id.toString(),
name: kb.name, name: kb.name,
description: kb.description || "", description: kb.description || "",
documentCount: kb.document_count, documentCount: kb.document_count,
enabled: true, // 默认启用 enabled: true,
isSystem: kb.is_system === true, // 明确检查是否为 true isSystem: kb.is_system === true,
})); }));
setSystemKnowledgeBases(formattedBases.filter(kb => kb.isSystem));
// 分离系统知识库和用户知识库 setUserKnowledgeBases(formattedBases.filter(kb => !kb.isSystem));
const systemBases = formattedBases.filter(kb => kb.isSystem === true);
const userBases = formattedBases.filter(kb => kb.isSystem !== true);
console.log("[DEBUG] 系统知识库:", systemBases.map(kb => ({ id: kb.id, name: kb.name })));
console.log("[DEBUG] 用户知识库:", userBases.map(kb => ({ id: kb.id, name: kb.name })));
setSystemKnowledgeBases(systemBases);
setUserKnowledgeBases(userBases);
} catch (error) { } catch (error) {
console.error("加载知识库失败:", error); console.error("加载知识库失败:", error);
} }
}; };
loadKnowledgeBases(); loadKnowledgeBases();
}, []); }, []);
const handleSendMessage = async () => { const handleSendMessage = async () => {
if (!inputMessage.trim() || isLoading || isStreaming) return; if (!inputMessage.trim() || isLoading || isStreaming) return;
const message = inputMessage.trim(); const message = inputMessage.trim();
setInputMessage(""); setInputMessage("");
setShowSettings(false);
console.log("[DEBUG-CHAT] 发送消息:", {
message: message,
chatMode: chatMode,
selectedKnowledgeBases: selectedKnowledgeBases,
selectedModel: selectedModel,
currentSession: currentSession?.id
});
// 如果没有当前会话,先创建一个新会话
if (!currentSession) { if (!currentSession) {
const newSession = await createSession("新对话"); const newSession = await createSession("新对话");
if (newSession) { if (newSession) {
await selectSession(newSession.id); await selectSession(newSession.id);
} }
} }
// 使用流式发送,传递聊天模式、选中的知识库ID和模型ID
await streamMessage(message, chatMode, selectedKnowledgeBases, selectedModel); await streamMessage(message, chatMode, selectedKnowledgeBases, selectedModel);
}; };
const handleKeyPress = (e: React.KeyboardEvent) => { const handleKeyPress = (e: React.KeyboardEvent) => {
if (e.key === "Enter" && !e.shiftKey) { if (e.key === "Enter" && !e.shiftKey && !isComposing) {
e.preventDefault(); e.preventDefault();
handleSendMessage(); handleSendMessage();
} }
@@ -128,48 +84,46 @@ export default function ChatInterface() {
const handleQuickQuestion = async (question: string) => { const handleQuickQuestion = async (question: string) => {
setInputMessage(question); setInputMessage(question);
setTimeout(() => inputRef.current?.focus(), 100);
// 聚焦到输入框
setTimeout(() => {
if (inputRef.current) {
inputRef.current.focus();
}
}, 100);
}; };
// Debug logging const scrollToBottom = () => {
console.log("[CHAT-INTERFACE] Current state:", { messagesEndRef.current?.scrollIntoView({ behavior: "smooth" });
hasSession: !!currentSession, };
sessionId: currentSession?.id,
messagesCount: messages.length
});
if (!currentSession) { const handleScroll = () => {
return ( if (!scrollContainerRef.current) return;
<div className="flex-1 flex flex-col bg-background h-full"> const { scrollTop, scrollHeight, clientHeight } = scrollContainerRef.current;
{/* 上部:欢迎内容区域 - 使用 flex-1 + 可滚动 */} setShowScrollDown(scrollHeight - scrollTop - clientHeight > 120);
<div className="flex-1 overflow-y-auto"> };
<div className="h-full flex items-center justify-center p-6">
<div className="text-center max-w-2xl px-4"> useEffect(() => {
<div className="w-16 h-16 bg-gradient-to-r from-blue-600 to-purple-600 rounded-xl flex items-center justify-center mx-auto mb-4 shadow-lg"> if (isStreaming) {
<Bot className="w-8 h-8 text-white" /> messagesEndRef.current?.scrollIntoView({ behavior: "auto" });
}
}, [messages, isStreaming]);
const renderWelcome = (size: "lg" | "sm") => (
<div className="text-center max-w-lg mx-auto px-4">
<div className="w-12 h-12 bg-primary/10 rounded-2xl flex items-center justify-center mx-auto mb-4">
<Bot className="w-6 h-6 text-primary" />
</div> </div>
<h3 className="text-2xl font-bold text-foreground mb-2"> <h1 className={cn("font-semibold text-foreground mb-1.5", size === "lg" ? "text-xl" : "text-lg")}>
使
</h3> </h1>
<p className="text-muted-foreground mb-6"> <p className="text-sm text-muted-foreground mb-6 max-w-sm mx-auto">
</p> </p>
<QuickQuestions onSelect={handleQuickQuestion} /> <QuickQuestions onSelect={handleQuickQuestion} />
</div> </div>
</div> );
</div>
{/* 底部:输入区域 - 不使用 sticky,直接作为 flex 子元素 */} const renderInputArea = () => (
<div className="border-t border-border bg-card p-4 lg:p-6 flex-shrink-0"> <div className="border-t border-border/40 bg-card/80 backdrop-blur-sm flex-shrink-0">
{/* 功能选择器 */} <div className="max-w-3xl mx-auto px-4 py-3">
<div className="mb-4 space-y-3"> {/* Settings panel — collapsible */}
<div className="flex items-center space-x-2"> {showSettings && (
<div className="flex items-center gap-2 mb-3 pb-3 border-b border-border/30">
<ModeSelector mode={chatMode} onModeChange={setChatMode} /> <ModeSelector mode={chatMode} onModeChange={setChatMode} />
<ModelSelector <ModelSelector
selectedModel={selectedModel} selectedModel={selectedModel}
@@ -183,240 +137,111 @@ export default function ChatInterface() {
userKnowledgeBases={userKnowledgeBases} userKnowledgeBases={userKnowledgeBases}
className="flex-shrink-0" className="flex-shrink-0"
/> />
<Button
variant="outline"
size="sm"
className="h-9 px-3 flex-shrink-0"
title="上传附件"
>
<Paperclip className="w-4 h-4" />
</Button>
</div>
</div> </div>
)}
{/* 输入框和发送按钮 */} {/* Input row */}
<div className="flex items-end space-x-3"> <div className="flex items-center gap-2">
<div className="flex-1"> <Button
<div className="relative"> variant="ghost"
<Input size="icon"
className={cn(
"h-9 w-9 shrink-0 rounded-lg text-muted-foreground",
showSettings && "bg-muted text-foreground"
)}
onClick={() => setShowSettings(!showSettings)}
title="设置"
>
<Settings2 className="w-4 h-4" />
</Button>
<div className="flex-1 relative">
<input
ref={inputRef} ref={inputRef}
value={inputMessage} value={inputMessage}
onChange={(e) => setInputMessage(e.target.value)} onChange={(e) => setInputMessage(e.target.value)}
onKeyPress={handleKeyPress} onKeyDown={handleKeyPress}
onCompositionStart={() => setIsComposing(true)} onCompositionStart={() => setIsComposing(true)}
onCompositionEnd={() => setIsComposing(false)} onCompositionEnd={() => setIsComposing(false)}
placeholder="输入您的问题..." placeholder="输入您的问题"
disabled={isLoading || isStreaming} disabled={isLoading || isStreaming}
className={cn( className={cn(
"min-h-[48px] resize-none touch-manipulation pr-12 rounded-xl", "w-full h-10 px-4 rounded-xl text-sm",
"border-border/50 bg-background/80 backdrop-blur-sm", "bg-muted/40 border border-border/30",
"focus:ring-2 focus:ring-blue-500/20 focus:border-blue-500/50", "placeholder:text-muted-foreground/60",
"transition-all duration-200" "focus:outline-none focus:ring-2 focus:ring-primary/20 focus:border-primary/30",
"transition-all duration-150"
)} )}
/> />
{isStreaming && (
<div className="absolute right-3 top-1/2 -translate-y-1/2">
<div className="flex items-center gap-1">
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" />
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" style={{ animationDelay: '0.1s' }} />
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" style={{ animationDelay: '0.2s' }} />
</div>
</div>
)}
</div>
</div>
<Button
onClick={handleSendMessage}
disabled={!inputMessage.trim() || isLoading || isStreaming}
size="icon"
className={cn(
"h-12 w-12 flex-shrink-0 touch-manipulation rounded-xl",
"bg-gradient-to-r from-blue-600 to-purple-600 hover:from-blue-700 hover:to-purple-700",
"shadow-lg hover:shadow-xl transition-all duration-200",
"disabled:opacity-50 disabled:cursor-not-allowed"
)}
>
{isLoading || isStreaming ? (
<Loader2 className="w-5 h-5 animate-spin" />
) : (
<Send className="w-5 h-5" />
)}
</Button>
</div> </div>
{/* 输入提示 */} <button
<div className="mt-3 flex items-center justify-between text-xs text-muted-foreground"> onClick={isStreaming ? stopGeneration : handleSendMessage}
<span className="hidden sm:block"> Enter Shift + Enter </span> disabled={!isStreaming && (!inputMessage.trim() || isLoading)}
<span className="sm:hidden"></span> className={cn(
{isStreaming && ( "h-9 w-9 shrink-0 rounded-lg inline-flex items-center justify-center",
<span className="flex items-center gap-1 text-blue-600"> "transition-all duration-150",
<div className="w-1.5 h-1.5 bg-blue-500 rounded-full animate-pulse" /> isStreaming
... ? "bg-destructive hover:bg-destructive/90 text-white"
</span> : "bg-[#2563eb] hover:bg-[#1d4ed8] text-white disabled:bg-[#2563eb]/40 disabled:cursor-not-allowed"
)} )}
>
{isStreaming ? (
<Square className="w-4 h-4 fill-current" />
) : (
<Send className="w-4 h-4" />
)}
</button>
</div> </div>
</div> </div>
</div> </div>
); );
// Empty state — no session selected
if (!currentSession) {
return (
<div className="flex-1 flex flex-col bg-background h-full">
<div className="flex-1 overflow-y-auto">
<div className="h-full flex items-center justify-center p-6">
{renderWelcome("lg")}
</div>
</div>
{renderInputArea()}
</div>
);
} }
return ( return (
<div className="flex-1 flex flex-col bg-background h-full"> <div className="flex-1 flex flex-col bg-background h-full relative">
{/* 聊天头部 - 不需要 sticky,作为 flex 子元素自然在顶部 */} {/* Scroll-to-bottom */}
<div className="border-b border-border bg-card p-4 lg:p-6 flex-shrink-0"> {showScrollDown && (
<div className="flex items-center justify-between"> <button
<div className="flex items-center space-x-3"> onClick={scrollToBottom}
<div className="w-8 h-8 bg-gradient-to-r from-blue-600 to-purple-600 rounded-lg flex items-center justify-center flex-shrink-0 shadow-sm"> className="absolute bottom-20 left-1/2 -translate-x-1/2 z-10 w-8 h-8 rounded-full bg-card border border-border/50 shadow-sm flex items-center justify-center hover:bg-muted/50 transition-colors"
<Bot className="w-5 h-5 text-white" />
</div>
<div className="flex-1 min-w-0">
<h2 className="text-lg font-semibold text-foreground truncate">
{currentSession.title}
</h2>
</div>
</div>
{/* 操作按钮 */}
<div className="flex items-center gap-2">
{isStreaming && (
<Button
variant="outline"
size="sm"
onClick={stopGeneration}
className="text-red-600 hover:text-red-700 hover:bg-red-50"
> >
<StopCircle className="w-4 h-4 mr-2" /> <ChevronDown className="w-4 h-4" />
</button>
</Button>
)} )}
<Button
variant="ghost"
size="sm"
onClick={() => messagesEndRef.current?.scrollIntoView({ behavior: "smooth" })}
className="opacity-60 hover:opacity-100"
>
<ArrowDown className="w-4 h-4" />
</Button>
</div>
</div>
</div>
{/* 消息列表 - 使用 flex-1 占据剩余空间 + 可滚动 */} {/* Messages */}
<div className="flex-1 overflow-y-auto"> <div
ref={scrollContainerRef}
className="flex-1 overflow-y-auto"
onScroll={handleScroll}
>
{messages.length === 0 ? ( {messages.length === 0 ? (
<div className="h-full flex items-center justify-center p-6"> <div className="h-full flex items-center justify-center p-6">
<div className="text-center max-w-2xl"> {renderWelcome("sm")}
<div className="w-16 h-16 bg-gradient-to-r from-blue-600 to-purple-600 rounded-xl flex items-center justify-center mx-auto mb-4 shadow-lg">
<Bot className="w-8 h-8 text-white" />
</div>
<h3 className="text-2xl font-bold text-foreground mb-2">
</h3>
<p className="text-muted-foreground mb-6 max-w-lg mx-auto">
</p>
<QuickQuestions onSelect={handleQuickQuestion} />
</div>
</div> </div>
) : ( ) : (
<> <>
<MessageList messages={messages} /> <MessageList messages={messages} selectedModel={selectedModel} />
<div ref={messagesEndRef} /> <div ref={messagesEndRef} />
</> </>
)} )}
</div> </div>
{/* 输入区域 - 不需要 sticky,作为 flex 子元素自然在底部 */} {renderInputArea()}
<div className="border-t border-border bg-card p-4 lg:p-6 flex-shrink-0">
{/* 功能选择器 */}
<div className="mb-4 space-y-3">
<div className="flex items-center space-x-2">
<ModeSelector mode={chatMode} onModeChange={setChatMode} />
<ModelSelector
selectedModel={selectedModel}
onModelChange={setSelectedModel}
className="flex-shrink-0"
/>
<KnowledgeSelector
selectedBases={selectedKnowledgeBases}
onBasesChange={setSelectedKnowledgeBases}
systemKnowledgeBases={systemKnowledgeBases}
userKnowledgeBases={userKnowledgeBases}
className="flex-shrink-0"
/>
<Button
variant="outline"
size="sm"
className="h-9 px-3 flex-shrink-0"
title="上传附件"
>
<Paperclip className="w-4 h-4" />
</Button>
</div>
</div>
{/* 输入框和发送按钮 */}
<div className="flex items-end space-x-3">
<div className="flex-1">
<div className="relative">
<Input
ref={inputRef}
value={inputMessage}
onChange={(e) => setInputMessage(e.target.value)}
onKeyPress={handleKeyPress}
onCompositionStart={() => setIsComposing(true)}
onCompositionEnd={() => setIsComposing(false)}
placeholder="输入您的问题..."
disabled={isLoading || isStreaming}
className={cn(
"min-h-[48px] resize-none touch-manipulation pr-12 rounded-xl",
"border-border/50 bg-background/80 backdrop-blur-sm",
"focus:ring-2 focus:ring-blue-500/20 focus:border-blue-500/50",
"transition-all duration-200"
)}
/>
{isStreaming && (
<div className="absolute right-3 top-1/2 -translate-y-1/2">
<div className="flex items-center gap-1">
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" />
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" style={{ animationDelay: '0.1s' }} />
<div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce" style={{ animationDelay: '0.2s' }} />
</div>
</div>
)}
</div>
</div>
<Button
onClick={handleSendMessage}
disabled={!inputMessage.trim() || isLoading || isStreaming}
size="icon"
className={cn(
"h-12 w-12 flex-shrink-0 touch-manipulation rounded-xl",
"bg-gradient-to-r from-blue-600 to-purple-600 hover:from-blue-700 hover:to-purple-700",
"shadow-lg hover:shadow-xl transition-all duration-200",
"disabled:opacity-50 disabled:cursor-not-allowed"
)}
>
{isLoading || isStreaming ? (
<Loader2 className="w-5 h-5 animate-spin" />
) : (
<Send className="w-5 h-5" />
)}
</Button>
</div>
{/* 输入提示 */}
<div className="mt-3 flex items-center justify-between text-xs text-muted-foreground">
<span className="hidden sm:block"> Enter Shift + Enter </span>
<span className="sm:hidden"></span>
{isStreaming && (
<span className="flex items-center gap-1 text-blue-600">
<div className="w-1.5 h-1.5 bg-blue-500 rounded-full animate-pulse" />
...
</span>
)}
</div>
</div>
</div> </div>
); );
} }
+79 -155
View File
@@ -1,7 +1,8 @@
"use client"; "use client";
import { ChatMessage, ThinkingStep } from "@/types"; import { ChatMessage, ThinkingStep } from "@/types";
import { User, Bot, MoreVertical, ThumbsUp, ThumbsDown, Copy, Edit, RotateCcw, Trash2, Loader2, CheckCircle2, FileSearch, Brain, Sparkles } from "lucide-react"; import { User, Bot, Copy, Edit, RotateCcw, ThumbsUp, ThumbsDown, Loader2, CheckCircle2, FileSearch, Brain, Sparkles } from "lucide-react";
import { cn } from "@/lib/utils";
import ReactMarkdown from "react-markdown"; import ReactMarkdown from "react-markdown";
import remarkGfm from "remark-gfm"; import remarkGfm from "remark-gfm";
import { Prism as SyntaxHighlighter } from "react-syntax-highlighter"; import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
@@ -9,12 +10,6 @@ import { tomorrow } from "react-syntax-highlighter/dist/esm/styles/prism";
import { formatDistanceToNow } from "date-fns"; import { formatDistanceToNow } from "date-fns";
import { zhCN } from "date-fns/locale"; import { zhCN } from "date-fns/locale";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import {
DropdownMenu,
DropdownMenuContent,
DropdownMenuItem,
DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { useChatStore } from "@/store/chat"; import { useChatStore } from "@/store/chat";
import SourceReferences from "./source-references"; import SourceReferences from "./source-references";
import { useState } from "react"; import { useState } from "react";
@@ -22,84 +17,64 @@ import { toast } from "sonner";
interface MessageItemProps { interface MessageItemProps {
message: ChatMessage; message: ChatMessage;
selectedModel?: string;
} }
// 思考过程组件
const ThinkingProcess = ({ thinking }: { thinking: ThinkingStep[] }) => { const ThinkingProcess = ({ thinking }: { thinking: ThinkingStep[] }) => {
if (!thinking || thinking.length === 0) return null; if (!thinking || thinking.length === 0) return null;
const getStageIcon = (stage: string) => { const getStageIcon = (stage: string) => {
switch (stage) { switch (stage) {
case 'understanding': return <Brain className="h-4 w-4" />; case 'understanding': return <Brain className="h-3.5 w-3.5" />;
case 'retrieving': return <FileSearch className="h-4 w-4 animate-spin" />; case 'retrieving': return <FileSearch className="h-3.5 w-3.5 animate-spin" />;
case 'retrieved': return <CheckCircle2 className="h-4 w-4 text-green-500" />; case 'retrieved': return <CheckCircle2 className="h-3.5 w-3.5 text-green-500" />;
case 'generating': return <Sparkles className="h-4 w-4 animate-pulse" />; case 'generating': return <Sparkles className="h-3.5 w-3.5 animate-pulse" />;
default: return <Loader2 className="h-4 w-4" />; default: return <Loader2 className="h-3.5 w-3.5" />;
} }
}; };
return ( return (
<div className="mb-3 space-y-2 text-sm text-muted-foreground bg-muted/50 rounded-lg p-3"> <div className="mb-2.5 space-y-1.5 text-xs text-muted-foreground bg-muted/30 rounded-lg p-2.5">
{thinking.map((step, index) => ( {thinking.map((step, index) => (
<div key={index} className="flex items-center gap-2"> <div key={index} className="flex items-center gap-2">
{getStageIcon(step.stage)} {getStageIcon(step.stage)}
<span>{step.message}</span> <span>{step.message}</span>
{step.time && ( {step.time && <span className="opacity-60">({step.time}s)</span>}
<span className="text-xs">({step.time}s)</span>
)}
</div> </div>
))} ))}
</div> </div>
); );
}; };
export default function MessageItem({ message }: MessageItemProps) { export default function MessageItem({ message, selectedModel }: MessageItemProps) {
const isUser = message.role === "user"; const isUser = message.role === "user";
const isAssistant = message.role === "assistant"; const isAssistant = message.role === "assistant";
const { editMessage, regenerateMessage, feedbackMessage, isStreaming } = useChatStore(); const { editMessage, regenerateMessage, feedbackMessage } = useChatStore();
const [isEditing, setIsEditing] = useState(false); const [isEditing, setIsEditing] = useState(false);
const [editContent, setEditContent] = useState(message.content); const [editContent, setEditContent] = useState(message.content);
// 添加调试信息
console.log(`[DEBUG-MESSAGE] 渲染消息 ${message.id}:`, {
role: message.role,
contentLength: message.content.length,
contentPreview: message.content.substring(0, 50) + "..."
});
const handleCopy = async () => { const handleCopy = async () => {
try { try {
await navigator.clipboard.writeText(message.content); await navigator.clipboard.writeText(message.content);
toast.success("已复制到剪贴板"); toast.success("已复制");
} catch (error) { } catch {
toast.error("复制失败"); toast.error("复制失败");
} }
}; };
const handleEdit = () => {
setIsEditing(true);
};
const handleSaveEdit = async () => { const handleSaveEdit = async () => {
try { try {
await editMessage(message.id, editContent); await editMessage(message.id, editContent);
setIsEditing(false); setIsEditing(false);
toast.success("消息已更新"); } catch {
} catch (error) {
toast.error("编辑失败"); toast.error("编辑失败");
} }
}; };
const handleCancelEdit = () => {
setEditContent(message.content);
setIsEditing(false);
};
const handleRegenerate = async () => { const handleRegenerate = async () => {
try { try {
await regenerateMessage(message.id); await regenerateMessage(message.id, selectedModel);
toast.success("正在重新生成回复"); } catch {
} catch (error) {
toast.error("重新生成失败"); toast.error("重新生成失败");
} }
}; };
@@ -107,50 +82,62 @@ export default function MessageItem({ message }: MessageItemProps) {
const handleFeedback = async (feedback: "like" | "dislike") => { const handleFeedback = async (feedback: "like" | "dislike") => {
try { try {
await feedbackMessage(message.id, feedback); await feedbackMessage(message.id, feedback);
toast.success("感谢您的反馈"); } catch {
} catch (error) { toast.error("反馈失败");
toast.error("反馈提交失败");
} }
}; };
return ( return (
<div className={`flex gap-3 p-4 ${isAssistant ? 'justify-start' : 'justify-end'}`}> <div className={cn(
{isAssistant && ( "flex gap-3 py-4 group",
<div className="flex-shrink-0"> isUser ? "flex-row-reverse" : "flex-row"
<div className="w-8 h-8 bg-gradient-to-r from-blue-600 to-blue-800 rounded-lg flex items-center justify-center shadow-sm"> )}>
<Bot className="w-4 h-4 text-white" /> {/* Avatar */}
</div> <div className="flex-shrink-0 pt-0.5">
</div> <div className={cn(
"w-7 h-7 rounded-full flex items-center justify-center",
isAssistant ? "bg-primary/10" : "bg-muted"
)}>
{isAssistant ? (
<Bot className="w-3.5 h-3.5 text-primary" />
) : (
<User className="w-3.5 h-3.5 text-muted-foreground" />
)} )}
</div>
</div>
<div className={`flex-1 max-w-[80%] ${isAssistant ? '' : 'flex justify-end'}`}> {/* Content */}
<div className={`mit-card ${isAssistant ? 'bg-card' : 'bg-primary text-primary-foreground'}`}> <div className={cn("min-w-0 max-w-[75%]", isUser && "flex flex-col items-end")}>
{isEditing ? ( {isEditing ? (
<div className="space-y-2"> <div className="space-y-2 w-full">
<textarea <textarea
value={editContent} value={editContent}
onChange={(e) => setEditContent(e.target.value)} onChange={(e) => setEditContent(e.target.value)}
className="w-full p-2 border rounded resize-none" className="w-full p-2.5 border rounded-lg resize-none text-sm bg-card"
rows={3} rows={3}
/> />
<div className="flex gap-2"> <div className="flex gap-2 justify-end">
<Button size="sm" onClick={handleSaveEdit}> <Button size="sm" variant="ghost" onClick={() => { setEditContent(message.content); setIsEditing(false); }}></Button>
<Button size="sm" onClick={handleSaveEdit}></Button>
</Button>
<Button size="sm" variant="outline" onClick={handleCancelEdit}>
</Button>
</div> </div>
</div> </div>
) : ( ) : (
<> <>
{/* 思考过程组件 */} {/* Bubble */}
<div className={cn(
"px-3.5 py-2.5 text-sm leading-relaxed",
isUser
? "bg-primary text-primary-foreground rounded-2xl rounded-tr-sm"
: "rounded-2xl rounded-tl-sm"
)}>
{isAssistant && message.thinking && ( {isAssistant && message.thinking && (
<ThinkingProcess thinking={message.thinking} /> <ThinkingProcess thinking={message.thinking} />
)} )}
{/* 消息内容 */} <div className={cn(
<div className="prose prose-sm max-w-none dark:prose-invert"> "prose prose-sm max-w-none",
isUser ? "prose-invert" : "dark:prose-invert"
)}>
<ReactMarkdown <ReactMarkdown
remarkPlugins={[remarkGfm]} remarkPlugins={[remarkGfm]}
components={{ components={{
@@ -161,33 +148,25 @@ export default function MessageItem({ message }: MessageItemProps) {
style={tomorrow} style={tomorrow}
language={match[1]} language={match[1]}
PreTag="div" PreTag="div"
className="rounded-md" className="rounded-md !text-xs"
{...props} {...props}
> >
{String(children).replace(/\n$/, "")} {String(children).replace(/\n$/, "")}
</SyntaxHighlighter> </SyntaxHighlighter>
) : ( ) : (
<code className={className} {...props}> <code className={cn("px-1 py-0.5 rounded text-xs", isUser ? "bg-primary-foreground/20" : "bg-muted")} {...props}>{children}</code>
{children}
</code>
); );
}, },
table: ({ children }) => ( table: ({ children }) => (
<div className="overflow-x-auto"> <div className="overflow-x-auto -mx-1">
<table className="min-w-full border-collapse border border-border"> <table className="min-w-full border-collapse border border-border text-xs">{children}</table>
{children}
</table>
</div> </div>
), ),
th: ({ children }) => ( th: ({ children }) => (
<th className="border border-border bg-muted px-3 py-2 text-left font-medium"> <th className="border border-border bg-muted px-2 py-1.5 text-left font-medium">{children}</th>
{children}
</th>
), ),
td: ({ children }) => ( td: ({ children }) => (
<td className="border border-border px-3 py-2"> <td className="border border-border px-2 py-1.5">{children}</td>
{children}
</td>
), ),
}} }}
> >
@@ -195,104 +174,49 @@ export default function MessageItem({ message }: MessageItemProps) {
</ReactMarkdown> </ReactMarkdown>
</div> </div>
{/* 知识来源 */}
{isAssistant && message.metadata?.sources && message.metadata.sources.length > 0 && ( {isAssistant && message.metadata?.sources && message.metadata.sources.length > 0 && (
<div className="mt-4 pt-4 border-t border-border"> <div className="mt-2.5 pt-2.5 border-t border-border/30">
<SourceReferences sources={message.metadata.sources} maxSources={5} /> <SourceReferences sources={message.metadata.sources} maxSources={3} />
</div> </div>
)} )}
</>
)}
</div> </div>
{/* 操作按钮 */} {/* Actions */}
<div className={`flex items-center gap-1 mt-2 opacity-0 group-hover:opacity-100 transition-opacity ${ <div className={cn(
isAssistant ? "flex-row" : "flex-row-reverse" "flex items-center gap-0.5 mt-1",
}`}> isUser ? "flex-row-reverse" : "flex-row"
{/* 复制按钮 */} )}>
<Button <Button size="sm" variant="ghost" onClick={handleCopy} className="h-6 w-6 p-0 text-muted-foreground hover:text-foreground">
size="sm"
variant="ghost"
onClick={handleCopy}
className="h-7 w-7 p-0 hover:bg-muted/50"
>
<Copy className="h-3 w-3" /> <Copy className="h-3 w-3" />
</Button> </Button>
{/* 用户消息操作 */}
{isUser && ( {isUser && (
<Button <Button size="sm" variant="ghost" onClick={() => setIsEditing(true)} className="h-6 w-6 p-0 text-muted-foreground hover:text-foreground">
size="sm"
variant="ghost"
onClick={handleEdit}
className="h-7 w-7 p-0 hover:bg-muted/50"
>
<Edit className="h-3 w-3" /> <Edit className="h-3 w-3" />
</Button> </Button>
)} )}
{/* 助手消息操作 */}
{isAssistant && ( {isAssistant && (
<> <>
<Button <Button size="sm" variant="ghost" onClick={handleRegenerate} className="h-6 w-6 p-0 text-muted-foreground hover:text-foreground">
size="sm"
variant="ghost"
onClick={handleRegenerate}
className="h-7 w-7 p-0 hover:bg-muted/50"
>
<RotateCcw className="h-3 w-3" /> <RotateCcw className="h-3 w-3" />
</Button> </Button>
<Button size="sm" variant="ghost" onClick={() => handleFeedback("like")} className={cn("h-6 w-6 p-0", message.feedback === "like" ? "text-green-600" : "text-muted-foreground hover:text-foreground")}>
{/* 反馈按钮 */}
<div className="flex gap-1">
<Button
size="sm"
variant="ghost"
onClick={() => handleFeedback("like")}
className={`h-7 w-7 p-0 hover:bg-muted/50 ${
message.feedback === "like" ? "text-green-600" : ""
}`}
>
<ThumbsUp className="h-3 w-3" /> <ThumbsUp className="h-3 w-3" />
</Button> </Button>
<Button <Button size="sm" variant="ghost" onClick={() => handleFeedback("dislike")} className={cn("h-6 w-6 p-0", message.feedback === "dislike" ? "text-red-500" : "text-muted-foreground hover:text-foreground")}>
size="sm"
variant="ghost"
onClick={() => handleFeedback("dislike")}
className={`h-7 w-7 p-0 hover:bg-muted/50 ${
message.feedback === "dislike" ? "text-red-600" : ""
}`}
>
<ThumbsDown className="h-3 w-3" /> <ThumbsDown className="h-3 w-3" />
</Button> </Button>
</>
)}
<span className="text-[10px] text-muted-foreground/50 mx-1">
{message.created_at ? formatDistanceToNow(
new Date(new Date(message.created_at).getTime() + 8 * 60 * 60 * 1000),
{ addSuffix: true, locale: zhCN }
) : ""}
</span>
</div> </div>
</> </>
)} )}
</div> </div>
{/* 时间戳和编辑标记 */}
<div className={`text-xs text-muted-foreground mt-1 ${
isUser ? "text-right" : "text-left"
}`}>
{message.created_at ? formatDistanceToNow(
new Date(new Date(message.created_at).getTime() + 8 * 60 * 60 * 1000),
{
addSuffix: true,
locale: zhCN
}) : '未知时间'}
{message.edited && (
<span className="ml-1 text-muted-foreground">()</span>
)}
</div>
</div>
{!isAssistant && (
<div className="flex-shrink-0">
<div className="w-8 h-8 bg-gradient-to-r from-gray-600 to-gray-800 rounded-lg flex items-center justify-center shadow-sm">
<User className="w-4 h-4 text-white" />
</div>
</div>
)}
</div> </div>
); );
} }
+4 -3
View File
@@ -5,13 +5,14 @@ import MessageItem from "./message-item";
interface MessageListProps { interface MessageListProps {
messages: ChatMessage[]; messages: ChatMessage[];
selectedModel?: string;
} }
export default function MessageList({ messages }: MessageListProps) { export default function MessageList({ messages, selectedModel }: MessageListProps) {
return ( return (
<div className="space-y-4 p-4"> <div className="max-w-3xl mx-auto px-4">
{messages.map((message) => ( {messages.map((message) => (
<MessageItem key={message.id} message={message} /> <MessageItem key={message.id} message={message} selectedModel={selectedModel} />
))} ))}
</div> </div>
); );
+7 -23
View File
@@ -9,39 +9,23 @@ interface QuickQuestionsProps {
const quickQuestions = [ const quickQuestions = [
"什么是国土空间规划?", "什么是国土空间规划?",
"国土空间规划的主要原则是什么?", "国土空间规划的三区三线是什么?",
"如何进行国土空间规划编制?",
"国土空间规划与城市规划的区别?", "国土空间规划与城市规划的区别?",
"国土空间规划中的三区三线是什么", "如何进行国土空间规划编制",
"如何评价国土空间规划的合理性?",
]; ];
export default function QuickQuestions({ onSelect }: QuickQuestionsProps) { export default function QuickQuestions({ onSelect }: QuickQuestionsProps) {
return ( return (
<div className="space-y-4"> <div className="grid grid-cols-1 sm:grid-cols-2 gap-2 max-w-lg mx-auto">
<div className="text-sm text-muted-foreground text-center">
</div>
<div className="grid grid-cols-1 sm:grid-cols-2 gap-3 max-w-4xl mx-auto">
{quickQuestions.map((question, index) => ( {quickQuestions.map((question, index) => (
<Button <button
key={index} key={index}
variant="outline"
size="sm"
className="justify-start text-left h-auto p-4 hover:bg-muted/50 transition-colors"
onClick={() => onSelect(question)} onClick={() => onSelect(question)}
className="text-left px-3.5 py-2.5 rounded-xl border border-border/40 bg-card/50 hover:bg-muted/50 hover:border-border/70 transition-colors text-sm text-foreground/80 leading-relaxed"
> >
<MessageSquare className="w-4 h-4 mr-3 flex-shrink-0" /> {question}
<span className="text-sm leading-relaxed">{question}</span> </button>
</Button>
))} ))}
</div> </div>
</div>
); );
} }
+136 -219
View File
@@ -1,8 +1,6 @@
"use client"; "use client";
import { useState } from "react"; import { useState } from "react";
import { useRouter } from "next/navigation";
import { useAuthStore } from "@/store/auth";
import { useChatStore } from "@/store/chat"; import { useChatStore } from "@/store/chat";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
import { Input } from "@/components/ui/input"; import { Input } from "@/components/ui/input";
@@ -10,25 +8,11 @@ import { ScrollArea } from "@/components/ui/scroll-area";
import { import {
Plus, Plus,
MessageSquare, MessageSquare,
Menu,
X,
Edit2, Edit2,
Trash2, Trash2,
Download, Download,
MoreVertical,
Search, Search,
Calendar,
Clock,
ChevronRight
} from "lucide-react"; } from "lucide-react";
import { formatDistanceToNow } from "date-fns";
import { zhCN } from "date-fns/locale";
import {
DropdownMenu,
DropdownMenuContent,
DropdownMenuItem,
DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu";
import { import {
Dialog, Dialog,
DialogContent, DialogContent,
@@ -40,27 +24,48 @@ import {
import { toast } from "sonner"; import { toast } from "sonner";
import ExportDialog from "./export-dialog"; import ExportDialog from "./export-dialog";
export default function Sidebar() { interface SidebarProps {
const [isMobileMenuOpen, setIsMobileMenuOpen] = useState(false); onClose?: () => void;
}
function groupSessionsByDate(sessions: any[]) {
const now = new Date();
const today = new Date(now.getFullYear(), now.getMonth(), now.getDate());
const yesterday = new Date(today.getTime() - 86400000);
const lastWeek = new Date(today.getTime() - 7 * 86400000);
const lastMonth = new Date(today.getTime() - 30 * 86400000);
const groups: { label: string; sessions: any[] }[] = [
{ label: "今天", sessions: [] },
{ label: "昨天", sessions: [] },
{ label: "过去 7 天", sessions: [] },
{ label: "过去 30 天", sessions: [] },
{ label: "更早", sessions: [] },
];
sessions.forEach(session => {
const date = new Date(session.created_at);
if (date >= today) groups[0].sessions.push(session);
else if (date >= yesterday) groups[1].sessions.push(session);
else if (date >= lastWeek) groups[2].sessions.push(session);
else if (date >= lastMonth) groups[3].sessions.push(session);
else groups[4].sessions.push(session);
});
return groups.filter(g => g.sessions.length > 0);
}
export default function Sidebar({ onClose }: SidebarProps) {
const [editingSession, setEditingSession] = useState<number | null>(null); const [editingSession, setEditingSession] = useState<number | null>(null);
const [editTitle, setEditTitle] = useState(""); const [editTitle, setEditTitle] = useState("");
const [deleteSessionId, setDeleteSessionId] = useState<number | null>(null); const [deleteSessionId, setDeleteSessionId] = useState<number | null>(null);
const [searchQuery, setSearchQuery] = useState(""); const [searchQuery, setSearchQuery] = useState("");
const [openDropdownId, setOpenDropdownId] = useState<number | null>(null);
const router = useRouter();
const { sessions, currentSession, selectSession, createSession, renameSession, deleteSession } = useChatStore(); const { sessions, currentSession, selectSession, createSession, renameSession, deleteSession } = useChatStore();
// 过滤会话(按创建时间排序)
const filteredSessions = () => { const filteredSessions = () => {
if (!searchQuery) { const sorted = [...sessions].sort((a, b) => new Date(b.created_at).getTime() - new Date(a.created_at).getTime());
return sessions.sort((a, b) => new Date(b.created_at).getTime() - new Date(a.created_at).getTime()); if (!searchQuery) return sorted;
} return sorted.filter(s => s.title.toLowerCase().includes(searchQuery.toLowerCase()));
return sessions
.filter(session =>
session.title.toLowerCase().includes(searchQuery.toLowerCase())
)
.sort((a, b) => new Date(b.created_at).getTime() - new Date(a.created_at).getTime());
}; };
const handleNewChat = async () => { const handleNewChat = async () => {
@@ -68,16 +73,12 @@ export default function Sidebar() {
if (newSession) { if (newSession) {
selectSession(newSession.id); selectSession(newSession.id);
} }
onClose?.();
}; };
const handleSelectSession = async (sessionId: number) => { const handleSelectSession = async (sessionId: number) => {
await selectSession(sessionId); await selectSession(sessionId);
setIsMobileMenuOpen(false); onClose?.();
};
const handleRenameSession = (sessionId: number, currentTitle: string) => {
setEditingSession(sessionId);
setEditTitle(currentTitle);
}; };
const handleSaveRename = async () => { const handleSaveRename = async () => {
@@ -86,246 +87,162 @@ export default function Sidebar() {
await renameSession(editingSession, editTitle.trim()); await renameSession(editingSession, editTitle.trim());
setEditingSession(null); setEditingSession(null);
setEditTitle(""); setEditTitle("");
toast.success("会话重命名成功"); toast.success("重命名成功");
} catch (error) { } catch {
toast.error("重命名失败"); toast.error("重命名失败");
} }
} }
}; };
const handleCancelRename = () => {
setEditingSession(null);
setEditTitle("");
};
const handleDeleteSession = (sessionId: number) => {
setDeleteSessionId(sessionId);
};
const handleConfirmDelete = async () => { const handleConfirmDelete = async () => {
if (deleteSessionId) { if (deleteSessionId) {
try { try {
await deleteSession(deleteSessionId); await deleteSession(deleteSessionId);
setDeleteSessionId(null); setDeleteSessionId(null);
toast.success("会话已删除"); toast.success("已删除");
} catch (error) { } catch {
toast.error("删除失败"); toast.error("删除失败");
} }
} }
}; };
const handleCancelDelete = () => { const groups = groupSessionsByDate(filteredSessions());
setDeleteSessionId(null);
};
const renderSessionItem = (session: any) => {
return (
<div
key={session.id}
className={`group relative flex items-center gap-3 px-3 py-2 rounded-lg cursor-pointer transition-all hover:bg-muted/50 ${
currentSession?.id === session.id ? 'bg-muted border border-border' : ''
}`}
onClick={() => handleSelectSession(session.id)}
>
<MessageSquare className="w-4 h-4 text-muted-foreground flex-shrink-0" />
<div className="flex-1 min-w-0">
<div className="text-sm font-medium text-foreground truncate">
{session.title}
</div>
<div className="text-xs text-muted-foreground">
{session.created_at ? (() => {
const rawDate = new Date(session.created_at);
const date = new Date(rawDate.getTime() + 8 * 60 * 60 * 1000);
const now = new Date();
const today = new Date(now.getFullYear(), now.getMonth(), now.getDate());
if (date >= today) {
// 今天:只显示时间
return date.toLocaleTimeString('zh-CN', { hour: '2-digit', minute: '2-digit', timeZone: 'Asia/Shanghai' });
} else {
// 昨天及更早:显示日期+时间
return date.toLocaleString('zh-CN', {
month: '2-digit',
day: '2-digit',
hour: '2-digit',
minute: '2-digit',
timeZone: 'Asia/Shanghai'
});
}
})() : '未知时间'}
</div>
</div>
{/* 操作按钮 */}
<DropdownMenu
open={openDropdownId === session.id}
onOpenChange={(open) => setOpenDropdownId(open ? session.id : null)}
>
<DropdownMenuTrigger asChild>
<Button
variant="ghost"
size="sm"
className="opacity-0 group-hover:opacity-100 transition-opacity h-6 w-6 p-0"
onClick={(e) => e.stopPropagation()}
>
<MoreVertical className="w-3 h-3" />
</Button>
</DropdownMenuTrigger>
<DropdownMenuContent align="end" onClick={(e) => e.stopPropagation()}>
<DropdownMenuItem
onSelect={() => {
handleRenameSession(session.id, session.title);
}}
>
<Edit2 className="w-4 h-4 mr-2" />
</DropdownMenuItem>
<ExportDialog
sessionId={session.id}
sessionTitle={session.title}
onClose={() => setOpenDropdownId(null)}
>
<DropdownMenuItem onSelect={(e) => e.preventDefault()}>
<Download className="w-4 h-4 mr-2" />
</DropdownMenuItem>
</ExportDialog>
<DropdownMenuItem
onSelect={() => {
handleDeleteSession(session.id);
}}
className="text-destructive"
>
<Trash2 className="w-4 h-4 mr-2" />
</DropdownMenuItem>
</DropdownMenuContent>
</DropdownMenu>
</div>
);
};
return ( return (
<> <>
{/* 移动端菜单按钮 */}
<div className="lg:hidden fixed top-4 left-4 z-50">
<Button
variant="outline"
size="icon"
onClick={() => setIsMobileMenuOpen(!isMobileMenuOpen)}
>
{isMobileMenuOpen ? <X className="w-4 h-4" /> : <Menu className="w-4 h-4" />}
</Button>
</div>
{/* 侧边栏 */}
<div className={`
fixed lg:static inset-y-0 left-0 z-40 w-80 max-w-[85vw] bg-card/50 backdrop-blur-sm border-r border-border transform transition-transform duration-300 ease-in-out mobile-safe-area lg:h-full
${isMobileMenuOpen ? 'translate-x-0' : '-translate-x-full lg:translate-x-0'}
`}>
<div className="flex flex-col h-full"> <div className="flex flex-col h-full">
{/* 头部 */} {/* Header */}
<div className="p-4 border-b"> <div className="px-3 pt-4 pb-3 space-y-3 border-b border-border/30">
<div className="flex items-center justify-between mb-4"> <div className="flex items-center justify-between px-1">
<h2 className="text-lg font-semibold text-foreground"></h2> <span className="text-sm font-semibold text-foreground"></span>
<Button <Button onClick={handleNewChat} size="sm" variant="ghost" className="h-7 w-7 p-0 hover:bg-primary/10 hover:text-primary">
onClick={handleNewChat} <Plus className="w-4 h-4" />
size="sm"
className="h-8 px-3"
>
<Plus className="w-4 h-4 mr-1" />
</Button> </Button>
</div> </div>
{/* 搜索框 */}
<div className="relative"> <div className="relative">
<Search className="absolute left-3 top-1/2 transform -translate-y-1/2 w-4 h-4 text-muted-foreground" /> <Search className="absolute left-2.5 top-1/2 -translate-y-1/2 w-3.5 h-3.5 text-muted-foreground" />
<Input <Input
placeholder="搜索对话..." placeholder="搜索"
value={searchQuery} value={searchQuery}
onChange={(e) => setSearchQuery(e.target.value)} onChange={(e) => setSearchQuery(e.target.value)}
className="pl-9 h-9" className="pl-8 h-8 text-sm bg-muted/30 border border-border/20 focus-visible:ring-1 focus-visible:border-primary/30"
/> />
</div> </div>
</div> </div>
{/* 会话列表 */} {/* Session list with date groups */}
<ScrollArea className="flex-1 px-2"> <ScrollArea className="flex-1 px-2">
<div className="py-4"> <div className="pb-4">
<div className="space-y-1"> {groups.map(group => (
{filteredSessions().map(renderSessionItem)} <div key={group.label} className="mb-3">
<div className="px-2 py-1.5 text-xs font-medium text-muted-foreground/70 uppercase tracking-wider">
{group.label}
</div>
<div className="space-y-0.5">
{group.sessions.map(session => (
<div
key={session.id}
className="relative rounded-lg cursor-pointer transition-colors hover:bg-muted/50"
style={{
backgroundColor: currentSession?.id === session.id ? 'var(--muted)' : undefined,
color: currentSession?.id === session.id ? 'var(--foreground)' : undefined
}}
onClick={() => handleSelectSession(session.id)}
>
{/* Title row */}
<div className="flex items-center gap-2 px-2.5 py-2 pr-28">
<MessageSquare className="w-3.5 h-3.5 text-muted-foreground flex-shrink-0" />
<span className="text-sm truncate">{session.title}</span>
</div> </div>
{/* Action buttons — absolute positioned, always visible */}
<div
className="absolute right-1.5 top-1/2 -translate-y-1/2 flex items-center gap-1"
onClick={(e) => e.stopPropagation()}
>
<button
className="inline-flex items-center justify-center w-7 h-7 rounded-md text-gray-500 hover:text-gray-800 hover:bg-gray-200 transition-colors"
title="重命名"
onClick={() => { setEditingSession(session.id); setEditTitle(session.title); }}
>
<Edit2 className="w-3.5 h-3.5" />
</button>
<ExportDialog sessionId={session.id} sessionTitle={session.title} onClose={() => {}}>
<button
className="inline-flex items-center justify-center w-7 h-7 rounded-md text-gray-500 hover:text-gray-800 hover:bg-gray-200 transition-colors"
title="导出"
>
<Download className="w-3.5 h-3.5" />
</button>
</ExportDialog>
<button
className="inline-flex items-center justify-center w-7 h-7 rounded-md text-gray-500 hover:text-red-600 hover:bg-red-50 transition-colors"
title="删除"
onClick={() => setDeleteSessionId(session.id)}
>
<Trash2 className="w-3.5 h-3.5" />
</button>
</div>
</div>
))}
</div>
</div>
))}
{sessions.length === 0 && ( {sessions.length === 0 && (
<div className="text-center py-8"> <div className="text-center py-12">
<MessageSquare className="w-12 h-12 text-muted-foreground mx-auto mb-3" /> <MessageSquare className="w-8 h-8 text-muted-foreground/40 mx-auto mb-2" />
<p className="text-sm text-muted-foreground mb-2"></p> <p className="text-xs text-muted-foreground"></p>
<p className="text-xs text-muted-foreground"></p>
</div> </div>
)} )}
</div> </div>
</ScrollArea> </ScrollArea>
{/* Footer — new chat button with distinct background */}
<div className="px-3 py-3 border-t border-border/30 bg-muted/20">
<Button
onClick={handleNewChat}
variant="outline"
className="w-full h-9 text-sm justify-center gap-2 border-dashed border-border/50 hover:bg-primary/5 hover:text-primary hover:border-primary/30"
>
<Plus className="w-4 h-4" />
</Button>
</div> </div>
</div> </div>
{/* 移动端遮罩 */} {/* Rename dialog */}
{isMobileMenuOpen && ( <Dialog open={!!editingSession} onOpenChange={() => { setEditingSession(null); setEditTitle(""); }}>
<div <DialogContent className="sm:max-w-sm">
className="lg:hidden fixed inset-0 bg-black bg-opacity-50 z-30"
onClick={() => setIsMobileMenuOpen(false)}
/>
)}
{/* 重命名对话框 */}
<Dialog open={!!editingSession} onOpenChange={handleCancelRename}>
<DialogContent>
<DialogHeader> <DialogHeader>
<DialogTitle></DialogTitle> <DialogTitle></DialogTitle>
<DialogDescription>
</DialogDescription>
</DialogHeader> </DialogHeader>
<div className="py-4"> <div className="py-2">
<Input <Input
value={editTitle} value={editTitle}
onChange={(e) => setEditTitle(e.target.value)} onChange={(e) => setEditTitle(e.target.value)}
placeholder="输入新的会话名称" placeholder="话名称"
onKeyPress={(e) => { onKeyDown={(e) => { if (e.key === 'Enter' && editTitle.trim()) handleSaveRename(); }}
if (e.key === 'Enter' && editTitle.trim()) {
handleSaveRename();
}
}}
autoFocus autoFocus
/> />
</div> </div>
<DialogFooter> <DialogFooter>
<Button variant="outline" onClick={handleCancelRename}> <Button variant="outline" onClick={() => { setEditingSession(null); setEditTitle(""); }}></Button>
<Button onClick={handleSaveRename} disabled={!editTitle.trim()}></Button>
</Button>
<Button onClick={handleSaveRename} disabled={!editTitle.trim()}>
</Button>
</DialogFooter> </DialogFooter>
</DialogContent> </DialogContent>
</Dialog> </Dialog>
{/* 删除确认对话框 */} {/* Delete dialog */}
<Dialog open={!!deleteSessionId} onOpenChange={() => setDeleteSessionId(null)}> <Dialog open={!!deleteSessionId} onOpenChange={() => setDeleteSessionId(null)}>
<DialogContent> <DialogContent className="sm:max-w-sm">
<DialogHeader> <DialogHeader>
<DialogTitle></DialogTitle> <DialogTitle></DialogTitle>
<DialogDescription> <DialogDescription></DialogDescription>
</DialogDescription>
</DialogHeader> </DialogHeader>
<DialogFooter> <DialogFooter>
<Button variant="outline" onClick={handleCancelDelete}> <Button variant="outline" onClick={() => setDeleteSessionId(null)}></Button>
<Button variant="destructive" onClick={handleConfirmDelete}></Button>
</Button>
<Button variant="destructive" onClick={handleConfirmDelete}>
</Button>
</DialogFooter> </DialogFooter>
</DialogContent> </DialogContent>
</Dialog> </Dialog>
+5 -18
View File
@@ -40,29 +40,16 @@ async function apiRequest<T>(
}; };
try { try {
console.log(`🌐 API请求: ${endpoint}`, { url, headers: defaultHeaders });
const response = await fetch(url, config); const response = await fetch(url, config);
console.log(`📡 API响应: ${endpoint}`, { status: response.status, ok: response.ok });
if (!response.ok) { if (!response.ok) {
// 401错误:token无效或过期,跳转到登录页
if (response.status === 401) { if (response.status === 401) {
console.warn("🚨 收到401错误,token可能无效或过期");
console.log("🔍 当前路径:", window.location.pathname);
console.log("🔍 当前token:", localStorage.getItem("auth_token") ? "存在" : "不存在");
// 清除本地存储的token
localStorage.removeItem("auth_token"); localStorage.removeItem("auth_token");
// 只在非登录/注册页面才跳转,避免死循环
if (typeof window !== 'undefined' && if (typeof window !== 'undefined' &&
!window.location.pathname.includes('/login') && !window.location.pathname.includes('/login') &&
!window.location.pathname.includes('/register')) { !window.location.pathname.includes('/register')) {
console.warn("🔄 Token无效或已过期,跳转到登录页");
// 使用replace而不是href,避免在历史记录中留下当前页面
window.location.replace("/login"); window.location.replace("/login");
} }
throw new Error("认证已过期,请重新登录"); throw new Error("认证已过期,请重新登录");
} }
@@ -72,7 +59,6 @@ async function apiRequest<T>(
return await response.json(); return await response.json();
} catch (error) { } catch (error) {
console.error(`API请求失败 ${endpoint}:`, error);
throw error; throw error;
} }
} }
@@ -443,7 +429,7 @@ export const chatAPI = {
knowledgeBaseIds?: string[], knowledgeBaseIds?: string[],
signal?: AbortSignal, signal?: AbortSignal,
onChunk?: (chunk: string) => void, onChunk?: (chunk: string) => void,
onComplete?: (sessionId: number) => void, onComplete?: (sessionId: number, messageId?: number, userMessageId?: number) => void,
onError?: (error: string) => void, onError?: (error: string) => void,
onStatus?: (status: string) => void, onStatus?: (status: string) => void,
onThinking?: (step: { step: string; message: string; details?: Record<string, any>; timestamp?: string }) => void, onThinking?: (step: { step: string; message: string; details?: Record<string, any>; timestamp?: string }) => void,
@@ -598,8 +584,8 @@ export const chatAPI = {
console.log("[DEBUG-STREAM] 接收chunk:", data.content); console.log("[DEBUG-STREAM] 接收chunk:", data.content);
onChunk?.(data.content); onChunk?.(data.content);
} else if (data.type === "done") { } else if (data.type === "done") {
console.log("[DEBUG-STREAM] 流式完成, session_id:", data.session_id); console.log("[DEBUG-STREAM] 流式完成, session_id:", data.session_id, "message_id:", data.message_id, "user_message_id:", data.user_message_id);
onComplete?.(data.session_id); onComplete?.(data.session_id, data.message_id, data.user_message_id);
} else if (data.error) { } else if (data.error) {
console.log("[DEBUG-STREAM] 流式错误:", data.error); console.log("[DEBUG-STREAM] 流式错误:", data.error);
onError?.(data.error); onError?.(data.error);
@@ -685,13 +671,14 @@ export const chatAPI = {
}); });
}, },
async regenerateMessage(messageId: number) { async regenerateMessage(messageId: number, model?: string) {
return apiRequest<{ return apiRequest<{
message: string; message: string;
new_message_id: number; new_message_id: number;
content: string; content: string;
}>(`/chat/messages/${messageId}/regenerate`, { }>(`/chat/messages/${messageId}/regenerate`, {
method: "POST", method: "POST",
body: JSON.stringify({ model: model || null }),
}); });
}, },
+23 -19
View File
@@ -28,7 +28,7 @@ interface ChatActions {
addMessage: (message: ChatMessage) => void; addMessage: (message: ChatMessage) => void;
updateMessage: (messageId: number, content: string) => void; updateMessage: (messageId: number, content: string) => void;
editMessage: (messageId: number, newContent: string) => Promise<void>; editMessage: (messageId: number, newContent: string) => Promise<void>;
regenerateMessage: (messageId: number) => Promise<void>; regenerateMessage: (messageId: number, model?: string) => Promise<void>;
feedbackMessage: (messageId: number, feedback: "like" | "dislike") => Promise<void>; feedbackMessage: (messageId: number, feedback: "like" | "dislike") => Promise<void>;
stopGeneration: () => void; stopGeneration: () => void;
@@ -228,12 +228,6 @@ export const useChatStore = create<ChatStore>((set, get) => ({
// 创建新的 AbortController // 创建新的 AbortController
const abortController = new AbortController(); const abortController = new AbortController();
console.log("[DEBUG-CHAT] 开始流式发送:", {
message: message,
sessionId: currentSession.id,
knowledgeBaseIds: knowledgeBaseIds
});
set({ isStreaming: true, error: null, abortController }); set({ isStreaming: true, error: null, abortController });
// 添加用户消息 // 添加用户消息
@@ -299,7 +293,6 @@ export const useChatStore = create<ChatStore>((set, get) => ({
// 处理内容chunk // 处理内容chunk
chunkCount++; chunkCount++;
totalChars += data.content.length; totalChars += data.content.length;
console.log(`[DEBUG-STREAM] 接收并显示chunk ${chunkCount}:`, data.content);
requestAnimationFrame(() => { requestAnimationFrame(() => {
set((state) => ({ set((state) => ({
@@ -315,7 +308,6 @@ export const useChatStore = create<ChatStore>((set, get) => ({
// 向后兼容:纯文本chunk // 向后兼容:纯文本chunk
chunkCount++; chunkCount++;
totalChars += chunk.length; totalChars += chunk.length;
console.log(`[DEBUG-STREAM] 接收并显示chunk ${chunkCount}:`, chunk);
requestAnimationFrame(() => { requestAnimationFrame(() => {
set((state) => ({ set((state) => ({
@@ -328,17 +320,30 @@ export const useChatStore = create<ChatStore>((set, get) => ({
}); });
} }
}, },
(sessionId: number) => { (sessionId: number, messageId?: number, userMessageId?: number) => {
// 流式完成,直接处理 if (messageId) {
console.log(`[DEBUG-STREAM] 流式完成 - 总chunk数: ${chunkCount}, 总字符数: ${totalChars}`); set((state) => ({
messages: state.messages.map(msg => {
if (msg.id === assistantMessage.id) return { ...msg, id: messageId };
if (userMessageId && msg.id === userMessage.id) return { ...msg, id: userMessageId };
return msg;
}),
isStreaming: false,
abortController: null,
}));
} else {
const { currentSession } = get();
if (currentSession) {
get().loadMessages(currentSession.id);
}
set({ isStreaming: false, abortController: null }); set({ isStreaming: false, abortController: null });
}
// 刷新会话列表以确保新会话显示在顶部 // 刷新会话列表以确保新会话显示在顶部
get().loadSessions(); get().loadSessions();
}, },
(error: string) => { (error: string) => {
// 流式错误 // 流式错误
console.log("[DEBUG-STREAM] 流式错误:", error);
set({ error, isStreaming: false, abortController: null }); set({ error, isStreaming: false, abortController: null });
}, },
undefined, // onStatus undefined, // onStatus
@@ -409,17 +414,17 @@ export const useChatStore = create<ChatStore>((set, get) => ({
} }
}, },
// 重新生成消息 // 重新生成消息(支持 assistant 消息 ID
regenerateMessage: async (messageId: number) => { regenerateMessage: async (messageId: number, model?: string) => {
try { try {
set({ isLoading: true, error: null }); set({ isLoading: true, error: null });
const response = await chatAPI.regenerateMessage(messageId); const response = await chatAPI.regenerateMessage(messageId, model);
// 删除该消息之后的所有消息 // 找到该消息,删除它及之后的所有消息
const messageIndex = get().messages.findIndex(msg => msg.id === messageId); const messageIndex = get().messages.findIndex(msg => msg.id === messageId);
if (messageIndex !== -1) { if (messageIndex !== -1) {
set((state) => ({ set((state) => ({
messages: state.messages.slice(0, messageIndex + 1), messages: state.messages.slice(0, messageIndex),
})); }));
} }
@@ -465,7 +470,6 @@ export const useChatStore = create<ChatStore>((set, get) => ({
// 如果存在 AbortController,调用 abort 中止请求 // 如果存在 AbortController,调用 abort 中止请求
if (abortController) { if (abortController) {
abortController.abort(); abortController.abort();
console.log("[DEBUG-STOP] 已中止流式请求");
} }
// 清理状态 // 清理状态