GenAIImpactError implem

genai_impact/__init__.py

@@ -1,3 +1,3 @@
-from .tracer import Tracer, TracerInitializationError
+from .tracer import Tracer
 
-__all__ = ["Tracer", "TracerInitializationError"]
+__all__ = ["Tracer"]

genai_impact/exceptions.py

@@ -0,0 +1,6 @@
+class GenAIImpactError(Exception):
+    pass
+
+class TracerInitializationError(GenAIImpactError):
+    "Tracer is initialized twice"
+    pass

genai_impact/tracer.py

@@ -1,5 +1,7 @@
 import importlib.util
 
+from genai_impact.exceptions import TracerInitializationError
+
 
 class Tracer:
     initialized = False
@@ -38,7 +40,3 @@ def init_mistralai_instrumentor() -> None:
 
         instrumentor = MistralAIInstrumentor()
         instrumentor.instrument()
-
-class TracerInitializationError(Exception):
-    "Tracer is initialized twice"
-    pass

tests/test_tracer.py

@@ -1,6 +1,7 @@
 import pytest
-from genai_impact import Tracer, TracerInitializationError
+from genai_impact import Tracer
+from genai_impact.exceptions import GenAIImpactError
 
 def test_double_init(tracer_init):
-    with pytest.raises(TracerInitializationError) as e:  
+    with pytest.raises(GenAIImpactError) as e:  
         Tracer.init() # Second initialization