|
|
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
|
|
|
|
|
|
|
|
|
models = {provider: config['available_models'] for provider, config in API_SETTINGS.items() if 'available_models' in config}
|
|
|
|
|
|
|
|
|
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
|
|
|
|
