from flask import Blueprint

from ..request_handler import before_request
from ... import opts
from ...logging import create_logger

_logger = create_logger('OpenAI')

openai_bp = Blueprint('openai/v1/', __name__)
openai_model_bp = Blueprint('openai/', __name__)


@openai_bp.before_request
@openai_model_bp.before_request
def before_oai_request():
    if not opts.enable_openi_compatible_backend:
        return 'The OpenAI-compatible backend is disabled.', 401
    return before_request()


@openai_bp.errorhandler(500)
@openai_model_bp.errorhandler(500)
def handle_error(e):
    """
    Found Codes:
    "auth_subrequest_error"
    """

    _logger(f'OAI returning error: {e}')
    return jsonify({
        "error": {
            "message": "Internal server error",
            "type": "auth_subrequest_error",
            "param": None,
            "code": "internal_error"
        }
    }), 500


from .models import openai_list_models
from .chat_completions import openai_chat_completions
from .info import get_openai_info
from .simulated import *
from .completions import openai_completions