239 lines
7.3 KiB
Python
239 lines
7.3 KiB
Python
|
from flask import Flask, jsonify, request
|
|||
|
|
from flask_cors import CORS
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
app = Flask(__name__)
|
|||
|
|
CORS(app) # 允许跨域请求
|
|||
|
|
|
|||
|
|
# 模拟数据库中的模板数据
|
|||
|
|
MOCK_TEMPLATES = [
|
|||
|
|
{
|
|||
|
|
"id": "1",
|
|||
|
|
"name": "后端开发工程师",
|
|||
|
|
"description": "专业的后端开发工程师模板",
|
|||
|
|
"category": "软件开发",
|
|||
|
|
"industry": "互联网",
|
|||
|
|
"profession": "开发工程师",
|
|||
|
|
"sub_category": "后端开发",
|
|||
|
|
"system_prompt": "你是一位专业的后端开发工程师..."
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
"id": "2",
|
|||
|
|
"name": "前端开发工程师",
|
|||
|
|
"description": "专业的前端开发工程师模板",
|
|||
|
|
"category": "软件开发",
|
|||
|
|
"industry": "互联网",
|
|||
|
|
"profession": "开发工程师",
|
|||
|
|
"sub_category": "前端开发",
|
|||
|
|
"system_prompt": "你是一位专业的前端开发工程师..."
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
# 身份验证装饰器
|
|||
|
|
def require_auth(f):
|
|||
|
|
from functools import wraps
|
|||
|
|
@wraps(f)
|
|||
|
|
def decorated(*args, **kwargs):
|
|||
|
|
auth_header = request.headers.get('Authorization')
|
|||
|
|
if not auth_header or not auth_header.startswith('Bearer '):
|
|||
|
|
return jsonify({'error': 'No authorization token provided'}), 401
|
|||
|
|
# 这里简单验证token，实际应该更严格
|
|||
|
|
token = auth_header.split(' ')[1]
|
|||
|
|
if token != 'test_token_123':
|
|||
|
|
return jsonify({'error': 'Invalid token'}), 401
|
|||
|
|
return f(*args, **kwargs)
|
|||
|
|
return decorated
|
|||
|
|
|
|||
|
|
# 获取模板列表接口
|
|||
|
|
@app.route('/api/v1/templates', methods=['GET'])
|
|||
|
|
@require_auth
|
|||
|
|
def get_templates():
|
|||
|
|
# 获取查询参数
|
|||
|
|
page = int(request.args.get('page', 1))
|
|||
|
|
size = int(request.args.get('size', 10))
|
|||
|
|
category = request.args.get('category')
|
|||
|
|
industry = request.args.get('industry')
|
|||
|
|
profession = request.args.get('profession')
|
|||
|
|
|
|||
|
|
# 过滤模板
|
|||
|
|
filtered_templates = MOCK_TEMPLATES
|
|||
|
|
if category:
|
|||
|
|
filtered_templates = [t for t in filtered_templates if t['category'] == category]
|
|||
|
|
if industry:
|
|||
|
|
filtered_templates = [t for t in filtered_templates if t['industry'] == industry]
|
|||
|
|
if profession:
|
|||
|
|
filtered_templates = [t for t in filtered_templates if t['profession'] == profession]
|
|||
|
|
|
|||
|
|
# 计算分页
|
|||
|
|
start = (page - 1) * size
|
|||
|
|
end = start + size
|
|||
|
|
paginated_templates = filtered_templates[start:end]
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'total': len(filtered_templates),
|
|||
|
|
'pages': (len(filtered_templates) + size - 1) // size,
|
|||
|
|
'current_page': page,
|
|||
|
|
'templates': paginated_templates
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 获取单个模板接口
|
|||
|
|
@app.route('/api/v1/templates/<template_id>', methods=['GET'])
|
|||
|
|
@require_auth
|
|||
|
|
def get_template(template_id):
|
|||
|
|
template = next((t for t in MOCK_TEMPLATES if t['id'] == template_id), None)
|
|||
|
|
if not template:
|
|||
|
|
return jsonify({'error': 'Template not found'}), 404
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': template
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 创建模板接口
|
|||
|
|
@app.route('/api/v1/templates', methods=['POST'])
|
|||
|
|
@require_auth
|
|||
|
|
def create_template():
|
|||
|
|
data = request.get_json()
|
|||
|
|
|
|||
|
|
# 验证必要字段
|
|||
|
|
required_fields = ['name', 'description', 'category', 'industry', 'profession']
|
|||
|
|
for field in required_fields:
|
|||
|
|
if field not in data:
|
|||
|
|
return jsonify({'error': f'Missing required field: {field}'}), 400
|
|||
|
|
|
|||
|
|
# 创建新模板
|
|||
|
|
new_template = {
|
|||
|
|
'id': str(len(MOCK_TEMPLATES) + 1),
|
|||
|
|
**data,
|
|||
|
|
'created_at': datetime.now().isoformat()
|
|||
|
|
}
|
|||
|
|
MOCK_TEMPLATES.append(new_template)
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'id': new_template['id'],
|
|||
|
|
'message': 'Template created successfully'
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 更新模板接口
|
|||
|
|
@app.route('/api/v1/templates/<template_id>', methods=['PUT'])
|
|||
|
|
@require_auth
|
|||
|
|
def update_template(template_id):
|
|||
|
|
template = next((t for t in MOCK_TEMPLATES if t['id'] == template_id), None)
|
|||
|
|
if not template:
|
|||
|
|
return jsonify({'error': 'Template not found'}), 404
|
|||
|
|
|
|||
|
|
data = request.get_json()
|
|||
|
|
template.update(data)
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'message': 'Template updated successfully'
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 删除模板接口
|
|||
|
|
@app.route('/api/v1/templates/<template_id>', methods=['DELETE'])
|
|||
|
|
@require_auth
|
|||
|
|
def delete_template(template_id):
|
|||
|
|
template_index = next((i for i, t in enumerate(MOCK_TEMPLATES) if t['id'] == template_id), None)
|
|||
|
|
if template_index is None:
|
|||
|
|
return jsonify({'error': 'Template not found'}), 404
|
|||
|
|
|
|||
|
|
MOCK_TEMPLATES.pop(template_index)
|
|||
|
|
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'message': 'Template deleted successfully'
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 模糊搜索提示词模板接口
|
|||
|
|
@app.route('/api/v1/templates/search', methods=['GET'])
|
|||
|
|
@require_auth
|
|||
|
|
def search_templates():
|
|||
|
|
"""模糊搜索提示词模板"""
|
|||
|
|
# 获取搜索参数
|
|||
|
|
keyword = request.args.get('keyword', '').strip()
|
|||
|
|
page = int(request.args.get('page', 1))
|
|||
|
|
size = int(request.args.get('size', 10))
|
|||
|
|
|
|||
|
|
# 如果没有关键词，返回空结果
|
|||
|
|
if not keyword:
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'total': 0,
|
|||
|
|
'pages': 0,
|
|||
|
|
'current_page': page,
|
|||
|
|
'templates': []
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 执行模糊搜索
|
|||
|
|
# 搜索范围：名称、描述、分类、行业、职业、子分类、系统提示词
|
|||
|
|
search_results = []
|
|||
|
|
for template in MOCK_TEMPLATES:
|
|||
|
|
# 在各个字段中搜索关键词
|
|||
|
|
if any(keyword.lower() in str(value).lower() for value in [
|
|||
|
|
template.get('name', ''),
|
|||
|
|
template.get('description', ''),
|
|||
|
|
template.get('category', ''),
|
|||
|
|
template.get('industry', ''),
|
|||
|
|
template.get('profession', ''),
|
|||
|
|
template.get('sub_category', ''),
|
|||
|
|
template.get('system_prompt', '')
|
|||
|
|
]):
|
|||
|
|
# 添加匹配度信息
|
|||
|
|
match_score = sum(
|
|||
|
|
str(value).lower().count(keyword.lower())
|
|||
|
|
for value in [
|
|||
|
|
template.get('name', ''),
|
|||
|
|
template.get('description', ''),
|
|||
|
|
template.get('category', ''),
|
|||
|
|
template.get('industry', ''),
|
|||
|
|
template.get('profession', ''),
|
|||
|
|
template.get('sub_category', ''),
|
|||
|
|
template.get('system_prompt', '')
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
search_results.append({
|
|||
|
|
**template,
|
|||
|
|
'_match_score': match_score
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
# 按匹配度排序
|
|||
|
|
search_results.sort(key=lambda x: x['_match_score'], reverse=True)
|
|||
|
|
|
|||
|
|
# 移除匹配度信息
|
|||
|
|
search_results = [{k: v for k, v in t.items() if k != '_match_score'}
|
|||
|
|
for t in search_results]
|
|||
|
|
|
|||
|
|
# 计算分页
|
|||
|
|
total = len(search_results)
|
|||
|
|
total_pages = (total + size - 1) // size
|
|||
|
|
start = (page - 1) * size
|
|||
|
|
end = start + size
|
|||
|
|
paginated_results = search_results[start:end]
|
|||
|
|
|
|||
|
|
# 返回结果
|
|||
|
|
return jsonify({
|
|||
|
|
'code': 200,
|
|||
|
|
'data': {
|
|||
|
|
'total': total,
|
|||
|
|
'pages': total_pages,
|
|||
|
|
'current_page': page,
|
|||
|
|
'keyword': keyword,
|
|||
|
|
'templates': paginated_results
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
app.run(debug=True, port=5000)
|