Add system stats and imported libraries tracking

agentops/sdk/core.py

@@ -2,6 +2,10 @@
 
 import atexit
 import threading
+import platform
+import sys
+import os
+import psutil
 from typing import List, Optional
 
 from opentelemetry import metrics, trace
@@ -24,6 +28,73 @@
 # No need to create shortcuts since we're using our own ResourceAttributes class now
 
 
+def get_imported_libraries():
+    """
+    Get the top-level imported libraries in the current script.
+
+    Returns:
+        list: List of imported libraries
+    """
+    user_libs = []
+
+    builtin_modules = {
+        'builtins', 'sys', 'os', '_thread', 'abc', 'io', 're', 'types', 
+        'collections', 'enum', 'math', 'datetime', 'time', 'warnings'
+    }
+
+    try:
+        main_module = sys.modules.get('__main__')
+        if main_module and hasattr(main_module, '__dict__'):
+            for name, obj in main_module.__dict__.items():
+                if isinstance(obj, type(sys)) and hasattr(obj, '__name__'):
+                    mod_name = obj.__name__.split('.')[0]
+                    if (mod_name and 
+                        not mod_name.startswith('_') and
+                        mod_name not in builtin_modules):
+                        user_libs.append(mod_name)
+    except Exception as e:
+        logger.debug(f"Error getting imports: {e}")
+
+    return user_libs
+
+
+def get_system_stats():
+    """
+    Get basic system stats including CPU and memory information.
+
+    Returns:
+        dict: Dictionary with system information
+    """
+    system_info = {
+        ResourceAttributes.HOST_MACHINE: platform.machine(),
+        ResourceAttributes.HOST_NAME: platform.node(),
+        ResourceAttributes.HOST_NODE: platform.node(),
+        ResourceAttributes.HOST_PROCESSOR: platform.processor(),
+        ResourceAttributes.HOST_SYSTEM: platform.system(),
+        ResourceAttributes.HOST_VERSION: platform.version(),
+        ResourceAttributes.HOST_OS_RELEASE: platform.release(),
+    }
+
+    # Add CPU stats
+    try:
+        system_info[ResourceAttributes.CPU_COUNT] = os.cpu_count() or 0
+        system_info[ResourceAttributes.CPU_PERCENT] = psutil.cpu_percent(interval=0.1)
+    except Exception as e:
+        logger.debug(f"Error getting CPU stats: {e}")
+
+    # Add memory stats
+    try:
+        memory = psutil.virtual_memory()
+        system_info[ResourceAttributes.MEMORY_TOTAL] = memory.total
+        system_info[ResourceAttributes.MEMORY_AVAILABLE] = memory.available
+        system_info[ResourceAttributes.MEMORY_USED] = memory.used
+        system_info[ResourceAttributes.MEMORY_PERCENT] = memory.percent
+    except Exception as e:
+        logger.debug(f"Error getting memory stats: {e}")
+
+    return system_info
+
+
 def setup_telemetry(
     service_name: str = "agentops",
     project_id: Optional[str] = None,
@@ -58,6 +129,14 @@
         # Add project_id as a custom resource attribute
         resource_attrs[ResourceAttributes.PROJECT_ID] = project_id
         logger.debug(f"Including project_id in resource attributes: {project_id}")
+
+    # Add system information
+    system_stats = get_system_stats()
+    resource_attrs.update(system_stats)
+
+    # Add imported libraries
+    imported_libraries = get_imported_libraries()
+    resource_attrs[ResourceAttributes.IMPORTED_LIBRARIES] = imported_libraries
 
     resource = Resource(resource_attrs)
     provider = TracerProvider(resource=resource)

agentops/semconv/resource.py

@@ -28,3 +28,25 @@ class ResourceAttributes:
     # SDK attributes
     SDK_NAME = "agentops.sdk.name"
     SDK_VERSION = "agentops.sdk.version"
+
+    # Host machine attributes
+    HOST_MACHINE = "host.machine"
+    HOST_NAME = "host.name"
+    HOST_NODE = "host.node"
+    HOST_OS_RELEASE = "host.os_release"
+    HOST_PROCESSOR = "host.processor"
+    HOST_SYSTEM = "host.system"
+    HOST_VERSION = "host.version"
+
+    # CPU attributes
+    CPU_COUNT = "cpu.count"
+    CPU_PERCENT = "cpu.percent"
+
+    # Memory attributes
+    MEMORY_TOTAL = "memory.total"
+    MEMORY_AVAILABLE = "memory.available"
+    MEMORY_USED = "memory.used"
+    MEMORY_PERCENT = "memory.percent"
+
+    # Libraries
+    IMPORTED_LIBRARIES = "imported_libraries"