diff --git a/utils/log_util/logger.py b/utils/log_util/logger.py index fc168c6..2d74148 100644 --- a/utils/log_util/logger.py +++ b/utils/log_util/logger.py @@ -15,10 +15,10 @@ def add_request_routes(app: FastAPI): @app.middleware("http") async def before_request(request: Request, call_next: Callable[[Request], Awaitable[JSONResponse]]): - # 先从header中获取X-Trace-Id,如果没有则生成新的 - trace_id = request.headers.get('X-Trace-Id') + # 先从header中获取x-request-id,如果没有则生成新的 + trace_id = request.headers.get('X-Request-ID') if not trace_id: - trace_id = "generate_" + str(uuid.uuid4()) + trace_id = "agent_" + str(uuid.uuid4()) # user_id = "未知的 user_id" @@ -26,7 +26,7 @@ def add_request_routes(app: FastAPI): # g.user_id = user_id response = await call_next(request) - response.headers['X-Trace-Id'] = g.trace_id + response.headers['X-Request-ID'] = g.trace_id return response