From 924d742a91f463e2de92d303862df69725b2b0d4 Mon Sep 17 00:00:00 2001 From: Dwij Patel Date: Sat, 7 Jun 2025 11:27:57 +0530 Subject: [PATCH] refactor: Update instrumentation manager usage and improve tracer provider access --- agentops/instrumentation/__init__.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/agentops/instrumentation/__init__.py b/agentops/instrumentation/__init__.py index 96966a8ad..cb4a6b88e 100644 --- a/agentops/instrumentation/__init__.py +++ b/agentops/instrumentation/__init__.py @@ -390,9 +390,6 @@ class InstrumentorConfig(TypedDict): # Combine all target packages for monitoring TARGET_PACKAGES = set(PROVIDERS.keys()) | set(AGENTIC_LIBRARIES.keys()) -# Create a single instance of the manager -# _manager = InstrumentationManager() # Removed - @dataclass class InstrumentorLoader: @@ -444,7 +441,8 @@ def instrument_one(loader: InstrumentorLoader) -> Optional[BaseInstrumentor]: instrumentor = loader.get_instance() try: - instrumentor.instrument(tracer_provider=tracer._provider) + # Use the provider directly from the global tracer instance + instrumentor.instrument(tracer_provider=tracer.provider) logger.info( f"AgentOps: Successfully instrumented '{loader.class_name}' for package '{loader.package_name or loader.module_name}'." )