From 3d056fcb10b6b6d9eb24dd75493d86be19ca4d75 Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Tue, 30 Jul 2024 10:53:40 -0400
Subject: [PATCH] capture image generation errors in Sentry

---
 backend/image_generation/core.py | 5 +++++
 1 file changed, 5 insertions(+)

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