dify/api/extensions/ext_sentry.py

20 lines
866 B
Python
Raw Normal View History

import openai
2023-05-15 08:51:32 +08:00
import sentry_sdk
from langfuse import parse_error
2023-05-15 08:51:32 +08:00
from sentry_sdk.integrations.celery import CeleryIntegration
from sentry_sdk.integrations.flask import FlaskIntegration
from werkzeug.exceptions import HTTPException
def init_app(app):
if app.config.get("SENTRY_DSN"):
2023-05-15 08:51:32 +08:00
sentry_sdk.init(
dsn=app.config.get("SENTRY_DSN"),
integrations=[FlaskIntegration(), CeleryIntegration()],
ignore_errors=[HTTPException, ValueError, openai.APIStatusError, parse_error.defaultErrorResponse],
traces_sample_rate=app.config.get("SENTRY_TRACES_SAMPLE_RATE", 1.0),
profiles_sample_rate=app.config.get("SENTRY_PROFILES_SAMPLE_RATE", 1.0),
environment=app.config.get("DEPLOY_ENV"),
release=f"dify-{app.config.get('CURRENT_VERSION')}-{app.config.get('COMMIT_SHA')}",
2023-05-15 08:51:32 +08:00
)