152 lines
5.5 KiB
Python
152 lines
5.5 KiB
Python
"""
|
|
AI模型 CRUD API
|
|
"""
|
|
import uuid
|
|
import json
|
|
from datetime import datetime
|
|
from flask import Blueprint, request, jsonify, Response
|
|
from config import MODELS_FILE
|
|
from utils import load_data, save_data, parse_date_to_timestamp, safe_sort_key
|
|
|
|
models_bp = Blueprint('api_models', __name__)
|
|
|
|
|
|
@models_bp.route('/api/models')
|
|
def api_models():
|
|
models = load_data(MODELS_FILE)
|
|
hide_hidden = request.args.get('all', '0') == '0'
|
|
if hide_hidden:
|
|
models = [m for m in models if m.get('visible', True)]
|
|
keyword = request.args.get('q', '').strip().lower()
|
|
if keyword:
|
|
models = [m for m in models if keyword in m.get('name', '').lower() or
|
|
keyword in m.get('organization', '').lower()]
|
|
sort_by = request.args.get('sort', 'default')
|
|
reverse = request.args.get('order', 'desc') == 'desc'
|
|
if sort_by == 'default':
|
|
models = sorted(models, key=lambda x: (
|
|
not x.get('is_pinned', False),
|
|
-(parse_date_to_timestamp(x.get('publish_date', '')) or
|
|
parse_date_to_timestamp(x.get('created_at', '')) or 0)
|
|
))
|
|
elif sort_by in ['name', 'parameters', 'context_length', 'mmlu', 'created_at',
|
|
'publish_date', 'views', 'updated_at']:
|
|
models = sorted(models, key=lambda x: safe_sort_key(x, sort_by), reverse=reverse)
|
|
return jsonify(models)
|
|
|
|
|
|
@models_bp.route('/api/models/<model_id>')
|
|
def api_model_detail(model_id):
|
|
models = load_data(MODELS_FILE)
|
|
model = next((m for m in models if m['id'] == model_id), None)
|
|
if not model:
|
|
return jsonify({'error': 'Model not found'}), 404
|
|
return jsonify(model)
|
|
|
|
|
|
@models_bp.route('/api/models', methods=['POST'])
|
|
def api_create_model():
|
|
data = request.get_json()
|
|
models = load_data(MODELS_FILE)
|
|
data['id'] = uuid.uuid4().hex[:12]
|
|
data['created_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
data['visible'] = data.get('visible', True)
|
|
data['publish_date'] = data.get('publish_date', '')
|
|
data['views'] = data.get('views', 0)
|
|
data['is_pinned'] = data.get('is_pinned', False)
|
|
models.append(data)
|
|
save_data(MODELS_FILE, models)
|
|
return jsonify(data)
|
|
|
|
|
|
@models_bp.route('/api/models/<model_id>', methods=['PUT'])
|
|
def api_update_model(model_id):
|
|
data = request.get_json()
|
|
models = load_data(MODELS_FILE)
|
|
model = next((m for m in models if m['id'] == model_id), None)
|
|
if not model:
|
|
return jsonify({'error': 'Model not found'}), 404
|
|
model.update(data)
|
|
model['updated_at'] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
|
save_data(MODELS_FILE, models)
|
|
return jsonify(model)
|
|
|
|
|
|
@models_bp.route('/api/models/<model_id>', methods=['DELETE'])
|
|
def api_delete_model(model_id):
|
|
models = load_data(MODELS_FILE)
|
|
models = [m for m in models if m['id'] != model_id]
|
|
save_data(MODELS_FILE, models)
|
|
return jsonify({'success': True})
|
|
|
|
|
|
@models_bp.route('/api/models/<model_id>/visible', methods=['POST'])
|
|
def api_toggle_model_visible(model_id):
|
|
models = load_data(MODELS_FILE)
|
|
model = next((m for m in models if m['id'] == model_id), None)
|
|
if not model:
|
|
return jsonify({'error': 'Model not found'}), 404
|
|
model['visible'] = not model.get('visible', True)
|
|
save_data(MODELS_FILE, models)
|
|
return jsonify({'success': True, 'visible': model['visible']})
|
|
|
|
|
|
# ─── 导出导入 ───────────────────────────────────────────────────────
|
|
|
|
@models_bp.route('/api/models/export', methods=['GET'])
|
|
def api_export_models():
|
|
"""导出所有模型数据"""
|
|
try:
|
|
models = load_data(MODELS_FILE)
|
|
export_data = {
|
|
'type': 'models',
|
|
'items': models,
|
|
'count': len(models),
|
|
'export_time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
|
|
'version': '1.0'
|
|
}
|
|
json_str = json.dumps(export_data, ensure_ascii=False, indent=2)
|
|
response = Response(
|
|
json_str,
|
|
mimetype='application/json',
|
|
headers={'Content-Disposition': f'attachment; filename=models-export-{datetime.now().strftime("%Y%m%d%H%M%S")}.json'}
|
|
)
|
|
return response
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|
|
|
|
|
|
@models_bp.route('/api/models/import', methods=['GET', 'POST'])
|
|
def api_import_models():
|
|
"""导入模型数据"""
|
|
try:
|
|
if request.method == 'GET':
|
|
return jsonify({'endpoint': '/api/models/import', 'method': 'POST', 'params': {'mode': 'merge 或 replace'}})
|
|
|
|
import_data = request.get_json()
|
|
if not import_data or 'items' not in import_data:
|
|
return jsonify({'error': '缺少 items 字段'}), 400
|
|
|
|
mode = request.args.get('mode', 'merge')
|
|
models = load_data(MODELS_FILE)
|
|
imported_items = import_data['items']
|
|
|
|
result = {'success': True, 'imported': 0, 'updated': 0, 'skipped': []}
|
|
|
|
for item in imported_items:
|
|
existing = next((m for m in models if m['id'] == item['id']), None)
|
|
if existing:
|
|
if mode == 'replace':
|
|
existing.update(item)
|
|
result['updated'] += 1
|
|
else:
|
|
result['skipped'].append(item['id'])
|
|
else:
|
|
models.append(item)
|
|
result['imported'] += 1
|
|
|
|
save_data(MODELS_FILE, models)
|
|
return jsonify(result)
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|