diff --git a/model_service/app.py b/model_service/app.py index 82b38fa..2e4661b 100644 --- a/model_service/app.py +++ b/model_service/app.py @@ -74,8 +74,19 @@ app.add_url_rule("/healthz", "healthcheck", view_func=lambda: health.run()) if __name__ == "__main__": app.run(debug=True) else: - import logging + import os - gunicorn_logger = logging.getLogger('gunicorn.error') - app.logger.handlers = gunicorn_logger.handlers - app.logger.setLevel(gunicorn_logger.level) + if "gunicorn" in os.environ.get("SERVER_SOFTWARE", ""): + import logging + + gunicorn_logger = logging.getLogger('gunicorn.error') + app.logger.handlers = gunicorn_logger.handlers + app.logger.setLevel(gunicorn_logger.level) + + jaeger_logger = logging.getLogger('jaeger_tracing') + jaeger_logger.handlers = gunicorn_logger.handlers + jaeger_logger.setLevel(gunicorn_logger.level) + + app.logger.info("Gunicorn environment detected!") + else: + app.logger.info("Not gunicorn")