openrunner-sdk 2.2.0__tar.gz → 2.3.0__tar.gz

{openrunner_sdk-2.2.0 → openrunner_sdk-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.2.0
+Version: 2.3.0
 Summary: OpenRunner SDK - W&B-compatible ML experiment tracking client
 Project-URL: Homepage, https://github.com/jqueguiner/openrunner
 Project-URL: Repository, https://github.com/jqueguiner/openrunner

{openrunner_sdk-2.2.0 → openrunner_sdk-2.3.0}/openrunner/api_client.py

@@ -684,21 +684,37 @@ class APIClient:
         presigned_url: str,
         data_bytes: bytes,
         content_type: str = "image/png",
+        run_id: str = "",
+        key: str = "",
     ) -> bool:
         """PUT bytes to a presigned URL with Content-Type header.
 
-        Uses httpx directly (not self._client) because presigned URLs
-        are absolute and should not have a base_url prefix.
+        Falls back to API proxy upload if presigned URL is unreachable
+        (e.g., MinIO not exposed publicly).
         """
         try:
-            resp = httpx.put(
-                presigned_url,
-                content=data_bytes,
-                headers={"Content-Type": content_type},
-                timeout=300.0,
-            )
-            resp.raise_for_status()
-            return True
+            try:
+                resp = httpx.put(
+                    presigned_url,
+                    content=data_bytes,
+                    headers={"Content-Type": content_type},
+                    timeout=300.0,
+                )
+                resp.raise_for_status()
+                return True
+            except (httpx.ConnectError, httpx.ConnectTimeout):
+                # Presigned URL unreachable — use API proxy
+                if run_id and key:
+                    import os
+                    fname = os.path.basename(key) or key.replace("/", "_")
+                    proxy_resp = self._request(
+                        "PUT",
+                        f"/runs/{run_id}/files/media/{fname}",
+                        content=data_bytes,
+                        headers={"Content-Type": content_type},
+                    )
+                    return proxy_resp.status_code == 200
+                return False
         except Exception as e:
             logger.warning("upload_media_bytes failed: %s", e)
             return False

{openrunner_sdk-2.2.0 → openrunner_sdk-2.3.0}/openrunner/sender.py

@@ -265,7 +265,8 @@ class Sender:
             if result and result.get("presigned_url"):
                 # Upload the bytes to the presigned URL
                 self._client.upload_media_bytes(
-                    result["presigned_url"], img_bytes, content_type
+                    result["presigned_url"], img_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "table":
@@ -293,7 +294,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], audio_bytes, content_type
+                    result["presigned_url"], audio_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "video":
@@ -311,7 +313,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], video_bytes, content_type
+                    result["presigned_url"], video_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "histogram":
@@ -408,7 +411,8 @@ class Sender:
             )
             if result and result.get("presigned_url"):
                 self._client.upload_media_bytes(
-                    result["presigned_url"], img_bytes, content_type
+                    result["presigned_url"], img_bytes, content_type,
+                    run_id=self._run_id, key=item["key"],
                 )
 
         elif media_type == "molecule":

{openrunner_sdk-2.2.0 → openrunner_sdk-2.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openrunner-sdk"
-version = "2.2.0"
+version = "2.3.0"
 description = "OpenRunner SDK - W&B-compatible ML experiment tracking client"
 readme = "README.md"
 license = {text = "MIT"}