fix: AG2 minor refactor for LIBRARY_NAME & LIBRARY_VERSION

agentops/instrumentation/ag2/__init__.py

@@ -5,7 +5,24 @@
 than individual message exchanges.
 """
 
-from agentops.instrumentation.ag2.instrumentor import AG2Instrumentor
-from agentops.instrumentation.ag2.version import LIBRARY_NAME, LIBRARY_VERSION
+from agentops.logging import logger
+
+
+def get_version() -> str:
+    """Get the version of the AG2 package, or 'unknown' if not found"""
+    try:
+        from importlib.metadata import version
+
+        return version("ag2")
+    except ImportError:
+        logger.debug("Could not find AG2 version")
+        return "unknown"
+
+
+LIBRARY_NAME = "ag2"
+LIBRARY_VERSION: str = get_version()
+
+# Import after defining constants to avoid circular imports
+from agentops.instrumentation.ag2.instrumentor import AG2Instrumentor  # noqa: E402
 
 __all__ = ["AG2Instrumentor", "LIBRARY_NAME", "LIBRARY_VERSION"]

agentops/instrumentation/ag2/instrumentor.py

@@ -13,7 +13,7 @@
 from wrapt import wrap_function_wrapper
 
 from agentops.logging import logger
-from agentops.instrumentation.ag2.version import LIBRARY_NAME, LIBRARY_VERSION
+from agentops.instrumentation.ag2 import LIBRARY_NAME, LIBRARY_VERSION
 from agentops.semconv import Meters
 from agentops.semconv.message import MessageAttributes
 from agentops.semconv.span_attributes import SpanAttributes