gxwebsoft/template-10556
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修复：AI问答模块

Browse Source
This commit is contained in:
赵忠林
2025-07-09 10:19:03 +08:00
parent ada779f880
commit 5deb2e96b5
6 changed files with 319 additions and 270 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
src/pages/ai/index.scss
View File

@@ -1,5 +1,19 @@
// ... existing code ...
html {
font-size: calc(100vw / 375 * 16); // 假设设计稿宽度为375px，设置基础字体大小为16px
@media screen and (max-width: 375px) {
font-size: 16px; // 最小字体大小
}
@media screen and (min-width: 768px) {
font-size: 24px; // 平板或大屏设备上的字体大小
}
}
.ai-chat {
height: 94vh;
height: 98vh;
display: flex;
flex-direction: column;
background-color: #f5f5f5;
@@ -9,7 +23,7 @@
color: white;
padding: 16px;
text-align: center;
font-size: 18px;
font-size: 1.2rem; // 使用 rem 单位
font-weight: bold;
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
}
@@ -71,13 +85,13 @@
}
.avatar {
width: 40px;
height: 40px;
width: 60px;
height: 60px;
border-radius: 50%;
display: flex;
align-items: center;
justify-content: center;
font-size: 18px;
font-size: 18px; // 可以保持不变或调整为 rem 单位
flex-shrink: 0;
&.user-avatar {
@@ -97,7 +111,7 @@
border-radius: 18px;
word-wrap: break-word;
line-height: 1.5;
font-size: 14px;
font-size: 1rem; // 使用 rem 单位
position: relative;
.typing-indicator {
@@ -136,16 +150,15 @@
.message-input {
width: 100%;
min-height: 40px;
min-height: 90px;
max-height: 120px;
padding: 10px 16px;
border: 1px solid #e0e0e0;
border-radius: 20px;
font-size: 14px;
line-height: 1.5;
font-size: 1.2rem; // 使用 rem 单位
line-height: 1.8;
resize: none;
outline: none;
background: #f9f9f9;
transition: all 0.3s ease;
&:focus {
@@ -197,7 +210,7 @@
border-top: 1px solid #e0e0e0;
.quick-title {
font-size: 14px;
font-size: 1rem; // 使用 rem 单位
color: #666;
margin-bottom: 12px;
display: flex;
@@ -215,7 +228,7 @@
background: #f9f9f9;
border: 1px solid #e0e0e0;
border-radius: 12px;
font-size: 14px;
font-size: 1rem; // 使用 rem 单位
color: #333;
cursor: pointer;
transition: all 0.3s ease;
@@ -246,20 +259,20 @@
text-align: center;
.empty-icon {
font-size: 64px;
font-size: 4rem; // 使用 rem 单位
margin-bottom: 16px;
opacity: 0.5;
}
.empty-title {
font-size: 18px;
font-size: 1.5rem; // 使用 rem 单位
font-weight: bold;
color: #333;
margin-bottom: 8px;
}
.empty-desc {
font-size: 14px;
font-size: 1rem; // 使用 rem 单位
color: #666;
line-height: 1.5;
}

480
src/pages/ai/index.tsx
View File

@@ -1,269 +1,273 @@
import { useEffect, useState, useRef } from "react";
import { View, Textarea } from '@tarojs/components';
// import Toast from '@tarojs/taro';
import {useEffect, useState, useRef} from "react";
import {View, Textarea} from '@tarojs/components';
import Taro from '@tarojs/taro';
import { Button } from '@nutui/nutui-react-taro';
import { User, ArrowUp } from '@nutui/icons-react-taro';
import { sendAiMessage, stopAiMessage } from '@/api/ai';
import { createWebSocket } from '@/utils/websocket';
import {Button} from '@nutui/nutui-react-taro';
import {User, ArrowUp} from '@nutui/icons-react-taro';
import {sendAiMessage, stopAiMessage} from '@/api/ai';
import {createWebSocket} from '@/utils/websocket';
import './index.scss';
import {APP_API_URL} from "@/utils/server";
import Header from "../index/Header";
import {WSS_API_URL} from "@/utils/server";
// 消息类型
interface Message {
id?: string;
type?: 'user' | 'ai';
query?: string;
timestamp?: number;
isTyping?: boolean;
user?: string;
responseMode?: string;
id?: string;
type?: 'user' | 'ai';
query?: string;
timestamp?: number;
isTyping?: boolean;
user?: string;
responseMode?: string;
}
/**
* AI问答页面
*/
const AiChat = () => {
const [messages, setMessages] = useState<Message[]>([]);
const [inputValue, setInputValue] = useState('');
const [isLoading, setIsLoading] = useState(false);
const [currentTaskId, setCurrentTaskId] = useState<string>('');
const messagesEndRef = useRef<HTMLDivElement>(null);
const wsRef = useRef<any>(null);
const [messages, setMessages] = useState<Message[]>([]);
const [inputValue, setInputValue] = useState('');
const [isLoading, setIsLoading] = useState(false);
const [currentTaskId, setCurrentTaskId] = useState<string>('');
const messagesEndRef = useRef<HTMLDivElement>(null);
const wsRef = useRef<any>(null);
// 快捷问题
const quickQuestions = [
"当兵的注意事项？",
"男兵应征报名对象：",
"当兵公务员考试？",
"请问您还有什么问题吗？"
];
// 快捷问题
const quickQuestions = [
"当兵的注意事项？",
"男兵应征报名对象：",
"当兵公务员考试？",
"请问您还有什么问题吗？"
];
// 滚动到底部
const scrollToBottom = () => {
setTimeout(() => {
messagesEndRef.current?.scrollIntoView({ behavior: 'smooth' });
}, 100);
};
// 初始化WebSocket连接
const initWebSocket = () => {
const wsUrl = APP_API_URL.replace('https', 'ws') + '/chat/' + Taro.getStorageSync('UserId') || 'token';
// const wsUrl = 'ws://127.0.0.1:9000/chat/test'
wsRef.current = createWebSocket(wsUrl);
wsRef.current.onMessage((data: any) => {
console.log('收到WebSocket消息:', data);
if (data.answer) {
if (data.answer === '__END__') {
// 消息结束，移除typing状态
setMessages(prev => prev.map(msg =>
msg.isTyping ? { ...msg, isTyping: false } : msg
));
setIsLoading(false);
setCurrentTaskId('');
} else {
// 更新AI回复消息
setMessages(prev => {
const lastMessage = prev[prev.length - 1];
if (lastMessage && lastMessage.type === 'ai' && lastMessage.isTyping) {
// 更新最后一条AI消息
return prev.map((msg, index) =>
index === prev.length - 1
? { ...msg, query: msg.query + data.answer }
: msg
);
} else {
// 创建新的AI消息
return [...prev, {
id: Date.now().toString(),
type: 'ai',
query: data.answer,
timestamp: Date.now(),
isTyping: true
}];
}
});
if (data.taskId) {
setCurrentTaskId(data.taskId);
}
// 滚动到底部
const scrollToBottom = () => {
if (!Taro.getStorageSync('UserId')) {
Taro.showToast({
title: '请先登录',
icon: 'none',
duration: 2000
});
Taro.navigateTo({
url: '/passport/sms-login',
})
return false;
}
setTimeout(() => {
messagesEndRef.current?.scrollIntoView({behavior: 'smooth'});
}, 100);
};
// 初始化WebSocket连接
const initWebSocket = () => {
// const wsUrl = APP_API_URL.replace('http', 'ws') + '/chat/' + Taro.getStorageSync('UserId') || 'token';
console.log(WSS_API_URL + "/chat/" + Taro.getStorageSync('UserId'), 'wsUrl')
wsRef.current = createWebSocket(WSS_API_URL + "/chat/" + Taro.getStorageSync('UserId'));
wsRef.current.onMessage((data: any) => {
console.log('收到WebSocket消息:', data);
if (data.answer) {
if (data.answer === '__END__') {
// 消息结束，移除typing状态
setMessages(prev => prev.map(msg =>
msg.isTyping ? {...msg, isTyping: false} : msg
));
setIsLoading(false);
setCurrentTaskId('');
} else {
// 更新AI回复消息
setMessages(prev => {
const lastMessage = prev[prev.length - 1];
if (lastMessage && lastMessage.type === 'ai' && lastMessage.isTyping) {
// 更新最后一条AI消息
return prev.map((msg, index) =>
index === prev.length - 1
? {...msg, query: msg.query + data.answer}
: msg
);
} else {
// 创建新的AI消息
return [...prev, {
id: Date.now().toString(),
type: 'ai',
query: data.answer,
timestamp: Date.now(),
isTyping: true
}];
}
});
if (data.taskId) {
setCurrentTaskId(data.taskId);
}
}
scrollToBottom();
}
});
wsRef.current.onOpen(() => {
console.log('WebSocket连接成功');
});
wsRef.current.onError((error: any) => {
console.error('WebSocket连接错误:', error);
});
wsRef.current.connect().catch((error: any) => {
console.error('WebSocket连接失败:', error);
});
};
useEffect(() => {
initWebSocket();
Taro.hideTabBar();
return () => {
if (wsRef.current) {
wsRef.current.close();
}
};
}, []);
useEffect(() => {
scrollToBottom();
}
});
}, [messages]);
wsRef.current.onOpen(() => {
console.log('WebSocket连接成功');
});
// 发送消息
const handleSendMessage = async (content: string) => {
if (!content.trim() || isLoading) return;
wsRef.current.onError((error: any) => {
console.error('WebSocket连接错误:', error);
// Toast.showToast({ title: '连接失败，请检查网络', icon: 'error'});
});
const userMessage: Message = {
id: Date.now().toString(),
type: 'user',
query: content.trim(),
timestamp: Date.now(),
user: `${Taro.getStorageSync('UserId')}`
};
setMessages(prev => [...prev, userMessage]);
setInputValue('');
setIsLoading(true);
wsRef.current.connect().catch((error: any) => {
console.error('WebSocket连接失败:', error);
});
};
useEffect(() => {
initWebSocket();
Taro.hideTabBar();
return () => {
if (wsRef.current) {
wsRef.current.close();
}
try {
await sendAiMessage({
query: content.trim(),
user: `${Taro.getStorageSync('UserId')}`,
responseMode: 'streaming',
});
} catch (error) {
console.error('发送消息失败:', error);
setIsLoading(false);
}
};
}, []);
useEffect(() => {
scrollToBottom();
}, [messages]);
// 发送消息
const handleSendMessage = async (content: string) => {
if (!content.trim() || isLoading) return;
const userMessage: Message = {
id: Date.now().toString(),
type: 'user',
query: content.trim(),
timestamp: Date.now(),
user: `${Taro.getStorageSync('UserId')}`
// 停止AI回复
const handleStopMessage = async () => {
if (currentTaskId) {
try {
await stopAiMessage({taskId: currentTaskId});
setIsLoading(false);
setCurrentTaskId('');
setMessages(prev => prev.map(msg =>
msg.isTyping ? {...msg, isTyping: false} : msg
));
} catch (error) {
console.error('停止消息失败:', error);
}
}
};
setMessages(prev => [...prev, userMessage]);
setInputValue('');
setIsLoading(true);
try {
await sendAiMessage({
query: content.trim(),
user: `${Taro.getStorageSync('UserId')}`,
responseMode: 'streaming',
});
} catch (error) {
console.error('发送消息失败:', error);
// Toast.showToast({ title: '发送失败，请重试', icon: 'error'});
setIsLoading(false);
}
};
// 处理快捷问题点击
const handleQuickQuestion = (question: string) => {
handleSendMessage(question);
};
// 停止AI回复
const handleStopMessage = async () => {
if (currentTaskId) {
try {
await stopAiMessage({ taskId: currentTaskId });
setIsLoading(false);
setCurrentTaskId('');
setMessages(prev => prev.map(msg =>
msg.isTyping ? { ...msg, isTyping: false } : msg
));
} catch (error) {
console.error('停止消息失败:', error);
}
}
};
// 处理快捷问题点击
const handleQuickQuestion = (question: string) => {
console.log(question,'qqq')
handleSendMessage(question);
};
return (
<View className="ai-chat">
{/*<View className="chat-header">*/}
{/* AI智能问答*/}
{/*</View>*/}
<View className="chat-container">
<View className="messages-container">
{messages.length === 0 ? (
<View className="empty-state">
<View className="empty-icon">🤖</View>
<View className="empty-title">您好！我是AI助手</View>
<View className="empty-desc">
我可以为您解答各种问题，请输入您想了解的内容
</View>
</View>
) : (
messages.map((message) => (
<View key={message.id} className={`message ${message.type}`}>
<View className={`avatar ${message.type}-avatar`}>
{message.type === 'user' ? <User size={20} /> : '🤖'}
return (
<View className="ai-chat">
<Header/>
<View className="chat-container">
<View className="messages-container">
{messages.length === 0 ? (
<View className="empty-state">
<View className="empty-icon">🤖</View>
<View className="empty-title">您好！我是AI助手</View>
<View className="empty-desc">
我可以为您解答各种问题，请输入您想了解的内容
</View>
</View>
) : (
messages.map((message) => (
<View key={message.id} className={`message ${message.type}`}>
<View className={`avatar ${message.type}-avatar`}>
{message.type === 'user' ? <User size={20}/> : '🤖'}
</View>
<View className="message-content">
{message.isTyping ? (
<View className="typing-indicator">
<View className="dot"></View>
<View className="dot"></View>
<View className="dot"></View>
</View>
) : (
message.query
)}
</View>
</View>
))
)}
<View ref={messagesEndRef}/>
</View>
<View className="message-content">
{message.isTyping ? (
<View className="typing-indicator">
<View className="dot"></View>
<View className="dot"></View>
<View className="dot"></View>
{messages.length === 0 && (
<View className="quick-questions">
<View className="quick-title">
💡 您可以问我：
</View>
<View className="questions-list">
{quickQuestions.map((question, index) => (
<View
key={index}
className="question-item"
onClick={() => handleQuickQuestion(question)}
>
{question}
</View>
))}
</View>
</View>
) : (
message.query
)}
)}
<View className="input-container">
<View className="input-wrapper">
<Textarea
className="message-input"
value={inputValue}
placeholder="请输入您的问题..."
onInput={(e) => setInputValue(e.detail.value)}
onConfirm={() => handleSendMessage(inputValue)}
autoHeight
maxlength={500}
/>
</View>
{isLoading ? (
<Button
onClick={handleStopMessage}
type="danger"
icon={<ArrowUp/>}
>
</Button>
) : (
<Button
type="danger"
onClick={() => handleSendMessage(inputValue)}
// disabled={!inputValue.trim()}
icon={<ArrowUp/>}
>
</Button>
)}
</View>
</View>
))
)}
<View ref={messagesEndRef} />
</View>
{messages.length === 0 && (
<View className="quick-questions">
<View className="quick-title">
💡 您可以问我：
</View>
<View className="questions-list">
{quickQuestions.map((question, index) => (
<View
key={index}
className="question-item"
onClick={() => handleQuickQuestion(question)}
>
{question}
</View>
))}
</View>
</View>
)}
<View className="input-container">
<View className="input-wrapper">
<Textarea
className="message-input"
value={inputValue}
placeholder="请输入您的问题..."
onInput={(e) => setInputValue(e.detail.value)}
onConfirm={() => handleSendMessage(inputValue)}
disabled={isLoading}
autoHeight
maxlength={500}
/>
</View>
{isLoading ? (
<Button
onClick={handleStopMessage}
type="danger"
icon={<ArrowUp />}
>
</Button>
) : (
<Button
type="danger"
onClick={() => handleSendMessage(inputValue)}
disabled={!inputValue.trim()}
icon={<ArrowUp />}
>
</Button>
)}
</View>
</View>
</View>
);
);
};
export default AiChat;

29
src/pages/index/Header.tsx
View File

@@ -1,7 +1,7 @@
import {useEffect, useState} from "react";
import Taro from '@tarojs/taro';
import {Button, Space} from '@nutui/nutui-react-taro'
import {TriangleDown} from '@nutui/icons-react-taro'
import {TriangleDown,ArrowLeft} from '@nutui/icons-react-taro'
import {Popup, Avatar, NavBar} from '@nutui/nutui-react-taro'
import {TenantId} from "@/utils/config";
@@ -10,15 +10,15 @@ const Header = (props: any) => {
const [showBasic, setShowBasic] = useState(false)
const [statusBarHeight, setStatusBarHeight] = useState<number>()
const onNav = () => {
if (!IsLogin) {
return false;
}
Taro.switchTab({
url: '/pages/user/user',
})
}
// const onNav = () => {
// if (!IsLogin) {
//
// return false;
// }
// Taro.switchTab({
// url: '/pages/user/user',
// })
// }
const reload = () => {
Taro.getSystemInfo({
success: (res) => {
@@ -92,13 +92,8 @@ const Header = (props: any) => {
<TriangleDown size={9}/>
</div>
) : (
<div style={{display: 'flex', alignItems: 'center', gap: '8px'}} onClick={onNav}>
<Avatar
size="22"
src={props.user?.avatar}
/>
{props.user?.nickname}
<TriangleDown size={9}/>
<div style={{display: 'flex', alignItems: 'center', gap: '8px'}} onClick={() => Taro.switchTab({url: '/pages/index/index'})}>
<Button icon={<ArrowLeft />} />
</div>
)}>
</NavBar>