From 863c414bb267fdd3feba1284a78d5637a0bf736f Mon Sep 17 00:00:00 2001
From: Stephan Fitzpatrick <knowsuchagency@gmail.com>
Date: Sun, 9 Jun 2024 19:16:59 -0700
Subject: [PATCH] feat: Add Sentry error tracking for improved error handling

---
 main.py          | 3 +++
 requirements.txt | 1 +
 2 files changed, 4 insertions(+)

diff --git a/main.py b/main.py
index bed97a1..4d22f33 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,7 @@
 from typing import List, Literal
 
 import gradio as gr
+import sentry_sdk
 from loguru import logger
 from openai import OpenAI
 from promptic import llm
@@ -12,6 +13,8 @@
 from pypdf import PdfReader
 from tenacity import retry, retry_if_exception_type
 
+sentry_sdk.init(os.getenv("SENTRY_DSN"))
+
 
 class DialogueItem(BaseModel):
     text: str
diff --git a/requirements.txt b/requirements.txt
index 90bf3d0..2362678 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ google-generativeai~=0.6
 loguru~=0.7
 pypdf~=4.1
 tenacity~=8.3
+sentry-sdk~=2.5