openrunner-sdk 2.4.3__tar.gz → 2.5.0__tar.gz

{openrunner_sdk-2.4.3 → openrunner_sdk-2.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.4.3
+Version: 2.5.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.4.3 → openrunner_sdk-2.5.0}/openrunner/__init__.py

@@ -120,7 +120,7 @@ launch.from_run = _launch_from_run  # type: ignore[attr-defined]
 # openrunner.trace.patch_openai() syntax
 trace.patch_openai = _patch_openai  # type: ignore[attr-defined]
 
-__version__ = "2.4.3"
+__version__ = "2.5.0"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.4.3 → openrunner_sdk-2.5.0}/openrunner/api_client.py

@@ -688,23 +688,43 @@ class APIClient:
     ) -> bool:
         """PUT raw file bytes to a presigned URL.
 
-        Falls back to API proxy upload if presigned URL is unreachable
-        (e.g., MinIO not exposed publicly).
+        Handles three cases:
+        1. Relative proxy URL (/api/v1/runs/...) — use authenticated client
+        2. Absolute URL — try direct PUT, fall back to proxy on ConnectError
+        3. ConnectError fallback — PUT via /runs/{run_id}/files/{name}
         """
+        import os
+
         try:
-            with open(file_path, "rb") as f:
-                data = f.read()
-            # Try presigned URL first
+            file_size = os.path.getsize(file_path)
+            # Use streaming for large files, read into memory for small ones
+            if file_size > 50 * 1024 * 1024:  # 50 MB
+                data = open(file_path, "rb")  # noqa: SIM115
+            else:
+                with open(file_path, "rb") as f:
+                    data = f.read()
+
+            # Scale timeout by file size (minimum 60s, ~30s per 100MB)
+            upload_timeout = max(60.0, file_size / (3 * 1024 * 1024))
+
+            # Case 1: relative proxy URL from server
+            if presigned_url.startswith("/"):
+                api_path = presigned_url.replace("/api/v1/", "/", 1) if presigned_url.startswith("/api/v1/") else presigned_url
+                resp = self._client.put(api_path, content=data, timeout=upload_timeout)
+                return resp.status_code == 200
+
+            # Case 2: absolute presigned URL (direct to S3)
             try:
-                resp = httpx.put(presigned_url, content=data, timeout=300.0)
+                resp = httpx.put(presigned_url, content=data, timeout=upload_timeout)
                 resp.raise_for_status()
                 return True
             except (httpx.ConnectError, httpx.ConnectTimeout):
-                # Presigned URL unreachable — use API proxy
+                # Case 3: fallback to API proxy
                 if run_id:
-                    import os
                     fname = os.path.basename(file_path)
-                    proxy_resp = self._request("PUT", f"/runs/{run_id}/files/{fname}", content=data)
+                    proxy_resp = self._client.put(
+                        f"/runs/{run_id}/files/{fname}", content=data, timeout=upload_timeout
+                    )
                     return proxy_resp.status_code == 200
                 return False
         except Exception as e:

{openrunner_sdk-2.4.3 → openrunner_sdk-2.5.0}/openrunner/run.py

@@ -825,13 +825,21 @@ class Run:
 
             # Upload files that need uploading (dedup may skip some)
             # Reference files and reused files never have upload URLs.
+            all_uploaded = True
             for url_entry in result.get("upload_urls", []):
                 local = next(
                     f for f in artifact._files if f["name"] == url_entry["name"]
                 )
-                self._client.upload_file_to_presigned_url(
+                ok = self._client.upload_file_to_presigned_url(
                     url_entry["presigned_url"], local["local_path"]
                 )
+                if not ok:
+                    logger.warning("upload failed for %s", url_entry["name"])
+                    all_uploaded = False
+
+            if not all_uploaded:
+                logger.error("artifact upload incomplete — not confirming version")
+                return None
 
             # Confirm the version
             self._client.confirm_artifact_version(result["version_id"])

{openrunner_sdk-2.4.3 → openrunner_sdk-2.5.0}/pyproject.toml

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