openrunner-sdk 2.41.3__tar.gz → 2.41.4__tar.gz

{openrunner_sdk-2.41.3 → openrunner_sdk-2.41.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrunner-sdk
-Version: 2.41.3
+Version: 2.41.4
 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.41.3 → openrunner_sdk-2.41.4}/openrunner/__init__.py

@@ -181,7 +181,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.41.3"
+__version__ = "2.41.4"
 
 logger = logging.getLogger("openrunner")
 

{openrunner_sdk-2.41.3 → openrunner_sdk-2.41.4}/openrunner/_native.py

@@ -9,7 +9,8 @@ single-implementation point, never a hard dependency.
 
 Covered surface today: projects (incl. linked repo), artifacts/versions,
 presigned downloads (selective, GIL released during transfers), container
-registries + broker login tokens.
+registries + broker login tokens, and dataset editable-overlay columns
+(add column / set cell / edit row / get row — source columns stay read-only).
 """
 
 from __future__ import annotations

{openrunner_sdk-2.41.3 → openrunner_sdk-2.41.4}/openrunner/datasets.py

@@ -194,14 +194,27 @@ class DatasetsClient:
                or _g.get("base_url") or "http://localhost:8000")
         if not key:
             raise ValueError("API key required. Set OPENRUNNER_API_KEY, pass api_key=, or run `openrunner login`.")
+        # Raw creds kept so the optional native core (Rust) can be built lazily —
+        # the crate appends its own /api/v1, so hand it the bare base URL.
+        self._base_url = url.rstrip("/")
+        self._api_key = key
+        self._native: Any = "unset"  # sentinel: native core not yet resolved
         from ._retry_transport import RetryTransport
         self._http = httpx.Client(
-            base_url=f"{url.rstrip('/')}/api/v1",
-            headers={"Authorization": f"Bearer {key}"},
+            base_url=f"{self._base_url}/api/v1",
+            headers={"Authorization": f"Bearer {self._api_key}"},
             timeout=timeout,
             transport=RetryTransport(),
         )
 
+    def _native_client(self):
+        """The native (Rust/PyO3) client if the ``openrunner-core`` wheel is
+        installed, else ``None`` — an accelerator with a pure-Python fallback."""
+        if self._native == "unset":
+            from openrunner import _native
+            self._native = _native.native_client(self._base_url, self._api_key)
+        return self._native
+
     # -- resolution ----------------------------------------------------------
     def _org_id(self, org: str) -> str:
         r = self._http.get("/organizations")
@@ -812,6 +825,9 @@ class DatasetsClient:
         rejected. Returns the editable column names.
         """
         ds = self._resolve(repo_id)
+        nat = self._native_client()
+        if nat is not None:
+            return nat.add_dataset_column(str(ds.id), name, default)
         r = self._http.post(f"/datasets/{ds.id}/columns",
                             json={"name": name, "default": default})
         _check(r)
@@ -826,6 +842,9 @@ class DatasetsClient:
         ``{ok, debug_id, column}``.
         """
         ds = self._resolve(repo_id)
+        nat = self._native_client()
+        if nat is not None:
+            return nat.set_dataset_cell(str(ds.id), debug_id, column, value)
         r = self._http.patch(f"/datasets/{ds.id}/cell",
                             json={"debug_id": debug_id, "column": column, "value": value})
         _check(r)
@@ -839,6 +858,9 @@ class DatasetsClient:
         a read-only source column. Returns ``{ok, debug_id, columns}``.
         """
         ds = self._resolve(repo_id)
+        nat = self._native_client()
+        if nat is not None:
+            return nat.edit_dataset_row(str(ds.id), debug_id, values)
         r = self._http.patch(f"/datasets/{ds.id}/rows/{debug_id}", json={"values": values})
         _check(r)
         return r.json()
@@ -849,6 +871,9 @@ class DatasetsClient:
         ``{debug_id, row, editable_columns}`` (everything in ``row`` not listed in
         ``editable_columns`` is a read-only parent column)."""
         ds = self._resolve(repo_id)
+        nat = self._native_client()
+        if nat is not None:
+            return nat.get_dataset_row(str(ds.id), debug_id)
         r = self._http.get(f"/datasets/{ds.id}/row", params={"debug_id": debug_id})
         _check(r)
         return r.json()

{openrunner_sdk-2.41.3 → openrunner_sdk-2.41.4}/pyproject.toml

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