diff --git a/app/log_config.py b/app/log_config.py index e9ebd2f..fc53449 100644 --- a/app/log_config.py +++ b/app/log_config.py @@ -1,3 +1,4 @@ +import logging import sys from loguru import logger @@ -5,6 +6,14 @@ from app.settings import settings +class EndpointFilter(logging.Filter): + def __init__(self, excluded_endpoints: list[str]): + self.excluded_endpoints = excluded_endpoints + + def filter(self, record: logging.LogRecord): + return not any(ep in record.getMessage() for ep in self.excluded_endpoints) + + def _format(record): if "classname" in record["extra"]: keyname = record["extra"]["classname"] @@ -19,8 +28,11 @@ def _format(record): ) -def setup_logger(): - # Remove loguru default handler +def setup_logging(): + logging.getLogger("uvicorn.access").addFilter( + EndpointFilter(excluded_endpoints=["/health"]) + ) + logger.remove() logger.add( diff --git a/app/main.py b/app/main.py index c01fb91..c8ceb00 100644 --- a/app/main.py +++ b/app/main.py @@ -14,10 +14,10 @@ from app.agent.tools import BDToolkit from app.api.main import api_router from app.db.database import engine, init_database -from app.log_config import setup_logger +from app.log_config import setup_logging from app.settings import settings -setup_logger() +setup_logging() @asynccontextmanager diff --git a/charts/basedosdados-chatbot/templates/deployment.yaml b/charts/basedosdados-chatbot/templates/deployment.yaml index b3fdea2..98d6f73 100644 --- a/charts/basedosdados-chatbot/templates/deployment.yaml +++ b/charts/basedosdados-chatbot/templates/deployment.yaml @@ -61,7 +61,7 @@ spec: path: /health port: 8000 initialDelaySeconds: 10 - periodSeconds: 10 + periodSeconds: 30 readinessProbe: httpGet: path: /health