43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
from flask import Blueprint
|
|
|
|
from ..request_handler import before_request
|
|
from ..server_error import handle_server_error
|
|
from ... import opts
|
|
|
|
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"
|
|
"""
|
|
|
|
print('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
|