Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

log to sentry when one generation fails

Browse Source
This commit is contained in:
Abi Raja 2024-09-17 16:00:17 +02:00
parent b41f3c65c7
commit 659b82b214
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/routes/generate_code.py
View File

@ -395,6 +395,10 @@ async def stream_code(websocket: WebSocket):
if isinstance(completion, Exception): if isinstance(completion, Exception):
completions[index] = "" completions[index] = ""
print("Generation failed for variant", 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) print("Models used for generation: ", variant_models)