attr checker

AgentOps-AI · Jan 3, 2024 · 02ae95a · 02ae95a
1 parent 08e433d
commit 02ae95a
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py
@@ -292,10 +292,11 @@ def override_api(self, api):
 
             # If openai 1.0.0+ is specified, patch the completions methods
             if api == 'openai':
-                module_version = parse(module.__version__)
-                if module_version >= parse('1.0.0'):
-                    self.override_openai_v1_completion()
-                    self.override_openai_v1_async_completion()
+                if hasattr(module, '__version__'):
+                    module_version = parse(module.__version__)
+                    if module_version >= parse('1.0.0'):
+                        self.override_openai_v1_completion()
+                        self.override_openai_v1_async_completion()
 
             # Patch all methods in every API
             if hasattr(module, '__version__'):