luminarycloud 0.22.1__py3-none-any.whl → 0.22.3__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
@@ -303,7 +304,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
 
@@ -331,11 +336,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
@@ -353,7 +362,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,
         *,
@@ -366,8 +389,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
 
@@ -377,16 +402,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
 
@@ -431,6 +475,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
@@ -465,6 +510,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,
         *,
@@ -483,9 +530,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
 
@@ -501,6 +549,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
@@ -516,6 +565,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,
         *,
@@ -526,8 +577,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
 
@@ -577,3 +629,265 @@ class CancelTrainingJobResponse(google.protobuf.message.Message):
     ) -> 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
+    EXPORT_CONFIG_FIELD_NUMBER: builtins.int
+    id: builtins.str
+    """Unique identifier for the dataset."""
+    created_by: builtins.str
+    """User ID who created the dataset (unset 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 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 | None = ...,
+        name: 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 = ...,
+        export_config: global___GetSolutionDataPhysicsAIRequest | None = ...,
+    ) -> 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
+
+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
+
+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

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

@@ -35,6 +35,16 @@ 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,
+                )
+        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):
@@ -69,6 +79,20 @@ 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 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 = {
@@ -92,6 +116,16 @@ 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,
+            ),
+            '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)
@@ -170,3 +204,37 @@ 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)
+
+    @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,6 +30,16 @@ 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"""
+    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."""
@@ -62,5 +72,19 @@ 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"""
+    @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: ...