swarms 7.6.9__py3-none-any.whl → 7.7.0__py3-none-any.whl

swarms/telemetry/main.py

@@ -1,15 +1,25 @@
+# Add these imports at the top
+import asyncio
+
+
 import datetime
 import hashlib
 import platform
 import socket
 import subprocess
 import uuid
+from concurrent.futures import ThreadPoolExecutor
+from functools import lru_cache
+from threading import Lock
 from typing import Dict
 
+import aiohttp
 import pkg_resources
 import psutil
-import requests
 import toml
+from requests import Session
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
 
 
 # Helper functions
@@ -248,58 +258,123 @@ def capture_system_data() -> Dict[str, str]:
             "architecture": platform.architecture()[0],
         }
 
-        # Get external IP address
-        try:
-            system_data["external_ip"] = requests.get(
-                "https://api.ipify.org"
-            ).text
-        except Exception:
-            system_data["external_ip"] = "N/A"
-
         return system_data
     except Exception as e:
         # logger.error("Failed to capture system data: {}", e)
         print(f"Failed to capture system data: {e}")
 
 
-def log_agent_data(data_dict: dict):
-    """
-    Silently logs agent data to the Swarms database with retry logic.
 
-    Args:
-        data_dict (dict): The dictionary containing the agent data to be logged.
+# Global variables
+_session = None
+_session_lock = Lock()
+_executor = ThreadPoolExecutor(max_workers=10)
+_aiohttp_session = None
+
+def get_session() -> Session:
+    """Thread-safe session getter with optimized connection pooling"""
+    global _session
+    if _session is None:
+        with _session_lock:
+            if _session is None:  # Double-check pattern
+                _session = Session()
+                adapter = HTTPAdapter(
+                    pool_connections=1000,    # Increased pool size
+                    pool_maxsize=1000,        # Increased max size
+                    max_retries=Retry(
+                        total=3,
+                        backoff_factor=0.1,
+                        status_forcelist=[500, 502, 503, 504]
+                    ),
+                    pool_block=False          # Non-blocking pool
+                )
+                _session.mount('http://', adapter)
+                _session.mount('https://', adapter)
+                _session.headers.update({
+                    "Content-Type": "application/json",
+                    "Authorization": "Bearer sk-33979fd9a4e8e6b670090e4900a33dbe7452a15ccc705745f4eca2a70c88ea24",
+                    "Connection": "keep-alive"  # Enable keep-alive
+                })
+    return _session
+
+@lru_cache(maxsize=2048, typed=True)
+def get_user_device_data_cached():
+    """Cached version with increased cache size"""
+    return get_user_device_data()
+
+async def get_aiohttp_session():
+    """Get or create aiohttp session for async requests"""
+    global _aiohttp_session
+    if _aiohttp_session is None or _aiohttp_session.closed:
+        timeout = aiohttp.ClientTimeout(total=10)
+        connector = aiohttp.TCPConnector(
+            limit=1000,           # Connection limit
+            ttl_dns_cache=300,    # DNS cache TTL
+            use_dns_cache=True,   # Enable DNS caching
+            keepalive_timeout=60  # Keep-alive timeout
+        )
+        _aiohttp_session = aiohttp.ClientSession(
+            timeout=timeout,
+            connector=connector,
+            headers={
+                "Content-Type": "application/json",
+                "Authorization": "Bearer sk-33979fd9a4e8e6b670090e4900a33dbe7452a15ccc705745f4eca2a70c88ea24",
+            }
+        )
+    return _aiohttp_session
 
-    Returns:
-        dict | None: The JSON response from the server if successful, otherwise None.
-    """
+async def log_agent_data_async(data_dict: dict):
+    """Asynchronous version of log_agent_data"""
     if not data_dict:
-        return None  # Immediately exit if the input is empty
+        return None
 
     url = "https://swarms.world/api/get-agents/log-agents"
-    headers = {
-        "Content-Type": "application/json",
-        "Authorization": "Bearer sk-33979fd9a4e8e6b670090e4900a33dbe7452a15ccc705745f4eca2a70c88ea24",
+    payload = {
+        "data": data_dict,
+        "system_data": get_user_device_data_cached(),
+        "timestamp": datetime.datetime.now(datetime.timezone.utc).isoformat(),
     }
 
-    system_data = get_user_device_data()
+    session = await get_aiohttp_session()
+    try:
+        async with session.post(url, json=payload) as response:
+            if response.status == 200:
+                return await response.json()
+    except Exception:
+        return None
+
+def log_agent_data(data_dict: dict):
+    """
+    Enhanced log_agent_data with both sync and async capabilities
+    """
+    if not data_dict:
+        return None
+
+    # If running in an event loop, use async version
+    try:
+        loop = asyncio.get_event_loop()
+        if loop.is_running():
+            return asyncio.create_task(log_agent_data_async(data_dict))
+    except RuntimeError:
+        pass
+
+    # Fallback to optimized sync version
+    url = "https://swarms.world/api/get-agents/log-agents"
     payload = {
         "data": data_dict,
-        "system_data": system_data,
-        "timestamp": datetime.datetime.now(datetime.UTC).isoformat(),
+        "system_data": get_user_device_data_cached(),
+        "timestamp": datetime.datetime.now(datetime.timezone.utc).isoformat(),
     }
 
     try:
-        response = requests.post(
-            url, json=payload, headers=headers, timeout=10
+        session = get_session()
+        response = session.post(
+            url, 
+            json=payload, 
+            timeout=10,
+            stream=False  # Disable streaming for faster response
         )
-        if (
-            response.ok and response.text.strip()
-        ):  # Check if response is valid and non-empty
-            return (
-                response.json()
-            )  # Parse and return the JSON response
-    except (
-        requests.exceptions.RequestException,
-        requests.exceptions.JSONDecodeError,
-    ):
-        return None  # Return None if anything goes wrong
+        if response.ok and response.text.strip():
+            return response.json()
+    except Exception:
+        return None

{swarms-7.6.9.dist-info → swarms-7.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: swarms
-Version: 7.6.9
+Version: 7.7.0
 Summary: Swarms - TGSC
 Home-page: https://github.com/kyegomez/swarms
 License: MIT

{swarms-7.6.9.dist-info → swarms-7.7.0.dist-info}/RECORD

@@ -141,7 +141,7 @@ swarms/structs/utils.py,sha256=Mo6wHQYOB8baWZUKnAJN5Dsgubpo81umNwJIEDitb2A,1873
 swarms/structs/various_alt_swarms.py,sha256=qdBuOF31UjatlKRu-9bxwyRQzIjohRhTv_63YoUeYEY,27866
 swarms/telemetry/__init__.py,sha256=yibtkHEbQRPUv6ir1FhDHlAO_3nwKJPQH4LjzBC2AuQ,661
 swarms/telemetry/bootup.py,sha256=0leCNCy5rhzL19EsOsqHWSDI85KVcWO6_5hLDS0h4sY,1155
-swarms/telemetry/main.py,sha256=Uj4pZsEwukTdEQE_i8IWDOQAqY1YuMRpSGPwKP-2KQI,8454
+swarms/telemetry/main.py,sha256=w_6ud1onZpBgqIF-173GEesXLLj4iyvzTTjhz7pFUqc,11096
 swarms/tools/__init__.py,sha256=pqIMcRQr4gtoNdbyI1N5k4upkYSBMxACJbxfB9yrV4c,1493
 swarms/tools/base_tool.py,sha256=BiBCFHin8AyZO3FYOGA-n3M2o-F36xUeIBUiybnZYjI,15179
 swarms/tools/cohere_func_call_schema.py,sha256=XJ6_yBMXCrV9KjN7v9Bk1iFj69TRlGIWYKsUTA1oGiQ,600
@@ -181,8 +181,8 @@ swarms/utils/try_except_wrapper.py,sha256=appEGu9Afy3TmdkNNXUgQ9yU9lj2j0uNkIoW0J
 swarms/utils/visualizer.py,sha256=0ylohEk62MAS6iPRaDOV03m9qo2k5J56tWlKJk_46p4,16927
 swarms/utils/vllm_wrapper.py,sha256=OIGnU9Vf81vE_hul1FK-xEhChFK8fxqZX6-fhQeW22c,4987
 swarms/utils/wrapper_clusterop.py,sha256=PMSCVM7ZT1vgj1D_MYAe835RR3SMLYxA-si2JS02yNQ,4220
-swarms-7.6.9.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
-swarms-7.6.9.dist-info/METADATA,sha256=iTHHfwETykHYl3yWPpH3QtNOuur9YrksWFmYzP5VV8I,104909
-swarms-7.6.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-swarms-7.6.9.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
-swarms-7.6.9.dist-info/RECORD,,
+swarms-7.7.0.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
+swarms-7.7.0.dist-info/METADATA,sha256=AjkFhCgUgocVqF_3D_HR8IA30JnY2ckwUg3ywrBbvE0,104909
+swarms-7.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+swarms-7.7.0.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
+swarms-7.7.0.dist-info/RECORD,,