Add AG2Instrumentor

agentops/instrumentation/__init__.py

@@ -77,6 +77,11 @@ def get_instance(self) -> BaseInstrumentor:
         class_name="GoogleGenerativeAIInstrumentor",
         provider_import_name="google.genai",
     ),
+    InstrumentorLoader(
+        module_name="agentops.instrumentation.ag2",
+        class_name="AG2Instrumentor",
+        provider_import_name="autogen",
+    ),
 ]
 
 

agentops/instrumentation/ag2/__init__.py

@@ -0,0 +1,17 @@
+"""AG2 Instrumentation for AgentOps
+
+This module provides instrumentation for AG2 (AutoGen), adding telemetry to track agent
+interactions, conversation flows, and tool usage while focusing on summary-level data rather
+than individual message exchanges.
+"""
+
+# Version string and package info
+LIBRARY_NAME = "ag2"
+LIBRARY_VERSION = "0.3.2"  # Update based on actual version requirement
+
+from typing import Collection
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+
+from agentops.instrumentation.ag2.instrumentor import AG2Instrumentor
+
+__all__ = ["AG2Instrumentor"]