Spaces:
Running
Running
File size: 1,921 Bytes
7b4b748 | 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 | """Central logging setup for CLI and app entrypoints."""
from __future__ import annotations
import logging
import os
import warnings
ALLOWED_LEVELS = {
"WARNING": logging.WARNING,
"ERROR": logging.ERROR,
"CRITICAL": logging.CRITICAL,
}
QUIET_LOGGER_NAMES = (
"httpx",
"httpcore",
"urllib3",
"transformers",
"datasets",
"huggingface_hub",
"filelock",
"gradio",
"openai",
"langchain",
"langchain_core",
"langchain_openai",
"evaluation",
"assistants",
"llm",
"memory",
)
def configure_logging(level: str | None = None, *, fmt: str | None = None) -> None:
"""Configure app logging to show only WARNING and ERROR by default."""
os.environ.setdefault("HF_HUB_DISABLE_PROGRESS_BARS", "1")
requested = (level or os.getenv("LOG_LEVEL", "WARNING")).upper()
log_level = ALLOWED_LEVELS.get(requested, logging.WARNING)
logging.basicConfig(
level=log_level,
format=fmt or "%(asctime)s | %(levelname)s | %(name)s | %(message)s",
force=True,
)
logging.getLogger().setLevel(log_level)
for name in QUIET_LOGGER_NAMES:
logging.getLogger(name).setLevel(log_level)
try:
from datasets.utils.logging import set_verbosity_warning
set_verbosity_warning()
import datasets
datasets.disable_progress_bars()
except ImportError:
pass
try:
from transformers.utils.logging import set_verbosity_warning
set_verbosity_warning()
except ImportError:
pass
warnings.filterwarnings("ignore", category=FutureWarning, module="transformers")
try:
from langchain_core._api.deprecation import LangChainDeprecationWarning
warnings.filterwarnings(
"ignore",
category=LangChainDeprecationWarning,
message=r".*\.text\(\).*",
)
except ImportError:
pass
|