ollama undo instrument (#397)

AgentOps-AI · Sep 17, 2024 · 51433d9 · 51433d9
1 parent 1ef969e
commit 51433d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agentops/llms/ollama.py b/agentops/llms/ollama.py
@@ -65,7 +65,7 @@ def override(self):
         self._override_chat_async_client()
 
     def undo_override(self):
-        if original_func is not None:
+        if original_func is not None and original_func != {}:
             import ollama
 
             ollama.chat = original_func["ollama.chat"]