314 lines
11 KiB
Python
Executable File
314 lines
11 KiB
Python
Executable File
from flask import Blueprint, render_template, request, jsonify
|
||
import requests
|
||
import json
|
||
import logging
|
||
|
||
feature_engineering_bp = Blueprint('feature_engineering', __name__)
|
||
|
||
# 配置日志
|
||
logging.basicConfig(level=logging.INFO)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
@feature_engineering_bp.route('/')
|
||
def feature_engineering():
|
||
"""特征工程主页面"""
|
||
return render_template('feature_engineering.html')
|
||
|
||
@feature_engineering_bp.route('/api/test-deepseek', methods=['POST'])
|
||
def test_deepseek_api():
|
||
"""测试 Deepseek API 连接"""
|
||
try:
|
||
api_key = request.headers.get('X-API-Key')
|
||
if not api_key:
|
||
return jsonify({'success': False, 'error': '需要提供 API 密钥'}), 400
|
||
|
||
# 通过简单请求测试 API
|
||
headers = {
|
||
'Authorization': f'Bearer {api_key}',
|
||
'Content-Type': 'application/json'
|
||
}
|
||
|
||
test_data = {
|
||
'model': 'deepseek-chat',
|
||
'messages': [
|
||
{'role': 'user', 'content': '你好，这是一条测试消息。'}
|
||
],
|
||
'max_tokens': 10
|
||
}
|
||
|
||
response = requests.post(
|
||
'https://api.deepseek.com/chat/completions ',
|
||
headers=headers,
|
||
json=test_data,
|
||
timeout=10
|
||
)
|
||
|
||
if response.status_code == 200:
|
||
return jsonify({'success': True, 'message': 'API 连接成功'})
|
||
else:
|
||
error_detail = response.text
|
||
return jsonify({'success': False, 'error': f'API 返回状态 {response.status_code}：{error_detail}'}), 400
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
logger.error(f"API 测试出错：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'网络错误：{str(e)}'}), 500
|
||
except Exception as e:
|
||
logger.error(f"API 测试中出现意外错误：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'未知错误：{str(e)}'}), 500
|
||
|
||
@feature_engineering_bp.route('/api/get-recommendations', methods=['POST'])
|
||
def get_feature_engineering_recommendations():
|
||
"""从 Deepseek API 获取特征工程建议"""
|
||
try:
|
||
api_key = request.headers.get('X-API-Key')
|
||
if not api_key:
|
||
return jsonify({'success': False, 'error': '需要提供 API 密钥'}), 400
|
||
|
||
data = request.get_json()
|
||
current_step = data.get('current_step', 1)
|
||
data_field = data.get('data_field', '')
|
||
previous_steps = data.get('previous_steps', [])
|
||
current_data_state = data.get('current_data_state', '原始数据')
|
||
|
||
if not data_field:
|
||
return jsonify({'success': False, 'error': '请提供数据字段描述'}), 400
|
||
|
||
# 构建系统提示
|
||
system_prompt = get_default_system_prompt_text()
|
||
|
||
# 构建用户提示
|
||
previous_steps_text = "无" if not previous_steps else "，".join([f"步骤 {i+1}：{step}" for i, step in enumerate(previous_steps)])
|
||
|
||
user_prompt = f"""上下文：
|
||
当前步骤：{current_step}
|
||
当前数据字段：{data_field}
|
||
之前的步骤及所用类别：{previous_steps_text}
|
||
当前数据状态：{current_data_state}"""
|
||
|
||
# 调用 Deepseek API
|
||
headers = {
|
||
'Authorization': f'Bearer {api_key}',
|
||
'Content-Type': 'application/json'
|
||
}
|
||
|
||
api_data = {
|
||
'model': 'deepseek-chat',
|
||
'messages': [
|
||
{'role': 'system', 'content': system_prompt},
|
||
{'role': 'user', 'content': user_prompt}
|
||
],
|
||
'max_tokens': 8192,
|
||
'temperature': 0.7
|
||
}
|
||
|
||
response = requests.post(
|
||
'https://api.deepseek.com/chat/completions ',
|
||
headers=headers,
|
||
json=api_data,
|
||
timeout=30
|
||
)
|
||
|
||
if response.status_code == 200:
|
||
response_data = response.json()
|
||
recommendations = response_data['choices'][0]['message']['content']
|
||
|
||
return jsonify({
|
||
'success': True,
|
||
'recommendations': recommendations,
|
||
'current_step': current_step,
|
||
'data_field': data_field,
|
||
'previous_steps': previous_steps,
|
||
'current_data_state': current_data_state
|
||
})
|
||
else:
|
||
error_detail = response.text
|
||
logger.error(f"Deepseek API 错误：{response.status_code} - {error_detail}")
|
||
return jsonify({'success': False, 'error': f'API 返回状态 {response.status_code}：{error_detail}'}), 400
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
logger.error(f"API 请求出错：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'网络错误：{str(e)}'}), 500
|
||
except Exception as e:
|
||
logger.error(f"获取建议时出现意外错误：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'未知错误：{str(e)}'}), 500
|
||
|
||
@feature_engineering_bp.route('/api/continue-conversation', methods=['POST'])
|
||
def continue_conversation():
|
||
"""继续对话，回答追问"""
|
||
try:
|
||
api_key = request.headers.get('X-API-Key')
|
||
if not api_key:
|
||
return jsonify({'success': False, 'error': '需要提供 API 密钥'}), 400
|
||
|
||
data = request.get_json()
|
||
conversation_history = data.get('conversation_history', [])
|
||
user_message = data.get('user_message', '')
|
||
custom_system_prompt = data.get('custom_system_prompt', None)
|
||
|
||
if not user_message:
|
||
return jsonify({'success': False, 'error': '请输入您的消息'}), 400
|
||
|
||
# 构建对话消息
|
||
messages = []
|
||
|
||
# 若提供了自定义系统提示则使用，否则用默认
|
||
if custom_system_prompt:
|
||
system_prompt = custom_system_prompt
|
||
else:
|
||
system_prompt = get_default_system_prompt_text()
|
||
|
||
messages.append({'role': 'system', 'content': system_prompt})
|
||
|
||
# 加入历史对话
|
||
for msg in conversation_history:
|
||
messages.append(msg)
|
||
|
||
# 加入新用户消息
|
||
messages.append({'role': 'user', 'content': user_message})
|
||
print(user_message)
|
||
# 调用 Deepseek API
|
||
headers = {
|
||
'Authorization': f'Bearer {api_key}',
|
||
'Content-Type': 'application/json'
|
||
}
|
||
|
||
api_data = {
|
||
'model': 'deepseek-chat',
|
||
'messages': messages,
|
||
'max_tokens': 8192,
|
||
'temperature': 0.7
|
||
}
|
||
|
||
response = requests.post(
|
||
'https://api.deepseek.com/chat/completions ',
|
||
headers=headers,
|
||
json=api_data,
|
||
timeout=30
|
||
)
|
||
|
||
if response.status_code == 200:
|
||
response_data = response.json()
|
||
assistant_response = response_data['choices'][0]['message']['content']
|
||
|
||
return jsonify({
|
||
'success': True,
|
||
'response': assistant_response
|
||
})
|
||
else:
|
||
error_detail = response.text
|
||
logger.error(f"Deepseek API 错误：{response.status_code} - {error_detail}")
|
||
return jsonify({'success': False, 'error': f'API 返回状态 {response.status_code}：{error_detail}'}), 400
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
logger.error(f"API 请求出错：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'网络错误：{str(e)}'}), 500
|
||
except Exception as e:
|
||
logger.error(f"继续对话时出现意外错误：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'未知错误：{str(e)}'}), 500
|
||
|
||
def get_default_system_prompt_text():
|
||
"""获取默认系统提示文本"""
|
||
return """你是一位特征工程专家助手。你的任务是针对给定的数据字段，设计一个最多包含 6 步的特征工程流水线。每一步都要根据当前数据状态、已完成的步骤以及总体目标，从 15 个类别中推荐最合适的特征工程类别。
|
||
|
||
说明：
|
||
|
||
每一步你都会收到：
|
||
|
||
当前步骤序号。
|
||
|
||
当前数据字段及其描述。
|
||
|
||
已完成的步骤及所用类别（如有）。
|
||
|
||
当前数据状态（例如：已归一化、已过滤等）。
|
||
|
||
你的任务是：
|
||
|
||
列出当前步骤最可行的特征工程类别，从以下 15 类中选择：
|
||
|
||
基础算术与数学运算
|
||
|
||
逻辑与条件运算
|
||
|
||
时间序列：变化检测与值比较
|
||
|
||
时间序列：统计特征工程
|
||
|
||
时间序列：排序、缩放与归一化
|
||
|
||
时间序列：衰减、平滑与翻转控制
|
||
|
||
时间序列：极值与位置识别
|
||
|
||
横截面：排序、缩放与归一化
|
||
|
||
横截面：回归与中性化
|
||
|
||
横截面：分布变换与截断
|
||
|
||
变换与过滤操作
|
||
|
||
分组聚合与统计摘要
|
||
|
||
分组排序、缩放与归一化
|
||
|
||
分组回归与中性化
|
||
|
||
分组插补与回填
|
||
|
||
对于每个推荐的类别，请按以下格式呈现：
|
||
|
||
重复完整上下文。
|
||
|
||
明确指出下一步选择的类别。
|
||
|
||
给出简要选择原因。
|
||
|
||
输出格式：
|
||
|
||
第 X 步可行类别：
|
||
|
||
第 X 步 选项 1：上下文：当前步骤：[序号] 当前数据字段：[描述] 之前步骤及所用类别：[列表] 当前数据状态：[详细描述之前步骤如何将数据转换到当前状态及其逻辑] 下一步选择：[类别名称] 原因：[解释]
|
||
|
||
第 X 步 选项 2：上下文：当前步骤：[序号] 当前数据字段：[描述] 之前步骤及所用类别：[列表] 当前数据状态：[详细描述之前步骤如何将数据转换到当前状态及其逻辑] 下一步选择：[类别名称] 原因：[解释]
|
||
|
||
...（继续列出所有可行选项）
|
||
|
||
额外说明：
|
||
|
||
仅推荐在当前数据状态及已完成的步骤下逻辑合理且有意义的类别。
|
||
|
||
若某类别在此步不适用，请勿列出。
|
||
|
||
解释需简洁明了。
|
||
|
||
除非特别要求，否则不要给出具体操作符。
|
||
|
||
每一步你都会收到如下输入：
|
||
|
||
上下文：
|
||
|
||
当前步骤：
|
||
|
||
当前数据字段：
|
||
|
||
之前步骤及所用类别：
|
||
|
||
当前数据状态：
|
||
|
||
收到输入后，请按上述格式回复。
|
||
重要：选项之后不要添加任何总结、建议、理由或其他解释。严格按照上述格式提供选项即可。不要添加“最推荐选择”、“理由”、“最佳选择”或“是否继续”等部分。列出所有选项后立即结束。
|
||
"""
|
||
|
||
@feature_engineering_bp.route('/api/get-default-system-prompt', methods=['GET'])
|
||
def get_default_system_prompt():
|
||
"""获取默认系统提示"""
|
||
try:
|
||
return jsonify({
|
||
'success': True,
|
||
'default_system_prompt': get_default_system_prompt_text()
|
||
})
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取默认系统提示出错：{str(e)}")
|
||
return jsonify({'success': False, 'error': f'未知错误：{str(e)}'}), 500 |