Merge pull request #15 from dataforgoodfr/feat/tracer_init

Double initialization

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,12 +1,16 @@
 import importlib.util
 
+from genai_impact.exceptions import TracerInitializationError
 
-class Tracer:
 
+class Tracer:
+    initialized = False
     @staticmethod
     def init() -> None:
+        if Tracer.initialized:
+            raise TracerInitializationError()
         init_instruments()
-
+        Tracer.initialized = True
 
 def init_instruments() -> None:
     init_openai_instrumentor()

tests/test_tracer.py

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