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:
+39
-12
@@ -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)
|
||||||
|
|||||||
@@ -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);
|
||||||
|
|||||||
@@ -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>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -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>
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -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
@@ -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
@@ -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
@@ -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] 已中止流式请求");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 清理状态
|
// 清理状态
|
||||||
|
|||||||
Reference in New Issue
Block a user