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

capture image generation errors in Sentry

Browse Source
This commit is contained in:
Abi Raja 2024-07-30 10:53:40 -04:00
parent 2d877c0d84
commit 3d056fcb10
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/image_generation/core.py
View File

@ -3,6 +3,7 @@ import re
from typing import Dict, List, Literal, Union from typing import Dict, List, Literal, Union
from openai import AsyncOpenAI from openai import AsyncOpenAI
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
import sentry_sdk
from image_generation.replicate import call_replicate from image_generation.replicate import call_replicate
@ -29,6 +30,10 @@ async def process_tasks(
for result in results: for result in results:
if isinstance(result, Exception): if isinstance(result, Exception):
print(f"An exception occurred: {result}") print(f"An exception occurred: {result}")
try:
raise result
except Exception:
sentry_sdk.capture_exception()
processed_results.append(None) processed_results.append(None)
else: else:
processed_results.append(result) processed_results.append(result)