Undo overrides

AgentOps-AI · Jun 15, 2024 · b74cf12 · b74cf12
1 parent 23d8162
commit b74cf12
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py
@@ -696,6 +696,7 @@ def override_api(self):
     def stop_instrumenting(self):
         self.undo_override_openai_v1_async_completion()
         self.undo_override_openai_v1_completion()
+        self.undo_override_ollama()
 
     def undo_override_openai_v1_completion(self):
         global original_create