File size: 1,591 Bytes
217acfe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
from flask import Blueprint, jsonify, request
setting_bp = Blueprint('setting', __name__)
@setting_bp.route('/setting', methods=['GET'])
def get_settings():
"""Get current settings and models"""
from config import API_SETTINGS, DEFAULT_MAIN_MODEL, DEFAULT_SUB_MODEL, MAX_THREAD_NUM, MAX_NOVEL_SUMMARY_LENGTH
# Get models grouped by provider
models = {provider: config['available_models'] for provider, config in API_SETTINGS.items() if 'available_models' in config}
# Combine all settings
settings = {
'models': models,
'MAIN_MODEL': DEFAULT_MAIN_MODEL,
'SUB_MODEL': DEFAULT_SUB_MODEL,
'MAX_THREAD_NUM': MAX_THREAD_NUM,
'MAX_NOVEL_SUMMARY_LENGTH': MAX_NOVEL_SUMMARY_LENGTH,
}
return jsonify(settings)
@setting_bp.route('/test_model', methods=['POST'])
def test_model():
"""Test if a model configuration works"""
try:
data = request.get_json()
provider_model = data.get('provider_model')
from backend_utils import get_model_config_from_provider_model
model_config = get_model_config_from_provider_model(provider_model)
from llm_api import test_stream_chat
response = None
for msg in test_stream_chat(model_config):
response = msg
return jsonify({
'success': True,
'response': response
})
except Exception as e:
return jsonify({
'success': False,
'error': str(e)
}), 500
|