Skip to content
Merged
6 changes: 3 additions & 3 deletions src/memos/api/middleware/request_context.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
2 changes: 1 addition & 1 deletion src/memos/log.py
Original file line number Diff line number Diff line change
Expand Up @@ -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": {
Expand Down
Loading