工具助手添加

2025-12-25 18:04:10 +08:00
parent f4bc9f6dab
commit 98081456b7
31 changed files with 4306 additions and 0 deletions
@@ -0,0 +1,156 @@
+"""
+AI助手Web客户端 - Flask应用
+100%还原docs/ai-assistant.html原型设计
+"""
+
+import os
+import sys
+import json
+import logging
+from flask import Flask, render_template, request, jsonify, Response
+from flask_cors import CORS
+
+# 添加父目录到路径，以便导入api_client
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+try:
+    from api_client import AIAssistantClient
+    from config import APPLICATIONS, AI_ASSISTANT_CONFIG
+except ImportError:
+    # 如果导入失败，使用模拟数据
+    AIAssistantClient = None
+    APPLICATIONS = {}
+    AI_ASSISTANT_CONFIG = {}
+
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# 创建Flask应用
+app = Flask(__name__)
+app.config['SECRET_KEY'] = 'ai-assistant-web-client-secret-key'
+CORS(app)  # 启用CORS
+
+# 初始化API客户端
+api_client = None
+if AIAssistantClient:
+    try:
+        api_client = AIAssistantClient()
+        logger.info("API客户端初始化成功")
+    except Exception as e:
+        logger.error(f"API客户端初始化失败: {e}")
+
+
+@app.route('/')
+def index():
+    """主页面"""
+    return render_template('index.html')
+
+
+@app.route('/api/applications', methods=['GET'])
+def get_applications():
+    """获取应用列表"""
+    try:
+        if api_client:
+            apps = api_client.get_app_list()
+            return jsonify({
+                'code': 200,
+                'message': 'success',
+                'data': apps
+            })
+        else:
+            # 返回模拟数据
+            return jsonify({
+                'code': 200,
+                'message': 'success',
+                'data': list(APPLICATIONS.values()) if APPLICATIONS else []
+            })
+    except Exception as e:
+        logger.error(f"获取应用列表失败: {e}")
+        return jsonify({
+            'code': 500,
+            'message': str(e),
+            'data': []
+        }), 500
+
+
+@app.route('/api/chat/send', methods=['POST'])
+def send_message():
+    """发送聊天消息"""
+    try:
+        data = request.get_json()
+        app_id = data.get('appId')
+        message = data.get('message')
+        chat_id = data.get('chatId')
+        stream = data.get('stream', False)
+        
+        if not message:
+            return jsonify({
+                'code': 400,
+                'message': '消息内容不能为空',
+                'data': None
+            }), 400
+        
+        if api_client and app_id:
+            # 调用真实API
+            response = api_client.send_message(
+                app_id=app_id,
+                message=message,
+                chat_id=chat_id,
+                stream=stream
+            )
+            
+            if stream:
+                # 流式响应
+                def generate():
+                    for chunk in response:
+                        yield f"data: {json.dumps(chunk, ensure_ascii=False)}\n\n"
+                    yield "data: [DONE]\n\n"
+                
+                return Response(generate(), mimetype='text/event-stream')
+            else:
+                # 普通响应
+                return jsonify(response)
+        else:
+            # 返回模拟响应
+            return jsonify({
+                'code': 200,
+                'message': 'success',
+                'data': {
+                    'answer': f'这是对"{message}"的模拟回复。实际使用时会调用真实的AI助手API。',
+                    'chatId': chat_id or 'mock_chat_id_001',
+                    'messageId': 'mock_message_id_001'
+                }
+            })
+    except Exception as e:
+        logger.error(f"发送消息失败: {e}")
+        return jsonify({
+            'code': 500,
+            'message': str(e),
+            'data': None
+        }), 500
+
+
+@app.route('/api/health', methods=['GET'])
+def health_check():
+    """健康检查"""
+    return jsonify({
+        'code': 200,
+        'message': 'OK',
+        'data': {
+            'status': 'healthy',
+            'api_client_available': api_client is not None
+        }
+    })
+
+
+if __name__ == '__main__':
+    port = int(os.environ.get('PORT', 5000))
+    debug = os.environ.get('DEBUG', 'True').lower() == 'true'
+    
+    logger.info(f"启动AI助手Web客户端，端口: {port}, 调试模式: {debug}")
+    app.run(host='0.0.0.0', port=port, debug=debug)
+