artificialbrains-sdk 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

ab_sdk/client.py

@@ -122,7 +122,15 @@ class ABClient:
             # derive host root from base_url (/api stripped)
             url = self.base_url[:-4] if self.base_url.endswith("/api") else self.base_url
 
-        socket = socketio.Client(reconnection=True, logger=False, engineio_logger=False)
+        socket = socketio.Client(
+            reconnection=True,
+            reconnection_attempts=0,  # infinite retries
+            reconnection_delay=1,
+            reconnection_delay_max=5,
+            logger=True,              # Enable to see ping/pong
+            engineio_logger=True,
+            request_timeout=3600,      # Allow long requests (1 hour)
+        )
 
         connect_headers: Dict[str, str] = {}
         auth_payload: Optional[Dict[str, Any]] = None
@@ -136,49 +144,23 @@ class ABClient:
          # Helpful visibility: log namespace connect/disconnect
         @socket.on("connect", namespace=ns)
         def _on_connect():
-            logger.info("Realtime connected to namespace %s (namespaces=%s)", ns, list(getattr(socket, "namespaces", {}).keys()))
+            logger.info("Realtime connected to namespace %s", ns)
 
         @socket.on("disconnect", namespace=ns)
         def _on_disconnect():
             logger.warning("Realtime disconnected from namespace %s", ns)
 
-        # Robust connect strategy:
-        #  1) Try websocket-only (does NOT require `requests`)
-        #  2) Fallback to default transports (polling+websocket) if needed
-        # Provide a clear error message if dependencies are missing.
-        try:
-            socket.connect(
-                url,
-                headers=connect_headers,
-                auth=auth_payload,
-                namespaces=[ns],
-                transports=["websocket"],
-                wait=True,
-                wait_timeout=self.timeout,
-            )
-        except Exception as e1:
-            # If websocket-client is missing, python-socketio will fail here.
-            # If polling is needed and `requests` is missing, it will fail on fallback.
-            try:
-                socket.connect(
-                    url,
-                    headers=connect_headers,
-                    auth=auth_payload,
-                    namespaces=[ns],
-                    wait=True,
-                    wait_timeout=self.timeout,
-                )
-            except Exception as e2:
-                msg = (
-                    "Realtime connection failed.\n\n"
-                    "Tried websocket-only then default transports.\n\n"
-                    "Common fixes:\n"
-                    "  - pip install websocket-client\n"
-                    "  - pip install requests\n\n"
-                    f"websocket error: {e1}\n"
-                    f"fallback error: {e2}"
-                )
-                raise socketio.exceptions.ConnectionError(msg)
+        # Use polling + websocket (not websocket-only)
+        # This maintains connection through NAT devices
+        socket.connect(
+            url,
+            headers=connect_headers,
+            auth=auth_payload,
+            namespaces=[ns],
+            transports=["polling", "websocket"],  # MUST have polling first!
+            wait=True,
+            wait_timeout=120,
+        )
             
 
         # HARD ASSERT: the namespace must actually be connected, or emits will fail with:

{artificialbrains_sdk-0.1.4.dist-info → artificialbrains_sdk-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: artificialbrains_sdk
-Version: 0.1.4
+Version: 0.1.5
 Summary: ArtificialBrains Python SDK
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: Other/Proprietary License

{artificialbrains_sdk-0.1.4.dist-info → artificialbrains_sdk-0.1.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ab_sdk/__init__.py,sha256=tYSMwbFzqsiZ7Bp7mYy2kQ0rX9I1NwM6wS1hEGYuze4,1831
-ab_sdk/client.py,sha256=TvANQpuEgjDYn51tIcXtGijHrqFUVXKPGXjKMRbWflc,10144
+ab_sdk/client.py,sha256=Aj5XgAEBtXeKg5DQgdG66SNtOvTYugd9DROkw34GmEY,9220
 ab_sdk/contract_scaffold.py,sha256=4zcsWhd9axxMu--FZ7U8B8u4QLUwluifyvqac2apDno,8930
 ab_sdk/endpoints.py,sha256=yc4h_E8vQA9TmWF0sW2GprgE0QnS3eBetH13ogeuw4E,2102
 ab_sdk/input_streamer.py,sha256=fRPlBimaTDBsm1gz9cdoYanVbmwYTBXSyI9IrXslXiM,6847
@@ -9,7 +9,7 @@ ab_sdk/plugins/__init__.py,sha256=veHfKqvymY8YPN9HF4E1S4pwRUcVVJHDAyBZptbLNlA,18
 ab_sdk/plugins/decoder.py,sha256=vHAubiAPOaOg3J006TLjn4qAoLIu2Lz2D2kjz421odc,10148
 ab_sdk/plugins/deviation.py,sha256=QLKOqOSa9HIj3Z3bO6oR79wY1Lni_P-j3GAGM7UYKVs,2019
 ab_sdk/plugins/reward.py,sha256=E5IsV29N6juyfIQ7ZksDa8JSxAAbVflaJOtLplilAD4,1796
-artificialbrains_sdk-0.1.4.dist-info/METADATA,sha256=m7pUtsi0xBlVHDzB2qEHkzxuHKTxm--2g0-9tLwIGFM,12177
-artificialbrains_sdk-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-artificialbrains_sdk-0.1.4.dist-info/top_level.txt,sha256=Ttsz2oF0qRh68OI9w3mu_LKkv2I_YvOWr3kudSpckpM,7
-artificialbrains_sdk-0.1.4.dist-info/RECORD,,
+artificialbrains_sdk-0.1.5.dist-info/METADATA,sha256=mu2eo9FmI8BgPKrOTE7h6INcLlwTNGFYKFQpaOCHFSo,12177
+artificialbrains_sdk-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+artificialbrains_sdk-0.1.5.dist-info/top_level.txt,sha256=Ttsz2oF0qRh68OI9w3mu_LKkv2I_YvOWr3kudSpckpM,7
+artificialbrains_sdk-0.1.5.dist-info/RECORD,,