luminarycloud 0.15.5__py3-none-any.whl → 0.16.1__py3-none-any.whl

luminarycloud/_client/client.py

@@ -32,6 +32,9 @@ from .._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2_grpc import (
     PhysicsAiServiceStub,
 )
 from .._proto.api.v0.luminarycloud.inference.inference_pb2_grpc import InferenceServiceStub
+from .._proto.api.v0.luminarycloud.thirdpartyintegration.onshape.onshape_pb2_grpc import (
+    OnshapeServiceStub,
+)
 from .._proto.api.v0.luminarycloud.solution.solution_pb2_grpc import SolutionServiceStub
 from .._proto.api.v0.luminarycloud.upload.upload_pb2_grpc import UploadServiceStub
 from .._proto.api.v0.luminarycloud.vis.vis_pb2_grpc import VisAPIServiceStub
@@ -60,6 +63,7 @@ class Client(
     NamedVariableSetServiceStub,
     PhysicsAiServiceStub,
     InferenceServiceStub,
+    OnshapeServiceStub,
 ):
     """
     Creates a Luminary API client.
@@ -206,6 +210,7 @@ class Client(
         PhysicsAiServiceStub.__init__(self, self._channel)
         InferenceServiceStub.__init__(self, self._channel)
         NamedVariableSetServiceStub.__init__(self, self._channel)
+        OnshapeServiceStub.__init__(self, self._channel)
         for name, value in self.__dict__.items():
             if isinstance(value, grpc.UnaryUnaryMultiCallable):
                 setattr(self, name, rpc_error(value))

luminarycloud/_helpers/__init__.py

@@ -11,6 +11,7 @@ from .download import (
     download_surface_deformation_template as download_surface_deformation_template,
     download_surface_sensitivity_data as download_surface_sensitivity_data,
     download_parameter_sensitivity_data as download_parameter_sensitivity_data,
+    download_solution_physics_ai as download_solution_physics_ai,
     save_file as save_file,
 )
 from .file_chunk_stream import (

luminarycloud/_helpers/_code_representation.py

@@ -1,5 +1,5 @@
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
-from typing import Any
+from typing import Any, Iterator, Tuple
 from google.protobuf.message import Message as _ProtoMessage
 
 
@@ -100,7 +100,7 @@ class CodeRepr:
             code = f" = {type(self).__name__}()\n"
 
         default = type(self)()
-        for field, val in vars(self).items():
+        for field, val in self._iterate_fields_for_code_gen():
             # Skip default values.
             if hide_defaults and val == vars(default).get(field):
                 continue
@@ -108,7 +108,8 @@ class CodeRepr:
             if isinstance(val, dict):
                 code += "# Clear the map and insert new items.\n"
                 code += f".{field} = {{}}\n"
-                for key, value in val.items():
+                # Sort dictionary items by key to ensure deterministic ordering
+                for key, value in sorted(val.items(), key=lambda x: map_key(x[0])):
                     # Simplify the code for maps of built-in types by not creating a temporary
                     # instance.
                     if isinstance(value, (bool, int, float)):
@@ -143,12 +144,28 @@ class CodeRepr:
                 code += generate(field, val)
         return code
 
+    # All fields yielded here will be included in code generation. By default, we iterate over all
+    # "public" instance variables (i.e. ones that do not begin with "_") and properties, but a class
+    # can override this if it's needs to do something funny like conditionally hide fields.
+    def _iterate_fields_for_code_gen(self) -> Iterator[Tuple[str, Any]]:
+        # Yield all instance variables that are not "private"
+        # Sort by field name to ensure deterministic ordering
+        for item in sorted(vars(self).items()):
+            if item[0].startswith("_"):
+                continue
+            yield item
+
+        # Yield all properties
+        for name in sorted(dir(self)):
+            if isinstance(getattr(type(self), name, None), property):
+                yield (name, getattr(self, name))
+
     # See the class description.
     def _to_code_helper(
         self, obj_name: str = "obj", hide_defaults: bool = True, use_tmp_objs: bool = True
     ) -> str:
         code = ""
-        for line in CodeRepr._to_code(self, hide_defaults, use_tmp_objs).split("\n"):
+        for line in self._to_code(hide_defaults, use_tmp_objs).split("\n"):
             if line:
                 if line.startswith(".new_"):
                     code += f"{line[1:]}\n"

luminarycloud/_helpers/download.py

@@ -4,7 +4,7 @@ import logging
 import os
 from pathlib import Path
 import requests
-from typing import Any, Iterator, Optional, Union, cast
+from typing import Any, Iterator, Optional, Union, cast, List
 
 from .file_chunk_stream import FileChunkStream
 from .._proto.api.v0.luminarycloud.common import common_pb2 as commonpb
@@ -15,7 +15,11 @@ from .._proto.api.v0.luminarycloud.solution.solution_pb2 import (
     GetSurfaceSensitivityDataRequest,
     GetParameterSensitivityDataRequest,
 )
+from .._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2 import (
+    GetSolutionDataPhysicsAIRequest,
+)
 from .._client import Client
+from ..enum.quantity_type import QuantityType
 
 logger = logging.getLogger(__name__)
 
@@ -126,6 +130,68 @@ def download_volume_solution(
     return _create_file_chunk_stream(client, solution_id, "volume solution", response.file)
 
 
+def download_solution_physics_ai(
+    client: Client,
+    solution_id: str,
+    exclude_surfaces: Optional[List[str]] = None,
+    fill_holes: float = -1.0,
+    surface_fields_to_keep: Optional[List[QuantityType]] = None,
+    volume_fields_to_keep: Optional[List[QuantityType]] = None,
+    process_volume: bool = False,
+    single_precision: bool = False,
+) -> Optional[FileChunkStream]:
+    """
+    Returns the download as a file-like object, or None if destination_url is provided.
+
+    The filename can be retrieved from the `filename` attribute of the returned object.
+
+    Parameters
+    ----------
+    client: Client
+        The client to use for the download.
+    solution_id: str
+        The ID of the solution to download.
+    exclude_surfaces: Optional[List[str]]
+        List of surfaces to exclude from surface solution during physics AI processing.
+    fill_holes: float
+        Sets the maximum size of the hole to be filled for the STL file, measured as the radius of the bounding circumsphere.
+        If fill_holes is negative or zero, no holes will be filled.
+    surface_fields_to_keep: List of QuantityType enum values for surface fields to keep in output.
+            If None, all available surface fields are included.
+    volume_fields_to_keep: List of QuantityType enum values for volume fields to keep in output.
+            If None, all available volume fields are included.
+    process_volume: bool
+        Whether to process volume meshes during physics AI processing.
+    single_precision: bool
+        If True, the solution will be downloaded in single precision.
+
+    Examples
+    --------
+    >>> with download_solution_physics_ai(client, "your-solution-id", process_volume=True) as dl:
+    ...     with open(dl.filename, "wb") as fp:
+    ...         fp.write(dl.read())
+    """
+
+    request = GetSolutionDataPhysicsAIRequest(
+        solution_id=solution_id,
+        exclude_surfaces=exclude_surfaces or [],
+        fill_holes=fill_holes,
+        surface_fields_to_keep=(
+            [x.value for x in surface_fields_to_keep] if surface_fields_to_keep else []
+        ),
+        volume_fields_to_keep=(
+            [x.value for x in volume_fields_to_keep] if volume_fields_to_keep else []
+        ),
+        process_volume=process_volume,
+        single_precision=single_precision,
+    )
+    response = client.GetSolutionDataPhysicsAI(request)
+
+    return _create_file_chunk_stream(
+        client, solution_id, "physics ai processed solution", response.file
+    )
+
+
 def download_surface_deformation_template(
     client: Client,
     solution_id: str,

luminarycloud/_proto/api/v0/luminarycloud/inference/inference_pb2.py

@@ -18,7 +18,7 @@ from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
 from luminarycloud._proto.inferenceservice import inferenceservice_pb2 as proto_dot_inferenceservice_dot_inferenceservice__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4proto/api/v0/luminarycloud/inference/inference.proto\x12-luminary.proto.api.v0.luminarycloud.inference\x1a\x1cgoogle/api/annotations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a-proto/inferenceservice/inferenceservice.proto\"q\n CreateInferenceServiceJobRequest\x12\x0f\n\x07stl_url\x18\x01 \x01(\t\x12\x11\n\tmodel_url\x18\x02 \x01(\t\x12\x13\n\x0b\x63onfig_name\x18\x03 \x01(\t\x12\x14\n\x0cstencil_size\x18\x04 \x01(\x05\"\x8e\x01\n!CreateInferenceServiceJobResponse\x12\x45\n\x06status\x18\x01 \x01(\x0e\x32\x35.luminary.proto.api.v0.luminarycloud.inference.Status\x12\x15\n\x08response\x18\x02 \x01(\x0cH\x00\x88\x01\x01\x42\x0b\n\t_response*D\n\x06Status\x12\x12\n\x0eSTATUS_PENDING\x10\x00\x12\x12\n\x0eSTATUS_SUCCESS\x10\x01\x12\x12\n\x0eSTATUS_FAILURE\x10\x02\x32\xea\x01\n\x10InferenceService\x12\xd5\x01\n\x19\x43reateInferenceServiceJob\x12O.luminary.proto.api.v0.luminarycloud.inference.CreateInferenceServiceJobRequest\x1aP.luminary.proto.api.v0.luminarycloud.inference.CreateInferenceServiceJobResponse\"\x15\x82\xd3\xe4\x93\x02\x0f\"\r/v0/inferenceB=Z;luminarycloud.com/core/proto/api/v0/luminarycloud/inferenceb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4proto/api/v0/luminarycloud/inference/inference.proto\x12-luminary.proto.api.v0.luminarycloud.inference\x1a\x1cgoogle/api/annotations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a-proto/inferenceservice/inferenceservice.proto\"\x8a\x01\n CreateInferenceServiceJobRequest\x12\x0f\n\x07stl_url\x18\x01 \x01(\t\x12\x16\n\x0e\x63heckpoint_url\x18\x02 \x01(\t\x12\x13\n\x0b\x63onfig_name\x18\x03 \x01(\t\x12\x14\n\x0cstencil_size\x18\x04 \x01(\x05\x12\x12\n\nproject_id\x18\x05 \x01(\t\"\x8e\x01\n!CreateInferenceServiceJobResponse\x12\x45\n\x06status\x18\x01 \x01(\x0e\x32\x35.luminary.proto.api.v0.luminarycloud.inference.Status\x12\x15\n\x08response\x18\x02 \x01(\x0cH\x00\x88\x01\x01\x42\x0b\n\t_response*D\n\x06Status\x12\x12\n\x0eSTATUS_PENDING\x10\x00\x12\x12\n\x0eSTATUS_SUCCESS\x10\x01\x12\x12\n\x0eSTATUS_FAILURE\x10\x02\x32\xea\x01\n\x10InferenceService\x12\xd5\x01\n\x19\x43reateInferenceServiceJob\x12O.luminary.proto.api.v0.luminarycloud.inference.CreateInferenceServiceJobRequest\x1aP.luminary.proto.api.v0.luminarycloud.inference.CreateInferenceServiceJobResponse\"\x15\x82\xd3\xe4\x93\x02\x0f\"\r/v0/inferenceB=Z;luminarycloud.com/core/proto/api/v0/luminarycloud/inferenceb\x06proto3')
 
 _STATUS = DESCRIPTOR.enum_types_by_name['Status']
 Status = enum_type_wrapper.EnumTypeWrapper(_STATUS)
@@ -50,12 +50,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._serialized_options = b'Z;luminarycloud.com/core/proto/api/v0/luminarycloud/inference'
   _INFERENCESERVICE.methods_by_name['CreateInferenceServiceJob']._options = None
   _INFERENCESERVICE.methods_by_name['CreateInferenceServiceJob']._serialized_options = b'\202\323\344\223\002\017\"\r/v0/inference'
-  _STATUS._serialized_start=469
-  _STATUS._serialized_end=537
-  _CREATEINFERENCESERVICEJOBREQUEST._serialized_start=209
-  _CREATEINFERENCESERVICEJOBREQUEST._serialized_end=322
-  _CREATEINFERENCESERVICEJOBRESPONSE._serialized_start=325
-  _CREATEINFERENCESERVICEJOBRESPONSE._serialized_end=467
-  _INFERENCESERVICE._serialized_start=540
-  _INFERENCESERVICE._serialized_end=774
+  _STATUS._serialized_start=495
+  _STATUS._serialized_end=563
+  _CREATEINFERENCESERVICEJOBREQUEST._serialized_start=210
+  _CREATEINFERENCESERVICEJOBREQUEST._serialized_end=348
+  _CREATEINFERENCESERVICEJOBRESPONSE._serialized_start=351
+  _CREATEINFERENCESERVICEJOBRESPONSE._serialized_end=493
+  _INFERENCESERVICE._serialized_start=566
+  _INFERENCESERVICE._serialized_end=800
 # @@protoc_insertion_point(module_scope)

luminarycloud/_proto/api/v0/luminarycloud/inference/inference_pb2.pyi

@@ -37,22 +37,25 @@ class CreateInferenceServiceJobRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     STL_URL_FIELD_NUMBER: builtins.int
-    MODEL_URL_FIELD_NUMBER: builtins.int
+    CHECKPOINT_URL_FIELD_NUMBER: builtins.int
     CONFIG_NAME_FIELD_NUMBER: builtins.int
     STENCIL_SIZE_FIELD_NUMBER: builtins.int
+    PROJECT_ID_FIELD_NUMBER: builtins.int
     stl_url: builtins.str
-    model_url: builtins.str
+    checkpoint_url: builtins.str
     config_name: builtins.str
     stencil_size: builtins.int
+    project_id: builtins.str
     def __init__(
         self,
         *,
         stl_url: builtins.str = ...,
-        model_url: builtins.str = ...,
+        checkpoint_url: builtins.str = ...,
         config_name: builtins.str = ...,
         stencil_size: builtins.int = ...,
+        project_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["config_name", b"config_name", "model_url", b"model_url", "stencil_size", b"stencil_size", "stl_url", b"stl_url"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["checkpoint_url", b"checkpoint_url", "config_name", b"config_name", "project_id", b"project_id", "stencil_size", b"stencil_size", "stl_url", b"stl_url"]) -> None: ...
 
 global___CreateInferenceServiceJobRequest = CreateInferenceServiceJobRequest
 

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

@@ -15,9 +15,11 @@ _sym_db = _symbol_database.Default()
 
 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
+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\"\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*\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\x32\xc1\x03\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_modelsB>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/api/annotations.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a.proto/api/v0/luminarycloud/common/common.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\"\xa3\x02\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\"b\n GetSolutionDataPhysicsAIResponse\x12>\n\x04\x66ile\x18\x01 \x01(\x0b\x32\x30.luminary.proto.api.v0.luminarycloud.common.File*\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\x32\xb9\x05\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*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)
@@ -36,6 +38,8 @@ _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']
+_GETSOLUTIONDATAPHYSICSAIREQUEST = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIRequest']
+_GETSOLUTIONDATAPHYSICSAIRESPONSE = DESCRIPTOR.message_types_by_name['GetSolutionDataPhysicsAIResponse']
 PhysicsAiArchitectureVersion = _reflection.GeneratedProtocolMessageType('PhysicsAiArchitectureVersion', (_message.Message,), {
   'DESCRIPTOR' : _PHYSICSAIARCHITECTUREVERSION,
   '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
@@ -92,6 +96,20 @@ ListPretrainedModelsResponse = _reflection.GeneratedProtocolMessageType('ListPre
   })
 _sym_db.RegisterMessage(ListPretrainedModelsResponse)
 
+GetSolutionDataPhysicsAIRequest = _reflection.GeneratedProtocolMessageType('GetSolutionDataPhysicsAIRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETSOLUTIONDATAPHYSICSAIREQUEST,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIRequest)
+  })
+_sym_db.RegisterMessage(GetSolutionDataPhysicsAIRequest)
+
+GetSolutionDataPhysicsAIResponse = _reflection.GeneratedProtocolMessageType('GetSolutionDataPhysicsAIResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETSOLUTIONDATAPHYSICSAIRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.physics_ai.GetSolutionDataPhysicsAIResponse)
+  })
+_sym_db.RegisterMessage(GetSolutionDataPhysicsAIResponse)
+
 _PHYSICSAISERVICE = DESCRIPTOR.services_by_name['PhysicsAiService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
@@ -101,24 +119,30 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PHYSICSAISERVICE.methods_by_name['ListArchitectures']._serialized_options = b'\202\323\344\223\002\036\022\034/v0/physics_ai/architectures'
   _PHYSICSAISERVICE.methods_by_name['ListPretrainedModels']._options = None
   _PHYSICSAISERVICE.methods_by_name['ListPretrainedModels']._serialized_options = b'\202\323\344\223\002\"\022 /v0/physics_ai/pretrained_models'
-  _PHYSICSAILIFECYCLESTATE._serialized_start=1114
-  _PHYSICSAILIFECYCLESTATE._serialized_end=1294
-  _PHYSICSAIARCHITECTUREVERSION._serialized_start=170
-  _PHYSICSAIARCHITECTUREVERSION._serialized_end=343
-  _PHYSICSAIARCHITECTURE._serialized_start=346
-  _PHYSICSAIARCHITECTURE._serialized_end=512
-  _LISTARCHITECTURESREQUEST._serialized_start=514
-  _LISTARCHITECTURESREQUEST._serialized_end=540
-  _LISTARCHITECTURESRESPONSE._serialized_start=542
-  _LISTARCHITECTURESRESPONSE._serialized_end=663
-  _PHYSICSAIMODELVERSION._serialized_start=666
-  _PHYSICSAIMODELVERSION._serialized_end=813
-  _PHYSICSAIMODEL._serialized_start=816
-  _PHYSICSAIMODEL._serialized_end=968
-  _LISTPRETRAINEDMODELSREQUEST._serialized_start=970
-  _LISTPRETRAINEDMODELSREQUEST._serialized_end=999
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1001
-  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1111
-  _PHYSICSAISERVICE._serialized_start=1297
-  _PHYSICSAISERVICE._serialized_end=1746
+  _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._options = None
+  _PHYSICSAISERVICE.methods_by_name['GetSolutionDataPhysicsAI']._serialized_options = b'\202\323\344\223\0020\"+/v0/physics_ai/solutions/{solution_id}/data:\001*'
+  _PHYSICSAILIFECYCLESTATE._serialized_start=1587
+  _PHYSICSAILIFECYCLESTATE._serialized_end=1767
+  _PHYSICSAIARCHITECTUREVERSION._serialized_start=249
+  _PHYSICSAIARCHITECTUREVERSION._serialized_end=422
+  _PHYSICSAIARCHITECTURE._serialized_start=425
+  _PHYSICSAIARCHITECTURE._serialized_end=591
+  _LISTARCHITECTURESREQUEST._serialized_start=593
+  _LISTARCHITECTURESREQUEST._serialized_end=619
+  _LISTARCHITECTURESRESPONSE._serialized_start=621
+  _LISTARCHITECTURESRESPONSE._serialized_end=742
+  _PHYSICSAIMODELVERSION._serialized_start=745
+  _PHYSICSAIMODELVERSION._serialized_end=892
+  _PHYSICSAIMODEL._serialized_start=895
+  _PHYSICSAIMODEL._serialized_end=1047
+  _LISTPRETRAINEDMODELSREQUEST._serialized_start=1049
+  _LISTPRETRAINEDMODELSREQUEST._serialized_end=1078
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_start=1080
+  _LISTPRETRAINEDMODELSRESPONSE._serialized_end=1190
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_start=1193
+  _GETSOLUTIONDATAPHYSICSAIREQUEST._serialized_end=1484
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_start=1486
+  _GETSOLUTIONDATAPHYSICSAIRESPONSE._serialized_end=1584
+  _PHYSICSAISERVICE._serialized_start=1770
+  _PHYSICSAISERVICE._serialized_end=2467
 # @@protoc_insertion_point(module_scope)

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

@@ -8,6 +8,8 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
+import luminarycloud._proto.api.v0.luminarycloud.common.common_pb2
+import luminarycloud._proto.quantity.quantity_pb2
 import sys
 import typing
 
@@ -225,3 +227,66 @@ class ListPretrainedModelsResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["models", b"models"]) -> None: ...
 
 global___ListPretrainedModelsResponse = ListPretrainedModelsResponse
+
+class GetSolutionDataPhysicsAIRequest(google.protobuf.message.Message):
+    """Request message for download and process solutions for physics ai"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOLUTION_ID_FIELD_NUMBER: builtins.int
+    EXCLUDE_SURFACES_FIELD_NUMBER: builtins.int
+    FILL_HOLES_FIELD_NUMBER: builtins.int
+    SURFACE_FIELDS_TO_KEEP_FIELD_NUMBER: builtins.int
+    VOLUME_FIELDS_TO_KEEP_FIELD_NUMBER: builtins.int
+    PROCESS_VOLUME_FIELD_NUMBER: builtins.int
+    SINGLE_PRECISION_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"""
+    fill_holes: builtins.float
+    """Size threshold for filling holes in the surface mesh (0 or negative value means no filling)"""
+    @property
+    def surface_fields_to_keep(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """List of surface fields to keep (all others will be removed). If left empty, all fields are kept."""
+    @property
+    def volume_fields_to_keep(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType]:
+        """List of volume fields to keep (all others will be removed). If left empty, all fields are kept."""
+    process_volume: builtins.bool
+    """Whether to process volume meshes during physics AI processing"""
+    single_precision: builtins.bool
+    """Whether to export the floating point fields in single precision"""
+    def __init__(
+        self,
+        *,
+        solution_id: builtins.str = ...,
+        exclude_surfaces: collections.abc.Iterable[builtins.str] | None = ...,
+        fill_holes: builtins.float = ...,
+        surface_fields_to_keep: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        volume_fields_to_keep: collections.abc.Iterable[luminarycloud._proto.quantity.quantity_pb2.QuantityType.ValueType] | None = ...,
+        process_volume: builtins.bool = ...,
+        single_precision: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["exclude_surfaces", b"exclude_surfaces", "fill_holes", b"fill_holes", "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
+
+class GetSolutionDataPhysicsAIResponse(google.protobuf.message.Message):
+    """Response message for download and process solutions for physics ai"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_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."""
+    def __init__(
+        self,
+        *,
+        file: luminarycloud._proto.api.v0.luminarycloud.common.common_pb2.File | None = ...,
+    ) -> 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: ...
+
+global___GetSolutionDataPhysicsAIResponse = GetSolutionDataPhysicsAIResponse

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

@@ -25,6 +25,11 @@ class PhysicsAiServiceStub(object):
                 request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListPretrainedModelsRequest.SerializeToString,
                 response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListPretrainedModelsResponse.FromString,
                 )
+        self.GetSolutionDataPhysicsAI = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService/GetSolutionDataPhysicsAI',
+                request_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIRequest.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.FromString,
+                )
 
 
 class PhysicsAiServiceServicer(object):
@@ -45,6 +50,13 @@ class PhysicsAiServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetSolutionDataPhysicsAI(self, request, context):
+        """Gets solution data with physics AI processing applied
+        """
+        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 = {
@@ -58,6 +70,11 @@ def add_PhysicsAiServiceServicer_to_server(servicer, server):
                     request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListPretrainedModelsRequest.FromString,
                     response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListPretrainedModelsResponse.SerializeToString,
             ),
+            'GetSolutionDataPhysicsAI': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetSolutionDataPhysicsAI,
+                    request_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIRequest.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'luminary.proto.api.v0.luminarycloud.physics_ai.PhysicsAiService', rpc_method_handlers)
@@ -102,3 +119,20 @@ class PhysicsAiService(object):
             proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.ListPretrainedModelsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetSolutionDataPhysicsAI(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/GetSolutionDataPhysicsAI',
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIRequest.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_physics__ai_dot_physics__ai__pb2.GetSolutionDataPhysicsAIResponse.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

@@ -20,6 +20,11 @@ class PhysicsAiServiceStub:
         luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListPretrainedModelsResponse,
     ]
     """Lists available pretrained physics ai models."""
+    GetSolutionDataPhysicsAI: grpc.UnaryUnaryMultiCallable[
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIRequest,
+        luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIResponse,
+    ]
+    """Gets solution data with physics AI processing applied"""
 
 class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
     """Manages physics ai architectures."""
@@ -38,5 +43,12 @@ class PhysicsAiServiceServicer(metaclass=abc.ABCMeta):
         context: grpc.ServicerContext,
     ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.ListPretrainedModelsResponse:
         """Lists available pretrained physics ai models."""
+    @abc.abstractmethod
+    def GetSolutionDataPhysicsAI(
+        self,
+        request: luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIRequest,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.physics_ai.physics_ai_pb2.GetSolutionDataPhysicsAIResponse:
+        """Gets solution data with physics AI processing applied"""
 
 def add_PhysicsAiServiceServicer_to_server(servicer: PhysicsAiServiceServicer, server: grpc.Server) -> None: ...