Let's also test instrument_llm_calls=True.

AgentOps-AI · Jun 14, 2024 · fc07963 · fc07963
1 parent cff9c23
commit fc07963
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_teardown.py b/tests/test_teardown.py
@@ -29,4 +29,10 @@ def test_exit(self):
     def test_can_stop_with_no_instrumentation(self):
         non_instrumented = Client(self.api_key, instrument_llm_calls=False,
                                   allow_multiple_instances=True)
+        assert non_instrumented.llm_tracker is None
         non_instrumented.stop_instrumenting()
+
+    def test_initializes_llm_tracker_when_enabled(self):
+        instrumented = Client(self.api_key, instrument_llm_calls=True,
+                              allow_multiple_instances=True)
+        assert instrumented.llm_tracker is not None