remove debug statements and import LIBRARY_VERSION

Author: dot-agi

agentops/instrumentation/openai_agents/__init__.py

@@ -22,10 +22,9 @@ def get_version() -> str:
         from importlib.metadata import version
 
         library_version = version("openai-agents")
-        logger.debug(f"OpenAI Agents SDK version: {library_version}")
         return library_version
     except ImportError:
-        logger.warning("Could not find OpenAI Agents SDK version")
+        logger.debug("Could not find OpenAI Agents SDK version")
         return "unknown"
 
 

agentops/instrumentation/openai_agents/instrumentor.py

@@ -24,65 +24,69 @@
 from agentops.logging import logger
 from agentops.instrumentation.openai_agents.processor import OpenAIAgentsProcessor
 from agentops.instrumentation.openai_agents.exporter import OpenAIAgentsExporter
+from agentops.instrumentation.openai_agents import LIBRARY_VERSION
 
 
 class OpenAIAgentsInstrumentor(BaseInstrumentor):
     """An instrumentor for OpenAI Agents SDK that primarily uses the built-in tracing API."""
-    
+
     _processor = None
     _exporter = None
     _default_processor = None
-    
+
     def instrumentation_dependencies(self) -> Collection[str]:
         """Return packages required for instrumentation."""
         return ["openai-agents >= 0.0.1"]
-    
+
     def _instrument(self, **kwargs):
         """Instrument the OpenAI Agents SDK."""
         tracer_provider = kwargs.get("tracer_provider")
-        
+
         try:
             # Check if Agents SDK is available
             try:
                 import agents  # type: ignore
-                logger.debug(f"Agents SDK detected, version: {getattr(agents, '__version__', 'unknown')}")
+
+                logger.debug(f"OpenAI Agents SDK detected with version: {LIBRARY_VERSION}")
             except ImportError as e:
-                logger.debug(f"Agents SDK import failed: {e}")
+                logger.debug(f"OpenAI Agents SDK import failed: {e}")
                 return
-            
+
             self._exporter = OpenAIAgentsExporter(tracer_provider=tracer_provider)
             self._processor = OpenAIAgentsProcessor(
                 exporter=self._exporter,
             )
-            
+
             # Replace the default processor with our processor
             from agents import set_trace_processors  # type: ignore
             from agents.tracing.processors import default_processor  # type: ignore
+
             # Store reference to default processor for later restoration
             self._default_processor = default_processor()
             set_trace_processors([self._processor])
             logger.debug("Replaced default processor with OpenAIAgentsProcessor in OpenAI Agents SDK")
-                
+
         except Exception as e:
             logger.warning(f"Failed to instrument OpenAI Agents SDK: {e}")
-    
+
     def _uninstrument(self, **kwargs):
         """Remove instrumentation from OpenAI Agents SDK."""
         try:
             # Clean up any active spans in the exporter
-            if hasattr(self, '_exporter') and self._exporter:
+            if hasattr(self, "_exporter") and self._exporter:
                 # Call cleanup to properly handle any active spans
-                if hasattr(self._exporter, 'cleanup'):
+                if hasattr(self._exporter, "cleanup"):
                     self._exporter.cleanup()
-            
+
             # Put back the default processor
             from agents import set_trace_processors
-            if hasattr(self, '_default_processor') and self._default_processor:
+
+            if hasattr(self, "_default_processor") and self._default_processor:
                 set_trace_processors([self._default_processor])
                 self._default_processor = None
             self._processor = None
             self._exporter = None
-            
+
             logger.info("Successfully removed OpenAI Agents SDK instrumentation")
         except Exception as e:
             logger.warning(f"Failed to uninstrument OpenAI Agents SDK: {e}")