airtrain 0.1.57__py3-none-any.whl → 0.1.58__py3-none-any.whl

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/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/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.dist-info → airtrain-0.1.58.dist-info}/METADATA

@@ -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.dist-info → airtrain-0.1.58.dist-info}/RECORD

@@ -1,5 +1,6 @@
-airtrain/__init__.py,sha256=jttOEv9X0ejW77bZrmIZJKCXTYcKR4RgO1VCsMRGQXQ,2816
+airtrain/__init__.py,sha256=8g5IGtnihwqBfltBMs9vMLsTNsnhwiYSXlWBkIOpqC0,3312
 airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
+airtrain/__pycache__/__init__.cpython-313.pyc,sha256=b5Xwx91d-BMvzIS-P4uZK8btsx2_mlxfUxvtoT4pVeA,3000
 airtrain/agents/__init__.py,sha256=r6v5_bblxamRgiaCT8CVhyzaDdWohGM7sSjLgIUpA5s,795
 airtrain/agents/example_agent.py,sha256=0dCS8QXIvUYYkxwyOEMLMdlQ4KgMAerQ56r7NcYGqTw,11681
 airtrain/agents/groq_agent.py,sha256=s-f-cGSrgsR4Jlrv6xzeg2Z0LAEHByhFMYVbTvu-Bkg,10558
@@ -18,6 +19,9 @@ airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
 airtrain/core/credentials.py,sha256=PgQotrQc46J5djidKnkK1znUv3fyNkUFDO-m2Kn_Gzo,4006
 airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
 airtrain/core/skills.py,sha256=kIkI1MwIzuAwyoxdnZ5MGOb70BOB-njbVb_csJEdVvc,9244
+airtrain/core/__pycache__/__init__.cpython-313.pyc,sha256=DEcKhvQGoYLFbdtCs9Fqp8sKXh8xM_viOh0drOix1cY,534
+airtrain/core/__pycache__/schemas.cpython-313.pyc,sha256=DX72FLcg5BWaj-xv6yjPFZvrhYhQY7jte89gIrNGEw0,8327
+airtrain/core/__pycache__/skills.cpython-313.pyc,sha256=lKR7C-bY2uS4Ad12UNSO8MLVxU-52OLJPC-nzV4u79w,11839
 airtrain/integrations/__init__.py,sha256=Y0yJt6c6uKOkMXdGPELSc6NYiLrEYr_38acViNPXWNk,2046
 airtrain/integrations/anthropic/__init__.py,sha256=K741w3v7fWsCknTo38ARqDL0D3HPlwDIvDuuBao9Tto,800
 airtrain/integrations/anthropic/credentials.py,sha256=hlTSw9HX66kYNaeQUtn0JjdZQBMNkzzFOJOoLOOzvcY,1246
@@ -77,16 +81,16 @@ airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9
 airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
 airtrain/integrations/together/skills.py,sha256=PU2z9YtiavZ8-ODOIXCi0stymaCsvyackTT1g2TOxZ0,11565
 airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
-airtrain/telemetry/__init__.py,sha256=338o1fYhEXGh5Sor-aotU_Lfp1z6GpjkZ8z6IIoCKuE,970
+airtrain/telemetry/__init__.py,sha256=_xDHzSmQvRCqihT0QTmF7bS9yKEl2LaZ3Zq05hXaI3k,1108
 airtrain/telemetry/service.py,sha256=6IB2-FL3fGsYYgMIVQ9MNCp9UlSE5cVhLlB3VBYkAnY,5592
-airtrain/telemetry/views.py,sha256=r6E5GnhETmBFogNcKqfWOlpIWiWmK99bo6-Y6e7oRH8,5583
+airtrain/telemetry/views.py,sha256=qob1swyLNEk_UIpDi5VTwMDsEsGZhJheFQrGbP8T5hw,8115
 airtrain/tools/__init__.py,sha256=RRqbnUqKorql5D1FDQArxpdwC5d8kUaeNzN9iZKY8V8,827
 airtrain/tools/command.py,sha256=-Qmto_PqivHQBasW8PtSYoTk_NR5Wl4zz4n7u71egMQ,6983
 airtrain/tools/filesystem.py,sha256=-YYdHj_KeSWPYXeRhWhIX9s_KujVA1R5tF3r93zRVTU,6324
 airtrain/tools/network.py,sha256=YR0AtMXDXkhCsXcx7_t2d12ItnKY8XXTmyP1kdj2M4c,3883
 airtrain/tools/registry.py,sha256=K-1H5EipYcDNDx2jdpsEY9gjfV4aNCGI1pY2UsgSpC0,10246
-airtrain-0.1.57.dist-info/METADATA,sha256=4r6uFows2tXs6mrVUblGy-wyzQdk9KmKs5D1bCo6HC8,6503
-airtrain-0.1.57.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-airtrain-0.1.57.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
-airtrain-0.1.57.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.57.dist-info/RECORD,,
+airtrain-0.1.58.dist-info/METADATA,sha256=3FCm8ujh-Lu1bEttg5i2GhsONwI5WwjQYAhSl2R10P0,6503
+airtrain-0.1.58.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+airtrain-0.1.58.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
+airtrain-0.1.58.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.58.dist-info/RECORD,,