From 659b82b214dd95ee1043801ce9dc9dea3b4b968d Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Tue, 17 Sep 2024 16:00:17 +0200
Subject: [PATCH] log to sentry when one generation fails

---
 backend/routes/generate_code.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index d01c309..65a33a7 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -395,6 +395,10 @@ async def stream_code(websocket: WebSocket):
                     if isinstance(completion, Exception):
                         completions[index] = ""
                         print("Generation failed for variant", index)
+                        try:
+                            raise Exception("One of the generations failed")
+                        except:
+                            sentry_sdk.capture_exception()
 
                 print("Models used for generation: ", variant_models)