feat: remove generate_trace_id to context

Author: harvey_xiang

src/memos/api/context/context.py

@@ -6,6 +6,7 @@
 and request isolation.
 """
 
+import os
 import uuid
 
 from collections.abc import Callable
@@ -117,6 +118,11 @@ def require_context() -> RequestContext:
 _trace_id_getter: TraceIdGetter | None = None
 
 
+def generate_trace_id() -> str:
+    """Generate a random trace_id."""
+    return os.urandom(16).hex()
+
+
 def set_trace_id_getter(getter: TraceIdGetter) -> None:
     """
     Set a custom trace_id getter function.

src/memos/api/middleware/request_context.py

@@ -3,25 +3,19 @@
 """
 
 import logging
-import os
 
 from collections.abc import Callable
 
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.requests import Request
 from starlette.responses import Response
 
-from memos.api.context.context import RequestContext, set_request_context
+from memos.api.context.context import RequestContext, generate_trace_id, set_request_context
 
 
 logger = logging.getLogger(__name__)
 
 
-def generate_trace_id() -> str:
-    """Generate a random trace_id."""
-    return os.urandom(16).hex()
-
-
 def extract_trace_id_from_headers(request: Request) -> str | None:
     """Extract trace_id from various possible headers with priority: g-trace-id > x-trace-id > trace-id."""
     trace_id = request.headers.get("g-trace-id")

src/memos/log.py

@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 
 from memos import settings
-from memos.api.context.context import get_current_trace_id
+from memos.api.context.context import generate_trace_id, get_current_trace_id
 
 
 # Load environment variables
@@ -33,9 +33,9 @@ class TraceIDFilter(logging.Filter):
     def filter(self, record):
         try:
             trace_id = get_current_trace_id()
-            record.trace_id = trace_id if trace_id else "no-trace-id"
+            record.trace_id = trace_id if trace_id else generate_trace_id()
         except Exception:
-            record.trace_id = "no-trace-id"
+            record.trace_id = generate_trace_id()
         return True