diff --git a/src/memos/api/middleware/request_context.py b/src/memos/api/middleware/request_context.py index 71f069f8..e9c6db6d 100644 --- a/src/memos/api/middleware/request_context.py +++ b/src/memos/api/middleware/request_context.py @@ -2,18 +2,18 @@ Request context middleware for automatic trace_id injection. """ -import logging - from collections.abc import Callable from starlette.middleware.base import BaseHTTPMiddleware from starlette.requests import Request from starlette.responses import Response +import memos.log + from memos.context.context import RequestContext, generate_trace_id, set_request_context -logger = logging.getLogger(__name__) +logger = memos.log.get_logger(__name__) def extract_trace_id_from_headers(request: Request) -> str | None: diff --git a/src/memos/log.py b/src/memos/log.py index 339d13f2..2d64a931 100644 --- a/src/memos/log.py +++ b/src/memos/log.py @@ -183,7 +183,7 @@ def close(self): }, "root": { # Root logger handles all logs "level": logging.DEBUG if settings.DEBUG else logging.INFO, - "handlers": ["console", "file"], + "handlers": ["console", "file", "custom_logger"], }, "loggers": { "memos": {