airtrain 0.1.57__tar.gz → 0.1.58__tar.gz

{airtrain-0.1.57 → airtrain-0.1.58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airtrain
-Version: 0.1.57
+Version: 0.1.58
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai

{airtrain-0.1.57 → airtrain-0.1.58}/airtrain/__init__.py

@@ -1,6 +1,8 @@
 """Airtrain - A platform for building and deploying AI agents with structured skills"""
 
-__version__ = "0.1.57"
+__version__ = "0.1.58"
+
+import sys
 
 # Core imports
 from .core.skills import Skill, ProcessingError
@@ -49,8 +51,9 @@ from .agents import (
     SharedMemory
 )
 
-# Telemetry import
+# Telemetry import - must be imported after version is defined
 from .telemetry import telemetry
+from .telemetry import PackageImportTelemetryEvent
 
 
 __all__ = [
@@ -98,3 +101,19 @@ __all__ = [
     # Telemetry - not directly exposed to users
     # but initialized at import time
 ]
+
+# Capture import telemetry
+try:
+    telemetry.capture(
+        PackageImportTelemetryEvent(
+            version=__version__,
+            python_version=(
+                f"{sys.version_info.major}."
+                f"{sys.version_info.minor}."
+                f"{sys.version_info.micro}"
+            )
+        )
+    )
+except Exception:
+    # Silently continue if telemetry fails
+    pass

{airtrain-0.1.57 → airtrain-0.1.58}/airtrain/telemetry/__init__.py

@@ -16,6 +16,8 @@ from airtrain.telemetry.views import (
     UserFeedbackTelemetryEvent,
     SkillInitTelemetryEvent,
     SkillProcessTelemetryEvent,
+    PackageInstallTelemetryEvent,
+    PackageImportTelemetryEvent,
 )
 
 __all__ = [
@@ -28,6 +30,8 @@ __all__ = [
     "UserFeedbackTelemetryEvent",
     "SkillInitTelemetryEvent",
     "SkillProcessTelemetryEvent",
+    "PackageInstallTelemetryEvent",
+    "PackageImportTelemetryEvent",
 ]
 
 # Create a singleton instance for easy import

{airtrain-0.1.57 → airtrain-0.1.58}/airtrain/telemetry/views.py

@@ -170,4 +170,68 @@ class SkillProcessTelemetryEvent(BaseTelemetryEvent):
     input_data: Optional[Dict[str, Any]] = None
     duration_seconds: float = 0.0
     error: Optional[str] = None
-    name: str = 'skill_process' 
+    name: str = 'skill_process'
+
+
+@dataclass
+class PackageInstallTelemetryEvent(BaseTelemetryEvent):
+    """Event type to capture package installation"""
+    version: str
+    python_version: str
+    install_method: Optional[str] = None  # pip, conda, source, etc.
+    platform: Optional[str] = None  # Operating system
+    dependencies: Optional[Dict[str, str]] = None  # Installed dependencies
+    name: str = 'package_install'
+    
+    def __post_init__(self):
+        # Collect platform info if not provided
+        if self.platform is None:
+            import platform
+            self.platform = platform.platform()
+            
+        # Collect dependency info if not provided
+        if self.dependencies is None:
+            # Try to get installed package versions for key dependencies
+            self.dependencies = {}
+            import importlib.metadata
+            try:
+                for package in ["openai", "anthropic", "groq", "together"]:
+                    try:
+                        self.dependencies[package] = importlib.metadata.version(package)
+                    except importlib.metadata.PackageNotFoundError:
+                        pass
+            except (ImportError, Exception):
+                pass
+
+
+@dataclass
+class PackageImportTelemetryEvent(BaseTelemetryEvent):
+    """Event type to capture package import"""
+    version: str
+    python_version: str
+    import_context: Optional[str] = None  # Information about what imported the package
+    platform: Optional[str] = None  # Operating system
+    name: str = 'package_import'
+    
+    def __post_init__(self):
+        # Collect platform info if not provided
+        if self.platform is None:
+            import platform
+            self.platform = platform.platform()
+            
+        # Try to get import context from traceback
+        if self.import_context is None:
+            try:
+                import inspect
+                frames = inspect.stack()
+                # Skip the first few frames which are inside our code
+                # Look for the first frame that's not in our module
+                import_frames = []
+                for frame in frames[3:10]:  # Skip first 3, take up to 7 more
+                    module = frame.frame.f_globals.get('__name__', '')
+                    if not module.startswith('airtrain'):
+                        import_frames.append(f"{module}:{frame.function}")
+                if import_frames:
+                    self.import_context = " -> ".join(import_frames)
+            except Exception:
+                pass 

{airtrain-0.1.57 → airtrain-0.1.58}/airtrain.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airtrain
-Version: 0.1.57
+Version: 0.1.58
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai

{airtrain-0.1.57 → airtrain-0.1.58}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "airtrain"
-version = "0.1.57"
+version = "0.1.58"
 description = "A platform for building and deploying AI agents with structured skills"
 readme = "README.md"
 requires-python = ">=3.8"

{airtrain-0.1.57 → airtrain-0.1.58}/setup.py

@@ -2,6 +2,7 @@ from setuptools import setup, find_packages  # type: ignore
 import os
 import sys
 import re
+from setuptools.command.install import install
 
 
 # Read version from __init__.py
@@ -35,6 +36,43 @@ def get_changelog() -> str:
     return "".join(changelog_content)
 
 
+# Custom install command that sends telemetry
+class CustomInstallCommand(install):
+    def run(self):
+        # Run the standard install
+        install.run(self)
+        
+        # Try to send telemetry after installation
+        try:
+            # We need to import here to avoid import errors
+            # before the package is installed
+            sys.path.insert(0, self.install_lib)
+            try:
+                from airtrain.telemetry import telemetry, PackageInstallTelemetryEvent
+                
+                # Get the version we just installed
+                version = get_version()
+                
+                # Send telemetry
+                telemetry.capture(
+                    PackageInstallTelemetryEvent(
+                        version=version,
+                        python_version=(
+                            f"{sys.version_info.major}."
+                            f"{sys.version_info.minor}."
+                            f"{sys.version_info.micro}"
+                        ),
+                        install_method="pip" if "pip" in sys.argv else "unknown"
+                    )
+                )
+            except ImportError:
+                # The package is not fully installed yet
+                pass
+        except Exception:
+            # Don't let telemetry failures interfere with installation
+            pass
+
+
 with open("README.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()
 
@@ -104,4 +142,7 @@ setup(
             "airtrain=airtrain.cli.main:main",
         ],
     },
+    cmdclass={
+        'install': CustomInstallCommand,
+    },
 )