kscale 0.3.0__tar.gz → 0.3.1__tar.gz

{kscale-0.3.0/kscale.egg-info → kscale-0.3.1}/PKG-INFO

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

{kscale-0.3.0 → kscale-0.3.1}/kscale/__init__.py

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

{kscale-0.3.0 → kscale-0.3.1}/kscale/web/cli/robot_class.py

@@ -16,6 +16,11 @@ from kscale.web.gen.api import RobotURDFMetadataInput
 logger = logging.getLogger(__name__)
 
 
+class RobotURDFMetadataInputStrict(RobotURDFMetadataInput):
+    class Config:
+        extra = "forbid"
+
+
 @click.group()
 def cli() -> None:
     """Get information about robot classes."""
@@ -83,7 +88,7 @@ async def update_metadata(name: str, json_path: str) -> None:
     """Updates the metadata of a robot class."""
     with open(json_path, "r", encoding="utf-8") as f:
         raw_metadata = json.load(f)
-    metadata = RobotURDFMetadataInput.model_validate(raw_metadata)
+    metadata = RobotURDFMetadataInputStrict.model_validate(raw_metadata)
     async with RobotClassClient() as client:
         robot_class = await client.update_robot_class(name, new_metadata=metadata)
     click.echo("Robot class metadata updated:")
@@ -91,6 +96,21 @@ async def update_metadata(name: str, json_path: str) -> None:
     click.echo(f"  Name: {click.style(robot_class.class_name, fg='green')}")
 
 
+@cli.command()
+@click.argument("name")
+@click.option("--json-path", type=click.Path(exists=False))
+@coro
+async def get_metadata(name: str, json_path: str | None = None) -> None:
+    """Gets the metadata of a robot class."""
+    async with RobotClassClient() as client:
+        robot_class = await client.get_robot_class(name)
+    if json_path is None:
+        click.echo(robot_class.metadata.model_dump_json(indent=2))
+    else:
+        with open(json_path, "w", encoding="utf-8") as f:
+            json.dump(robot_class.model_dump(), f)
+
+
 @cli.command()
 @click.argument("name")
 @coro

{kscale-0.3.0 → kscale-0.3.1}/kscale/web/clients/robot_class.py

@@ -35,6 +35,14 @@ class RobotClassClient(BaseClient):
         )
         return [RobotClass.model_validate(item) for item in data]
 
+    async def get_robot_class(self, class_name: str) -> RobotClass:
+        data = await self._request(
+            "GET",
+            f"/robots/name/{class_name}",
+            auth=True,
+        )
+        return RobotClass.model_validate(data)
+
     async def create_robot_class(self, class_name: str, description: str | None = None) -> RobotClass:
         data = {}
         if description is not None:

{kscale-0.3.0 → kscale-0.3.1}/kscale/web/gen/api.py

@@ -2,7 +2,7 @@
 
 # generated by datamodel-codegen:
 #   filename:  openapi.json
-#   timestamp: 2025-01-21T07:10:22+00:00
+#   timestamp: 2025-01-22T04:20:37+00:00
 
 from __future__ import annotations
 
@@ -11,6 +11,10 @@ from typing import Dict, List, Optional, Union
 from pydantic import BaseModel, Field
 
 
+class APIKeyRequest(BaseModel):
+    num_hours: Optional[int] = Field(24, title="Num Hours")
+
+
 class APIKeyResponse(BaseModel):
     api_key: str = Field(..., title="Api Key")
 
@@ -29,8 +33,7 @@ class JointMetadataInput(BaseModel):
     kp: Optional[Union[float, str]] = Field(None, title="Kp")
     kd: Optional[Union[float, str]] = Field(None, title="Kd")
     offset: Optional[Union[float, str]] = Field(None, title="Offset")
-    lower_limit: Optional[Union[float, str]] = Field(None, title="Lower Limit")
-    upper_limit: Optional[Union[float, str]] = Field(None, title="Upper Limit")
+    flipped: Optional[bool] = Field(None, title="Flipped")
 
 
 class JointMetadataOutput(BaseModel):
@@ -38,8 +41,7 @@ class JointMetadataOutput(BaseModel):
     kp: Optional[str] = Field(None, title="Kp")
     kd: Optional[str] = Field(None, title="Kd")
     offset: Optional[str] = Field(None, title="Offset")
-    lower_limit: Optional[str] = Field(None, title="Lower Limit")
-    upper_limit: Optional[str] = Field(None, title="Upper Limit")
+    flipped: Optional[bool] = Field(None, title="Flipped")
 
 
 class OICDInfo(BaseModel):

{kscale-0.3.0 → kscale-0.3.1/kscale.egg-info}/PKG-INFO

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