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

luminarycloud/_proto/api/v0/luminarycloud/geometry/geometry_pb2.pyi

@@ -72,6 +72,7 @@ class Geometry(google.protobuf.message.Message):
     USES_TAGS_FIELD_NUMBER: builtins.int
     PROJECT_ID_FIELD_NUMBER: builtins.int
     STATUS_FIELD_NUMBER: builtins.int
+    CHECK_FIELD_NUMBER: builtins.int
     VERSIONS_FIELD_NUMBER: builtins.int
     id: builtins.str
     name: builtins.str
@@ -84,6 +85,10 @@ class Geometry(google.protobuf.message.Message):
     uses_tags: builtins.bool
     project_id: builtins.str
     status: global___Geometry.Status.ValueType
+    """Status for the latest version."""
+    @property
+    def check(self) -> global___GetCheckGeometryResponse:
+        """Geometry check result for the latest version."""
     @property
     def versions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GeometryVersion]:
         """Versions are not populated by default, but can be included by request."""
@@ -99,10 +104,11 @@ class Geometry(google.protobuf.message.Message):
         uses_tags: builtins.bool = ...,
         project_id: builtins.str = ...,
         status: global___Geometry.Status.ValueType | None = ...,
+        check: global___GetCheckGeometryResponse | None = ...,
         versions: collections.abc.Iterable[global___GeometryVersion] | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["_status", b"_status", "create_time", b"create_time", "status", b"status", "update_time", b"update_time"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_status", b"_status", "create_time", b"create_time", "deleted", b"deleted", "id", b"id", "last_version_id", b"last_version_id", "name", b"name", "project_id", b"project_id", "status", b"status", "update_time", b"update_time", "uses_tags", b"uses_tags", "versions", b"versions"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_status", b"_status", "check", b"check", "create_time", b"create_time", "status", b"status", "update_time", b"update_time"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_status", b"_status", "check", b"check", "create_time", b"create_time", "deleted", b"deleted", "id", b"id", "last_version_id", b"last_version_id", "name", b"name", "project_id", b"project_id", "status", b"status", "update_time", b"update_time", "uses_tags", b"uses_tags", "versions", b"versions"]) -> None: ...
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_status", b"_status"]) -> typing_extensions.Literal["status"] | None: ...
 
 global___Geometry = Geometry
@@ -116,6 +122,8 @@ class GeometryVersion(google.protobuf.message.Message):
     NAME_FIELD_NUMBER: builtins.int
     KERNEL_TYPE_FIELD_NUMBER: builtins.int
     NAMED_VARIABLE_SET_VERSION_ID_FIELD_NUMBER: builtins.int
+    CHECK_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
     id: builtins.str
     geometry_id: builtins.str
     @property
@@ -123,6 +131,11 @@ class GeometryVersion(google.protobuf.message.Message):
     name: builtins.str
     kernel_type: luminarycloud._proto.geometry.geometry_pb2.KernelType.ValueType
     named_variable_set_version_id: builtins.str
+    @property
+    def check(self) -> global___GetCheckGeometryResponse:
+        """Optional. Included by some RPCs on request."""
+    status: global___Geometry.Status.ValueType
+    """Optional. Included by some RPCs on request."""
     def __init__(
         self,
         *,
@@ -132,9 +145,12 @@ class GeometryVersion(google.protobuf.message.Message):
         name: builtins.str = ...,
         kernel_type: luminarycloud._proto.geometry.geometry_pb2.KernelType.ValueType = ...,
         named_variable_set_version_id: builtins.str = ...,
+        check: global___GetCheckGeometryResponse | None = ...,
+        status: global___Geometry.Status.ValueType | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["create_time", b"create_time"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["create_time", b"create_time", "geometry_id", b"geometry_id", "id", b"id", "kernel_type", b"kernel_type", "name", b"name", "named_variable_set_version_id", b"named_variable_set_version_id"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_status", b"_status", "check", b"check", "create_time", b"create_time", "status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_status", b"_status", "check", b"check", "create_time", b"create_time", "geometry_id", b"geometry_id", "id", b"id", "kernel_type", b"kernel_type", "name", b"name", "named_variable_set_version_id", b"named_variable_set_version_id", "status", b"status"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_status", b"_status"]) -> typing_extensions.Literal["status"] | None: ...
 
 global___GeometryVersion = GeometryVersion
 
@@ -1154,6 +1170,36 @@ class GetTagsResponse(google.protobuf.message.Message):
 
 global___GetTagsResponse = GetTagsResponse
 
+class GetTagsBatchRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REQUESTS_FIELD_NUMBER: builtins.int
+    @property
+    def requests(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GetTagsRequest]: ...
+    def __init__(
+        self,
+        *,
+        requests: collections.abc.Iterable[global___GetTagsRequest] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["requests", b"requests"]) -> None: ...
+
+global___GetTagsBatchRequest = GetTagsBatchRequest
+
+class GetTagsBatchResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RESPONSES_FIELD_NUMBER: builtins.int
+    @property
+    def responses(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GetTagsResponse]: ...
+    def __init__(
+        self,
+        *,
+        responses: collections.abc.Iterable[global___GetTagsResponse] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["responses", b"responses"]) -> None: ...
+
+global___GetTagsBatchResponse = GetTagsBatchResponse
+
 class Tag(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -1235,14 +1281,19 @@ class ListGeometryVersionsRequest(google.protobuf.message.Message):
 
     GEOMETRY_ID_FIELD_NUMBER: builtins.int
     UNFILTERED_FIELD_NUMBER: builtins.int
+    INCLUDE_CHECK_RESULTS_FIELD_NUMBER: builtins.int
     PAGE_SIZE_FIELD_NUMBER: builtins.int
     PAGE_TOKEN_FIELD_NUMBER: builtins.int
     geometry_id: builtins.str
     """Required. The ID of the geometry to list versions for."""
     unfiltered: builtins.bool
     """Optional. By default, the server only returns versions that are named OR have an associated
-    Mesh OR are the latest version of the geometry. If `unfiltered` is true, the server returns all
-    versions.
+    Mesh OR are the latest version of the geometry (i.e. the "interesting" versions). If
+    `unfiltered` is true, the server returns all versions.
+    """
+    include_check_results: builtins.bool
+    """Optional. If true, the server will include the geometry check results for
+    each version if available.
     """
     page_size: builtins.int
     """Optional. The maximum number of versions to return. The service may return fewer than this
@@ -1260,11 +1311,12 @@ class ListGeometryVersionsRequest(google.protobuf.message.Message):
         *,
         geometry_id: builtins.str = ...,
         unfiltered: builtins.bool = ...,
+        include_check_results: builtins.bool = ...,
         page_size: builtins.int | None = ...,
         page_token: builtins.str | None = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["_page_size", b"_page_size", "_page_token", b"_page_token", "page_size", b"page_size", "page_token", b"page_token"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["_page_size", b"_page_size", "_page_token", b"_page_token", "geometry_id", b"geometry_id", "page_size", b"page_size", "page_token", b"page_token", "unfiltered", b"unfiltered"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_page_size", b"_page_size", "_page_token", b"_page_token", "geometry_id", b"geometry_id", "include_check_results", b"include_check_results", "page_size", b"page_size", "page_token", b"page_token", "unfiltered", b"unfiltered"]) -> None: ...
     @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["_page_size", b"_page_size"]) -> typing_extensions.Literal["page_size"] | None: ...
     @typing.overload
@@ -1302,13 +1354,19 @@ class GetGeometryVersionRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     GEOMETRY_VERSION_ID_FIELD_NUMBER: builtins.int
+    INCLUDE_CHECK_RESULTS_FIELD_NUMBER: builtins.int
     geometry_version_id: builtins.str
+    include_check_results: builtins.bool
+    """Optional. If true, the server will include the geometry check result if
+    available.
+    """
     def __init__(
         self,
         *,
         geometry_version_id: builtins.str = ...,
+        include_check_results: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["geometry_version_id", b"geometry_version_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["geometry_version_id", b"geometry_version_id", "include_check_results", b"include_check_results"]) -> None: ...
 
 global___GetGeometryVersionRequest = GetGeometryVersionRequest
 

luminarycloud/_proto/api/v0/luminarycloud/geometry/geometry_pb2_grpc.py

@@ -120,6 +120,11 @@ class GeometryServiceStub(object):
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsRequest.SerializeToString,
                 response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsResponse.FromString,
                 )
+        self.GetTagsBatch = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.geometry.GeometryService/GetTagsBatch',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchResponse.FromString,
+                )
         self.ListTags = channel.unary_unary(
                 '/luminary.proto.api.v0.luminarycloud.geometry.GeometryService/ListTags',
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.ListTagsRequest.SerializeToString,
@@ -307,6 +312,13 @@ class GeometryServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetTagsBatch(self, request, context):
+        """Same as GetTags but for multiple geometry versions.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def ListTags(self, request, context):
         """List the tags associated with a geometry. Used by SDK clients since it
         contains more carefully crafted datastructures since SDK clients expose
@@ -449,6 +461,11 @@ def add_GeometryServiceServicer_to_server(servicer, server):
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsRequest.FromString,
                     response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsResponse.SerializeToString,
             ),
+            'GetTagsBatch': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetTagsBatch,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchResponse.SerializeToString,
+            ),
             'ListTags': grpc.unary_unary_rpc_method_handler(
                     servicer.ListTags,
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.ListTagsRequest.FromString,
@@ -841,6 +858,23 @@ class GeometryService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def GetTagsBatch(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.geometry.GeometryService/GetTagsBatch',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_geometry_dot_geometry__pb2.GetTagsBatchResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def ListTags(request,
             target,

luminarycloud/_proto/api/v0/luminarycloud/geometry/geometry_pb2_grpc.pyi

@@ -132,6 +132,11 @@ class GeometryServiceStub:
     clients since it contains more data about the associativity of faces and
     bodies.
     """
+    GetTagsBatch: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.GetTagsBatchRequest,
+        luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.GetTagsBatchResponse,
+    ]
+    """Same as GetTags but for multiple geometry versions."""
     ListTags: grpc.UnaryUnaryMultiCallable[
         luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.ListTagsRequest,
         luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.ListTagsResponse,
@@ -324,6 +329,13 @@ class GeometryServiceServicer(metaclass=abc.ABCMeta):
         bodies.
         """
     @abc.abstractmethod
+    def GetTagsBatch(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.GetTagsBatchRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.GetTagsBatchResponse:
+        """Same as GetTags but for multiple geometry versions."""
+    @abc.abstractmethod
     def ListTags(
         self,
         request: luminarycloud._proto.api.v0.luminarycloud.geometry.geometry_pb2.ListTagsRequest,

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

@@ -13,6 +13,7 @@ from google.protobuf import symbol_database as _symbol_database
 _sym_db = _symbol_database.Default()
 
 
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from luminarycloud._proto.api.v0.luminarycloud.common import common_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_common_dot_common__pb2
@@ -20,7 +21,7 @@ from luminarycloud._proto.base import base_pb2 as proto_dot_base_dot_base__pb2
 from luminarycloud._proto.quantity import quantity_pb2 as proto_dot_quantity_dot_quantity__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6proto/api/v0/luminarycloud/physics_ai/physics_ai.proto\x12.luminary.proto.api.v0.luminarycloud.physics_ai\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.proto/api/v0/luminarycloud/common/common.proto\x1a\x15proto/base/base.proto\x1a\x1dproto/quantity/quantity.proto\"\xad\x01\n\x1cPhysicsAiArchitectureVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x11\n\tchangelog\x18\x03 \x01(\t\x12`\n\x0flifecycle_state\x18\x04 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\xa6\x01\n\x15PhysicsAiArchitecture\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12^\n\x08versions\x18\x04 \x03(\x0b\x32L.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitectureVersion\"\x1a\n\x18ListArchitecturesRequest\"y\n\x19ListArchitecturesResponse\x12\\\n\rarchitectures\x18\x01 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitecture\"\x93\x01\n\x15PhysicsAiModelVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12`\n\x0flifecycle_state\x18\x03 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\x98\x01\n\x0ePhysicsAiModel\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12W\n\x08versions\x18\x04 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModelVersion\"\x1d\n\x1bListPretrainedModelsRequest\"n\n\x1cListPretrainedModelsResponse\x12N\n\x06models\x18\x01 \x03(\x0b\x32>.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModel\"\xdb\x03\n\x1fGetSolutionDataPhysicsAIRequest\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x18\n\x10\x65xclude_surfaces\x18\x02 \x03(\t\x12\x12\n\nfill_holes\x18\x03 \x01(\x02\x12\x45\n\x16surface_fields_to_keep\x18\x04 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x44\n\x15volume_fields_to_keep\x18\x05 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x16\n\x0eprocess_volume\x18\x06 \x01(\x08\x12\x18\n\x10single_precision\x18\x07 \x01(\x08\x12~\n\x10internal_options\x18\x08 \x03(\x0b\x32\x64.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest.InternalOptionsEntry\x1a\x36\n\x14InternalOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"b\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File\";\n\x10TrainingSolution\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x12\n\ndata_split\x18\x02 \x01(\t\"\x9a\x05\n\x14PhysicsAiTrainingJob\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1f\n\x17\x61rchitecture_version_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x17\n\x0ftraining_config\x18\x04 \x01(\t\x12i\n\x19training_data_source_type\x18\x05 \x01(\x0e\x32\x46.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingDataSourceType\x12\x1c\n\x14training_description\x18\x06 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x07 \x01(\t\x12\x64\n\x13initialization_type\x18\x08 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\t \x01(\t\x12.\n\x06status\x18\n \x01(\x0b\x32\x1e.luminary.proto.base.JobStatus\x12\x15\n\rerror_message\x18\x0b \x01(\t\x12\x1f\n\x17output_model_version_id\x18\x0c \x01(\t\x12\x31\n\rcreation_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x33\n\x0f\x63ompletion_time\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xf3\x02\n\x18SubmitTrainingJobRequest\x12\x1f\n\x17\x61rchitecture_version_id\x18\x01 \x01(\t\x12\x1c\n\x14training_description\x18\x02 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x03 \x01(\t\x12\\\n\x12training_solutions\x18\x04 \x03(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingSolution\x12\x17\n\x0ftraining_config\x18\x05 \x01(\t\x12\x64\n\x13initialization_type\x18\x06 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\x07 \x01(\t\"w\n\x19SubmitTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob*\xb4\x01\n\x17PhysicsAiLifecycleState\x12\x1f\n\x1bLIFECYCLE_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1bLIFECYCLE_STATE_DEVELOPMENT\x10\x01\x12\x1a\n\x16LIFECYCLE_STATE_ACTIVE\x10\x02\x12\x1e\n\x1aLIFECYCLE_STATE_DEPRECATED\x10\x03\x12\x1b\n\x17LIFECYCLE_STATE_RETIRED\x10\x04*\xa7\x01\n\x16TrainingDataSourceType\x12)\n%TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\x32\n.TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS\x10\x01\x12.\n*TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET\x10\x02*\xc1\x01\n\x17ModelInitializationType\x12)\n%MODEL_INITIALIZATION_TYPE_UNSPECIFIED\x10\x00\x12$\n MODEL_INITIALIZATION_TYPE_RANDOM\x10\x01\x12+\n\'MODEL_INITIALIZATION_TYPE_MODEL_VERSION\x10\x02\x12(\n$MODEL_INITIALIZATION_TYPE_CHECKPOINT\x10\x03\x32\x8d\x07\n\x10PhysicsAiService\x12\xce\x01\n\x11ListArchitectures\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesResponse\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/v0/physics_ai/architectures\x12\xdb\x01\n\x14ListPretrainedModels\x12K.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsRequest\x1aL.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsResponse\"(\x82\xd3\xe4\x93\x02\"\x12 /v0/physics_ai/pretrained_models\x12\xf5\x01\n\x18GetSolutionDataPhysicsAI\x12O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\x1aP.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIResponse\"6\x82\xd3\xe4\x93\x02\x30\"+/v0/physics_ai/solutions/{solution_id}/data:\x01*\x12\xd1\x01\n\x11SubmitTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse\"\'\x82\xd3\xe4\x93\x02!\"\x1c/v0/physics_ai/training/jobs:\x01*B>Z<luminarycloud.com/core/proto/api/v0/luminarycloud/physics_aib\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6proto/api/v0/luminarycloud/physics_ai/physics_ai.proto\x12.luminary.proto.api.v0.luminarycloud.physics_ai\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1cgoogle/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.proto/api/v0/luminarycloud/common/common.proto\x1a\x15proto/base/base.proto\x1a\x1dproto/quantity/quantity.proto\"\xad\x01\n\x1cPhysicsAiArchitectureVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x11\n\tchangelog\x18\x03 \x01(\t\x12`\n\x0flifecycle_state\x18\x04 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\xa6\x01\n\x15PhysicsAiArchitecture\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12^\n\x08versions\x18\x04 \x03(\x0b\x32L.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitectureVersion\"\x1a\n\x18ListArchitecturesRequest\"y\n\x19ListArchitecturesResponse\x12\\\n\rarchitectures\x18\x01 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiArchitecture\"\x93\x01\n\x15PhysicsAiModelVersion\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12`\n\x0flifecycle_state\x18\x03 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiLifecycleState\"\x98\x01\n\x0ePhysicsAiModel\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12W\n\x08versions\x18\x04 \x03(\x0b\x32\x45.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModelVersion\"\x1d\n\x1bListPretrainedModelsRequest\"n\n\x1cListPretrainedModelsResponse\x12N\n\x06models\x18\x01 \x03(\x0b\x32>.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiModel\".\n\x0cSurfaceGroup\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08surfaces\x18\x02 \x03(\t\"\xb8\x04\n\x1fGetSolutionDataPhysicsAIRequest\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x18\n\x10\x65xclude_surfaces\x18\x02 \x03(\t\x12\x12\n\nfill_holes\x18\x03 \x01(\x02\x12\x45\n\x16surface_fields_to_keep\x18\x04 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x44\n\x15volume_fields_to_keep\x18\x05 \x03(\x0e\x32%.luminary.proto.quantity.QuantityType\x12\x16\n\x0eprocess_volume\x18\x06 \x01(\x08\x12\x18\n\x10single_precision\x18\x07 \x01(\x08\x12~\n\x10internal_options\x18\x08 \x03(\x0b\x32\x64.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest.InternalOptionsEntry\x12[\n\x15\x65xport_surface_groups\x18\t \x03(\x0b\x32<.luminary.proto.api.v0.luminarycloud.physics_ai.SurfaceGroup\x1a\x36\n\x14InternalOptionsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"b\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File\";\n\x10TrainingSolution\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x12\n\ndata_split\x18\x02 \x01(\t\"\xae\x05\n\x14PhysicsAiTrainingJob\x12\n\n\x02id\x18\x01 \x01(\t\x12\x1f\n\x17\x61rchitecture_version_id\x18\x02 \x01(\t\x12\x0f\n\x07user_id\x18\x03 \x01(\t\x12\x17\n\x0ftraining_config\x18\x04 \x01(\t\x12i\n\x19training_data_source_type\x18\x05 \x01(\x0e\x32\x46.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingDataSourceType\x12\x1c\n\x14training_description\x18\x06 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x07 \x01(\t\x12\x64\n\x13initialization_type\x18\x08 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\t \x01(\t\x12.\n\x06status\x18\n \x01(\x0b\x32\x1e.luminary.proto.base.JobStatus\x12\x15\n\rerror_message\x18\x0b \x01(\t\x12\x1f\n\x17output_model_version_id\x18\x0c \x01(\t\x12\x31\n\rcreation_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x0e \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x33\n\x0f\x63ompletion_time\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ndataset_id\x18\x10 \x01(\t\"\x87\x03\n\x18SubmitTrainingJobRequest\x12\x1f\n\x17\x61rchitecture_version_id\x18\x01 \x01(\t\x12\x1c\n\x14training_description\x18\x02 \x01(\t\x12\x1c\n\x14\x65xternal_dataset_uri\x18\x03 \x01(\t\x12\\\n\x12training_solutions\x18\x04 \x03(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.TrainingSolution\x12\x17\n\x0ftraining_config\x18\x05 \x01(\t\x12\x64\n\x13initialization_type\x18\x06 \x01(\x0e\x32G.luminary.proto.api.v0.luminarycloud.physics_ai.ModelInitializationType\x12\x1d\n\x15\x62\x61se_model_version_id\x18\x07 \x01(\t\x12\x12\n\ndataset_id\x18\x08 \x01(\t\"w\n\x19SubmitTrainingJobResponse\x12Z\n\x0ctraining_job\x18\x01 \x01(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiTrainingJob\"3\n\x18\x43\x61ncelTrainingJobRequest\x12\x17\n\x0ftraining_job_id\x18\x01 \x01(\t\"\x1b\n\x19\x43\x61ncelTrainingJobResponse\"\xb8\x03\n\x10PhysicsAiDataset\x12\n\n\x02id\x18\x01 \x01(\t\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\x12\x11\n\tis_locked\x18\x05 \x01(\x08\x12\x31\n\rcreation_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12/\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12-\n\tlocked_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12S\n\x05\x63\x61ses\x18\t \x03(\x0b\x32\x44.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCase\x12\x66\n\rexport_config\x18\n \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\";\n\x18PhysicsAiDatasetCaseFile\x12\x11\n\tfile_type\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\"t\n\x19PhysicsAiDatasetCaseFiles\x12W\n\x05\x66iles\x18\x01 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\"\x87\x02\n\x14PhysicsAiDatasetCase\x12\x0f\n\x07\x63\x61se_id\x18\x01 \x01(\t\x12\x13\n\x0bsolution_id\x18\x02 \x01(\t\x12\x15\n\rsimulation_id\x18\x03 \x01(\t\x12\'\n\x06params\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x05 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x06 \x01(\x0b\x32\x17.google.protobuf.Struct\"\x81\x02\n\x1f\x43reatePhysicsAiDatasetCaseInput\x12\x13\n\x0bsolution_id\x18\x01 \x01(\t\x12\x15\n\rsimulation_id\x18\x02 \x01(\t\x12\'\n\x06params\x18\x03 \x01(\x0b\x32\x17.google.protobuf.Struct\x12W\n\x05\x66iles\x18\x04 \x03(\x0b\x32H.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile\x12\x30\n\x0fsource_metadata\x18\x05 \x01(\x0b\x32\x17.google.protobuf.Struct\"\xb4\x02\n\x14\x43reateDatasetRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12^\n\x05\x63\x61ses\x18\x03 \x03(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.CreatePhysicsAiDatasetCaseInput\x12\x31\n\x10parameter_schema\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\x12\x66\n\rexport_config\x18\x05 \x01(\x0b\x32O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\"j\n\x15\x43reateDatasetResponse\x12Q\n\x07\x64\x61taset\x18\x01 \x01(\x0b\x32@.luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset*\xb4\x01\n\x17PhysicsAiLifecycleState\x12\x1f\n\x1bLIFECYCLE_STATE_UNSPECIFIED\x10\x00\x12\x1f\n\x1bLIFECYCLE_STATE_DEVELOPMENT\x10\x01\x12\x1a\n\x16LIFECYCLE_STATE_ACTIVE\x10\x02\x12\x1e\n\x1aLIFECYCLE_STATE_DEPRECATED\x10\x03\x12\x1b\n\x17LIFECYCLE_STATE_RETIRED\x10\x04*\xa7\x01\n\x16TrainingDataSourceType\x12)\n%TRAINING_DATA_SOURCE_TYPE_UNSPECIFIED\x10\x00\x12\x32\n.TRAINING_DATA_SOURCE_TYPE_SIMULATION_SOLUTIONS\x10\x01\x12.\n*TRAINING_DATA_SOURCE_TYPE_EXTERNAL_DATASET\x10\x02*\xc1\x01\n\x17ModelInitializationType\x12)\n%MODEL_INITIALIZATION_TYPE_UNSPECIFIED\x10\x00\x12$\n MODEL_INITIALIZATION_TYPE_RANDOM\x10\x01\x12+\n\'MODEL_INITIALIZATION_TYPE_MODEL_VERSION\x10\x02\x12(\n$MODEL_INITIALIZATION_TYPE_CHECKPOINT\x10\x03\x32\xd0\x08\n\x10PhysicsAiService\x12\xce\x01\n\x11ListArchitectures\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.ListArchitecturesResponse\"$\x82\xd3\xe4\x93\x02\x1e\x12\x1c/v0/physics_ai/architectures\x12\xdb\x01\n\x14ListPretrainedModels\x12K.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsRequest\x1aL.luminary.proto.api.v0.luminarycloud.physics_ai.ListPretrainedModelsResponse\"(\x82\xd3\xe4\x93\x02\"\x12 /v0/physics_ai/pretrained_models\x12\xf5\x01\n\x18GetSolutionDataPhysicsAI\x12O.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest\x1aP.luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIResponse\"6\x82\xd3\xe4\x93\x02\x30\"+/v0/physics_ai/solutions/{solution_id}/data:\x01*\x12\xd1\x01\n\x11SubmitTrainingJob\x12H.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobRequest\x1aI.luminary.proto.api.v0.luminarycloud.physics_ai.SubmitTrainingJobResponse\"\'\x82\xd3\xe4\x93\x02!\"\x1c/v0/physics_ai/training/jobs:\x01*\x12\xc0\x01\n\rCreateDataset\x12\x44.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetRequest\x1a\x45.luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetResponse\"\"\x82\xd3\xe4\x93\x02\x1c\"\x17/v0/physics_ai/datasets:\x01*B>Z<luminarycloud.com/core/proto/api/v0/luminarycloud/physics_aib\x06proto3')
 
 _PHYSICSAILIFECYCLESTATE = DESCRIPTOR.enum_types_by_name['PhysicsAiLifecycleState']
 PhysicsAiLifecycleState = enum_type_wrapper.EnumTypeWrapper(_PHYSICSAILIFECYCLESTATE)
@@ -50,6 +51,7 @@ _PHYSICSAIMODELVERSION = DESCRIPTOR.message_types_by_name['PhysicsAiModelVersion
 _PHYSICSAIMODEL = DESCRIPTOR.message_types_by_name['PhysicsAiModel']
 _LISTPRETRAINEDMODELSREQUEST = DESCRIPTOR.message_types_by_name['ListPretrainedModelsRequest']
 _LISTPRETRAINEDMODELSRESPONSE = DESCRIPTOR.message_types_by_name['ListPretrainedModelsResponse']
+_SURFACEGROUP = DESCRIPTOR.message_types_by_name['SurfaceGroup']
 _GETSOLUTIONDATAPHYSICSAIREQUEST = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIRequest']
 _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY = _GETSOLUTIONDATAPHYSICSAIREQUEST.nested_types_by_name['InternalOptionsEntry']
 _GETSOLUTIONDATAPHYSICSAIRESPONSE = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIResponse']
@@ -57,6 +59,15 @@ _TRAININGSOLUTION = DESCRIPTOR.message_types_by_name['TrainingSolution']
 _PHYSICSAITRAININGJOB = DESCRIPTOR.message_types_by_name['PhysicsAiTrainingJob']
 _SUBMITTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['SubmitTrainingJobRequest']
 _SUBMITTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['SubmitTrainingJobResponse']
+_CANCELTRAININGJOBREQUEST = DESCRIPTOR.message_types_by_name['CancelTrainingJobRequest']
+_CANCELTRAININGJOBRESPONSE = DESCRIPTOR.message_types_by_name['CancelTrainingJobResponse']
+_PHYSICSAIDATASET = DESCRIPTOR.message_types_by_name['PhysicsAiDataset']
+_PHYSICSAIDATASETCASEFILE = DESCRIPTOR.message_types_by_name['PhysicsAiDatasetCaseFile']
+_PHYSICSAIDATASETCASEFILES = DESCRIPTOR.message_types_by_name['PhysicsAiDatasetCaseFiles']
+_PHYSICSAIDATASETCASE = DESCRIPTOR.message_types_by_name['PhysicsAiDatasetCase']
+_CREATEPHYSICSAIDATASETCASEINPUT = DESCRIPTOR.message_types_by_name['CreatePhysicsAiDatasetCaseInput']
+_CREATEDATASETREQUEST = DESCRIPTOR.message_types_by_name['CreateDatasetRequest']
+_CREATEDATASETRESPONSE = DESCRIPTOR.message_types_by_name['CreateDatasetResponse']
 PhysicsAiArchitectureVersion = _reflection.GeneratedProtocolMessageType('PhysicsAiArchitectureVersion', (_message.Message,), {
   'DESCRIPTOR' : _PHYSICSAIARCHITECTUREVERSION,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -113,6 +124,13 @@ ListPretrainedModelsResponse = _reflection.GeneratedProtocolMessageType('ListPre
   })
 _sym_db.RegisterMessage(ListPretrainedModelsResponse)
 
+SurfaceGroup = _reflection.GeneratedProtocolMessageType('SurfaceGroup', (_message.Message,), {
+  'DESCRIPTOR' : _SURFACEGROUP,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.SurfaceGroup)
+  })
+_sym_db.RegisterMessage(SurfaceGroup)
+
 GetSolutionDataPhysicsAIRequest = _reflection.GeneratedProtocolMessageType('GetSolutionDataPhysicsAIRequest', (_message.Message,), {
 
   'InternalOptionsEntry' : _reflection.GeneratedProtocolMessageType('InternalOptionsEntry', (_message.Message,), {
@@ -163,6 +181,69 @@ SubmitTrainingJobResponse = _reflection.GeneratedProtocolMessageType('SubmitTrai
   })
 _sym_db.RegisterMessage(SubmitTrainingJobResponse)
 
+CancelTrainingJobRequest = _reflection.GeneratedProtocolMessageType('CancelTrainingJobRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELTRAININGJOBREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobRequest)
+  })
+_sym_db.RegisterMessage(CancelTrainingJobRequest)
+
+CancelTrainingJobResponse = _reflection.GeneratedProtocolMessageType('CancelTrainingJobResponse', (_message.Message,), {
+  'DESCRIPTOR' : _CANCELTRAININGJOBRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.CancelTrainingJobResponse)
+  })
+_sym_db.RegisterMessage(CancelTrainingJobResponse)
+
+PhysicsAiDataset = _reflection.GeneratedProtocolMessageType('PhysicsAiDataset', (_message.Message,), {
+  'DESCRIPTOR' : _PHYSICSAIDATASET,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDataset)
+  })
+_sym_db.RegisterMessage(PhysicsAiDataset)
+
+PhysicsAiDatasetCaseFile = _reflection.GeneratedProtocolMessageType('PhysicsAiDatasetCaseFile', (_message.Message,), {
+  'DESCRIPTOR' : _PHYSICSAIDATASETCASEFILE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFile)
+  })
+_sym_db.RegisterMessage(PhysicsAiDatasetCaseFile)
+
+PhysicsAiDatasetCaseFiles = _reflection.GeneratedProtocolMessageType('PhysicsAiDatasetCaseFiles', (_message.Message,), {
+  'DESCRIPTOR' : _PHYSICSAIDATASETCASEFILES,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCaseFiles)
+  })
+_sym_db.RegisterMessage(PhysicsAiDatasetCaseFiles)
+
+PhysicsAiDatasetCase = _reflection.GeneratedProtocolMessageType('PhysicsAiDatasetCase', (_message.Message,), {
+  'DESCRIPTOR' : _PHYSICSAIDATASETCASE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiDatasetCase)
+  })
+_sym_db.RegisterMessage(PhysicsAiDatasetCase)
+
+CreatePhysicsAiDatasetCaseInput = _reflection.GeneratedProtocolMessageType('CreatePhysicsAiDatasetCaseInput', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEPHYSICSAIDATASETCASEINPUT,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.CreatePhysicsAiDatasetCaseInput)
+  })
+_sym_db.RegisterMessage(CreatePhysicsAiDatasetCaseInput)
+
+CreateDatasetRequest = _reflection.GeneratedProtocolMessageType('CreateDatasetRequest', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEDATASETREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetRequest)
+  })
+_sym_db.RegisterMessage(CreateDatasetRequest)
+
+CreateDatasetResponse = _reflection.GeneratedProtocolMessageType('CreateDatasetResponse', (_message.Message,), {
+  'DESCRIPTOR' : _CREATEDATASETRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.CreateDatasetResponse)
+  })
+_sym_db.RegisterMessage(CreateDatasetResponse)
+
 _PHYSICSAISERVICE = DESCRIPTOR.services_by_name['PhysicsAiService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -178,42 +259,64 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._serialized_options = b'\202\323\344\223\0020\"+/v0/physics_ai/solutions/{solution_id}/data:\001*'
   _PHYSICSAISERVICE.methods_by_name['SubmitTrainingJob']._options = None
   _PHYSICSAISERVICE.methods_by_name['SubmitTrainingJob']._serialized_options = b'\202\323\344\223\002!\"\034/v0/physics_ai/training/jobs:\001*'
-  _PHYSICSAILIFECYCLESTATE._serialized_start=3019
-  _PHYSICSAILIFECYCLESTATE._serialized_end=3199
-  _TRAININGDATASOURCETYPE._serialized_start=3202
-  _TRAININGDATASOURCETYPE._serialized_end=3369
-  _MODELINITIALIZATIONTYPE._serialized_start=3372
-  _MODELINITIALIZATIONTYPE._serialized_end=3565
-  _PHYSICSAIARCHITECTUREVERSION._serialized_start=272
-  _PHYSICSAIARCHITECTUREVERSION._serialized_end=445
-  _PHYSICSAIARCHITECTURE._serialized_start=448
-  _PHYSICSAIARCHITECTURE._serialized_end=614
-  _LISTARCHITECTURESREQUEST._serialized_start=616
-  _LISTARCHITECTURESREQUEST._serialized_end=642
-  _LISTARCHITECTURESRESPONSE._serialized_start=644
-  _LISTARCHITECTURESRESPONSE._serialized_end=765
-  _PHYSICSAIMODELVERSION._serialized_start=768
-  _PHYSICSAIMODELVERSION._serialized_end=915
-  _PHYSICSAIMODEL._serialized_start=918
-  _PHYSICSAIMODEL._serialized_end=1070
-  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1072
-  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1101
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1103
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1213
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1216
-  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1691
-  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_start=1637
-  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_end=1691
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1693
-  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=1791
-  _TRAININGSOLUTION._serialized_start=1793
-  _TRAININGSOLUTION._serialized_end=1852
-  _PHYSICSAITRAININGJOB._serialized_start=1855
-  _PHYSICSAITRAININGJOB._serialized_end=2521
-  _SUBMITTRAININGJOBREQUEST._serialized_start=2524
-  _SUBMITTRAININGJOBREQUEST._serialized_end=2895
-  _SUBMITTRAININGJOBRESPONSE._serialized_start=2897
-  _SUBMITTRAININGJOBRESPONSE._serialized_end=3016
-  _PHYSICSAISERVICE._serialized_start=3568
-  _PHYSICSAISERVICE._serialized_end=4477
+  _PHYSICSAISERVICE.methods_by_name['CreateDataset']._options = None
+  _PHYSICSAISERVICE.methods_by_name['CreateDataset']._serialized_options = b'\202\323\344\223\002\034\"\027/v0/physics_ai/datasets:\001*'
+  _PHYSICSAILIFECYCLESTATE._serialized_start=4879
+  _PHYSICSAILIFECYCLESTATE._serialized_end=5059
+  _TRAININGDATASOURCETYPE._serialized_start=5062
+  _TRAININGDATASOURCETYPE._serialized_end=5229
+  _MODELINITIALIZATIONTYPE._serialized_start=5232
+  _MODELINITIALIZATIONTYPE._serialized_end=5425
+  _PHYSICSAIARCHITECTUREVERSION._serialized_start=302
+  _PHYSICSAIARCHITECTUREVERSION._serialized_end=475
+  _PHYSICSAIARCHITECTURE._serialized_start=478
+  _PHYSICSAIARCHITECTURE._serialized_end=644
+  _LISTARCHITECTURESREQUEST._serialized_start=646
+  _LISTARCHITECTURESREQUEST._serialized_end=672
+  _LISTARCHITECTURESRESPONSE._serialized_start=674
+  _LISTARCHITECTURESRESPONSE._serialized_end=795
+  _PHYSICSAIMODELVERSION._serialized_start=798
+  _PHYSICSAIMODELVERSION._serialized_end=945
+  _PHYSICSAIMODEL._serialized_start=948
+  _PHYSICSAIMODEL._serialized_end=1100
+  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1102
+  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1131
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1133
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1243
+  _SURFACEGROUP._serialized_start=1245
+  _SURFACEGROUP._serialized_end=1291
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1294
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1862
+  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_start=1808
+  _GETSOLUTIONDATAPHYSICSAIREQUEST_INTERNALOPTIONSENTRY._serialized_end=1862
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1864
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=1962
+  _TRAININGSOLUTION._serialized_start=1964
+  _TRAININGSOLUTION._serialized_end=2023
+  _PHYSICSAITRAININGJOB._serialized_start=2026
+  _PHYSICSAITRAININGJOB._serialized_end=2712
+  _SUBMITTRAININGJOBREQUEST._serialized_start=2715
+  _SUBMITTRAININGJOBREQUEST._serialized_end=3106
+  _SUBMITTRAININGJOBRESPONSE._serialized_start=3108
+  _SUBMITTRAININGJOBRESPONSE._serialized_end=3227
+  _CANCELTRAININGJOBREQUEST._serialized_start=3229
+  _CANCELTRAININGJOBREQUEST._serialized_end=3280
+  _CANCELTRAININGJOBRESPONSE._serialized_start=3282
+  _CANCELTRAININGJOBRESPONSE._serialized_end=3309
+  _PHYSICSAIDATASET._serialized_start=3312
+  _PHYSICSAIDATASET._serialized_end=3752
+  _PHYSICSAIDATASETCASEFILE._serialized_start=3754
+  _PHYSICSAIDATASETCASEFILE._serialized_end=3813
+  _PHYSICSAIDATASETCASEFILES._serialized_start=3815
+  _PHYSICSAIDATASETCASEFILES._serialized_end=3931
+  _PHYSICSAIDATASETCASE._serialized_start=3934
+  _PHYSICSAIDATASETCASE._serialized_end=4197
+  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_start=4200
+  _CREATEPHYSICSAIDATASETCASEINPUT._serialized_end=4457
+  _CREATEDATASETREQUEST._serialized_start=4460
+  _CREATEDATASETREQUEST._serialized_end=4768
+  _CREATEDATASETRESPONSE._serialized_start=4770
+  _CREATEDATASETRESPONSE._serialized_end=4876
+  _PHYSICSAISERVICE._serialized_start=5428
+  _PHYSICSAISERVICE._serialized_end=6532
 # @@protoc_insertion_point(module_scope)