ensure telemetry hits your server

confident-ai · Oct 12, 2023 · 8dc3214 · 8dc3214
1 parent 4dacaa5
commit 8dc3214
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/deepeval/__init__.py b/deepeval/__init__.py
@@ -13,7 +13,7 @@
 import re
 
 # Optionally add telemtry
-from . import telemetry
+from .telemetry import *
 from ._version import __version__
 
 

diff --git a/deepeval/_version.py b/deepeval/_version.py
@@ -1 +1 @@
-__version__: str = "0.20.5"
+__version__: str = "0.20.6"
diff --git a/deepeval/telemetry.py b/deepeval/telemetry.py
@@ -1,5 +1,6 @@
 import os
 import socket
+import sys
 
 
 def check_firewall():
@@ -24,6 +25,15 @@ def check_firewall():
             # We recommend adjusting this value in production.
             profiles_sample_rate=1.0,
         )
+
+        # Add a global error handler
+        def handle_exception(exc_type, exc_value, exc_traceback):
+            print({"exc_type": exc_type, "exc_value": exc_value})
+            sentry_sdk.capture_exception(exc_value)
+            sys.__excepthook__(exc_type, exc_value, exc_traceback)
+
+        sys.excepthook = handle_exception
+
     except ModuleNotFoundError:
         # sentry_sdk not installed
         pass