From b27ee68f734606c36196bc72121235d22d82dc5a Mon Sep 17 00:00:00 2001 From: larry-neil-1206 Date: Tue, 28 Jan 2025 06:47:43 -0700 Subject: [PATCH] feat: sentry sdk integration --- requirements.txt | 3 ++- services/image_generation_service.py | 5 +++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 55f2db8..99107e7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,4 +12,5 @@ transformers jinja2==3.1.0 python-dotenv==1.0.1 PyJWT==2.9.0 -stripe \ No newline at end of file +stripe +sentry-sdk[fastapi] \ No newline at end of file diff --git a/services/image_generation_service.py b/services/image_generation_service.py index 28b4e21..9dcee52 100644 --- a/services/image_generation_service.py +++ b/services/image_generation_service.py @@ -20,6 +20,11 @@ from utils.data_types import Prompt, TextPrompt, TextToImage, ImageToImage, ValidatorInfo, ChatCompletion from fastapi.middleware.cors import CORSMiddleware from transformers import AutoTokenizer +import sentry_sdk + +sentry_sdk.init( + dsn="https://c29179b17e940dd744ef5987e54076ff@o4508692980170752.ingest.us.sentry.io/4508721596334080", +) # Define a list of allowed origins (domains) allowed_origins = [