81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
|
"""
|
|||
|
|
Text2SQL API 仓库级日志：控制台 + 单一磁盘文件（每次进程启动清空 logs/*.log 后重写）。
|
|||
|
|
|
|||
|
|
由 ``api_server`` 在导入早期调用；与 ``uvicorn.run(..., log_config=False)`` 配合，
|
|||
|
|
使 uvicorn / FastAPI 的日志经 root 统一落到文件。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
def configure_text2sql_api_logging(repo_root: Path) -> Path:
|
|||
|
|
"""
|
|||
|
|
配置 root logger：控制台（INFO，简短格式）+ ``logs/text2sql_api.log``（INFO，含文件名/行号/函数）。
|
|||
|
|
|
|||
|
|
每次调用会删除 ``logs`` 目录下所有 ``*.log``，再新建 ``text2sql_api.log``（覆盖写），
|
|||
|
|
保证一次运行仅保留一个日志文件。
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
主日志文件绝对路径。
|
|||
|
|
"""
|
|||
|
|
log_dir = (repo_root / "logs").resolve()
|
|||
|
|
log_dir.mkdir(parents=True, exist_ok=True)
|
|||
|
|
for p in log_dir.glob("*.log"):
|
|||
|
|
try:
|
|||
|
|
p.unlink()
|
|||
|
|
except OSError:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
log_path = log_dir / "text2sql_api.log"
|
|||
|
|
|
|||
|
|
root = logging.getLogger()
|
|||
|
|
root.setLevel(logging.INFO)
|
|||
|
|
for h in list(root.handlers):
|
|||
|
|
root.removeHandler(h)
|
|||
|
|
try:
|
|||
|
|
h.close()
|
|||
|
|
except Exception:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
fmt_console = logging.Formatter(
|
|||
|
|
"%(asctime)s [%(levelname)s] %(name)s: %(message)s",
|
|||
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|||
|
|
)
|
|||
|
|
fmt_file = logging.Formatter(
|
|||
|
|
"%(asctime)s %(levelname)-7s [%(name)s] %(filename)s:%(lineno)d %(funcName)s() | %(message)s",
|
|||
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
fh = logging.FileHandler(log_path, mode="w", encoding="utf-8")
|
|||
|
|
fh.setLevel(logging.INFO)
|
|||
|
|
fh.setFormatter(fmt_file)
|
|||
|
|
|
|||
|
|
ch = logging.StreamHandler()
|
|||
|
|
ch.setLevel(logging.INFO)
|
|||
|
|
ch.setFormatter(fmt_console)
|
|||
|
|
|
|||
|
|
root.addHandler(fh)
|
|||
|
|
root.addHandler(ch)
|
|||
|
|
|
|||
|
|
# 依赖库降噪（避免 httpx 每条请求刷屏；Chroma 遥测与新版 posthog 不兼容时会打 ERROR）
|
|||
|
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
|||
|
|
logging.getLogger("httpcore").setLevel(logging.WARNING)
|
|||
|
|
for _name in (
|
|||
|
|
"chromadb.telemetry",
|
|||
|
|
"chromadb.telemetry.product.posthog",
|
|||
|
|
"posthog",
|
|||
|
|
):
|
|||
|
|
logging.getLogger(_name).disabled = True
|
|||
|
|
|
|||
|
|
# Uvicorn 默认自带 handler；与 log_config=False 合用时改为只往 root 冒泡
|
|||
|
|
for name in ("uvicorn", "uvicorn.error", "uvicorn.access"):
|
|||
|
|
lg = logging.getLogger(name)
|
|||
|
|
lg.handlers.clear()
|
|||
|
|
lg.propagate = True
|
|||
|
|
|
|||
|
|
logging.getLogger(__name__).info("日志文件: %s", log_path)
|
|||
|
|
return log_path
|