From 23b585154864f8fdbc9cb5830cb481f2e549e6f5 Mon Sep 17 00:00:00 2001
From: Adrien Banse <adrien.banse@uclouvain.be>
Date: Tue, 19 Mar 2024 15:39:08 +0100
Subject: [PATCH] GenAIImpactError implem

---
 genai_impact/__init__.py   | 4 ++--
 genai_impact/exceptions.py | 6 ++++++
 genai_impact/tracer.py     | 6 ++----
 tests/test_tracer.py       | 5 +++--
 4 files changed, 13 insertions(+), 8 deletions(-)
 create mode 100644 genai_impact/exceptions.py

diff --git a/genai_impact/__init__.py b/genai_impact/__init__.py
index 502c9ad..a135463 100644
--- a/genai_impact/__init__.py
+++ b/genai_impact/__init__.py
@@ -1,3 +1,3 @@
-from .tracer import Tracer, TracerInitializationError
+from .tracer import Tracer
 
-__all__ = ["Tracer", "TracerInitializationError"]
+__all__ = ["Tracer"]
diff --git a/genai_impact/exceptions.py b/genai_impact/exceptions.py
new file mode 100644
index 0000000..b14fc9b
--- /dev/null
+++ b/genai_impact/exceptions.py
@@ -0,0 +1,6 @@
+class GenAIImpactError(Exception):
+    pass
+
+class TracerInitializationError(GenAIImpactError):
+    "Tracer is initialized twice"
+    pass
\ No newline at end of file
diff --git a/genai_impact/tracer.py b/genai_impact/tracer.py
index 272ad9c..5be6c73 100644
--- a/genai_impact/tracer.py
+++ b/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
diff --git a/tests/test_tracer.py b/tests/test_tracer.py
index ecd14c2..2dcc6c9 100644
--- a/tests/test_tracer.py
+++ b/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