local-llm-server/llm_server/routes/v1/__init__.py

from flask import Blueprint, request

from ..helpers.http import require_api_key
from ..server_error import handle_server_error
from ... import opts

bp = Blueprint('v1', __name__)


@bp.before_request
def before_request():
    if not opts.http_host:
        opts.http_host = request.headers.get("Host")
    if not opts.base_client_api:
        opts.base_client_api = f'{request.headers.get("Host")}/{opts.frontend_api_client.strip("/")}'
    if request.endpoint != 'v1.get_stats':
        response = require_api_key()
        if response is not None:
            return response


@bp.errorhandler(500)
def handle_error(e):
    return handle_server_error(e)


from . import generate, info, proxy, generate_stream