luminarycloud 0.22.2__py3-none-any.whl → 0.23.0__py3-none-any.whl

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2.pyi

@@ -107,6 +107,56 @@ MODEL_INITIALIZATION_TYPE_CHECKPOINT: ModelInitializationType.ValueType  # 3
 """Initialize from checkpoint."""
 global___ModelInitializationType = ModelInitializationType
 
+class ParameterDefinition(google.protobuf.message.Message):
+    """Structured definition of a parameter in a dataset's parameter schema."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    DIMENSION_FIELD_NUMBER: builtins.int
+    DEFAULT_REFERENCE_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    type: builtins.str
+    """"scalar" or "vector" """
+    dimension: builtins.int
+    """1 for scalar, N for N-dimensional vector"""
+    @property
+    def default_reference(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]:
+        """optional default normalization reference"""
+    description: builtins.str
+    """optional description"""
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        type: builtins.str = ...,
+        dimension: builtins.int = ...,
+        default_reference: collections.abc.Iterable[builtins.float] | None = ...,
+        description: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["default_reference", b"default_reference", "description", b"description", "dimension", b"dimension", "name", b"name", "type", b"type"]) -> None: ...
+
+global___ParameterDefinition = ParameterDefinition
+
+class ParameterSchema(google.protobuf.message.Message):
+    """Container for parameter definitions."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PARAMETERS_FIELD_NUMBER: builtins.int
+    @property
+    def parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ParameterDefinition]: ...
+    def __init__(
+        self,
+        *,
+        parameters: collections.abc.Iterable[global___ParameterDefinition] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["parameters", b"parameters"]) -> None: ...
+
+global___ParameterSchema = ParameterSchema
+
 class PhysicsAiArchitectureVersion(google.protobuf.message.Message):
     """A physics ai architecture version."""
 
@@ -304,7 +354,11 @@ class SurfaceGroup(google.protobuf.message.Message):
 global___SurfaceGroup = SurfaceGroup
 
 class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
-    """Request message for download and process solutions for physics ai"""
+    """Request message for download and process solutions for physics ai.
+
+    Surfaces from exclude_tags are merged with exclude_surfaces to form the final
+    list of excluded surfaces.
+    """
 
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -332,11 +386,15 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
     SINGLE_PRECISION_FIELD_NUMBER: builtins.int
     INTERNAL_OPTIONS_FIELD_NUMBER: builtins.int
     EXPORT_SURFACE_GROUPS_FIELD_NUMBER: builtins.int
+    SKIP_TAR_CREATION_FIELD_NUMBER: builtins.int
+    EXCLUDE_TAGS_FIELD_NUMBER: builtins.int
     solution_id: builtins.str
     """Required. The globally unique identifier for the solution."""
     @property
     def exclude_surfaces(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """List of surfaces to exclude from surface solution during physics AI processing"""
+        """List of mesh boundary names to exclude from surface solution during physics AI processing.
+        See message-level comment for how this interacts with exclude_tags.
+        """
     fill_holes: builtins.float
     """Size threshold for filling holes in the surface mesh (0 or negative value means no filling)"""
     @property
@@ -354,7 +412,21 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
         """Internal options for physics AI processing only available for staff users."""
     @property
     def export_surface_groups(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SurfaceGroup]:
-        """List of surface groups to export as individual STL files"""
+        """List of surface groups to export as individual STL files
+        Excluded surfaces will not appear in surface groups.
+        """
+    skip_tar_creation: builtins.bool
+    """If true, skip tar.gz creation and only produce folder output with individual files.
+    Internal use only - for dataset export where tar.gz is not needed.
+    """
+    @property
+    def exclude_tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of geometry tag names (e.g., "Farfield") to exclude from surface solution.
+        Tags are resolved to mesh boundary names server-side. Returns an error if
+        exclude_tags is used on a simulation without geometry tags. Individual tags
+        that are not found are skipped gracefully to support heterogeneous datasets.
+        See message-level comment for how this interacts with exclude_surfaces.
+        """
     def __init__(
         self,
         *,
@@ -367,8 +439,10 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
         single_precision: builtins.bool = ...,
         internal_options: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
         export_surface_groups: collections.abc.Iterable[global___SurfaceGroup] | None = ...,
+        skip_tar_creation: builtins.bool = ...,
+        exclude_tags: collections.abc.Iterable[builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["exclude_surfaces", b"exclude_surfaces", "export_surface_groups", b"export_surface_groups", "fill_holes", b"fill_holes", "internal_options", b"internal_options", "process_volume", b"process_volume", "single_precision", b"single_precision", "solution_id", b"solution_id", "surface_fields_to_keep", b"surface_fields_to_keep", "volume_fields_to_keep", b"volume_fields_to_keep"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["exclude_surfaces", b"exclude_surfaces", "exclude_tags", b"exclude_tags", "export_surface_groups", b"export_surface_groups", "fill_holes", b"fill_holes", "internal_options", b"internal_options", "process_volume", b"process_volume", "single_precision", b"single_precision", "skip_tar_creation", b"skip_tar_creation", "solution_id", b"solution_id", "surface_fields_to_keep", b"surface_fields_to_keep", "volume_fields_to_keep", b"volume_fields_to_keep"]) -> None: ...
 
 global___GetSolutionDataPhysicsAIRequest = GetSolutionDataPhysicsAIRequest
 
@@ -378,16 +452,35 @@ class GetSolutionDataPhysicsAIResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     FILE_FIELD_NUMBER: builtins.int
+    PHYSICS_AI_FILES_FIELD_NUMBER: builtins.int
+    PHYSICS_AI_FOLDER_URL_FIELD_NUMBER: builtins.int
     @property
     def file(self) -> luminarycloud._proto.api.v0.luminarycloud.common.common_pb2.File:
         """The processed solution data. Either a signed URL or a file ID."""
+    @property
+    def physics_ai_files(self) -> global___PhysicsAiDatasetCaseFiles:
+        """TODO: Need to implement operations/operation IDs for long-running processes.
+        Users should be able to poll for progress and then request the files/their location
+        rather than having synchronous blocking calls for potentially long-running exports.
+
+        Physics AI processed files. Contains individual files (merged_surfaces.vtp,
+        merged_surfaces.stl, etc.) that were generated during processing.
+        Internal use only - for server-side file operations.
+        """
+    physics_ai_folder_url: builtins.str
+    """GCS folder URL containing the physics AI processed files (e.g., "gs://bucket/path/").
+    Internal use only - for server-side file operations. Use with physics_ai_files
+    to construct full file paths: folder_url + file.name + "." + file.file_type
+    """
     def __init__(
         self,
         *,
         file: luminarycloud._proto.api.v0.luminarycloud.common.common_pb2.File | None = ...,
+        physics_ai_files: global___PhysicsAiDatasetCaseFiles | None = ...,
+        physics_ai_folder_url: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["file", b"file"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["file", b"file"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["file", b"file", "physics_ai_files", b"physics_ai_files"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file", b"file", "physics_ai_files", b"physics_ai_files", "physics_ai_folder_url", b"physics_ai_folder_url"]) -> None: ...
 
 global___GetSolutionDataPhysicsAIResponse = GetSolutionDataPhysicsAIResponse
 
@@ -494,6 +587,30 @@ class PhysicsAiTrainingJob(google.protobuf.message.Message):
 
 global___PhysicsAiTrainingJob = PhysicsAiTrainingJob
 
+class TrainingInputParameter(google.protobuf.message.Message):
+    """Input parameter specification for training with reference values for normalization.
+    Type is inherited from the dataset's ParameterSchema.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    REFERENCE_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Parameter name (must exist in dataset's ParameterSchema)"""
+    @property
+    def reference(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]:
+        """Reference value(s) for normalization (optional, falls back to schema default)"""
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        reference: collections.abc.Iterable[builtins.float] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "reference", b"reference"]) -> None: ...
+
+global___TrainingInputParameter = TrainingInputParameter
+
 class SubmitTrainingJobRequest(google.protobuf.message.Message):
     """Request message for submitting a training job"""
 
@@ -507,6 +624,9 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     INITIALIZATION_TYPE_FIELD_NUMBER: builtins.int
     BASE_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
     DATASET_ID_FIELD_NUMBER: builtins.int
+    SURFACE_OUTPUTS_FIELD_NUMBER: builtins.int
+    VOLUME_OUTPUTS_FIELD_NUMBER: builtins.int
+    INPUT_PARAMETERS_FIELD_NUMBER: builtins.int
     architecture_version_id: builtins.str
     """Architecture version ID to train."""
     training_description: builtins.str
@@ -517,13 +637,29 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     def training_solutions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingSolution]:
         """Training solutions"""
     training_config: builtins.str
-    """Training configuration in JSON format."""
+    """Training configuration in JSON format. Used for internal/advanced config
+    (resources, priority_class, mode, custom_args). For model I/O configuration,
+    prefer the typed fields below.
+    """
     initialization_type: global___ModelInitializationType.ValueType
     """Model initialization type."""
     base_model_version_id: builtins.str
     """Base model version ID (if initializing from existing model)."""
     dataset_id: builtins.str
     """Dataset ID to use for training (if using dataset-based training)."""
+    @property
+    def surface_outputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """Model output configuration (typed fields, preferred over JSON training_config).
+        Surface quantities to predict (e.g., PRESSURE_TIME_AVERAGE, WALL_SHEAR_STRESS_TIME_AVERAGE).
+        """
+    @property
+    def volume_outputs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """Volume quantities to predict (e.g., VELOCITY_TIME_AVERAGE, PRESSURE_TIME_AVERAGE)."""
+    @property
+    def input_parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrainingInputParameter]:
+        """Input parameters for the model with reference values for normalization.
+        Parameter names must exist in the dataset's ParameterSchema.
+        """
     def __init__(
         self,
         *,
@@ -535,8 +671,11 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
         initialization_type: global___ModelInitializationType.ValueType = ...,
         base_model_version_id: builtins.str = ...,
         dataset_id: builtins.str = ...,
+        surface_outputs: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        volume_outputs: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        input_parameters: collections.abc.Iterable[global___TrainingInputParameter] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "dataset_id", b"dataset_id", "external_dataset_uri", b"external_dataset_uri", "initialization_type", b"initialization_type", "training_config", b"training_config", "training_description", b"training_description", "training_solutions", b"training_solutions"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "dataset_id", b"dataset_id", "external_dataset_uri", b"external_dataset_uri", "initialization_type", b"initialization_type", "input_parameters", b"input_parameters", "surface_outputs", b"surface_outputs", "training_config", b"training_config", "training_description", b"training_description", "training_solutions", b"training_solutions", "volume_outputs", b"volume_outputs"]) -> None: ...
 
 global___SubmitTrainingJobRequest = SubmitTrainingJobRequest
 
@@ -587,6 +726,42 @@ class CancelTrainingJobResponse(google.protobuf.message.Message):
 
 global___CancelTrainingJobResponse = CancelTrainingJobResponse
 
+class GetTrainingJobRequest(google.protobuf.message.Message):
+    """Request message for getting a training job by ID."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_ID_FIELD_NUMBER: builtins.int
+    training_job_id: builtins.str
+    """Training job ID to retrieve."""
+    def __init__(
+        self,
+        *,
+        training_job_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_job_id", b"training_job_id"]) -> None: ...
+
+global___GetTrainingJobRequest = GetTrainingJobRequest
+
+class GetTrainingJobResponse(google.protobuf.message.Message):
+    """Response message for getting a training job."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_FIELD_NUMBER: builtins.int
+    @property
+    def training_job(self) -> global___PhysicsAiTrainingJob:
+        """The training job."""
+    def __init__(
+        self,
+        *,
+        training_job: global___PhysicsAiTrainingJob | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["training_job", b"training_job"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_job", b"training_job"]) -> None: ...
+
+global___GetTrainingJobResponse = GetTrainingJobResponse
+
 class PhysicsAiDataset(google.protobuf.message.Message):
     """A Physics AI dataset"""
 
@@ -600,12 +775,11 @@ class PhysicsAiDataset(google.protobuf.message.Message):
     CREATION_TIME_FIELD_NUMBER: builtins.int
     UPDATE_TIME_FIELD_NUMBER: builtins.int
     LOCKED_AT_FIELD_NUMBER: builtins.int
-    CASES_FIELD_NUMBER: builtins.int
     EXPORT_CONFIG_FIELD_NUMBER: builtins.int
     id: builtins.str
     """Unique identifier for the dataset."""
     created_by: builtins.str
-    """User ID who created the dataset (empty for platform-curated datasets)."""
+    """User ID who created the dataset (unset for platform-curated datasets)."""
     name: builtins.str
     """Name of the dataset."""
     description: builtins.str
@@ -622,9 +796,6 @@ class PhysicsAiDataset(google.protobuf.message.Message):
     def locked_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """Dataset locked time (if locked)."""
     @property
-    def cases(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiDatasetCase]:
-        """Cases in this dataset."""
-    @property
     def export_config(self) -> global___GetSolutionDataPhysicsAIRequest:
         """Export configuration for extracting solution data for cases. The field
         solution_id is ignored.
@@ -633,18 +804,21 @@ class PhysicsAiDataset(google.protobuf.message.Message):
         self,
         *,
         id: builtins.str = ...,
-        created_by: builtins.str = ...,
+        created_by: builtins.str | None = ...,
         name: builtins.str = ...,
-        description: builtins.str = ...,
+        description: builtins.str | None = ...,
         is_locked: builtins.bool = ...,
         creation_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         update_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         locked_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
-        cases: collections.abc.Iterable[global___PhysicsAiDatasetCase] | None = ...,
         export_config: global___GetSolutionDataPhysicsAIRequest | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["creation_time", b"creation_time", "export_config", b"export_config", "locked_at", b"locked_at", "update_time", b"update_time"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["cases", b"cases", "created_by", b"created_by", "creation_time", b"creation_time", "description", b"description", "export_config", b"export_config", "id", b"id", "is_locked", b"is_locked", "locked_at", b"locked_at", "name", b"name", "update_time", b"update_time"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_created_by", b"_created_by", "_description", b"_description", "created_by", b"created_by", "creation_time", b"creation_time", "description", b"description", "export_config", b"export_config", "locked_at", b"locked_at", "update_time", b"update_time"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_created_by", b"_created_by", "_description", b"_description", "created_by", b"created_by", "creation_time", b"creation_time", "description", b"description", "export_config", b"export_config", "id", b"id", "is_locked", b"is_locked", "locked_at", b"locked_at", "name", b"name", "update_time", b"update_time"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_created_by", b"_created_by"]) -> typing_extensions.Literal["created_by"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_description", b"_description"]) -> typing_extensions.Literal["description"] | None: ...
 
 global___PhysicsAiDataset = PhysicsAiDataset
 
@@ -782,11 +956,11 @@ class CreateDatasetRequest(google.protobuf.message.Message):
     def cases(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CreatePhysicsAiDatasetCaseInput]:
         """Cases to add to the dataset (case_id will be server-generated)."""
     @property
-    def parameter_schema(self) -> google.protobuf.struct_pb2.Struct:
-        """JSON schema defining allowed parameters for cases in this dataset."""
+    def parameter_schema(self) -> global___ParameterSchema:
+        """Structured parameter schema defining allowed parameters for cases in this dataset."""
     @property
     def export_config(self) -> global___GetSolutionDataPhysicsAIRequest:
-        """Export configuration for extracting solution data forthe cases. The
+        """Export configuration for extracting solution data for the cases. The
         solution_id field is ignored.
         """
     def __init__(
@@ -795,7 +969,7 @@ class CreateDatasetRequest(google.protobuf.message.Message):
         name: builtins.str = ...,
         description: builtins.str = ...,
         cases: collections.abc.Iterable[global___CreatePhysicsAiDatasetCaseInput] | None = ...,
-        parameter_schema: google.protobuf.struct_pb2.Struct | None = ...,
+        parameter_schema: global___ParameterSchema | None = ...,
         export_config: global___GetSolutionDataPhysicsAIRequest | None = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["export_config", b"export_config", "parameter_schema", b"parameter_schema"]) -> builtins.bool: ...
@@ -821,3 +995,59 @@ class CreateDatasetResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["dataset", b"dataset"]) -> None: ...
 
 global___CreateDatasetResponse = CreateDatasetResponse
+
+class ListDatasetsRequest(google.protobuf.message.Message):
+    """Request message for listing datasets for the current user"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListDatasetsRequest = ListDatasetsRequest
+
+class ListDatasetsResponse(google.protobuf.message.Message):
+    """Response message for listing datasets"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATASETS_FIELD_NUMBER: builtins.int
+    @property
+    def datasets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiDataset]: ...
+    def __init__(
+        self,
+        *,
+        datasets: collections.abc.Iterable[global___PhysicsAiDataset] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["datasets", b"datasets"]) -> None: ...
+
+global___ListDatasetsResponse = ListDatasetsResponse
+
+class ListTrainingJobsRequest(google.protobuf.message.Message):
+    """Request message for listing training jobs for the current user"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListTrainingJobsRequest = ListTrainingJobsRequest
+
+class ListTrainingJobsResponse(google.protobuf.message.Message):
+    """Response message for listing training jobs"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOBS_FIELD_NUMBER: builtins.int
+    @property
+    def training_jobs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiTrainingJob]: ...
+    def __init__(
+        self,
+        *,
+        training_jobs: collections.abc.Iterable[global___PhysicsAiTrainingJob] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["training_jobs", b"training_jobs"]) -> None: ...
+
+global___ListTrainingJobsResponse = ListTrainingJobsResponse

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2_grpc.py

@@ -35,11 +35,26 @@ class PhysicsAiServiceStub(object):
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.SerializeToString,
                 response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.FromString,
                 )
+        self.CancelTrainingJob = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/CancelTrainingJob',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.FromString,
+                )
+        self.GetTrainingJob = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/GetTrainingJob',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.FromString,
+                )
         self.CreateDataset = channel.unary_unary(
                 '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/CreateDataset',
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.SerializeToString,
                 response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.FromString,
                 )
+        self.ListDatasets = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/ListDatasets',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsResponse.FromString,
+                )
 
 
 class PhysicsAiServiceServicer(object):
@@ -74,6 +89,20 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CancelTrainingJob(self, request, context):
+        """Cancels a Physics AI training job
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetTrainingJob(self, request, context):
+        """Gets a Physics AI training job by ID
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def CreateDataset(self, request, context):
         """Creates a new Physics AI dataset
         """
@@ -81,6 +110,13 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListDatasets(self, request, context):
+        """Lists datasets for the current user
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_PhysicsAiServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -104,11 +140,26 @@ def add_PhysicsAiServiceServicer_to_server(servicer, server):
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobRequest.FromString,
                     response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.SerializeToString,
             ),
+            'CancelTrainingJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.CancelTrainingJob,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.SerializeToString,
+            ),
+            'GetTrainingJob': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetTrainingJob,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.SerializeToString,
+            ),
             'CreateDataset': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateDataset,
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.FromString,
                     response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.SerializeToString,
             ),
+            'ListDatasets': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListDatasets,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService', rpc_method_handlers)
@@ -188,6 +239,40 @@ class PhysicsAiService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def CancelTrainingJob(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/CancelTrainingJob',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CancelTrainingJobResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetTrainingJob(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/GetTrainingJob',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetTrainingJobResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateDataset(request,
             target,
@@ -204,3 +289,20 @@ class PhysicsAiService(object):
             proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListDatasets(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/ListDatasets',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListDatasetsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

luminarycloud/_proto/api/v0/luminarycloud/physics_ai/physics_ai_pb2_grpc.pyi

@@ -30,11 +30,26 @@ class PhysicsAiServiceStub:
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse,
     ]
     """Submits a training job for a Physics AI architecture"""
+    CancelTrainingJob: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobResponse,
+    ]
+    """Cancels a Physics AI training job"""
+    GetTrainingJob: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetTrainingJobRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetTrainingJobResponse,
+    ]
+    """Gets a Physics AI training job by ID"""
     CreateDataset: grpc.UnaryUnaryMultiCallable[
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetRequest,
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetResponse,
     ]
     """Creates a new Physics AI dataset"""
+    ListDatasets: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListDatasetsRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListDatasetsResponse,
+    ]
+    """Lists datasets for the current user"""
 
 class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
     """Manages physics ai architectures."""
@@ -68,11 +83,32 @@ class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse:
         """Submits a training job for a Physics AI architecture"""
     @abc.abstractmethod
+    def CancelTrainingJob(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CancelTrainingJobResponse:
+        """Cancels a Physics AI training job"""
+    @abc.abstractmethod
+    def GetTrainingJob(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetTrainingJobRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetTrainingJobResponse:
+        """Gets a Physics AI training job by ID"""
+    @abc.abstractmethod
     def CreateDataset(
         self,
         request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetRequest,
         context: grpc.ServicerContext,
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.CreateDatasetResponse:
         """Creates a new Physics AI dataset"""
+    @abc.abstractmethod
+    def ListDatasets(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListDatasetsRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListDatasetsResponse:
+        """Lists datasets for the current user"""
 
 def add_PhysicsAiServiceServicer_to_server(servicer: PhysicsAiServiceServicer, server: grpc.Server) -> None: ...