runbios-sdk 0.2.2__tar.gz → 0.2.4__tar.gz

{runbios_sdk-0.2.2 → runbios_sdk-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: runbios-sdk
-Version: 0.2.2
+Version: 0.2.4
 Summary: Official Python SDK for the Run BiOS training and deployment platform API
 Project-URL: Homepage, https://runbios.ai
 Project-URL: Documentation, https://runbios.ai/docs/sdk-python

{runbios_sdk-0.2.2 → runbios_sdk-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "runbios-sdk"
-version = "0.2.2"
+version = "0.2.4"
 description = "Official Python SDK for the Run BiOS training and deployment platform API"
 readme = "README.md"
 license = "MIT"

{runbios_sdk-0.2.2 → runbios_sdk-0.2.4}/src/bios/_client.py

@@ -5,7 +5,7 @@ from typing import Any, Dict, Optional, Tuple
 
 import requests
 
-VERSION = "0.2.2"
+VERSION = "0.2.4"
 
 
 class ApiError(Exception):
@@ -120,6 +120,28 @@ class HttpClient:
         self._session = requests.Session()
         self._session.headers.update(self._build_base_headers())
 
+    def resolve_workspace_id(self) -> str:
+        """Return the configured workspace or the API key's bound workspace.
+
+        Multipart dataset endpoints require workspace_id in the form body while
+        ordinary JSON routes can rely on gateway identity headers. A bound API
+        key must therefore remain sufficient for upload: introspect once and
+        cache the authoritative binding instead of requiring a duplicate ID.
+        """
+        if self._workspace_id:
+            return self._workspace_id
+        data = self.get("/api/api-keys/introspect")
+        workspace = data.get("workspace", {}) if isinstance(data, dict) else {}
+        workspace_id = workspace.get("id") if isinstance(workspace, dict) else None
+        if not isinstance(workspace_id, str) or not workspace_id.strip():
+            raise ValueError(
+                "Run BiOS: the API key is not bound to a workspace; "
+                "dataset operations require a workspace-bound key"
+            )
+        self._workspace_id = workspace_id.strip()
+        self._session.headers["X-Workspace-ID"] = self._workspace_id
+        return self._workspace_id
+
     def _build_base_headers(self) -> Dict[str, str]:
         headers: Dict[str, str] = {
             "User-Agent": "bios-python/{}".format(VERSION),

{runbios_sdk-0.2.2 → runbios_sdk-0.2.4}/src/bios/datasets.py

@@ -145,7 +145,7 @@ class Datasets:
 
         ws = workspace_id or self._http._workspace_id
         if not ws:
-            raise ValueError("Run BiOS: workspace_id is required for dataset uploads")
+            ws = self._http.resolve_workspace_id()
         with open(file_path, "rb") as f:
             files = {"file": (file_name, f, content_type)}
             data: Dict[str, str] = {"name": name}
@@ -323,9 +323,7 @@ class Datasets:
         """Register a Hub source for background validation and optional materialization."""
         ws = workspace_id or self._http._workspace_id
         if not ws:
-            raise ValueError(
-                "Run BiOS: workspace_id is required to register a HuggingFace dataset"
-            )
+            ws = self._http.resolve_workspace_id()
         body: Dict[str, Any] = {
             "hf_dataset_id": repo_id,
             "name": name or repo_id.split("/")[-1],

runbios_sdk-0.2.4/tests/test_workspace_resolution.py

@@ -0,0 +1,59 @@
+import os
+import tempfile
+import unittest
+from unittest.mock import Mock
+
+from bios._client import HttpClient
+from bios.datasets import Datasets
+
+
+class WorkspaceResolutionTests(unittest.TestCase):
+    def test_http_client_introspects_once_and_caches_bound_workspace(self) -> None:
+        client = HttpClient(api_key="bios-test", base_url="https://api.test")
+        client.get = Mock(return_value={"workspace": {"id": "ws-from-key"}})  # type: ignore[method-assign]
+
+        self.assertEqual(client.resolve_workspace_id(), "ws-from-key")
+        self.assertEqual(client.resolve_workspace_id(), "ws-from-key")
+        client.get.assert_called_once_with("/api/api-keys/introspect")
+        self.assertEqual(client._session.headers["X-Workspace-ID"], "ws-from-key")
+
+    def test_dataset_upload_uses_resolved_workspace_when_not_explicit(self) -> None:
+        http = Mock()
+        http._workspace_id = None
+        http.resolve_workspace_id.return_value = "ws-from-key"
+        http.upload.return_value = {"id": "ds-1", "name": "sample", "status": "processing"}
+        datasets = Datasets(http)
+
+        with tempfile.NamedTemporaryFile(suffix=".jsonl", delete=False) as handle:
+            handle.write(b'{"messages":[{"role":"user","content":"hi"},{"role":"assistant","content":"hello"}]}\n')
+            path = handle.name
+        try:
+            result = datasets.upload(path, "sample")
+        finally:
+            os.unlink(path)
+
+        self.assertEqual(result["id"], "ds-1")
+        http.resolve_workspace_id.assert_called_once_with()
+        data = http.upload.call_args.kwargs["data"]
+        self.assertEqual(data["workspace_id"], "ws-from-key")
+
+    def test_explicit_workspace_does_not_introspect(self) -> None:
+        http = Mock()
+        http._workspace_id = None
+        http.upload.return_value = {"id": "ds-1", "name": "sample", "status": "processing"}
+        datasets = Datasets(http)
+
+        with tempfile.NamedTemporaryFile(suffix=".jsonl", delete=False) as handle:
+            handle.write(b'{"messages":[{"role":"user","content":"hi"},{"role":"assistant","content":"hello"}]}\n')
+            path = handle.name
+        try:
+            datasets.upload(path, "sample", workspace_id="ws-explicit")
+        finally:
+            os.unlink(path)
+
+        http.resolve_workspace_id.assert_not_called()
+        self.assertEqual(http.upload.call_args.kwargs["data"]["workspace_id"], "ws-explicit")
+
+
+if __name__ == "__main__":
+    unittest.main()