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

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

@@ -8,6 +8,7 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import google.protobuf.struct_pb2
 import google.protobuf.timestamp_pb2
 import luminarycloud._proto.api.v0.luminarycloud.common.common_pb2
 import luminarycloud._proto.base.base_pb2
@@ -282,6 +283,26 @@ class ListPretrainedModelsResponse(google.protobuf.message.Message):
 
 global___ListPretrainedModelsResponse = ListPretrainedModelsResponse
 
+class SurfaceGroup(google.protobuf.message.Message):
+    """Represents a named group of surfaces to export as a single STL file"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    SURFACES_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    @property
+    def surfaces(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        surfaces: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["name", b"name", "surfaces", b"surfaces"]) -> None: ...
+
+global___SurfaceGroup = SurfaceGroup
+
 class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
     """Request message for download and process solutions for physics ai"""
 
@@ -310,6 +331,7 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
     PROCESS_VOLUME_FIELD_NUMBER: builtins.int
     SINGLE_PRECISION_FIELD_NUMBER: builtins.int
     INTERNAL_OPTIONS_FIELD_NUMBER: builtins.int
+    EXPORT_SURFACE_GROUPS_FIELD_NUMBER: builtins.int
     solution_id: builtins.str
     """Required. The globally unique identifier for the solution."""
     @property
@@ -330,6 +352,9 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
     @property
     def internal_options(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
         """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"""
     def __init__(
         self,
         *,
@@ -341,8 +366,9 @@ class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
         process_volume: builtins.bool = ...,
         single_precision: builtins.bool = ...,
         internal_options: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        export_surface_groups: collections.abc.Iterable[global___SurfaceGroup] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["exclude_surfaces", b"exclude_surfaces", "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", "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: ...
 
 global___GetSolutionDataPhysicsAIRequest = GetSolutionDataPhysicsAIRequest
 
@@ -406,6 +432,7 @@ class PhysicsAiTrainingJob(google.protobuf.message.Message):
     CREATION_TIME_FIELD_NUMBER: builtins.int
     UPDATE_TIME_FIELD_NUMBER: builtins.int
     COMPLETION_TIME_FIELD_NUMBER: builtins.int
+    DATASET_ID_FIELD_NUMBER: builtins.int
     id: builtins.str
     """Unique identifier for the training job."""
     architecture_version_id: builtins.str
@@ -440,6 +467,8 @@ class PhysicsAiTrainingJob(google.protobuf.message.Message):
     @property
     def completion_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """Job completion time, set upon job completion. (mutable)"""
+    dataset_id: builtins.str
+    """Dataset ID used for training (if using dataset-based training)."""
     def __init__(
         self,
         *,
@@ -458,9 +487,10 @@ class PhysicsAiTrainingJob(google.protobuf.message.Message):
         creation_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         update_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         completion_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        dataset_id: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["completion_time", b"completion_time", "creation_time", b"creation_time", "status", b"status", "update_time", b"update_time"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["architecture_version_id", b"architecture_version_id", "base_model_version_id", b"base_model_version_id", "completion_time", b"completion_time", "creation_time", b"creation_time", "error_message", b"error_message", "external_dataset_uri", b"external_dataset_uri", "id", b"id", "initialization_type", b"initialization_type", "output_model_version_id", b"output_model_version_id", "status", b"status", "training_config", b"training_config", "training_data_source_type", b"training_data_source_type", "training_description", b"training_description", "update_time", b"update_time", "user_id", b"user_id"]) -> 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", "completion_time", b"completion_time", "creation_time", b"creation_time", "dataset_id", b"dataset_id", "error_message", b"error_message", "external_dataset_uri", b"external_dataset_uri", "id", b"id", "initialization_type", b"initialization_type", "output_model_version_id", b"output_model_version_id", "status", b"status", "training_config", b"training_config", "training_data_source_type", b"training_data_source_type", "training_description", b"training_description", "update_time", b"update_time", "user_id", b"user_id"]) -> None: ...
 
 global___PhysicsAiTrainingJob = PhysicsAiTrainingJob
 
@@ -476,6 +506,7 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     TRAINING_CONFIG_FIELD_NUMBER: builtins.int
     INITIALIZATION_TYPE_FIELD_NUMBER: builtins.int
     BASE_MODEL_VERSION_ID_FIELD_NUMBER: builtins.int
+    DATASET_ID_FIELD_NUMBER: builtins.int
     architecture_version_id: builtins.str
     """Architecture version ID to train."""
     training_description: builtins.str
@@ -491,6 +522,8 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
     """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)."""
     def __init__(
         self,
         *,
@@ -501,8 +534,9 @@ class SubmitTrainingJobRequest(google.protobuf.message.Message):
         training_config: builtins.str = ...,
         initialization_type: global___ModelInitializationType.ValueType = ...,
         base_model_version_id: builtins.str = ...,
+        dataset_id: builtins.str = ...,
     ) -> 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", "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", "training_config", b"training_config", "training_description", b"training_description", "training_solutions", b"training_solutions"]) -> None: ...
 
 global___SubmitTrainingJobRequest = SubmitTrainingJobRequest
 
@@ -524,3 +558,266 @@ class SubmitTrainingJobResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["training_job", b"training_job"]) -> None: ...
 
 global___SubmitTrainingJobResponse = SubmitTrainingJobResponse
+
+class CancelTrainingJobRequest(google.protobuf.message.Message):
+    """Request message for canceling a training job."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TRAINING_JOB_ID_FIELD_NUMBER: builtins.int
+    training_job_id: builtins.str
+    """Training job ID to cancel."""
+    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___CancelTrainingJobRequest = CancelTrainingJobRequest
+
+class CancelTrainingJobResponse(google.protobuf.message.Message):
+    """Response message for canceling a training job."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___CancelTrainingJobResponse = CancelTrainingJobResponse
+
+class PhysicsAiDataset(google.protobuf.message.Message):
+    """A Physics AI dataset"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ID_FIELD_NUMBER: builtins.int
+    CREATED_BY_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    IS_LOCKED_FIELD_NUMBER: builtins.int
+    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)."""
+    name: builtins.str
+    """Name of the dataset."""
+    description: builtins.str
+    """Description of the dataset."""
+    is_locked: builtins.bool
+    """Whether the dataset is locked (immutable after first training)."""
+    @property
+    def creation_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Dataset creation time."""
+    @property
+    def update_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Dataset last update time."""
+    @property
+    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.
+        """
+    def __init__(
+        self,
+        *,
+        id: builtins.str = ...,
+        created_by: builtins.str = ...,
+        name: builtins.str = ...,
+        description: builtins.str = ...,
+        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: ...
+
+global___PhysicsAiDataset = PhysicsAiDataset
+
+class PhysicsAiDatasetCaseFile(google.protobuf.message.Message):
+    """A file in a Physics AI dataset case"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_TYPE_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    file_type: builtins.str
+    """File type (e.g., "vtp", "stl", "vtu")."""
+    name: builtins.str
+    """File name (e.g., "merged_surfaces")."""
+    def __init__(
+        self,
+        *,
+        file_type: builtins.str = ...,
+        name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file_type", b"file_type", "name", b"name"]) -> None: ...
+
+global___PhysicsAiDatasetCaseFile = PhysicsAiDatasetCaseFile
+
+class PhysicsAiDatasetCaseFiles(google.protobuf.message.Message):
+    """Wrapper message for storing repeated PhysicsAiDatasetCaseFile in JSONB."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILES_FIELD_NUMBER: builtins.int
+    @property
+    def files(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiDatasetCaseFile]: ...
+    def __init__(
+        self,
+        *,
+        files: collections.abc.Iterable[global___PhysicsAiDatasetCaseFile] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["files", b"files"]) -> None: ...
+
+global___PhysicsAiDatasetCaseFiles = PhysicsAiDatasetCaseFiles
+
+class PhysicsAiDatasetCase(google.protobuf.message.Message):
+    """A case in a Physics AI dataset"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CASE_ID_FIELD_NUMBER: builtins.int
+    SOLUTION_ID_FIELD_NUMBER: builtins.int
+    SIMULATION_ID_FIELD_NUMBER: builtins.int
+    PARAMS_FIELD_NUMBER: builtins.int
+    FILES_FIELD_NUMBER: builtins.int
+    SOURCE_METADATA_FIELD_NUMBER: builtins.int
+    case_id: builtins.str
+    """Unique identifier for this case within the dataset (server-generated)."""
+    solution_id: builtins.str
+    """Solution ID if this case was created from a simulation solution."""
+    simulation_id: builtins.str
+    """Simulation ID if this case was created from a simulation."""
+    @property
+    def params(self) -> google.protobuf.struct_pb2.Struct:
+        """Parameters for this case as JSON string."""
+    @property
+    def files(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiDatasetCaseFile]:
+        """Files associated with this case."""
+    @property
+    def source_metadata(self) -> google.protobuf.struct_pb2.Struct:
+        """Source metadata as JSON string."""
+    def __init__(
+        self,
+        *,
+        case_id: builtins.str = ...,
+        solution_id: builtins.str = ...,
+        simulation_id: builtins.str = ...,
+        params: google.protobuf.struct_pb2.Struct | None = ...,
+        files: collections.abc.Iterable[global___PhysicsAiDatasetCaseFile] | None = ...,
+        source_metadata: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["params", b"params", "source_metadata", b"source_metadata"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["case_id", b"case_id", "files", b"files", "params", b"params", "simulation_id", b"simulation_id", "solution_id", b"solution_id", "source_metadata", b"source_metadata"]) -> None: ...
+
+global___PhysicsAiDatasetCase = PhysicsAiDatasetCase
+
+class CreatePhysicsAiDatasetCaseInput(google.protobuf.message.Message):
+    """Input for creating a case in a Physics AI dataset (case_id is server-generated)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOLUTION_ID_FIELD_NUMBER: builtins.int
+    SIMULATION_ID_FIELD_NUMBER: builtins.int
+    PARAMS_FIELD_NUMBER: builtins.int
+    FILES_FIELD_NUMBER: builtins.int
+    SOURCE_METADATA_FIELD_NUMBER: builtins.int
+    solution_id: builtins.str
+    """Solution ID if this case was created from a simulation solution."""
+    simulation_id: builtins.str
+    """Simulation ID if this case was created from a simulation."""
+    @property
+    def params(self) -> google.protobuf.struct_pb2.Struct:
+        """Parameters for this case as JSON string."""
+    @property
+    def files(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PhysicsAiDatasetCaseFile]:
+        """Files associated with this case."""
+    @property
+    def source_metadata(self) -> google.protobuf.struct_pb2.Struct:
+        """Source metadata as JSON string."""
+    def __init__(
+        self,
+        *,
+        solution_id: builtins.str = ...,
+        simulation_id: builtins.str = ...,
+        params: google.protobuf.struct_pb2.Struct | None = ...,
+        files: collections.abc.Iterable[global___PhysicsAiDatasetCaseFile] | None = ...,
+        source_metadata: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["params", b"params", "source_metadata", b"source_metadata"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["files", b"files", "params", b"params", "simulation_id", b"simulation_id", "solution_id", b"solution_id", "source_metadata", b"source_metadata"]) -> None: ...
+
+global___CreatePhysicsAiDatasetCaseInput = CreatePhysicsAiDatasetCaseInput
+
+class CreateDatasetRequest(google.protobuf.message.Message):
+    """Request message for creating a dataset"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    DESCRIPTION_FIELD_NUMBER: builtins.int
+    CASES_FIELD_NUMBER: builtins.int
+    PARAMETER_SCHEMA_FIELD_NUMBER: builtins.int
+    EXPORT_CONFIG_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    """Name of the dataset."""
+    description: builtins.str
+    """Optional description of the dataset."""
+    @property
+    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."""
+    @property
+    def export_config(self) -> global___GetSolutionDataPhysicsAIRequest:
+        """Export configuration for extracting solution data forthe cases. The
+        solution_id field is ignored.
+        """
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        description: builtins.str = ...,
+        cases: collections.abc.Iterable[global___CreatePhysicsAiDatasetCaseInput] | None = ...,
+        parameter_schema: google.protobuf.struct_pb2.Struct | 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: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cases", b"cases", "description", b"description", "export_config", b"export_config", "name", b"name", "parameter_schema", b"parameter_schema"]) -> None: ...
+
+global___CreateDatasetRequest = CreateDatasetRequest
+
+class CreateDatasetResponse(google.protobuf.message.Message):
+    """Response message for creating a dataset"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATASET_FIELD_NUMBER: builtins.int
+    @property
+    def dataset(self) -> global___PhysicsAiDataset:
+        """The created dataset."""
+    def __init__(
+        self,
+        *,
+        dataset: global___PhysicsAiDataset | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["dataset", b"dataset"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["dataset", b"dataset"]) -> None: ...
+
+global___CreateDatasetResponse = CreateDatasetResponse

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

@@ -35,6 +35,11 @@ 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.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,
+                )
 
 
 class PhysicsAiServiceServicer(object):
@@ -69,6 +74,13 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def CreateDataset(self, request, context):
+        """Creates a new Physics AI dataset
+        """
+        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 = {
@@ -92,6 +104,11 @@ 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,
             ),
+            '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,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService', rpc_method_handlers)
@@ -170,3 +187,20 @@ class PhysicsAiService(object):
             proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.SubmitTrainingJobResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CreateDataset(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/CreateDataset',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.CreateDatasetRequest.SerializeToString,
+            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)

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

@@ -30,6 +30,11 @@ class PhysicsAiServiceStub:
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse,
     ]
     """Submits a training job for a Physics AI architecture"""
+    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"""
 
 class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
     """Manages physics ai architectures."""
@@ -62,5 +67,12 @@ class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.SubmitTrainingJobResponse:
         """Submits a training job for a Physics AI architecture"""
+    @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"""
 
 def add_PhysicsAiServiceServicer_to_server(servicer: PhysicsAiServiceServicer, server: grpc.Server) -> None: ...