kscale 0.3.10__py3-none-any.whl → 0.3.12__py3-none-any.whl

kscale/__init__.py

@@ -1,6 +1,6 @@
 """Defines the common interface for the K-Scale Python API."""
 
-__version__ = "0.3.10"
+__version__ = "0.3.12"
 
 from pathlib import Path
 

kscale/web/cli/robot_class.py

@@ -41,10 +41,11 @@ async def list() -> None:
                 click.style(rc.id, fg="blue"),
                 click.style(rc.class_name, fg="green"),
                 rc.description or "N/A",
+                "N/A" if rc.num_downloads is None else f"{rc.num_downloads:,}",
             ]
             for rc in robot_classes
         ]
-        click.echo(tabulate(table_data, headers=["ID", "Name", "Description"], tablefmt="simple"))
+        click.echo(tabulate(table_data, headers=["ID", "Name", "Description", "Downloads"], tablefmt="simple"))
     else:
         click.echo(click.style("No robot classes found", fg="red"))
 

kscale/web/clients/base.py

@@ -9,7 +9,7 @@ import sys
 import time
 import webbrowser
 from types import TracebackType
-from typing import Any, Self, Type
+from typing import Any, Mapping, Self, Type
 from urllib.parse import urljoin
 
 import aiohttp
@@ -363,6 +363,7 @@ class BaseClient:
         params: dict[str, Any] | None = None,
         data: BaseModel | dict[str, Any] | None = None,
         files: dict[str, Any] | None = None,
+        error_code_suggestions: dict[int, str] | None = None,
     ) -> dict[str, Any]:
         url = urljoin(self.base_url, endpoint)
         kwargs: dict[str, Any] = {}
@@ -380,12 +381,27 @@ class BaseClient:
         response = await client.request(method, url, **kwargs)
 
         if response.is_error:
-            logger.error("Got %d error K-Scale: %s", response.status_code, response.text)
-            if verbose_error():
+            error_code = response.status_code
+            error_json = response.json()
+            use_verbose_error = verbose_error()
+
+            if not use_verbose_error:
+                logger.info("Use KSCALE_VERBOSE_ERROR=1 to see the full error message")
+                logger.info("If this persists, please create an issue here: https://github.com/kscalelabs/kscale")
+
+            logger.error("Got error %d from the K-Scale API", error_code)
+            if isinstance(error_json, Mapping):
+                for key, value in error_json.items():
+                    logger.error("  [%s] %s", key, value)
+            else:
+                logger.error("  %s", error_json)
+
+            if error_code_suggestions is not None and error_code in error_code_suggestions:
+                logger.error("Hint: %s", error_code_suggestions[error_code])
+
+            if use_verbose_error:
                 response.raise_for_status()
             else:
-                logger.error("Use KSCALE_VERBOSE_ERROR=1 to see the full error message")
-                logger.error("If this persists, please create an issue here: https://github.com/kscalelabs/kscale")
                 sys.exit(1)
 
         return response.json()

kscale/web/clients/robot_class.py

@@ -31,7 +31,7 @@ class RobotClassClient(BaseClient):
         data = await self._request(
             "GET",
             "/robots/",
-            auth=True,
+            auth=False,
         )
         return [RobotClass.model_validate(item) for item in data]
 
@@ -39,7 +39,7 @@ class RobotClassClient(BaseClient):
         data = await self._request(
             "GET",
             f"/robots/name/{class_name}",
-            auth=True,
+            auth=False,
         )
         return RobotClass.model_validate(data)
 
@@ -117,7 +117,12 @@ class RobotClassClient(BaseClient):
         cache_path = get_robots_dir() / class_name / "robot.tgz"
         if cache and cache_path.exists() and not should_refresh_file(cache_path):
             return cache_path
-        data = await self._request("GET", f"/robots/urdf/{class_name}", auth=True)
+        data = await self._request(
+            "GET",
+            f"/robots/urdf/{class_name}",
+            auth=False,
+            error_code_suggestions={404: "Use `kscale robot list` to view classes."},
+        )
         response = RobotDownloadURDFResponse.model_validate(data)
         expected_hash = response.md5_hash
         cache_path.parent.mkdir(parents=True, exist_ok=True)

kscale/web/gen/api.py

@@ -2,7 +2,7 @@
 
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2025-01-22T23:12:27+00:00
+#   timestamp: 2025-05-03T21:02:12+00:00
 
 from __future__ import annotations
 
@@ -37,8 +37,8 @@ class JointMetadataInput(BaseModel):
     offset: Optional[Union[float, str]] = Field(None, title="Offset")
     flipped: Optional[bool] = Field(None, title="Flipped")
     actuator_type: Optional[str] = Field(None, title="Actuator Type")
-    nn_id: Optional[int] = Field(None, title="Neural Network Id")
-    soft_torque_limit: Optional[str] = Field(None, title="Soft Torque Limit")
+    nn_id: Optional[int] = Field(None, title="Nn Id")
+    soft_torque_limit: Optional[Union[float, str]] = Field(None, title="Soft Torque Limit")
 
 
 class JointMetadataOutput(BaseModel):
@@ -50,6 +50,8 @@ class JointMetadataOutput(BaseModel):
     offset: Optional[str] = Field(None, title="Offset")
     flipped: Optional[bool] = Field(None, title="Flipped")
     actuator_type: Optional[str] = Field(None, title="Actuator Type")
+    nn_id: Optional[int] = Field(None, title="Nn Id")
+    soft_torque_limit: Optional[str] = Field(None, title="Soft Torque Limit")
 
 
 class OICDInfo(BaseModel):
@@ -139,3 +141,4 @@ class RobotClass(BaseModel):
     description: str = Field(..., title="Description")
     user_id: str = Field(..., title="User Id")
     metadata: Optional[RobotURDFMetadataOutput] = None
+    num_downloads: Optional[int] = Field(0, title="Num Downloads")

{kscale-0.3.10.dist-info → kscale-0.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kscale
-Version: 0.3.10
+Version: 0.3.12
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte

{kscale-0.3.10.dist-info → kscale-0.3.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kscale/__init__.py,sha256=OrJ0sOz8gJCA-Z3BRE8pwaZr9SYTmtm1K4gKbvdtpI8,201
+kscale/__init__.py,sha256=dvs5DUTbLSWgAycehACzl60WVsJdFN7yCMHoTvxFI38,201
 kscale/cli.py,sha256=JvaPtmWvF7s0D4I3K98eZAItf3oOi2ULsn5aPGxDcu4,795
 kscale/conf.py,sha256=dm35XSnzJp93St-ixVtYN4Nvqvb5upPGBrWkSI6Yb-4,1743
 kscale/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,19 +15,19 @@ kscale/web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/web/utils.py,sha256=Mme-FAQ0_zbjjOQeX8wyq8F4kL4i9fH7ytri16U6qOA,1046
 kscale/web/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/web/cli/robot.py,sha256=rI-A4_0uvJPeA71Apl4Z3mV5fIfWkgmzT9JRmJYxz3A,3307
-kscale/web/cli/robot_class.py,sha256=DLB5ok69JAeUUeA7Yy80VmyZGFt9dKf8kD0vSBMmqP8,18987
+kscale/web/cli/robot_class.py,sha256=Gzg1NTD1iDf04jPVRz_fCKVAjnQ3pmA7dB3KmUjxXwk,19080
 kscale/web/cli/user.py,sha256=9IGsJBPyhjsmT04mZ2RGOs35ePfqB2RltYP0Ty5zF5o,1290
 kscale/web/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kscale/web/clients/base.py,sha256=ThQakuvGh7CquiGQ4wiCyCWAgqmKInsJcANo910X9Uw,15585
+kscale/web/clients/base.py,sha256=P5AZdoawNpocgx_8j5v0eauowG5suUf8SnnD7q1_gx0,16213
 kscale/web/clients/client.py,sha256=rzW2s8T7bKVuybOSQ65-ghl02rcXBoOxnx_nUDwgEPw,362
 kscale/web/clients/robot.py,sha256=PI8HHkU-4Re9I5rLpp6dGbekRE-rBNVfXZxR_mO2MqE,1485
-kscale/web/clients/robot_class.py,sha256=G8Nk6V7LGJE9Wpg9tyyCkIfz1fRTsxXQRgHtleiUVqo,6834
+kscale/web/clients/robot_class.py,sha256=mgA-95P8lR0eCIF7-9f7oPE5tQ8tYLOIU7a83dFiqCs,6970
 kscale/web/clients/user.py,sha256=jsa1_s6qXRM-AGBbHlPhd1NierUtynjY9tVAPNr6_Os,568
 kscale/web/gen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kscale/web/gen/api.py,sha256=ZzOtY6zC8mgLfzmCcb7tN2rj8Bt9XUC_JBMCfJPt5ww,4946
-kscale-0.3.10.dist-info/licenses/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
-kscale-0.3.10.dist-info/METADATA,sha256=adNko0KwPbaTni8Bk3yi67kB3bUmgROXp0sgtWInVIs,2343
-kscale-0.3.10.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-kscale-0.3.10.dist-info/entry_points.txt,sha256=N_0pCpPnwGDYVzOeuaSOrbJkS5L3lS9d8CxpJF1f8UI,62
-kscale-0.3.10.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
-kscale-0.3.10.dist-info/RECORD,,
+kscale/web/gen/api.py,sha256=VSTHOLFtsjH_9mGYChW0iDbtX5FlLU8Itz5MzDb4maE,5147
+kscale-0.3.12.dist-info/licenses/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
+kscale-0.3.12.dist-info/METADATA,sha256=Pqgb5F9FCKrg6aoVgzHzmqKTT229TOMWOBqOa1e9SfM,2343
+kscale-0.3.12.dist-info/WHEEL,sha256=GHB6lJx2juba1wDgXDNlMTyM13ckjBMKf-OnwgKOCtA,91
+kscale-0.3.12.dist-info/entry_points.txt,sha256=N_0pCpPnwGDYVzOeuaSOrbJkS5L3lS9d8CxpJF1f8UI,62
+kscale-0.3.12.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
+kscale-0.3.12.dist-info/RECORD,,

{kscale-0.3.10.dist-info → kscale-0.3.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any