cyberdesk 2.1.11__py3-none-any.whl → 2.1.13__py3-none-any.whl

openapi_client/cyberdesk_cloud_client/models/run_response.py

@@ -29,15 +29,15 @@ class RunResponse:
         machine_id (Union[None, UUID]):
         id (UUID):
         status (RunStatus):
-        error (Union[None, list[str]]):
-        output_data (Union['RunResponseOutputDataType0', None]):
-        input_attachment_ids (Union[None, list[str]]):
-        output_attachment_ids (Union[None, list[str]]):
-        run_message_history (Union[None, list['RunResponseRunMessageHistoryType0Item']]):
-        input_values (Union['RunResponseInputValuesType0', None]):
         created_at (datetime.datetime):
         user_id (Union[None, UUID, Unset]):
         organization_id (Union[None, Unset, str]):
+        error (Union[None, Unset, list[str]]):
+        output_data (Union['RunResponseOutputDataType0', None, Unset]):
+        input_attachment_ids (Union[None, Unset, list[str]]):
+        output_attachment_ids (Union[None, Unset, list[str]]):
+        run_message_history (Union[None, Unset, list['RunResponseRunMessageHistoryType0Item']]):
+        input_values (Union['RunResponseInputValuesType0', None, Unset]):
         pool_ids (Union[None, Unset, list[UUID]]):
         sensitive_input_aliases (Union['RunResponseSensitiveInputAliasesType0', None, Unset]):
         session_id (Union[None, UUID, Unset]):
@@ -49,15 +49,15 @@ class RunResponse:
     machine_id: Union[None, UUID]
     id: UUID
     status: RunStatus
-    error: Union[None, list[str]]
-    output_data: Union["RunResponseOutputDataType0", None]
-    input_attachment_ids: Union[None, list[str]]
-    output_attachment_ids: Union[None, list[str]]
-    run_message_history: Union[None, list["RunResponseRunMessageHistoryType0Item"]]
-    input_values: Union["RunResponseInputValuesType0", None]
     created_at: datetime.datetime
     user_id: Union[None, UUID, Unset] = UNSET
     organization_id: Union[None, Unset, str] = UNSET
+    error: Union[None, Unset, list[str]] = UNSET
+    output_data: Union["RunResponseOutputDataType0", None, Unset] = UNSET
+    input_attachment_ids: Union[None, Unset, list[str]] = UNSET
+    output_attachment_ids: Union[None, Unset, list[str]] = UNSET
+    run_message_history: Union[None, Unset, list["RunResponseRunMessageHistoryType0Item"]] = UNSET
+    input_values: Union["RunResponseInputValuesType0", None, Unset] = UNSET
     pool_ids: Union[None, Unset, list[UUID]] = UNSET
     sensitive_input_aliases: Union["RunResponseSensitiveInputAliasesType0", None, Unset] = UNSET
     session_id: Union[None, UUID, Unset] = UNSET
@@ -82,35 +82,61 @@ class RunResponse:
 
         status = self.status.value
 
-        error: Union[None, list[str]]
-        if isinstance(self.error, list):
+        created_at = self.created_at.isoformat()
+
+        user_id: Union[None, Unset, str]
+        if isinstance(self.user_id, Unset):
+            user_id = UNSET
+        elif isinstance(self.user_id, UUID):
+            user_id = str(self.user_id)
+        else:
+            user_id = self.user_id
+
+        organization_id: Union[None, Unset, str]
+        if isinstance(self.organization_id, Unset):
+            organization_id = UNSET
+        else:
+            organization_id = self.organization_id
+
+        error: Union[None, Unset, list[str]]
+        if isinstance(self.error, Unset):
+            error = UNSET
+        elif isinstance(self.error, list):
             error = self.error
 
         else:
             error = self.error
 
-        output_data: Union[None, dict[str, Any]]
-        if isinstance(self.output_data, RunResponseOutputDataType0):
+        output_data: Union[None, Unset, dict[str, Any]]
+        if isinstance(self.output_data, Unset):
+            output_data = UNSET
+        elif isinstance(self.output_data, RunResponseOutputDataType0):
             output_data = self.output_data.to_dict()
         else:
             output_data = self.output_data
 
-        input_attachment_ids: Union[None, list[str]]
-        if isinstance(self.input_attachment_ids, list):
+        input_attachment_ids: Union[None, Unset, list[str]]
+        if isinstance(self.input_attachment_ids, Unset):
+            input_attachment_ids = UNSET
+        elif isinstance(self.input_attachment_ids, list):
             input_attachment_ids = self.input_attachment_ids
 
         else:
             input_attachment_ids = self.input_attachment_ids
 
-        output_attachment_ids: Union[None, list[str]]
-        if isinstance(self.output_attachment_ids, list):
+        output_attachment_ids: Union[None, Unset, list[str]]
+        if isinstance(self.output_attachment_ids, Unset):
+            output_attachment_ids = UNSET
+        elif isinstance(self.output_attachment_ids, list):
             output_attachment_ids = self.output_attachment_ids
 
         else:
             output_attachment_ids = self.output_attachment_ids
 
-        run_message_history: Union[None, list[dict[str, Any]]]
-        if isinstance(self.run_message_history, list):
+        run_message_history: Union[None, Unset, list[dict[str, Any]]]
+        if isinstance(self.run_message_history, Unset):
+            run_message_history = UNSET
+        elif isinstance(self.run_message_history, list):
             run_message_history = []
             for run_message_history_type_0_item_data in self.run_message_history:
                 run_message_history_type_0_item = run_message_history_type_0_item_data.to_dict()
@@ -119,28 +145,14 @@ class RunResponse:
         else:
             run_message_history = self.run_message_history
 
-        input_values: Union[None, dict[str, Any]]
-        if isinstance(self.input_values, RunResponseInputValuesType0):
+        input_values: Union[None, Unset, dict[str, Any]]
+        if isinstance(self.input_values, Unset):
+            input_values = UNSET
+        elif isinstance(self.input_values, RunResponseInputValuesType0):
             input_values = self.input_values.to_dict()
         else:
             input_values = self.input_values
 
-        created_at = self.created_at.isoformat()
-
-        user_id: Union[None, Unset, str]
-        if isinstance(self.user_id, Unset):
-            user_id = UNSET
-        elif isinstance(self.user_id, UUID):
-            user_id = str(self.user_id)
-        else:
-            user_id = self.user_id
-
-        organization_id: Union[None, Unset, str]
-        if isinstance(self.organization_id, Unset):
-            organization_id = UNSET
-        else:
-            organization_id = self.organization_id
-
         pool_ids: Union[None, Unset, list[str]]
         if isinstance(self.pool_ids, Unset):
             pool_ids = UNSET
@@ -189,12 +201,6 @@ class RunResponse:
                 "machine_id": machine_id,
                 "id": id,
                 "status": status,
-                "error": error,
-                "output_data": output_data,
-                "input_attachment_ids": input_attachment_ids,
-                "output_attachment_ids": output_attachment_ids,
-                "run_message_history": run_message_history,
-                "input_values": input_values,
                 "created_at": created_at,
             }
         )
@@ -202,6 +208,18 @@ class RunResponse:
             field_dict["user_id"] = user_id
         if organization_id is not UNSET:
             field_dict["organization_id"] = organization_id
+        if error is not UNSET:
+            field_dict["error"] = error
+        if output_data is not UNSET:
+            field_dict["output_data"] = output_data
+        if input_attachment_ids is not UNSET:
+            field_dict["input_attachment_ids"] = input_attachment_ids
+        if output_attachment_ids is not UNSET:
+            field_dict["output_attachment_ids"] = output_attachment_ids
+        if run_message_history is not UNSET:
+            field_dict["run_message_history"] = run_message_history
+        if input_values is not UNSET:
+            field_dict["input_values"] = input_values
         if pool_ids is not UNSET:
             field_dict["pool_ids"] = pool_ids
         if sensitive_input_aliases is not UNSET:
@@ -244,9 +262,39 @@ class RunResponse:
 
         status = RunStatus(d.pop("status"))
 
-        def _parse_error(data: object) -> Union[None, list[str]]:
+        created_at = isoparse(d.pop("created_at"))
+
+        def _parse_user_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()
+                user_id_type_0 = UUID(data)
+
+                return user_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        user_id = _parse_user_id(d.pop("user_id", UNSET))
+
+        def _parse_organization_id(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)
+
+        organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
+
+        def _parse_error(data: object) -> Union[None, Unset, list[str]]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, list):
                     raise TypeError()
@@ -255,13 +303,15 @@ class RunResponse:
                 return error_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[None, list[str]], data)
+            return cast(Union[None, Unset, list[str]], data)
 
-        error = _parse_error(d.pop("error"))
+        error = _parse_error(d.pop("error", UNSET))
 
-        def _parse_output_data(data: object) -> Union["RunResponseOutputDataType0", None]:
+        def _parse_output_data(data: object) -> Union["RunResponseOutputDataType0", None, Unset]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, dict):
                     raise TypeError()
@@ -270,13 +320,15 @@ class RunResponse:
                 return output_data_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union["RunResponseOutputDataType0", None], data)
+            return cast(Union["RunResponseOutputDataType0", None, Unset], data)
 
-        output_data = _parse_output_data(d.pop("output_data"))
+        output_data = _parse_output_data(d.pop("output_data", UNSET))
 
-        def _parse_input_attachment_ids(data: object) -> Union[None, list[str]]:
+        def _parse_input_attachment_ids(data: object) -> Union[None, Unset, list[str]]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, list):
                     raise TypeError()
@@ -285,13 +337,15 @@ class RunResponse:
                 return input_attachment_ids_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[None, list[str]], data)
+            return cast(Union[None, Unset, list[str]], data)
 
-        input_attachment_ids = _parse_input_attachment_ids(d.pop("input_attachment_ids"))
+        input_attachment_ids = _parse_input_attachment_ids(d.pop("input_attachment_ids", UNSET))
 
-        def _parse_output_attachment_ids(data: object) -> Union[None, list[str]]:
+        def _parse_output_attachment_ids(data: object) -> Union[None, Unset, list[str]]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, list):
                     raise TypeError()
@@ -300,13 +354,17 @@ class RunResponse:
                 return output_attachment_ids_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[None, list[str]], data)
+            return cast(Union[None, Unset, list[str]], data)
 
-        output_attachment_ids = _parse_output_attachment_ids(d.pop("output_attachment_ids"))
+        output_attachment_ids = _parse_output_attachment_ids(d.pop("output_attachment_ids", UNSET))
 
-        def _parse_run_message_history(data: object) -> Union[None, list["RunResponseRunMessageHistoryType0Item"]]:
+        def _parse_run_message_history(
+            data: object,
+        ) -> Union[None, Unset, list["RunResponseRunMessageHistoryType0Item"]]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, list):
                     raise TypeError()
@@ -322,13 +380,15 @@ class RunResponse:
                 return run_message_history_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union[None, list["RunResponseRunMessageHistoryType0Item"]], data)
+            return cast(Union[None, Unset, list["RunResponseRunMessageHistoryType0Item"]], data)
 
-        run_message_history = _parse_run_message_history(d.pop("run_message_history"))
+        run_message_history = _parse_run_message_history(d.pop("run_message_history", UNSET))
 
-        def _parse_input_values(data: object) -> Union["RunResponseInputValuesType0", None]:
+        def _parse_input_values(data: object) -> Union["RunResponseInputValuesType0", None, Unset]:
             if data is None:
                 return data
+            if isinstance(data, Unset):
+                return data
             try:
                 if not isinstance(data, dict):
                     raise TypeError()
@@ -337,37 +397,9 @@ class RunResponse:
                 return input_values_type_0
             except:  # noqa: E722
                 pass
-            return cast(Union["RunResponseInputValuesType0", None], data)
-
-        input_values = _parse_input_values(d.pop("input_values"))
+            return cast(Union["RunResponseInputValuesType0", None, Unset], data)
 
-        created_at = isoparse(d.pop("created_at"))
-
-        def _parse_user_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()
-                user_id_type_0 = UUID(data)
-
-                return user_id_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[None, UUID, Unset], data)
-
-        user_id = _parse_user_id(d.pop("user_id", UNSET))
-
-        def _parse_organization_id(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)
-
-        organization_id = _parse_organization_id(d.pop("organization_id", UNSET))
+        input_values = _parse_input_values(d.pop("input_values", UNSET))
 
         def _parse_pool_ids(data: object) -> Union[None, Unset, list[UUID]]:
             if data is None:
@@ -448,15 +480,15 @@ class RunResponse:
             machine_id=machine_id,
             id=id,
             status=status,
+            created_at=created_at,
+            user_id=user_id,
+            organization_id=organization_id,
             error=error,
             output_data=output_data,
             input_attachment_ids=input_attachment_ids,
             output_attachment_ids=output_attachment_ids,
             run_message_history=run_message_history,
             input_values=input_values,
-            created_at=created_at,
-            user_id=user_id,
-            organization_id=organization_id,
             pool_ids=pool_ids,
             sensitive_input_aliases=sensitive_input_aliases,
             session_id=session_id,

openapi_client/cyberdesk_cloud_client/models/run_retry.py

@@ -0,0 +1,312 @@
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, TypeVar, Union, cast
+from uuid import UUID
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, Unset
+
+if TYPE_CHECKING:
+    from ..models.file_input import FileInput
+    from ..models.run_retry_input_values_type_0 import RunRetryInputValuesType0
+    from ..models.run_retry_sensitive_input_values_type_0 import RunRetrySensitiveInputValuesType0
+
+
+T = TypeVar("T", bound="RunRetry")
+
+
+@_attrs_define
+class RunRetry:
+    """Options for retrying an existing run in-place (same run_id).
+
+    Notes:
+    - If `file_inputs` are provided, existing input attachments are replaced.
+    - Prior outputs, history, and output attachments are always cleared as part of retry.
+    - Retry is only allowed for terminal runs (success, error, or cancelled).
+
+        Attributes:
+            input_values (Union['RunRetryInputValuesType0', None, Unset]): Override input values for workflow variables
+            sensitive_input_values (Union['RunRetrySensitiveInputValuesType0', None, Unset]): Provide new sensitive inputs;
+                stored in vault and mapped to aliases
+            file_inputs (Union[None, Unset, list['FileInput']]): Provide new input files for this retry; replaces existing
+                input attachments
+            machine_id (Union[None, UUID, Unset]): Override specific machine for this retry
+            pool_ids (Union[None, Unset, list[UUID]]): Override pool filters if not using a specific machine
+            reuse_session (Union[None, Unset, bool]): Keep existing session_id. If false and no session_id provided, clears
+                session fields Default: True.
+            session_id (Union[None, UUID, Unset]): Set/override session_id for this retry
+            release_session_after (Union[None, Unset, bool]): Override release_session_after behavior for this retry
+    """
+
+    input_values: Union["RunRetryInputValuesType0", None, Unset] = UNSET
+    sensitive_input_values: Union["RunRetrySensitiveInputValuesType0", None, Unset] = UNSET
+    file_inputs: Union[None, Unset, list["FileInput"]] = UNSET
+    machine_id: Union[None, UUID, Unset] = UNSET
+    pool_ids: Union[None, Unset, list[UUID]] = UNSET
+    reuse_session: Union[None, Unset, bool] = True
+    session_id: Union[None, UUID, Unset] = UNSET
+    release_session_after: Union[None, Unset, bool] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.run_retry_input_values_type_0 import RunRetryInputValuesType0
+        from ..models.run_retry_sensitive_input_values_type_0 import RunRetrySensitiveInputValuesType0
+
+        input_values: Union[None, Unset, dict[str, Any]]
+        if isinstance(self.input_values, Unset):
+            input_values = UNSET
+        elif isinstance(self.input_values, RunRetryInputValuesType0):
+            input_values = self.input_values.to_dict()
+        else:
+            input_values = self.input_values
+
+        sensitive_input_values: Union[None, Unset, dict[str, Any]]
+        if isinstance(self.sensitive_input_values, Unset):
+            sensitive_input_values = UNSET
+        elif isinstance(self.sensitive_input_values, RunRetrySensitiveInputValuesType0):
+            sensitive_input_values = self.sensitive_input_values.to_dict()
+        else:
+            sensitive_input_values = self.sensitive_input_values
+
+        file_inputs: Union[None, Unset, list[dict[str, Any]]]
+        if isinstance(self.file_inputs, Unset):
+            file_inputs = UNSET
+        elif isinstance(self.file_inputs, list):
+            file_inputs = []
+            for file_inputs_type_0_item_data in self.file_inputs:
+                file_inputs_type_0_item = file_inputs_type_0_item_data.to_dict()
+                file_inputs.append(file_inputs_type_0_item)
+
+        else:
+            file_inputs = self.file_inputs
+
+        machine_id: Union[None, Unset, str]
+        if isinstance(self.machine_id, Unset):
+            machine_id = UNSET
+        elif isinstance(self.machine_id, UUID):
+            machine_id = str(self.machine_id)
+        else:
+            machine_id = self.machine_id
+
+        pool_ids: Union[None, Unset, list[str]]
+        if isinstance(self.pool_ids, Unset):
+            pool_ids = UNSET
+        elif isinstance(self.pool_ids, list):
+            pool_ids = []
+            for pool_ids_type_0_item_data in self.pool_ids:
+                pool_ids_type_0_item = str(pool_ids_type_0_item_data)
+                pool_ids.append(pool_ids_type_0_item)
+
+        else:
+            pool_ids = self.pool_ids
+
+        reuse_session: Union[None, Unset, bool]
+        if isinstance(self.reuse_session, Unset):
+            reuse_session = UNSET
+        else:
+            reuse_session = self.reuse_session
+
+        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
+
+        release_session_after: Union[None, Unset, bool]
+        if isinstance(self.release_session_after, Unset):
+            release_session_after = UNSET
+        else:
+            release_session_after = self.release_session_after
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if input_values is not UNSET:
+            field_dict["input_values"] = input_values
+        if sensitive_input_values is not UNSET:
+            field_dict["sensitive_input_values"] = sensitive_input_values
+        if file_inputs is not UNSET:
+            field_dict["file_inputs"] = file_inputs
+        if machine_id is not UNSET:
+            field_dict["machine_id"] = machine_id
+        if pool_ids is not UNSET:
+            field_dict["pool_ids"] = pool_ids
+        if reuse_session is not UNSET:
+            field_dict["reuse_session"] = reuse_session
+        if session_id is not UNSET:
+            field_dict["session_id"] = session_id
+        if release_session_after is not UNSET:
+            field_dict["release_session_after"] = release_session_after
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: type[T], src_dict: Mapping[str, Any]) -> T:
+        from ..models.file_input import FileInput
+        from ..models.run_retry_input_values_type_0 import RunRetryInputValuesType0
+        from ..models.run_retry_sensitive_input_values_type_0 import RunRetrySensitiveInputValuesType0
+
+        d = dict(src_dict)
+
+        def _parse_input_values(data: object) -> Union["RunRetryInputValuesType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                input_values_type_0 = RunRetryInputValuesType0.from_dict(data)
+
+                return input_values_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["RunRetryInputValuesType0", None, Unset], data)
+
+        input_values = _parse_input_values(d.pop("input_values", UNSET))
+
+        def _parse_sensitive_input_values(data: object) -> Union["RunRetrySensitiveInputValuesType0", None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                sensitive_input_values_type_0 = RunRetrySensitiveInputValuesType0.from_dict(data)
+
+                return sensitive_input_values_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["RunRetrySensitiveInputValuesType0", None, Unset], data)
+
+        sensitive_input_values = _parse_sensitive_input_values(d.pop("sensitive_input_values", UNSET))
+
+        def _parse_file_inputs(data: object) -> Union[None, Unset, list["FileInput"]]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                file_inputs_type_0 = []
+                _file_inputs_type_0 = data
+                for file_inputs_type_0_item_data in _file_inputs_type_0:
+                    file_inputs_type_0_item = FileInput.from_dict(file_inputs_type_0_item_data)
+
+                    file_inputs_type_0.append(file_inputs_type_0_item)
+
+                return file_inputs_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, list["FileInput"]], data)
+
+        file_inputs = _parse_file_inputs(d.pop("file_inputs", UNSET))
+
+        def _parse_machine_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()
+                machine_id_type_0 = UUID(data)
+
+                return machine_id_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, UUID, Unset], data)
+
+        machine_id = _parse_machine_id(d.pop("machine_id", UNSET))
+
+        def _parse_pool_ids(data: object) -> Union[None, Unset, list[UUID]]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                pool_ids_type_0 = []
+                _pool_ids_type_0 = data
+                for pool_ids_type_0_item_data in _pool_ids_type_0:
+                    pool_ids_type_0_item = UUID(pool_ids_type_0_item_data)
+
+                    pool_ids_type_0.append(pool_ids_type_0_item)
+
+                return pool_ids_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, list[UUID]], data)
+
+        pool_ids = _parse_pool_ids(d.pop("pool_ids", UNSET))
+
+        def _parse_reuse_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)
+
+        reuse_session = _parse_reuse_session(d.pop("reuse_session", 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_release_session_after(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)
+
+        release_session_after = _parse_release_session_after(d.pop("release_session_after", UNSET))
+
+        run_retry = cls(
+            input_values=input_values,
+            sensitive_input_values=sensitive_input_values,
+            file_inputs=file_inputs,
+            machine_id=machine_id,
+            pool_ids=pool_ids,
+            reuse_session=reuse_session,
+            session_id=session_id,
+            release_session_after=release_session_after,
+        )
+
+        run_retry.additional_properties = d
+        return run_retry
+
+    @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