chore: update models and api types

Author: nfrasser

cryosparc/api.pyi

@@ -1349,7 +1349,7 @@ class JobsAPI(APINamespace):
 
         """
         ...
-    def set_param(self, project_uid: str, job_uid: str, param: str, /, *, value: Any) -> Job:
+    def set_param(self, project_uid: str, job_uid: str, param: str, /, value: Any) -> Job:
         """
         Sets the given job parameter to the value
 
@@ -1710,9 +1710,12 @@ class JobsAPI(APINamespace):
         self, project_uid: str, job_uid: str, output_name: str, /, result_names: Optional[List[str]] = None
     ) -> None:
         """
-        Prepares a job's output for import to another project or instance. Creates a folder in the project directory → exports subfolder,
-        then links the output's associated files there..
-        Note that the returned .csg file's parent folder must be manually copied with symlinks resolved into the target project folder before importing.
+        Prepares a job's output for import to another project or instance.
+        Creates a folder in the project directory → exports subfolder,
+        then links the output's associated files there.
+
+        Note that the returned .csg file's parent folder must be manually copied
+        with symlinks resolved into the target project folder before importing.
 
         Args:
             project_uid (str): Project UID, e.g., "P3"
@@ -2688,6 +2691,31 @@ class SessionsAPI(APINamespace):
         Returns:
             Session: Successful Response
 
+        """
+        ...
+    def clone(
+        self,
+        project_uid: str,
+        session_uid: str,
+        /,
+        *,
+        title: Optional[str] = None,
+        description: Optional[str] = None,
+        created_by_job_uid: Optional[str] = None,
+    ) -> Session:
+        """
+        Clones an existing session, copying session configuration, parameters, and exposure groups.
+
+        Args:
+            project_uid (str): Project UID, e.g., "P3"
+            session_uid (str): Session UID, e.g., "S3"
+            title (str, optional): Defaults to None
+            description (str, optional): Defaults to None
+            created_by_job_uid (str, optional): Defaults to None
+
+        Returns:
+            Session: Successful Response
+
         """
         ...
     def find_exposure_groups(self, project_uid: str, session_uid: str, /) -> List[ExposureGroup]:
@@ -2801,6 +2829,27 @@ class SessionsAPI(APINamespace):
         Returns:
             Session: Successful Response
 
+        """
+        ...
+    def configure_auto_pause(
+        self,
+        project_uid: str,
+        session_uid: str,
+        /,
+        *,
+        auto_pause: Literal["disabled", "graceful", "immediate"],
+        auto_pause_after_idle_minutes: int = 10,
+    ) -> Session:
+        """
+        Args:
+            project_uid (str): Project UID, e.g., "P3"
+            session_uid (str): Session UID, e.g., "S3"
+            auto_pause (Literal['disabled', 'graceful', 'immediate']):
+            auto_pause_after_idle_minutes (int, optional): Defaults to 10
+
+        Returns:
+            Session: Successful Response
+
         """
         ...
     def update_compute_configuration(
@@ -3117,6 +3166,22 @@ class SessionsAPI(APINamespace):
         Returns:
             Session: Successful Response
 
+        """
+        ...
+    def set_session_phase_one_wait_for_exposures(
+        self, project_uid: str, session_uid: str, /, *, phase_one_wait_for_exposures: bool
+    ) -> Session:
+        """
+        Sets whether to wait until exposures are available before queuing the session's preprocessing worker jobs.
+
+        Args:
+            project_uid (str): Project UID, e.g., "P3"
+            session_uid (str): Session UID, e.g., "S3"
+            phase_one_wait_for_exposures (bool):
+
+        Returns:
+            Session: Successful Response
+
         """
         ...
     def update_picking_threshold_values(
@@ -3914,6 +3979,20 @@ class ExposuresAPI(APINamespace):
         Returns:
             Exposure: Unrejected exposure
 
+        """
+        ...
+    def mark_failed(self, project_uid: str, session_uid: str, exposure_uid: int, /) -> Exposure:
+        """
+        Mark an exposure as failed.
+
+        Args:
+            project_uid (str): Project UID, e.g., "P3"
+            session_uid (str): Session UID, e.g., "S3"
+            exposure_uid (int):
+
+        Returns:
+            Exposure: Successful Response
+
         """
         ...
     def toggle_manual_reject_exposure(self, project_uid: str, session_uid: str, exposure_uid: int, /) -> Exposure:
@@ -4521,6 +4600,7 @@ class ProjectsAPI(APINamespace):
         clear_categories: List[Category] = [],
         clear_types: List[str] = [],
         clear_statuses: List[JobStatus] = [],
+        clear_preprocessing: bool = False,
     ) -> None:
         """
         Cleanup project or workspace data, clearing/deleting jobs based on final result status, sections, types, or job status
@@ -4534,6 +4614,7 @@ class ProjectsAPI(APINamespace):
             clear_categories (List[Category], optional): Defaults to []
             clear_types (List[str], optional): Defaults to []
             clear_statuses (List[JobStatus], optional): Defaults to []
+            clear_preprocessing (bool, optional): Defaults to False
 
         """
         ...

cryosparc/models/session.py

@@ -91,6 +91,7 @@ class LiveComputeResources(BaseModel):
     auxiliary_gpus: int = 1
     auxiliary_ssd: bool = True
     priority: int = 0
+    phase_one_workers_per_gpu: int = 1
 
 
 class LiveVolumeInfo(BaseModel):
@@ -343,6 +344,7 @@ class Session(BaseModel):
     compute_resources: LiveComputeResources = LiveComputeResources()
     notes: str = ""
     notes_lock: Optional[str] = None
+    phase_one_wait_for_exposures: bool = False
     phase_one_workers: Dict[str, RtpWorkerState] = {}
     """
     Example: {'J1': RtpWorkerState(status=<JobStatus.BUILDING: 'building'>, errors=[]),
@@ -431,11 +433,13 @@ class Session(BaseModel):
     phase2_refine_ready_partial: bool = False
     phase2_refine_num_particles_in: int = 0
     phase2_refine_last_updated: Optional[datetime.datetime] = None
-    athena_epu_run_id: Optional[str] = None
-    is_multigrid_epu_run: bool = False
+    auto_pause: Literal["disabled", "graceful", "immediate"] = "disabled"
+    auto_pause_after_idle_minutes: int = 10
     is_gracefully_pausing: bool = False
     computed_stats_last_run_time: Optional[datetime.datetime] = None
     last_processed_exposure_priority: Literal["normal", "oldest", "latest", "alternate"] = "oldest"
+    athena_epu_run_id: Optional[str] = None
+    is_multigrid_epu_run: bool = False
     ecl: ECLSessionProperties = ECLSessionProperties()
     uid_num: int
     project_uid_num: int

cryosparc/models/session_params.py

@@ -265,7 +265,7 @@ def __getattr__(self, key: str) -> Any: ...
 class LiveAbinitParams(BaseModel):
     abinit_symmetry: str = "C1"
     abinit_K: int = 1
-    abinit_num_particles: Optional[int] = None
+    abinit_num_particles: Optional[int] = 100000
 
     model_config = ConfigDict(extra="allow")
     if TYPE_CHECKING: