Skip to content

Commit 806e581

Browse files
authored
Add log statements when we 400, to make it clearer what's happening. (#729)
1 parent 8901a01 commit 806e581

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

pyproject.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "truss"
3-
version = "0.7.16"
3+
version = "0.7.17rc1"
44
description = "A seamless bridge from model development to model delivery"
55
license = "MIT"
66
readme = "README.md"

truss/templates/server/common/truss_server.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,9 @@ async def parse_body(request: Request) -> bytes:
4545
try:
4646
return await request.body()
4747
except ClientDisconnect as exc:
48-
raise HTTPException(status_code=499, detail="Client disconnected") from exc
48+
error_message = "Client disconnected"
49+
logging.error(error_message)
50+
raise HTTPException(status_code=499, detail=error_message) from exc
4951

5052

5153
FORMAT = "%(asctime)s.%(msecs)03d %(name)s %(levelname)s [%(funcName)s():%(lineno)s] %(message)s"
@@ -135,9 +137,9 @@ async def predict(
135137
try:
136138
body = json.loads(body_raw)
137139
except json.JSONDecodeError as e:
138-
raise HTTPException(
139-
status_code=400, detail=f"Invalid JSON payload: {str(e)}"
140-
)
140+
error_message = f"Invalid JSON payload: {str(e)}"
141+
logging.error(error_message)
142+
raise HTTPException(status_code=400, detail=error_message)
141143

142144
# calls ModelWrapper.__call__, which runs validate, preprocess, predict, and postprocess
143145
response: Union[Dict, Generator] = await model(

0 commit comments

Comments
 (0)