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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.4.4
+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.4 → 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.4"
+__version__ = "2.5.0"
 
 logger = logging.getLogger("openrunner")
 

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

@@ -693,27 +693,38 @@ class APIClient:
         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()
+            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._request("put", api_path, content=data)
+                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):
                 # 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.4 → 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.4 → openrunner_sdk-2.5.0}/pyproject.toml

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