Update .gitignore and add files
This commit is contained in:
jrhlh
265
back/blueprints/aiask.py
Normal file
265
back/blueprints/aiask.py
Normal file
@ -0,0 +1,265 @@
|
||||
from flask import Blueprint, request, jsonify, current_app
|
||||
import json
|
||||
import datetime
|
||||
import threading
|
||||
import time
|
||||
from werkzeug.exceptions import BadRequest, InternalServerError
|
||||
from ai_processor import process_ai_message
|
||||
import uuid
|
||||
|
||||
bp = Blueprint('aiask', __name__)
|
||||
|
||||
# 使用字典存储对话历史和状态
|
||||
conversation_data = {}
|
||||
data_lock = threading.Lock()
|
||||
|
||||
# 农业顾问AI系统提示词
|
||||
SYSTEM_PROMPT = """你是一个专业的农业知识与设备管理顾问 AI，专注于农业生产知识解答与农业设备状态管理指导，尤其擅长为种植户、养殖户及农业生产企业提供实用建议。
|
||||
|
||||
你的任务是：
|
||||
1. 解答农业生产相关的知识疑问，包括但不限于作物种植、畜禽养殖、病虫害防治、土壤改良、农资使用等内容。
|
||||
2. 提供农业设备（如播种机、收割机、灌溉设备、养殖设备等）的状态监测、日常维护、常见故障排查及管理建议。
|
||||
3. 鼓励用户采用科学的农业生产方式和规范的设备管理流程，提升生产效率与安全性。
|
||||
|
||||
你需要遵循的原则：
|
||||
- 始终保持专业、严谨、科学的态度，基于农业技术规范和设备管理标准提供建议。
|
||||
- 不提供未经证实的农业技术或设备操作方法，对于复杂的设备故障或特殊农业问题，建议用户咨询专业技术人员或农业机构。
|
||||
- 只返回相关建议，不要返回其他无关内容"""
|
||||
|
||||
|
||||
def get_conversation_data(user_id):
|
||||
"""获取用户的对话数据"""
|
||||
with data_lock:
|
||||
if user_id not in conversation_data:
|
||||
conversation_data[user_id] = {
|
||||
'history': [],
|
||||
'processing': False,
|
||||
'last_active': time.time(),
|
||||
'request_id': None
|
||||
}
|
||||
return conversation_data[user_id]
|
||||
|
||||
|
||||
def cleanup_inactive_sessions():
|
||||
"""清理超过1小时不活跃的会话"""
|
||||
with data_lock:
|
||||
current_time = time.time()
|
||||
inactive_users = [
|
||||
user_id for user_id, data in conversation_data.items()
|
||||
if current_time - data['last_active'] > 3600 # 1小时
|
||||
]
|
||||
for user_id in inactive_users:
|
||||
del conversation_data[user_id]
|
||||
|
||||
|
||||
def add_to_history(user_id, role, content):
|
||||
"""添加消息到对话历史，确保内容不为空"""
|
||||
data = get_conversation_data(user_id)
|
||||
data['history'].append({
|
||||
"role": role,
|
||||
"content": content or "（空回复）",
|
||||
"timestamp": datetime.datetime.now().isoformat()
|
||||
})
|
||||
data['last_active'] = time.time()
|
||||
|
||||
# 限制历史记录长度（仅保留最近10条交互）
|
||||
if len(data['history']) > 10:
|
||||
data['history'] = data['history'][-10:]
|
||||
|
||||
|
||||
def format_question(history):
|
||||
"""格式化问题，包含系统提示和历史对话"""
|
||||
# 获取最近的历史记录（最多10条交互）
|
||||
recent_history = history[-10:] if len(history) > 10 else history
|
||||
|
||||
# 构建包含系统提示的完整对话上下文
|
||||
formatted_context = [
|
||||
{"role": "system", "content": SYSTEM_PROMPT}
|
||||
]
|
||||
|
||||
# 添加用户与助手的历史对话
|
||||
formatted_context.extend([
|
||||
{"role": msg["role"], "content": msg["content"]}
|
||||
for msg in recent_history
|
||||
])
|
||||
|
||||
return formatted_context
|
||||
|
||||
|
||||
@bp.route('/ask', methods=['POST'])
|
||||
def ask():
|
||||
"""向AI提问，改进异步处理流程"""
|
||||
data = request.json
|
||||
question = data.get('question')
|
||||
user_id = data.get('user_id', 'anonymous')
|
||||
|
||||
if not question:
|
||||
return jsonify({"code": 400, "message": "问题不能为空"}), 400
|
||||
|
||||
# 获取用户数据
|
||||
user_data = get_conversation_data(user_id)
|
||||
|
||||
# 检查是否已有处理中的请求
|
||||
if user_data['processing']:
|
||||
return jsonify({
|
||||
"code": 429,
|
||||
"message": "已有处理中的请求，请稍后再试",
|
||||
"request_id": user_data['request_id']
|
||||
}), 429
|
||||
|
||||
# 添加用户问题到历史
|
||||
add_to_history(user_id, "user", question)
|
||||
|
||||
# 格式化问题（包含系统提示和历史对话）
|
||||
full_question = format_question(user_data['history'])
|
||||
|
||||
# 生成唯一请求ID
|
||||
request_id = f"req_{uuid.uuid4().hex[:8]}"
|
||||
user_data['request_id'] = request_id
|
||||
user_data['processing'] = True
|
||||
|
||||
# 调用AI回答函数（异步处理）
|
||||
try:
|
||||
appid = current_app.config.get('APPID')
|
||||
api_key = current_app.config.get('API_KEY')
|
||||
api_secret = current_app.config.get('API_SECRET')
|
||||
spark_url = current_app.config.get('SPARK_URL')
|
||||
domain = current_app.config.get('DOMAIN', 'x1')
|
||||
|
||||
app = current_app._get_current_object()
|
||||
|
||||
def process_ai_request():
|
||||
try:
|
||||
with app.app_context():
|
||||
start_time = time.time()
|
||||
current_app.logger.info(f"开始处理AI请求 {request_id}")
|
||||
|
||||
ai_answer = process_ai_message(
|
||||
appid, api_key, api_secret, spark_url, domain, full_question
|
||||
)
|
||||
|
||||
# 记录处理时间
|
||||
process_time = time.time() - start_time
|
||||
current_app.logger.info(
|
||||
f"AI请求 {request_id} 处理完成，耗时 {process_time:.2f}秒"
|
||||
)
|
||||
|
||||
add_to_history(user_id, "assistant", ai_answer)
|
||||
except Exception as e:
|
||||
with app.app_context():
|
||||
current_app.logger.error(f"AI请求 {request_id} 处理错误: {str(e)}")
|
||||
add_to_history(user_id, "assistant", f"处理请求时出错: {str(e)}")
|
||||
finally:
|
||||
# 更新处理状态
|
||||
user_data['processing'] = False
|
||||
user_data['request_id'] = None
|
||||
cleanup_inactive_sessions()
|
||||
|
||||
# 启动处理线程
|
||||
threading.Thread(target=process_ai_request, daemon=True).start()
|
||||
|
||||
return jsonify({
|
||||
"code": 200,
|
||||
"message": "请求已接收，正在处理中",
|
||||
"request_id": request_id
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
user_data['processing'] = False
|
||||
user_data['request_id'] = None
|
||||
current_app.logger.error(f"AI请求初始化错误: {str(e)}")
|
||||
return jsonify({
|
||||
"code": 500,
|
||||
"message": "服务器内部错误",
|
||||
"request_id": request_id
|
||||
}), 500
|
||||
|
||||
|
||||
@bp.route('/history', methods=['GET'])
|
||||
def get_history():
|
||||
"""获取对话历史，优化性能"""
|
||||
user_id = request.args.get('user_id')
|
||||
|
||||
if not user_id:
|
||||
return jsonify({
|
||||
"code": 400,
|
||||
"message": "用户ID不能为空",
|
||||
"history": []
|
||||
}), 400
|
||||
|
||||
try:
|
||||
user_data = get_conversation_data(user_id)
|
||||
history = user_data.get('history', [])
|
||||
|
||||
current_app.logger.info(
|
||||
f"获取历史记录，用户ID: {user_id}，记录数量: {len(history)}"
|
||||
)
|
||||
|
||||
return jsonify({
|
||||
"code": 200,
|
||||
"message": "获取历史记录成功",
|
||||
"history": history,
|
||||
"processing": user_data['processing'],
|
||||
"request_id": user_data['request_id']
|
||||
})
|
||||
except Exception as e:
|
||||
current_app.logger.error(f"获取历史记录错误: {str(e)}")
|
||||
return jsonify({
|
||||
"code": 500,
|
||||
"message": "获取历史记录失败",
|
||||
"history": []
|
||||
}), 500
|
||||
|
||||
|
||||
@bp.route('/clear', methods=['POST'])
|
||||
def clear_history():
|
||||
"""清除对话历史"""
|
||||
user_id = request.json.get('user_id', 'anonymous')
|
||||
with data_lock:
|
||||
if user_id in conversation_data:
|
||||
conversation_data[user_id]['history'] = []
|
||||
return jsonify({
|
||||
"code": 200,
|
||||
"message": "对话历史已清除"
|
||||
})
|
||||
|
||||
|
||||
@bp.route('/status', methods=['GET'])
|
||||
def check_status():
|
||||
"""检查AI处理状态，优化准确性"""
|
||||
user_id = request.args.get('user_id')
|
||||
if not user_id:
|
||||
return jsonify({
|
||||
"code": 400,
|
||||
"message": "用户ID不能为空"
|
||||
}), 400
|
||||
|
||||
try:
|
||||
user_data = get_conversation_data(user_id)
|
||||
return jsonify({
|
||||
"code": 200,
|
||||
"processing": user_data['processing'],
|
||||
"request_id": user_data['request_id'],
|
||||
"last_active": user_data['last_active']
|
||||
})
|
||||
except Exception as e:
|
||||
current_app.logger.error(f"状态检查错误: {str(e)}")
|
||||
return jsonify({
|
||||
"code": 500,
|
||||
"message": "状态检查失败"
|
||||
}), 500
|
||||
|
||||
|
||||
# 定时清理不活跃会话
|
||||
def start_cleanup_scheduler():
|
||||
def cleanup_task():
|
||||
while True:
|
||||
time.sleep(3600) # 每小时清理一次
|
||||
cleanup_inactive_sessions()
|
||||
|
||||
thread = threading.Thread(target=cleanup_task, daemon=True)
|
||||
thread.start()
|
||||
|
||||
|
||||
# 启动时开始清理任务
|
||||
start_cleanup_scheduler()
|
||||
Reference in New Issue
Block a user