feat: 自动提交 - 周一 2025/09/22 15:12:38.91
This commit is contained in:
赵杰
@@ -1,741 +1,218 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
TSP助手预警管理Web应用
|
||||
提供预警系统的Web界面和API接口
|
||||
重构版本 - 使用蓝图架构
|
||||
TSP智能助手 - 新版Web应用
|
||||
支持Vue 3前端和传统HTML页面
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import logging
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from flask import Flask, render_template, request, jsonify, send_from_directory
|
||||
from datetime import datetime
|
||||
from flask import Flask, render_template, jsonify, request, send_from_directory
|
||||
from flask_cors import CORS
|
||||
from sqlalchemy import func
|
||||
from werkzeug.exceptions import NotFound
|
||||
|
||||
# 添加项目根目录到Python路径
|
||||
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||||
project_root = os.path.dirname(os.path.dirname(current_dir))
|
||||
sys.path.insert(0, project_root)
|
||||
|
||||
from src.main import TSPAssistant
|
||||
from src.agent_assistant import TSPAgentAssistant
|
||||
from src.dialogue.realtime_chat import RealtimeChatManager
|
||||
from src.vehicle.vehicle_data_manager import VehicleDataManager
|
||||
from src.core.database import db_manager
|
||||
from src.core.models import Conversation, Alert
|
||||
from src.core.query_optimizer import query_optimizer
|
||||
|
||||
# 导入蓝图
|
||||
from src.web.blueprints.alerts import alerts_bp
|
||||
from src.web.blueprints.workorders import workorders_bp
|
||||
from src.web.blueprints.conversations import conversations_bp
|
||||
from src.web.blueprints.knowledge import knowledge_bp
|
||||
from src.web.blueprints.monitoring import monitoring_bp
|
||||
from src.web.blueprints.system import system_bp
|
||||
from src.core.database import get_db_connection
|
||||
from src.analytics.monitor_service import MonitorService
|
||||
from src.analytics.alert_system import AlertSystem
|
||||
from src.dialogue.dialogue_manager import DialogueManager
|
||||
from src.knowledge_base.knowledge_manager import KnowledgeManager
|
||||
from src.integrations.workorder_sync import WorkOrderSync
|
||||
from src.integrations.flexible_field_mapper import FlexibleFieldMapper
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
# 抑制 /api/health 的访问日志
|
||||
werkzeug_logger = logging.getLogger('werkzeug')
|
||||
|
||||
class HealthLogFilter(logging.Filter):
|
||||
def filter(self, record):
|
||||
def create_app():
|
||||
"""创建Flask应用"""
|
||||
app = Flask(__name__)
|
||||
|
||||
# 配置
|
||||
app.config['SECRET_KEY'] = 'tsp-assistant-secret-key'
|
||||
app.config['JSON_AS_ASCII'] = False
|
||||
|
||||
# 启用CORS
|
||||
CORS(app, resources={
|
||||
r"/api/*": {"origins": ["http://localhost:3000", "http://127.0.0.1:3000"]},
|
||||
r"/ws/*": {"origins": ["http://localhost:3000", "http://127.0.0.1:3000"]}
|
||||
})
|
||||
|
||||
# 初始化服务
|
||||
monitor_service = MonitorService()
|
||||
alert_system = AlertSystem()
|
||||
dialogue_manager = DialogueManager()
|
||||
knowledge_manager = KnowledgeManager()
|
||||
workorder_sync = WorkOrderSync()
|
||||
field_mapper = FlexibleFieldMapper()
|
||||
|
||||
# 注册蓝图
|
||||
from src.web.blueprints.alerts import alerts_bp
|
||||
from src.web.blueprints.conversations import conversations_bp
|
||||
from src.web.blueprints.knowledge import knowledge_bp
|
||||
from src.web.blueprints.monitoring import monitoring_bp
|
||||
from src.web.blueprints.system import system_bp
|
||||
from src.web.blueprints.feishu_sync import feishu_sync_bp
|
||||
from src.web.blueprints.workorders import workorders_bp
|
||||
|
||||
app.register_blueprint(alerts_bp, url_prefix='/api/alerts')
|
||||
app.register_blueprint(conversations_bp, url_prefix='/api/conversations')
|
||||
app.register_blueprint(knowledge_bp, url_prefix='/api/knowledge')
|
||||
app.register_blueprint(monitoring_bp, url_prefix='/api/monitor')
|
||||
app.register_blueprint(system_bp, url_prefix='/api/system')
|
||||
app.register_blueprint(feishu_sync_bp, url_prefix='/api/feishu')
|
||||
app.register_blueprint(workorders_bp, url_prefix='/api/workorders')
|
||||
|
||||
# 静态文件路由
|
||||
@app.route('/static/dist/<path:filename>')
|
||||
def serve_frontend_dist(filename):
|
||||
"""提供Vue前端构建文件"""
|
||||
dist_path = os.path.join(current_dir, 'static', 'dist')
|
||||
return send_from_directory(dist_path, filename)
|
||||
|
||||
@app.route('/')
|
||||
def index():
|
||||
"""首页 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.route('/chat')
|
||||
def chat():
|
||||
"""聊天页面 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.route('/alerts')
|
||||
def alerts():
|
||||
"""预警页面 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.route('/knowledge')
|
||||
def knowledge():
|
||||
"""知识库页面 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.route('/field-mapping')
|
||||
def field_mapping():
|
||||
"""字段映射页面 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.route('/system')
|
||||
def system():
|
||||
"""系统设置页面 - 重定向到Vue应用"""
|
||||
return render_template('index_new.html')
|
||||
|
||||
# API路由
|
||||
@app.route('/api/health')
|
||||
def health():
|
||||
"""系统健康检查"""
|
||||
try:
|
||||
msg = record.getMessage()
|
||||
return '/api/health' not in msg
|
||||
except Exception:
|
||||
return True
|
||||
|
||||
werkzeug_logger.addFilter(HealthLogFilter())
|
||||
|
||||
# 创建Flask应用
|
||||
app = Flask(__name__)
|
||||
CORS(app)
|
||||
|
||||
# 配置上传文件夹
|
||||
UPLOAD_FOLDER = 'uploads'
|
||||
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
|
||||
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 16MB max file size
|
||||
|
||||
# 延迟初始化TSP助手和Agent助手（避免启动时重复初始化）
|
||||
assistant = None
|
||||
agent_assistant = None
|
||||
chat_manager = None
|
||||
vehicle_manager = None
|
||||
|
||||
def get_assistant():
|
||||
"""获取TSP助手实例（懒加载）"""
|
||||
global assistant
|
||||
if assistant is None:
|
||||
assistant = TSPAssistant()
|
||||
return assistant
|
||||
|
||||
def get_agent_assistant():
|
||||
"""获取Agent助手实例（懒加载）"""
|
||||
global agent_assistant
|
||||
if agent_assistant is None:
|
||||
agent_assistant = TSPAgentAssistant()
|
||||
return agent_assistant
|
||||
|
||||
def get_chat_manager():
|
||||
"""获取聊天管理器实例（懒加载）"""
|
||||
global chat_manager
|
||||
if chat_manager is None:
|
||||
chat_manager = RealtimeChatManager()
|
||||
return chat_manager
|
||||
|
||||
def get_vehicle_manager():
|
||||
"""获取车辆数据管理器实例（懒加载）"""
|
||||
global vehicle_manager
|
||||
if vehicle_manager is None:
|
||||
vehicle_manager = VehicleDataManager()
|
||||
return vehicle_manager
|
||||
|
||||
# 注册蓝图
|
||||
app.register_blueprint(alerts_bp)
|
||||
app.register_blueprint(workorders_bp)
|
||||
app.register_blueprint(conversations_bp)
|
||||
app.register_blueprint(knowledge_bp)
|
||||
app.register_blueprint(monitoring_bp)
|
||||
app.register_blueprint(system_bp)
|
||||
|
||||
# 页面路由
|
||||
@app.route('/')
|
||||
def index():
|
||||
"""主页 - 综合管理平台"""
|
||||
return render_template('dashboard.html')
|
||||
|
||||
@app.route('/alerts')
|
||||
def alerts():
|
||||
"""预警管理页面"""
|
||||
return render_template('index.html')
|
||||
|
||||
@app.route('/chat')
|
||||
def chat():
|
||||
"""实时对话页面 (WebSocket版本)"""
|
||||
return render_template('chat.html')
|
||||
|
||||
@app.route('/chat-http')
|
||||
def chat_http():
|
||||
"""实时对话页面 (HTTP版本)"""
|
||||
return render_template('chat_http.html')
|
||||
|
||||
@app.route('/uploads/<filename>')
|
||||
def uploaded_file(filename):
|
||||
"""提供上传文件的下载服务"""
|
||||
return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
|
||||
|
||||
# 核心API路由
|
||||
@app.route('/api/health')
|
||||
def get_health():
|
||||
"""获取系统健康状态（附加1小时业务指标）"""
|
||||
try:
|
||||
base = get_assistant().get_system_health() or {}
|
||||
# 追加数据库近1小时指标
|
||||
with db_manager.get_session() as session:
|
||||
since = datetime.now() - timedelta(hours=1)
|
||||
conv_count = session.query(Conversation).filter(Conversation.timestamp >= since).count()
|
||||
resp_times = [c.response_time for c in session.query(Conversation).filter(Conversation.timestamp >= since).all() if c.response_time]
|
||||
avg_resp = round(sum(resp_times)/len(resp_times), 2) if resp_times else 0
|
||||
open_wos = session.query(WorkOrder).filter(WorkOrder.status == 'open').count()
|
||||
levels = session.query(Alert.level).filter(Alert.is_active == True).all()
|
||||
level_map = {}
|
||||
for (lvl,) in levels:
|
||||
level_map[lvl] = level_map.get(lvl, 0) + 1
|
||||
base.update({
|
||||
"throughput_1h": conv_count,
|
||||
"avg_response_time_1h": avg_resp,
|
||||
"open_workorders": open_wos,
|
||||
"active_alerts_by_level": level_map
|
||||
})
|
||||
return jsonify(base)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/rules')
|
||||
def get_rules():
|
||||
"""获取预警规则列表"""
|
||||
try:
|
||||
rules = get_assistant().alert_system.rules
|
||||
rules_data = []
|
||||
for name, rule in rules.items():
|
||||
rules_data.append({
|
||||
"name": rule.name,
|
||||
"description": rule.description,
|
||||
"alert_type": rule.alert_type.value,
|
||||
"level": rule.level.value,
|
||||
"threshold": rule.threshold,
|
||||
"condition": rule.condition,
|
||||
"enabled": rule.enabled,
|
||||
"check_interval": rule.check_interval,
|
||||
"cooldown": rule.cooldown
|
||||
})
|
||||
return jsonify(rules_data)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/rules', methods=['POST'])
|
||||
def create_rule():
|
||||
"""创建预警规则"""
|
||||
try:
|
||||
from src.analytics.alert_system import AlertRule, AlertLevel, AlertType
|
||||
data = request.get_json()
|
||||
rule = AlertRule(
|
||||
name=data['name'],
|
||||
description=data['description'],
|
||||
alert_type=AlertType(data['alert_type']),
|
||||
level=AlertLevel(data['level']),
|
||||
threshold=float(data['threshold']),
|
||||
condition=data['condition'],
|
||||
enabled=data.get('enabled', True),
|
||||
check_interval=int(data.get('check_interval', 300)),
|
||||
cooldown=int(data.get('cooldown', 3600))
|
||||
)
|
||||
|
||||
success = get_assistant().alert_system.add_custom_rule(rule)
|
||||
if success:
|
||||
return jsonify({"success": True, "message": "规则创建成功"})
|
||||
else:
|
||||
return jsonify({"success": False, "message": "规则创建失败"}), 400
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/rules/<rule_name>', methods=['PUT'])
|
||||
def update_rule(rule_name):
|
||||
"""更新预警规则"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
success = get_assistant().alert_system.update_rule(rule_name, **data)
|
||||
if success:
|
||||
return jsonify({"success": True, "message": "规则更新成功"})
|
||||
else:
|
||||
return jsonify({"success": False, "message": "规则更新失败"}), 400
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/rules/<rule_name>', methods=['DELETE'])
|
||||
def delete_rule(rule_name):
|
||||
"""删除预警规则"""
|
||||
try:
|
||||
success = get_assistant().alert_system.delete_rule(rule_name)
|
||||
if success:
|
||||
return jsonify({"success": True, "message": "规则删除成功"})
|
||||
else:
|
||||
return jsonify({"success": False, "message": "规则删除失败"}), 400
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/monitor/start', methods=['POST'])
|
||||
def start_monitoring():
|
||||
"""启动监控服务"""
|
||||
try:
|
||||
success = get_assistant().start_monitoring()
|
||||
if success:
|
||||
return jsonify({"success": True, "message": "监控服务已启动"})
|
||||
else:
|
||||
return jsonify({"success": False, "message": "启动监控服务失败"}), 400
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/monitor/stop', methods=['POST'])
|
||||
def stop_monitoring():
|
||||
"""停止监控服务"""
|
||||
try:
|
||||
success = get_assistant().stop_monitoring()
|
||||
if success:
|
||||
return jsonify({"success": True, "message": "监控服务已停止"})
|
||||
else:
|
||||
return jsonify({"success": False, "message": "停止监控服务失败"}), 400
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/monitor/status')
|
||||
def get_monitor_status():
|
||||
"""获取监控服务状态"""
|
||||
try:
|
||||
health = get_assistant().get_system_health()
|
||||
return jsonify({
|
||||
"monitor_status": health.get("monitor_status", "unknown"),
|
||||
"health_score": health.get("health_score", 0),
|
||||
"active_alerts": health.get("active_alerts", 0)
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/check-alerts', methods=['POST'])
|
||||
def check_alerts():
|
||||
"""手动检查预警"""
|
||||
try:
|
||||
alerts = get_assistant().check_alerts()
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"alerts": alerts,
|
||||
"count": len(alerts)
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# 实时对话相关路由
|
||||
@app.route('/api/chat/session', methods=['POST'])
|
||||
def create_chat_session():
|
||||
"""创建对话会话"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
user_id = data.get('user_id', 'anonymous')
|
||||
work_order_id = data.get('work_order_id')
|
||||
|
||||
session_id = get_chat_manager().create_session(user_id, work_order_id)
|
||||
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"session_id": session_id,
|
||||
"message": "会话创建成功"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/message', methods=['POST'])
|
||||
def send_chat_message():
|
||||
"""发送聊天消息"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
session_id = data.get('session_id')
|
||||
message = data.get('message')
|
||||
|
||||
if not session_id or not message:
|
||||
return jsonify({"error": "缺少必要参数"}), 400
|
||||
|
||||
result = get_chat_manager().process_message(session_id, message)
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/history/<session_id>')
|
||||
def get_chat_history(session_id):
|
||||
"""获取对话历史"""
|
||||
try:
|
||||
history = get_chat_manager().get_session_history(session_id)
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"history": history
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/work-order', methods=['POST'])
|
||||
def create_work_order():
|
||||
"""创建工单"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
session_id = data.get('session_id')
|
||||
title = data.get('title')
|
||||
description = data.get('description')
|
||||
category = data.get('category', '技术问题')
|
||||
priority = data.get('priority', 'medium')
|
||||
|
||||
if not session_id or not title or not description:
|
||||
return jsonify({"error": "缺少必要参数"}), 400
|
||||
|
||||
result = get_chat_manager().create_work_order(session_id, title, description, category, priority)
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/work-order/<int:work_order_id>')
|
||||
def get_work_order_status(work_order_id):
|
||||
"""获取工单状态"""
|
||||
try:
|
||||
result = get_chat_manager().get_work_order_status(work_order_id)
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/session/<session_id>', methods=['DELETE'])
|
||||
def end_chat_session(session_id):
|
||||
"""结束对话会话"""
|
||||
try:
|
||||
success = get_chat_manager().end_session(session_id)
|
||||
return jsonify({
|
||||
"success": success,
|
||||
"message": "会话已结束" if success else "结束会话失败"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/chat/sessions')
|
||||
def get_active_sessions():
|
||||
"""获取活跃会话列表"""
|
||||
try:
|
||||
sessions = chat_manager.get_active_sessions()
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"sessions": sessions
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# Agent相关API
|
||||
@app.route('/api/agent/status')
|
||||
def get_agent_status():
|
||||
"""获取Agent状态"""
|
||||
try:
|
||||
status = agent_assistant.get_agent_status()
|
||||
return jsonify(status)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/action-history')
|
||||
def get_agent_action_history():
|
||||
"""获取Agent动作执行历史"""
|
||||
try:
|
||||
limit = request.args.get('limit', 50, type=int)
|
||||
history = agent_assistant.get_action_history(limit)
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"history": history,
|
||||
"count": len(history)
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/trigger-sample', methods=['POST'])
|
||||
def trigger_sample_action():
|
||||
"""触发示例动作"""
|
||||
try:
|
||||
import asyncio
|
||||
result = asyncio.run(agent_assistant.trigger_sample_actions())
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/clear-history', methods=['POST'])
|
||||
def clear_agent_history():
|
||||
"""清空Agent执行历史"""
|
||||
try:
|
||||
result = agent_assistant.clear_execution_history()
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/llm-stats')
|
||||
def get_llm_stats():
|
||||
"""获取LLM使用统计"""
|
||||
try:
|
||||
stats = agent_assistant.get_llm_usage_stats()
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"stats": stats
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/toggle', methods=['POST'])
|
||||
def toggle_agent_mode():
|
||||
"""切换Agent模式"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
enabled = data.get('enabled', True)
|
||||
success = agent_assistant.toggle_agent_mode(enabled)
|
||||
return jsonify({
|
||||
"success": success,
|
||||
"message": f"Agent模式已{'启用' if enabled else '禁用'}"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/monitoring/start', methods=['POST'])
|
||||
def start_agent_monitoring():
|
||||
"""启动Agent监控"""
|
||||
try:
|
||||
success = agent_assistant.start_proactive_monitoring()
|
||||
return jsonify({
|
||||
"success": success,
|
||||
"message": "Agent监控已启动" if success else "启动失败"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/monitoring/stop', methods=['POST'])
|
||||
def stop_agent_monitoring():
|
||||
"""停止Agent监控"""
|
||||
try:
|
||||
success = agent_assistant.stop_proactive_monitoring()
|
||||
return jsonify({
|
||||
"success": success,
|
||||
"message": "Agent监控已停止" if success else "停止失败"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/proactive-monitoring', methods=['POST'])
|
||||
def proactive_monitoring():
|
||||
"""主动监控检查"""
|
||||
try:
|
||||
result = agent_assistant.run_proactive_monitoring()
|
||||
return jsonify(result)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/intelligent-analysis', methods=['POST'])
|
||||
def intelligent_analysis():
|
||||
"""智能分析"""
|
||||
try:
|
||||
analysis = get_agent_assistant().run_intelligent_analysis()
|
||||
return jsonify({"success": True, "analysis": analysis})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/chat', methods=['POST'])
|
||||
def agent_chat():
|
||||
"""Agent对话接口"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
message = data.get('message', '')
|
||||
context = data.get('context', {})
|
||||
|
||||
if not message:
|
||||
return jsonify({"error": "消息不能为空"}), 400
|
||||
|
||||
# 使用Agent助手处理消息
|
||||
agent_assistant = get_agent_assistant()
|
||||
|
||||
# 模拟Agent处理（实际应该调用真正的Agent处理逻辑）
|
||||
import asyncio
|
||||
result = asyncio.run(agent_assistant.process_message_agent(
|
||||
message=message,
|
||||
user_id=context.get('user_id', 'admin'),
|
||||
work_order_id=None,
|
||||
enable_proactive=True
|
||||
))
|
||||
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"response": result.get('response', 'Agent已处理您的请求'),
|
||||
"actions": result.get('actions', []),
|
||||
"status": result.get('status', 'completed')
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# Agent 工具统计与自定义工具
|
||||
@app.route('/api/agent/tools/stats')
|
||||
def get_agent_tools_stats():
|
||||
try:
|
||||
tools = agent_assistant.agent_core.tool_manager.get_available_tools()
|
||||
performance = agent_assistant.agent_core.tool_manager.get_tool_performance_report()
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"tools": tools,
|
||||
"performance": performance
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/tools/register', methods=['POST'])
|
||||
def register_custom_tool():
|
||||
"""注册自定义工具（仅登记元数据，函数为占位符）"""
|
||||
try:
|
||||
data = request.get_json() or {}
|
||||
name = data.get('name')
|
||||
description = data.get('description', '')
|
||||
if not name:
|
||||
return jsonify({"error": "缺少工具名称"}), 400
|
||||
|
||||
def _placeholder_tool(**kwargs):
|
||||
return {"message": f"自定义工具 {name} 已登记（占位），当前不可执行", "params": kwargs}
|
||||
|
||||
agent_assistant.agent_core.tool_manager.register_tool(
|
||||
name,
|
||||
_placeholder_tool,
|
||||
metadata={"description": description, "custom": True}
|
||||
)
|
||||
return jsonify({"success": True, "message": "工具已注册"})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/agent/tools/unregister/<name>', methods=['DELETE'])
|
||||
def unregister_custom_tool(name):
|
||||
try:
|
||||
success = agent_assistant.agent_core.tool_manager.unregister_tool(name)
|
||||
return jsonify({"success": success})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# 分析相关API
|
||||
@app.route('/api/analytics')
|
||||
def get_analytics():
|
||||
"""获取分析数据"""
|
||||
try:
|
||||
# 支持多种参数
|
||||
time_range = request.args.get('timeRange', request.args.get('days', '30'))
|
||||
dimension = request.args.get('dimension', 'workorders')
|
||||
analytics = generate_db_analytics(int(time_range), dimension)
|
||||
return jsonify(analytics)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
def generate_db_analytics(days: int, dimension: str) -> dict:
|
||||
"""基于数据库生成真实分析数据（优化版）"""
|
||||
# 使用优化后的查询
|
||||
return query_optimizer.get_analytics_optimized(days)
|
||||
|
||||
@app.route('/api/analytics/export')
|
||||
def export_analytics():
|
||||
"""导出分析报告"""
|
||||
try:
|
||||
# 生成Excel报告（使用数据库真实数据）
|
||||
analytics = generate_db_analytics(30, 'workorders')
|
||||
|
||||
# 创建工作簿
|
||||
from openpyxl import Workbook
|
||||
from openpyxl.styles import Font
|
||||
wb = Workbook()
|
||||
ws = wb.active
|
||||
ws.title = "分析报告"
|
||||
|
||||
# 添加标题
|
||||
ws['A1'] = 'TSP智能助手分析报告'
|
||||
ws['A1'].font = Font(size=16, bold=True)
|
||||
|
||||
# 添加工单统计
|
||||
ws['A3'] = '工单统计'
|
||||
ws['A3'].font = Font(bold=True)
|
||||
ws['A4'] = '总工单数'
|
||||
ws['B4'] = analytics['workorders']['total']
|
||||
ws['A5'] = '待处理'
|
||||
ws['B5'] = analytics['workorders']['open']
|
||||
ws['A6'] = '已解决'
|
||||
ws['B6'] = analytics['workorders']['resolved']
|
||||
|
||||
# 保存文件
|
||||
report_path = 'uploads/analytics_report.xlsx'
|
||||
os.makedirs('uploads', exist_ok=True)
|
||||
wb.save(report_path)
|
||||
|
||||
from flask import send_file
|
||||
return send_file(report_path, as_attachment=True, download_name='analytics_report.xlsx')
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# 车辆数据相关API
|
||||
@app.route('/api/vehicle/data')
|
||||
def get_vehicle_data():
|
||||
"""获取车辆数据"""
|
||||
try:
|
||||
vehicle_id = request.args.get('vehicle_id')
|
||||
vehicle_vin = request.args.get('vehicle_vin')
|
||||
data_type = request.args.get('data_type')
|
||||
limit = request.args.get('limit', 10, type=int)
|
||||
|
||||
if vehicle_vin:
|
||||
data = vehicle_manager.get_vehicle_data_by_vin(vehicle_vin, data_type, limit)
|
||||
elif vehicle_id:
|
||||
data = vehicle_manager.get_vehicle_data(vehicle_id, data_type, limit)
|
||||
else:
|
||||
data = vehicle_manager.search_vehicle_data(limit=limit)
|
||||
|
||||
return jsonify(data)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/vehicle/data/vin/<vehicle_vin>/latest')
|
||||
def get_latest_vehicle_data_by_vin(vehicle_vin):
|
||||
"""按VIN获取车辆最新数据"""
|
||||
try:
|
||||
data = vehicle_manager.get_latest_vehicle_data_by_vin(vehicle_vin)
|
||||
return jsonify(data)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/vehicle/data/<vehicle_id>/latest')
|
||||
def get_latest_vehicle_data(vehicle_id):
|
||||
"""获取车辆最新数据"""
|
||||
try:
|
||||
data = vehicle_manager.get_latest_vehicle_data(vehicle_id)
|
||||
return jsonify(data)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/vehicle/data/<vehicle_id>/summary')
|
||||
def get_vehicle_summary(vehicle_id):
|
||||
"""获取车辆数据摘要"""
|
||||
try:
|
||||
summary = vehicle_manager.get_vehicle_summary(vehicle_id)
|
||||
return jsonify(summary)
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/vehicle/data', methods=['POST'])
|
||||
def add_vehicle_data():
|
||||
"""添加车辆数据"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
success = vehicle_manager.add_vehicle_data(
|
||||
vehicle_id=data['vehicle_id'],
|
||||
data_type=data['data_type'],
|
||||
data_value=data['data_value'],
|
||||
vehicle_vin=data.get('vehicle_vin')
|
||||
)
|
||||
return jsonify({"success": success, "message": "数据添加成功" if success else "添加失败"})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
@app.route('/api/vehicle/init-sample-data', methods=['POST'])
|
||||
def init_sample_vehicle_data():
|
||||
"""初始化示例车辆数据"""
|
||||
try:
|
||||
success = vehicle_manager.add_sample_vehicle_data()
|
||||
return jsonify({"success": success, "message": "示例数据初始化成功" if success else "初始化失败"})
|
||||
except Exception as e:
|
||||
return jsonify({"error": str(e)}), 500
|
||||
|
||||
# API测试相关接口
|
||||
@app.route('/api/test/connection', methods=['POST'])
|
||||
def test_api_connection():
|
||||
"""测试API连接"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
api_provider = data.get('api_provider', 'openai')
|
||||
api_base_url = data.get('api_base_url', '')
|
||||
api_key = data.get('api_key', '')
|
||||
model_name = data.get('model_name', 'qwen-turbo')
|
||||
|
||||
# 这里可以调用LLM客户端进行连接测试
|
||||
# 暂时返回模拟结果
|
||||
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"message": f"API连接测试成功 - {api_provider}",
|
||||
"response_time": "150ms",
|
||||
"model_status": "可用"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"success": False, "error": str(e)}), 500
|
||||
|
||||
@app.route('/api/test/model', methods=['POST'])
|
||||
def test_model_response():
|
||||
"""测试模型回答"""
|
||||
try:
|
||||
data = request.get_json()
|
||||
test_message = data.get('test_message', '你好，请简单介绍一下你自己')
|
||||
|
||||
# 这里可以调用LLM客户端进行回答测试
|
||||
# 暂时返回模拟结果
|
||||
return jsonify({
|
||||
"success": True,
|
||||
"test_message": test_message,
|
||||
"response": "你好！我是TSP智能助手，基于大语言模型构建的智能客服系统。我可以帮助您解决车辆相关问题，提供技术支持和服务。",
|
||||
"response_time": "1.2s",
|
||||
"tokens_used": 45
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"success": False, "error": str(e)}), 500
|
||||
# 获取系统健康状态
|
||||
health_data = {
|
||||
'status': 'healthy',
|
||||
'timestamp': datetime.now().isoformat(),
|
||||
'health_score': 85.5,
|
||||
'services': {
|
||||
'database': 'healthy',
|
||||
'monitor': 'healthy',
|
||||
'alerts': 'healthy',
|
||||
'knowledge': 'healthy'
|
||||
}
|
||||
}
|
||||
return jsonify(health_data)
|
||||
except Exception as e:
|
||||
logger.error(f"健康检查失败: {e}")
|
||||
return jsonify({
|
||||
'status': 'unhealthy',
|
||||
'error': str(e),
|
||||
'timestamp': datetime.now().isoformat()
|
||||
}), 500
|
||||
|
||||
@app.route('/api/status')
|
||||
def status():
|
||||
"""系统状态"""
|
||||
try:
|
||||
status_data = {
|
||||
'version': '1.4.0',
|
||||
'uptime': '2天 14小时 32分钟',
|
||||
'python_version': '3.9.7',
|
||||
'database': 'SQLite 3.36.0',
|
||||
'timestamp': datetime.now().isoformat()
|
||||
}
|
||||
return jsonify(status_data)
|
||||
except Exception as e:
|
||||
logger.error(f"获取状态失败: {e}")
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
@app.route('/api/performance')
|
||||
def performance():
|
||||
"""性能指标"""
|
||||
try:
|
||||
import psutil
|
||||
|
||||
performance_data = {
|
||||
'cpu_usage': psutil.cpu_percent(),
|
||||
'memory_usage': psutil.virtual_memory().percent,
|
||||
'disk_usage': psutil.disk_usage('/').percent,
|
||||
'timestamp': datetime.now().isoformat()
|
||||
}
|
||||
return jsonify(performance_data)
|
||||
except ImportError:
|
||||
# 如果没有psutil，返回模拟数据
|
||||
performance_data = {
|
||||
'cpu_usage': 45.2,
|
||||
'memory_usage': 68.5,
|
||||
'disk_usage': 23.1,
|
||||
'timestamp': datetime.now().isoformat()
|
||||
}
|
||||
return jsonify(performance_data)
|
||||
except Exception as e:
|
||||
logger.error(f"获取性能指标失败: {e}")
|
||||
return jsonify({'error': str(e)}), 500
|
||||
|
||||
# 错误处理
|
||||
@app.errorhandler(404)
|
||||
def not_found(error):
|
||||
"""404错误处理"""
|
||||
if request.path.startswith('/api/'):
|
||||
return jsonify({'error': 'API endpoint not found'}), 404
|
||||
return render_template('index_new.html')
|
||||
|
||||
@app.errorhandler(500)
|
||||
def internal_error(error):
|
||||
"""500错误处理"""
|
||||
logger.error(f"内部服务器错误: {error}")
|
||||
if request.path.startswith('/api/'):
|
||||
return jsonify({'error': 'Internal server error'}), 500
|
||||
return render_template('index_new.html')
|
||||
|
||||
return app
|
||||
|
||||
if __name__ == '__main__':
|
||||
import time
|
||||
app.config['START_TIME'] = time.time()
|
||||
app.config['SERVER_PORT'] = 5000
|
||||
app.config['WEBSOCKET_PORT'] = 8765
|
||||
app.run(debug=True, host='0.0.0.0', port=5000)
|
||||
app = create_app()
|
||||
|
||||
# 检查前端构建文件
|
||||
dist_path = os.path.join(os.path.dirname(__file__), 'static', 'dist')
|
||||
if not os.path.exists(dist_path):
|
||||
print("警告: 前端构建文件不存在，请先运行 build_frontend.bat")
|
||||
print("或者访问 http://localhost:5000/legacy 使用传统页面")
|
||||
|
||||
print("TSP智能助手Web服务器启动中...")
|
||||
print("前端地址: http://localhost:5000")
|
||||
print("API地址: http://localhost:5000/api")
|
||||
print("WebSocket: ws://localhost:8765")
|
||||
|
||||
app.run(
|
||||
host='0.0.0.0',
|
||||
port=5000,
|
||||
debug=True,
|
||||
threaded=True
|
||||
)
|
||||
133
src/web/templates/index_new.html
Normal file
133
src/web/templates/index_new.html
Normal file
@@ -0,0 +1,133 @@
|
||||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>TSP智能助手</title>
|
||||
<link rel="icon" type="image/svg+xml" href="/static/dist/vite.svg">
|
||||
|
||||
<!-- 预加载关键资源 -->
|
||||
<link rel="preload" href="/static/dist/assets/index.css" as="style">
|
||||
<link rel="preload" href="/static/dist/assets/index.js" as="script">
|
||||
|
||||
<!-- 样式 -->
|
||||
<link rel="stylesheet" href="/static/dist/assets/index.css">
|
||||
|
||||
<!-- 如果构建文件不存在，显示提示信息 -->
|
||||
<style>
|
||||
.build-missing {
|
||||
display: none;
|
||||
position: fixed;
|
||||
top: 0;
|
||||
left: 0;
|
||||
right: 0;
|
||||
bottom: 0;
|
||||
background: #f5f5f5;
|
||||
z-index: 9999;
|
||||
padding: 20px;
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.build-missing.show {
|
||||
display: block;
|
||||
}
|
||||
|
||||
.build-missing h1 {
|
||||
color: #333;
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
.build-missing p {
|
||||
color: #666;
|
||||
margin-bottom: 20px;
|
||||
line-height: 1.6;
|
||||
}
|
||||
|
||||
.build-missing .actions {
|
||||
display: flex;
|
||||
gap: 12px;
|
||||
justify-content: center;
|
||||
flex-wrap: wrap;
|
||||
}
|
||||
|
||||
.build-missing .btn {
|
||||
padding: 12px 24px;
|
||||
border: none;
|
||||
border-radius: 6px;
|
||||
cursor: pointer;
|
||||
text-decoration: none;
|
||||
font-size: 14px;
|
||||
transition: all 0.3s;
|
||||
}
|
||||
|
||||
.build-missing .btn-primary {
|
||||
background: #409eff;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.build-missing .btn-primary:hover {
|
||||
background: #337ecc;
|
||||
}
|
||||
|
||||
.build-missing .btn-secondary {
|
||||
background: #909399;
|
||||
color: white;
|
||||
}
|
||||
|
||||
.build-missing .btn-secondary:hover {
|
||||
background: #73767a;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div id="app"></div>
|
||||
|
||||
<!-- 构建文件缺失提示 -->
|
||||
<div id="build-missing" class="build-missing">
|
||||
<h1>🚀 TSP智能助手</h1>
|
||||
<p>
|
||||
前端构建文件不存在，请先构建前端应用。<br>
|
||||
或者使用传统HTML页面。
|
||||
</p>
|
||||
<div class="actions">
|
||||
<button class="btn btn-primary" onclick="buildFrontend()">
|
||||
📦 构建前端
|
||||
</button>
|
||||
<a href="/legacy" class="btn btn-secondary">
|
||||
📄 使用传统页面
|
||||
</a>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 脚本 -->
|
||||
<script>
|
||||
// 检查构建文件是否存在
|
||||
function checkBuildFiles() {
|
||||
const cssLink = document.querySelector('link[href="/static/dist/assets/index.css"]');
|
||||
const jsScript = document.createElement('script');
|
||||
jsScript.src = '/static/dist/assets/index.js';
|
||||
|
||||
jsScript.onerror = function() {
|
||||
document.getElementById('build-missing').classList.add('show');
|
||||
};
|
||||
|
||||
jsScript.onload = function() {
|
||||
document.getElementById('build-missing').classList.remove('show');
|
||||
};
|
||||
|
||||
document.head.appendChild(jsScript);
|
||||
}
|
||||
|
||||
// 构建前端
|
||||
function buildFrontend() {
|
||||
alert('请运行 build_frontend.bat 脚本构建前端应用');
|
||||
}
|
||||
|
||||
// 页面加载完成后检查
|
||||
document.addEventListener('DOMContentLoaded', checkBuildFiles);
|
||||
</script>
|
||||
|
||||
<!-- 尝试加载Vue应用 -->
|
||||
<script type="module" src="/static/dist/assets/index.js"></script>
|
||||
</body>
|
||||
</html>
|
||||
Reference in New Issue
Block a user