main.py

import argparse
import logging
import uvicorn

from gate import AIBudgetGate
from server import create_app
from utils import load_config

def main() -> None:
    parser = argparse.ArgumentParser(description = "AI API proxy with budget control")
    parser.add_argument("--settings", default = "settings.yaml")
    args = parser.parse_args()

    logging_config = load_config("logging.yaml")
    logging.config.dictConfig(logging_config)

    settings_config = load_config(args.settings)
    gate = AIBudgetGate(settings_config)
    gate.start_background_sync()

    app = create_app(gate)
    uvicorn.run(
        app,
        host = settings_config["server"]["host"],
        port = settings_config["server"]["port"],
        log_config = logging_config
    )

if __name__ == "__main__":
    main()