fix: use stdout instead richHandler in loguru improve performance #7018 (#7024)

* Update py_autofix.yml

* fix: use stdout instead richHandler in loguru improve performance

* Revert "Update py_autofix.yml"

This reverts commit 7aed45e98602f57c2d6d33532080bad30cf02507.

* log to stdout use env vairable control

* feat:  env  LAGFLOW_PRETTY_LOGS

---------

Co-authored-by: tianzhipeng <tzpabc@gmail.com>
Co-authored-by: caojianwei-jk <caojianwei-jk@360shuke.com>
This commit is contained in:
jeevi cao 2025-03-27 21:13:01 +08:00 committed by GitHub
commit 858f2f3cd0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -231,17 +231,21 @@ def configure(
if log_format is None or not is_valid_log_format(log_format):
log_format = DEFAULT_LOG_FORMAT
# Configure loguru to use RichHandler
logger.configure(
handlers=[
{
"sink": RichHandler(rich_tracebacks=True, markup=True),
"format": log_format,
"level": log_level.upper(),
}
]
)
# pretty print to rich stdout development-friendly but poor performance, It's better for debugger.
# suggest directly print to stdout in production
log_stdout_pretty = os.getenv("LAGFLOW_PRETTY_LOGS", "true").lower() == "true"
if log_stdout_pretty:
logger.configure(
handlers=[
{
"sink": RichHandler(rich_tracebacks=True, markup=True),
"format": log_format,
"level": log_level.upper(),
}
]
)
else:
logger.add(sys.stdout, level=log_level.upper(), format=log_format, backtrace=True, diagnose=True)
if not log_file:
cache_dir = Path(user_cache_dir("langflow"))