inference-gateway/logging_config.py at main · auroraGPT-ANL/inference-gateway · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
import os

# Determine logging behaviour for different environments
environment = os.getenv("ENV", "production")
log_to_stdout = os.getenv("LOG_TO_STDOUT", "false").lower() in ("true", "1", "t")

handlers = {}
logger_handlers = {
    "uvicorn.error": None,
    "uvicorn.access": None,
    "resource_server_async": None,
    "utils": None,
}

if log_to_stdout:
    handlers["default"] = {
        "formatter": "default",
        "class": "logging.StreamHandler",
        "stream": "ext://sys.stderr",
    }
    handlers["access"] = {
        "formatter": "access",
        "class": "logging.StreamHandler",
        "stream": "ext://sys.stdout",
    }

    logger_handlers["uvicorn.error"] = ["default"]
    logger_handlers["uvicorn.access"] = ["access"]
    logger_handlers["resource_server_async"] = ["default"]
    logger_handlers["utils"] = ["default"]
    root_handlers = ["default"]
else:
    if environment == "development":
        accesslog = "./logs/backend_gateway.access.log"
        errorlog = "./logs/backend_gateway.error.log"
    else:
        accesslog = "/var/log/inference-service/backend_gateway.access.log"
        errorlog = "/var/log/inference-service/backend_gateway.error.log"

    handlers["default"] = {
        "formatter": "default",
        "class": "logging.FileHandler",
        "filename": errorlog,
    }
    handlers["access"] = {
        "formatter": "access",
        "class": "logging.FileHandler",
        "filename": accesslog,
    }
    handlers["console"] = {
        "formatter": "default",
        "class": "logging.StreamHandler",
    }

    logger_handlers["uvicorn.error"] = ["default"]
    logger_handlers["uvicorn.access"] = ["access"]
    logger_handlers["resource_server_async"] = ["default", "console"]
    logger_handlers["utils"] = ["default", "console"]
    root_handlers = ["default", "console"]


LOGGING_CONFIG = {
    "version": 1,
    "disable_existing_loggers": False,
    "formatters": {
        "default": {
            "()": "uvicorn.logging.DefaultFormatter",
            "fmt": "%(asctime)s - %(levelname)s - %(message)s",
            "datefmt": "%Y-%m-%d %H:%M:%S",
        },
        "access": {
            "()": "uvicorn.logging.AccessFormatter",
            "fmt": '%(asctime)s - %(client_addr)s - "%(request_line)s" %(status_code)s',
            "datefmt": "%Y-%m-%d %H:%M:%S",
        },
    },
    "handlers": handlers,
    "loggers": {
        "uvicorn.error": {
            "handlers": logger_handlers["uvicorn.error"],
            "level": "INFO",
        },
        "uvicorn.access": {
            "handlers": logger_handlers["uvicorn.access"],
            "level": "INFO",
            "propagate": False,
        },
        "resource_server_async": {
            "handlers": logger_handlers["resource_server_async"],
            "level": "INFO",
            "propagate": False,
        },
        "utils": {
            "handlers": logger_handlers["utils"],
            "level": "INFO",
            "propagate": False,
        },
    },
    "root": {
        "level": "INFO",
        "handlers": root_handlers,
    },
}