Skip to content

Commit a19584f

Browse files
author
harvey_xiang
committed
feat: add chat_time
1 parent 01547e1 commit a19584f

File tree

2 files changed

+12
-5
lines changed

2 files changed

+12
-5
lines changed

src/memos/api/middleware/request_context.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,13 +68,20 @@ async def dispatch(self, request: Request, call_next: Callable) -> Response:
6868
response = await call_next(request)
6969
end_time = time.time()
7070

71+
try:
72+
print(f"response.json(): {response}")
73+
response_json = response.json()
74+
except Exception as e:
75+
response_json = None
76+
logger.error(f"Error getting response body: {e}")
77+
7178
if response.status_code == 200:
7279
logger.info(
73-
f"Request completed: {request.url.path}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
80+
f"Request completed: {request.url.path}, response: {response_json}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
7481
)
7582
else:
7683
logger.error(
77-
f"Request Failed: {request.url.path}, response: {response.json()}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
84+
f"Request Failed: {request.url.path}, response: {response_json}, status: {response.status_code}, cost: {(end_time - start_time) * 1000:.2f}ms"
7885
)
7986

8087
return response

src/memos/log.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -171,13 +171,13 @@ def close(self):
171171
"disable_existing_loggers": False,
172172
"formatters": {
173173
"standard": {
174-
"format": "%(asctime)s | trace_id=%(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s |%(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(funcName)s - %(message)s"
174+
"format": "%(asctime)s | %(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s |%(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(funcName)s - %(message)s"
175175
},
176176
"no_datetime": {
177-
"format": "trace_id=%(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s | %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(funcName)s - %(message)s"
177+
"format": "%(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s | %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(funcName)s - %(message)s"
178178
},
179179
"simplified": {
180-
"format": "%(asctime)s | trace_id=%(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s | % %(levelname)s | %(filename)s:%(lineno)d: %(funcName)s | %(message)s"
180+
"format": "%(asctime)s | %(trace_id)s | env=%(env)s | user_type=%(user_type)s | user_name=%(user_name)s | % %(levelname)s | %(filename)s:%(lineno)d: %(funcName)s | %(message)s"
181181
},
182182
},
183183
"filters": {

0 commit comments

Comments
 (0)