From 7b08507b61e6ef402332a950a8fa3bc36e94f860 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Tue, 17 Sep 2024 16:56:09 +0200 Subject: [PATCH] capture underlying exception when only one of the generations fails --- backend/routes/generate_code.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 65a33a7..c71ceff 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -396,7 +396,9 @@ async def stream_code(websocket: WebSocket): completions[index] = "" print("Generation failed for variant", index) try: - raise Exception("One of the generations failed") + raise Exception( + "One of the generations failed" + ) from completion except: sentry_sdk.capture_exception()