cyberdesk 2.1.5__py3-none-any.whl → 2.1.6__py3-none-any.whl

openapi_client/cyberdesk_cloud_client/models/chain_step_inputs_type_0.py

@@ -0,0 +1,74 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.ref_value import RefValue
+
+
+T = TypeVar("T", bound="ChainStepInputsType0")
+
+
+@_attrs_define
+class ChainStepInputsType0:
+    """ """
+
+    additional_properties: dict[str, Union["RefValue", str]] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.ref_value import RefValue
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            if isinstance(prop, RefValue):
+                field_dict[prop_name] = prop.to_dict()
+            else:
+                field_dict[prop_name] = prop
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.ref_value import RefValue
+
+        d = dict(src_dict)
+        chain_step_inputs_type_0 = cls()
+
+        additional_properties = {}
+        for prop_name, prop_dict in d.items():
+
+            def _parse_additional_property(data: object) -> Union["RefValue", str]:
+                try:
+                    if not isinstance(data, dict):
+                        raise TypeError()
+                    additional_property_type_1 = RefValue.from_dict(data)
+
+                    return additional_property_type_1
+                except:  # noqa: E722
+                    pass
+                return cast(Union["RefValue", str], data)
+
+            additional_property = _parse_additional_property(prop_dict)
+
+            additional_properties[prop_name] = additional_property
+
+        chain_step_inputs_type_0.additional_properties = additional_properties
+        return chain_step_inputs_type_0
+
+    @property
+    def additional_keys(self) -> list[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Union["RefValue", str]:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Union["RefValue", str]) -> 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

openapi_client/cyberdesk_cloud_client/models/machine_response.py

@@ -35,6 +35,7 @@ class MachineResponse:
         os_info (Union[None, Unset, str]):
         user_id (Union[None, UUID, Unset]):
         organization_id (Union[None, Unset, str]):
+        reserved_session_id (Union[None, UUID, Unset]):
         pools (Union[None, Unset, list['PoolResponse']]):
     """
 
@@ -51,6 +52,7 @@ class MachineResponse:
     os_info: Union[None, Unset, str] = UNSET
     user_id: Union[None, UUID, Unset] = UNSET
     organization_id: Union[None, Unset, str] = UNSET
+    reserved_session_id: Union[None, UUID, Unset] = UNSET
     pools: Union[None, Unset, list["PoolResponse"]] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -107,6 +109,14 @@ class MachineResponse:
         else:
             organization_id = self.organization_id
 
+        reserved_session_id: Union[None, Unset, str]
+        if isinstance(self.reserved_session_id, Unset):
+            reserved_session_id = UNSET
+        elif isinstance(self.reserved_session_id, UUID):
+            reserved_session_id = str(self.reserved_session_id)
+        else:
+            reserved_session_id = self.reserved_session_id
+
         pools: Union[None, Unset, list[dict[str, Any]]]
         if isinstance(self.pools, Unset):
             pools = UNSET
@@ -144,6 +154,8 @@ class MachineResponse:
             field_dict["user_id"] = user_id
         if organization_id is not UNSET:
             field_dict["organization_id"] = organization_id
+        if reserved_session_id is not UNSET:
+            field_dict["reserved_session_id"] = reserved_session_id
         if pools is not UNSET:
             field_dict["pools"] = pools
 
@@ -230,6 +242,23 @@ class MachineResponse:
 
         organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
 
+        def _parse_reserved_session_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                reserved_session_id_type_0 = UUID(data)
+
+                return reserved_session_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        reserved_session_id = _parse_reserved_session_id(d.pop("reserved_session_id", UNSET))
+
         def _parse_pools(data: object) -> Union[None, Unset, list["PoolResponse"]]:
             if data is None:
                 return data
@@ -266,6 +295,7 @@ class MachineResponse:
             os_info=os_info,
             user_id=user_id,
             organization_id=organization_id,
+            reserved_session_id=reserved_session_id,
             pools=pools,
         )
 

openapi_client/cyberdesk_cloud_client/models/machine_update.py

@@ -1,6 +1,7 @@
 import datetime
 from collections.abc import Mapping
 from typing import Any, TypeVar, Union, cast
+from uuid import UUID
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
@@ -24,6 +25,8 @@ class MachineUpdate:
         status (Union[MachineStatus, None, Unset]):
         is_available (Union[None, Unset, bool]):
         last_seen (Union[None, Unset, datetime.datetime]):
+        reserved_session_id (Union[None, UUID, Unset]): Set to null to clear reservation; server will cancel
+            queued/running session runs and clear
     """
 
     name: Union[None, Unset, str] = UNSET
@@ -33,6 +36,7 @@ class MachineUpdate:
     status: Union[MachineStatus, None, Unset] = UNSET
     is_available: Union[None, Unset, bool] = UNSET
     last_seen: Union[None, Unset, datetime.datetime] = UNSET
+    reserved_session_id: Union[None, UUID, Unset] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -82,6 +86,14 @@ class MachineUpdate:
         else:
             last_seen = self.last_seen
 
+        reserved_session_id: Union[None, Unset, str]
+        if isinstance(self.reserved_session_id, Unset):
+            reserved_session_id = UNSET
+        elif isinstance(self.reserved_session_id, UUID):
+            reserved_session_id = str(self.reserved_session_id)
+        else:
+            reserved_session_id = self.reserved_session_id
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update({})
@@ -99,6 +111,8 @@ class MachineUpdate:
             field_dict["is_available"] = is_available
         if last_seen is not UNSET:
             field_dict["last_seen"] = last_seen
+        if reserved_session_id is not UNSET:
+            field_dict["reserved_session_id"] = reserved_session_id
 
         return field_dict
 
@@ -185,6 +199,23 @@ class MachineUpdate:
 
         last_seen = _parse_last_seen(d.pop("last_seen", UNSET))
 
+        def _parse_reserved_session_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                reserved_session_id_type_0 = UUID(data)
+
+                return reserved_session_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        reserved_session_id = _parse_reserved_session_id(d.pop("reserved_session_id", UNSET))
+
         machine_update = cls(
             name=name,
             version=version,
@@ -193,6 +224,7 @@ class MachineUpdate:
             status=status,
             is_available=is_available,
             last_seen=last_seen,
+            reserved_session_id=reserved_session_id,
         )
 
         machine_update.additional_properties = d

openapi_client/cyberdesk_cloud_client/models/ref_value.py

@@ -0,0 +1,61 @@
+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="RefValue")
+
+
+@_attrs_define
+class RefValue:
+    """Reference to a prior step's output within the same session.
+    The wire shape is {"$ref": "alias.outputs.path"}.
+
+        Attributes:
+            ref (str):
+    """
+
+    ref: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        ref = self.ref
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "$ref": ref,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        d = dict(src_dict)
+        ref = d.pop("$ref")
+
+        ref_value = cls(
+            ref=ref,
+        )
+
+        ref_value.additional_properties = d
+        return ref_value
+
+    @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

openapi_client/cyberdesk_cloud_client/models/run_bulk_create.py

@@ -31,6 +31,9 @@ class RunBulkCreate:
         file_inputs (Union[None, Unset, list['FileInput']]): Files to upload to the machine
         sensitive_input_values (Union['RunBulkCreateSensitiveInputValuesType0', None, Unset]): Sensitive input values to
             store in the secure vault per run. Not persisted in our database.
+        session_id (Union[None, UUID, Unset]): Join an existing session; overrides machine_id/pool_ids for all runs
+        start_session (Union[None, Unset, bool]): Start a new session for these runs; a new UUID will be generated and
+            set on all runs. The first run will attempt to reserve a machine. Default: False.
     """
 
     workflow_id: UUID
@@ -40,6 +43,8 @@ class RunBulkCreate:
     input_values: Union["RunBulkCreateInputValuesType0", None, Unset] = UNSET
     file_inputs: Union[None, Unset, list["FileInput"]] = UNSET
     sensitive_input_values: Union["RunBulkCreateSensitiveInputValuesType0", None, Unset] = UNSET
+    session_id: Union[None, UUID, Unset] = UNSET
+    start_session: Union[None, Unset, bool] = False
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -98,6 +103,20 @@ class RunBulkCreate:
         else:
             sensitive_input_values = self.sensitive_input_values
 
+        session_id: Union[None, Unset, str]
+        if isinstance(self.session_id, Unset):
+            session_id = UNSET
+        elif isinstance(self.session_id, UUID):
+            session_id = str(self.session_id)
+        else:
+            session_id = self.session_id
+
+        start_session: Union[None, Unset, bool]
+        if isinstance(self.start_session, Unset):
+            start_session = UNSET
+        else:
+            start_session = self.start_session
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -116,6 +135,10 @@ class RunBulkCreate:
             field_dict["file_inputs"] = file_inputs
         if sensitive_input_values is not UNSET:
             field_dict["sensitive_input_values"] = sensitive_input_values
+        if session_id is not UNSET:
+            field_dict["session_id"] = session_id
+        if start_session is not UNSET:
+            field_dict["start_session"] = start_session
 
         return field_dict
 
@@ -225,6 +248,32 @@ class RunBulkCreate:
 
         sensitive_input_values = _parse_sensitive_input_values(d.pop("sensitive_input_values", UNSET))
 
+        def _parse_session_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                session_id_type_0 = UUID(data)
+
+                return session_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        session_id = _parse_session_id(d.pop("session_id", UNSET))
+
+        def _parse_start_session(data: object) -> Union[None, Unset, bool]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, bool], data)
+
+        start_session = _parse_start_session(d.pop("start_session", UNSET))
+
         run_bulk_create = cls(
             workflow_id=workflow_id,
             count=count,
@@ -233,6 +282,8 @@ class RunBulkCreate:
             input_values=input_values,
             file_inputs=file_inputs,
             sensitive_input_values=sensitive_input_values,
+            session_id=session_id,
+            start_session=start_session,
         )
 
         run_bulk_create.additional_properties = d

openapi_client/cyberdesk_cloud_client/models/run_create.py

@@ -30,6 +30,10 @@ class RunCreate:
         file_inputs (Union[None, Unset, list['FileInput']]): Files to upload to the machine
         sensitive_input_values (Union['RunCreateSensitiveInputValuesType0', None, Unset]): Sensitive input values. These
             are not stored and will be written to a secure vault and referenced by alias only.
+        session_id (Union[None, UUID, Unset]): Join an existing session; overrides machine_id/pool_ids
+        start_session (Union[None, Unset, bool]): Start a new session on the machine used by this run. Session ID will
+            be this run's ID. Default: False.
+        session_alias (Union[None, Unset, str]): Persist outputs under this alias for $ref in this session
     """
 
     workflow_id: UUID
@@ -38,6 +42,9 @@ class RunCreate:
     input_values: Union["RunCreateInputValuesType0", None, Unset] = UNSET
     file_inputs: Union[None, Unset, list["FileInput"]] = UNSET
     sensitive_input_values: Union["RunCreateSensitiveInputValuesType0", None, Unset] = UNSET
+    session_id: Union[None, UUID, Unset] = UNSET
+    start_session: Union[None, Unset, bool] = False
+    session_alias: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -94,6 +101,26 @@ class RunCreate:
         else:
             sensitive_input_values = self.sensitive_input_values
 
+        session_id: Union[None, Unset, str]
+        if isinstance(self.session_id, Unset):
+            session_id = UNSET
+        elif isinstance(self.session_id, UUID):
+            session_id = str(self.session_id)
+        else:
+            session_id = self.session_id
+
+        start_session: Union[None, Unset, bool]
+        if isinstance(self.start_session, Unset):
+            start_session = UNSET
+        else:
+            start_session = self.start_session
+
+        session_alias: Union[None, Unset, str]
+        if isinstance(self.session_alias, Unset):
+            session_alias = UNSET
+        else:
+            session_alias = self.session_alias
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -111,6 +138,12 @@ class RunCreate:
             field_dict["file_inputs"] = file_inputs
         if sensitive_input_values is not UNSET:
             field_dict["sensitive_input_values"] = sensitive_input_values
+        if session_id is not UNSET:
+            field_dict["session_id"] = session_id
+        if start_session is not UNSET:
+            field_dict["start_session"] = start_session
+        if session_alias is not UNSET:
+            field_dict["session_alias"] = session_alias
 
         return field_dict
 
@@ -218,6 +251,41 @@ class RunCreate:
 
         sensitive_input_values = _parse_sensitive_input_values(d.pop("sensitive_input_values", UNSET))
 
+        def _parse_session_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                session_id_type_0 = UUID(data)
+
+                return session_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        session_id = _parse_session_id(d.pop("session_id", UNSET))
+
+        def _parse_start_session(data: object) -> Union[None, Unset, bool]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, bool], data)
+
+        start_session = _parse_start_session(d.pop("start_session", UNSET))
+
+        def _parse_session_alias(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        session_alias = _parse_session_alias(d.pop("session_alias", UNSET))
+
         run_create = cls(
             workflow_id=workflow_id,
             machine_id=machine_id,
@@ -225,6 +293,9 @@ class RunCreate:
             input_values=input_values,
             file_inputs=file_inputs,
             sensitive_input_values=sensitive_input_values,
+            session_id=session_id,
+            start_session=start_session,
+            session_alias=session_alias,
         )
 
         run_create.additional_properties = d

openapi_client/cyberdesk_cloud_client/models/run_response.py

@@ -40,6 +40,8 @@ class RunResponse:
         organization_id (Union[None, Unset, str]):
         pool_ids (Union[None, Unset, list[UUID]]):
         sensitive_input_aliases (Union['RunResponseSensitiveInputAliasesType0', None, Unset]):
+        session_id (Union[None, UUID, Unset]):
+        session_alias (Union[None, Unset, str]):
     """
 
     workflow_id: UUID
@@ -57,6 +59,8 @@ class RunResponse:
     organization_id: Union[None, Unset, str] = UNSET
     pool_ids: Union[None, Unset, list[UUID]] = UNSET
     sensitive_input_aliases: Union["RunResponseSensitiveInputAliasesType0", None, Unset] = UNSET
+    session_id: Union[None, UUID, Unset] = UNSET
+    session_alias: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -155,6 +159,20 @@ class RunResponse:
         else:
             sensitive_input_aliases = self.sensitive_input_aliases
 
+        session_id: Union[None, Unset, str]
+        if isinstance(self.session_id, Unset):
+            session_id = UNSET
+        elif isinstance(self.session_id, UUID):
+            session_id = str(self.session_id)
+        else:
+            session_id = self.session_id
+
+        session_alias: Union[None, Unset, str]
+        if isinstance(self.session_alias, Unset):
+            session_alias = UNSET
+        else:
+            session_alias = self.session_alias
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -180,6 +198,10 @@ class RunResponse:
             field_dict["pool_ids"] = pool_ids
         if sensitive_input_aliases is not UNSET:
             field_dict["sensitive_input_aliases"] = sensitive_input_aliases
+        if session_id is not UNSET:
+            field_dict["session_id"] = session_id
+        if session_alias is not UNSET:
+            field_dict["session_alias"] = session_alias
 
         return field_dict
 
@@ -376,6 +398,32 @@ class RunResponse:
 
         sensitive_input_aliases = _parse_sensitive_input_aliases(d.pop("sensitive_input_aliases", UNSET))
 
+        def _parse_session_id(data: object) -> Union[None, UUID, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                session_id_type_0 = UUID(data)
+
+                return session_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        session_id = _parse_session_id(d.pop("session_id", UNSET))
+
+        def _parse_session_alias(data: object) -> Union[None, Unset, str]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, str], data)
+
+        session_alias = _parse_session_alias(d.pop("session_alias", UNSET))
+
         run_response = cls(
             workflow_id=workflow_id,
             machine_id=machine_id,
@@ -392,6 +440,8 @@ class RunResponse:
             organization_id=organization_id,
             pool_ids=pool_ids,
             sensitive_input_aliases=sensitive_input_aliases,
+            session_id=session_id,
+            session_alias=session_alias,
         )
 
         run_response.additional_properties = d