robodock-cli 0.1.0__py3-none-any.whl

fast_api_client/models/project_create_request.py

@@ -0,0 +1,174 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.project_create_request_default_configs import ProjectCreateRequestDefaultConfigs
+
+
+T = TypeVar("T", bound="ProjectCreateRequest")
+
+
+@_attrs_define
+class ProjectCreateRequest:
+    """
+    Attributes:
+        owner (str): 创建人
+        train_arc (str): 模型架构: Gr00t N1.6/Fii_FALCON_0/FII_FALCON_SMT
+        default_configs (ProjectCreateRequestDefaultConfigs): Default configurations including dataset
+        description (None | str | Unset): 项目描述
+        label (None | str | Unset): 标签
+        robot_type (None | str | Unset): 机器人类型: fii_w2/actibot/umi
+        first_run_name (str | Unset): Initial run name Default: 'first_run'.
+        extra_info (None | str | Unset): 额外信息
+    """
+
+    owner: str
+    train_arc: str
+    default_configs: ProjectCreateRequestDefaultConfigs
+    description: None | str | Unset = UNSET
+    label: None | str | Unset = UNSET
+    robot_type: None | str | Unset = UNSET
+    first_run_name: str | Unset = "first_run"
+    extra_info: None | str | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        owner = self.owner
+
+        train_arc = self.train_arc
+
+        default_configs = self.default_configs.to_dict()
+
+        description: None | str | Unset
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        label: None | str | Unset
+        if isinstance(self.label, Unset):
+            label = UNSET
+        else:
+            label = self.label
+
+        robot_type: None | str | Unset
+        if isinstance(self.robot_type, Unset):
+            robot_type = UNSET
+        else:
+            robot_type = self.robot_type
+
+        first_run_name = self.first_run_name
+
+        extra_info: None | str | Unset
+        if isinstance(self.extra_info, Unset):
+            extra_info = UNSET
+        else:
+            extra_info = self.extra_info
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "owner": owner,
+                "train_arc": train_arc,
+                "default_configs": default_configs,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+        if label is not UNSET:
+            field_dict["label"] = label
+        if robot_type is not UNSET:
+            field_dict["robot_type"] = robot_type
+        if first_run_name is not UNSET:
+            field_dict["first_run_name"] = first_run_name
+        if extra_info is not UNSET:
+            field_dict["extra_info"] = extra_info
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.project_create_request_default_configs import ProjectCreateRequestDefaultConfigs  # noqa: PLC0415
+
+        d = dict(src_dict)
+        owner = d.pop("owner")
+
+        train_arc = d.pop("train_arc")
+
+        default_configs = ProjectCreateRequestDefaultConfigs.from_dict(d.pop("default_configs"))
+
+        def _parse_description(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        def _parse_label(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        label = _parse_label(d.pop("label", UNSET))
+
+        def _parse_robot_type(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        robot_type = _parse_robot_type(d.pop("robot_type", UNSET))
+
+        first_run_name = d.pop("first_run_name", UNSET)
+
+        def _parse_extra_info(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        extra_info = _parse_extra_info(d.pop("extra_info", UNSET))
+
+        project_create_request = cls(
+            owner=owner,
+            train_arc=train_arc,
+            default_configs=default_configs,
+            description=description,
+            label=label,
+            robot_type=robot_type,
+            first_run_name=first_run_name,
+            extra_info=extra_info,
+        )
+
+        project_create_request.additional_properties = d
+        return project_create_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/project_create_request_default_configs.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ProjectCreateRequestDefaultConfigs")
+
+
+@_attrs_define
+class ProjectCreateRequestDefaultConfigs:
+    """Default configurations including dataset"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        project_create_request_default_configs = cls()
+
+        project_create_request_default_configs.additional_properties = d
+        return project_create_request_default_configs
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/project_update_request.py

@@ -0,0 +1,175 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.project_update_request_default_configs_type_0 import ProjectUpdateRequestDefaultConfigsType0
+
+
+T = TypeVar("T", bound="ProjectUpdateRequest")
+
+
+@_attrs_define
+class ProjectUpdateRequest:
+    """
+    Attributes:
+        description (None | str | Unset):
+        label (None | str | Unset):
+        train_arc (None | str | Unset):
+        robot_type (None | str | Unset):
+        default_configs (None | ProjectUpdateRequestDefaultConfigsType0 | Unset):
+    """
+
+    description: None | str | Unset = UNSET
+    label: None | str | Unset = UNSET
+    train_arc: None | str | Unset = UNSET
+    robot_type: None | str | Unset = UNSET
+    default_configs: None | ProjectUpdateRequestDefaultConfigsType0 | Unset = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.project_update_request_default_configs_type_0 import (
+            ProjectUpdateRequestDefaultConfigsType0,  # noqa: PLC0415
+        )
+
+        description: None | str | Unset
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        label: None | str | Unset
+        if isinstance(self.label, Unset):
+            label = UNSET
+        else:
+            label = self.label
+
+        train_arc: None | str | Unset
+        if isinstance(self.train_arc, Unset):
+            train_arc = UNSET
+        else:
+            train_arc = self.train_arc
+
+        robot_type: None | str | Unset
+        if isinstance(self.robot_type, Unset):
+            robot_type = UNSET
+        else:
+            robot_type = self.robot_type
+
+        default_configs: dict[str, Any] | None | Unset
+        if isinstance(self.default_configs, Unset):
+            default_configs = UNSET
+        elif isinstance(self.default_configs, ProjectUpdateRequestDefaultConfigsType0):
+            default_configs = self.default_configs.to_dict()
+        else:
+            default_configs = self.default_configs
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if description is not UNSET:
+            field_dict["description"] = description
+        if label is not UNSET:
+            field_dict["label"] = label
+        if train_arc is not UNSET:
+            field_dict["train_arc"] = train_arc
+        if robot_type is not UNSET:
+            field_dict["robot_type"] = robot_type
+        if default_configs is not UNSET:
+            field_dict["default_configs"] = default_configs
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.project_update_request_default_configs_type_0 import (
+            ProjectUpdateRequestDefaultConfigsType0,  # noqa: PLC0415
+        )
+
+        d = dict(src_dict)
+
+        def _parse_description(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        def _parse_label(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        label = _parse_label(d.pop("label", UNSET))
+
+        def _parse_train_arc(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        train_arc = _parse_train_arc(d.pop("train_arc", UNSET))
+
+        def _parse_robot_type(data: object) -> None | str | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(None | str | Unset, data)
+
+        robot_type = _parse_robot_type(d.pop("robot_type", UNSET))
+
+        def _parse_default_configs(data: object) -> None | ProjectUpdateRequestDefaultConfigsType0 | Unset:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                default_configs_type_0 = ProjectUpdateRequestDefaultConfigsType0.from_dict(data)
+
+                return default_configs_type_0
+            except (TypeError, ValueError, AttributeError, KeyError):
+                pass
+            return cast(None | ProjectUpdateRequestDefaultConfigsType0 | Unset, data)
+
+        default_configs = _parse_default_configs(d.pop("default_configs", UNSET))
+
+        project_update_request = cls(
+            description=description,
+            label=label,
+            train_arc=train_arc,
+            robot_type=robot_type,
+            default_configs=default_configs,
+        )
+
+        project_update_request.additional_properties = d
+        return project_update_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/project_update_request_default_configs_type_0.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ProjectUpdateRequestDefaultConfigsType0")
+
+
+@_attrs_define
+class ProjectUpdateRequestDefaultConfigsType0:
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        project_update_request_default_configs_type_0 = cls()
+
+        project_update_request_default_configs_type_0.additional_properties = d
+        return project_update_request_default_configs_type_0
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/refresh_request.py

@@ -0,0 +1,61 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RefreshRequest")
+
+
+@_attrs_define
+class RefreshRequest:
+    """
+    Attributes:
+        refresh_token (str):
+    """
+
+    refresh_token: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        refresh_token = self.refresh_token
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "refresh_token": refresh_token,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        refresh_token = d.pop("refresh_token")
+
+        refresh_request = cls(
+            refresh_token=refresh_token,
+        )
+
+        refresh_request.additional_properties = d
+        return refresh_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/reset_password_request.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ResetPasswordRequest")
+
+
+@_attrs_define
+class ResetPasswordRequest:
+    """
+    Attributes:
+        token (str):
+        new_password (str):
+    """
+
+    token: str
+    new_password: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        token = self.token
+
+        new_password = self.new_password
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "token": token,
+                "new_password": new_password,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        token = d.pop("token")
+
+        new_password = d.pop("new_password")
+
+        reset_password_request = cls(
+            token=token,
+            new_password=new_password,
+        )
+
+        reset_password_request.additional_properties = d
+        return reset_password_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/rl_stage_create_request.py

@@ -0,0 +1,75 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.rl_stage_create_request_configs import RLStageCreateRequestConfigs
+
+
+T = TypeVar("T", bound="RLStageCreateRequest")
+
+
+@_attrs_define
+class RLStageCreateRequest:
+    """
+    Attributes:
+        checkpoint_path (str): 用于发起 RL 的 SFT checkpoint path
+        configs (RLStageCreateRequestConfigs): RL stage configurations
+    """
+
+    checkpoint_path: str
+    configs: RLStageCreateRequestConfigs
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        checkpoint_path = self.checkpoint_path
+
+        configs = self.configs.to_dict()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "checkpoint_path": checkpoint_path,
+                "configs": configs,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.rl_stage_create_request_configs import RLStageCreateRequestConfigs  # noqa: PLC0415
+
+        d = dict(src_dict)
+        checkpoint_path = d.pop("checkpoint_path")
+
+        configs = RLStageCreateRequestConfigs.from_dict(d.pop("configs"))
+
+        rl_stage_create_request = cls(
+            checkpoint_path=checkpoint_path,
+            configs=configs,
+        )
+
+        rl_stage_create_request.additional_properties = d
+        return rl_stage_create_request
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

fast_api_client/models/rl_stage_create_request_configs.py

@@ -0,0 +1,47 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from typing import Any, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="RLStageCreateRequestConfigs")
+
+
+@_attrs_define
+class RLStageCreateRequestConfigs:
+    """RL stage configurations"""
+
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        rl_stage_create_request_configs = cls()
+
+        rl_stage_create_request_configs.additional_properties = d
+        return rl_stage_create_request_configs
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties