luminarycloud 0.21.1__tar.gz → 0.22.0__tar.gz

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: luminarycloud
-Version: 0.21.1
+Version: 0.22.0
 Summary: Luminary Cloud SDK
 Project-URL: Homepage, https://www.luminarycloud.com/
 Project-URL: Documentation, https://app.luminarycloud.com/docs/api/

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/luminarycloud/__init__.py

@@ -13,12 +13,14 @@ from . import (
     types as types,
     vis as vis,
 )
-
 from ._client import (
     Client as Client,
     get_default_client as get_default_client,
     set_default_client as set_default_client,
 )
+from ._feature_flag import (
+    _get_feature_flags as _get_feature_flags,
+)
 from .geometry import (
     get_geometry as get_geometry,
     Geometry as Geometry,

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/luminarycloud/_client/client.py

@@ -44,6 +44,9 @@ from .._proto.api.v0.luminarycloud.project_ui_state.project_ui_state_pb2_grpc im
 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
+from .._proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2_grpc import (
+    FeatureFlagServiceStub,
+)
 from .authentication_plugin import AuthenticationPlugin
 from .config import LC_DOMAIN, LC_API_KEY
 from .logging_interceptor import LoggingInterceptor
@@ -71,6 +74,7 @@ class Client(
     InferenceServiceStub,
     OnshapeServiceStub,
     ProjectUIStateServiceStub,
+    FeatureFlagServiceStub,
 ):
     """
     Creates a Luminary API client.
@@ -248,6 +252,7 @@ class Client(
         NamedVariableSetServiceStub.__init__(self, self._channel)
         OnshapeServiceStub.__init__(self, self._channel)
         ProjectUIStateServiceStub.__init__(self, self._channel)
+        FeatureFlagServiceStub.__init__(self, self._channel)
         for name, value in self.__dict__.items():
             if isinstance(value, grpc.UnaryUnaryMultiCallable):
                 setattr(self, name, rpc_error(value))

luminarycloud-0.22.0/luminarycloud/_feature_flag.py

@@ -0,0 +1,22 @@
+# Copyright 2023-2025 Luminary Cloud, Inc. All Rights Reserved.
+from google.protobuf import empty_pb2
+
+from ._proto.api.v0.luminarycloud.feature_flag import feature_flag_pb2
+from ._client import get_default_client
+
+
+def _get_feature_flags() -> dict[int, str]:
+    """Get enabled feature flags for the authenticated user.
+
+    Feature flags are used to control access to experimental and internal features.
+    Each flag is identified by an experiment ID and a feature flag name.
+
+    Returns
+    -------
+    dict[int, str]
+        A dictionary mapping experiment IDs (uint64) to feature flag names.
+        Returns an empty dictionary if no feature flags are enabled for the user.
+    """
+    req = empty_pb2.Empty()
+    res = get_default_client().GetFeatureFlags(req)
+    return dict(res.enabled_feature_flags)

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/luminarycloud/_helpers/_create_simulation.py

@@ -7,6 +7,9 @@ from .._proto.api.v0.luminarycloud.simulation.simulation_pb2 import (
     SimulationOptions,
     CreateSimulationRequest,
 )
+from .._proto.api.v0.luminarycloud.simulation_template.simulation_template_pb2 import (
+    SimulationTemplate,
+)
 from .._client import Client
 from ..enum import GPUType
 
@@ -18,8 +21,9 @@ def create_simulation(
     project_id: str,
     mesh_id: str,
     name: str,
-    simulation_template_id: str,
+    simulation_template_id: Optional[str] = None,
     *,
+    simulation_template: Optional[SimulationTemplate] = None,
     named_variable_set_version_id: Optional[str] = None,
     description: str = "",
     batch_processing: bool = True,
@@ -32,7 +36,8 @@ def create_simulation(
         CreateSimulationRequest(
             project_id=project_id,
             mesh_id=mesh_id,
-            simulation_template_id=simulation_template_id,
+            simulation_template_id=simulation_template_id or "",
+            simulation_template=simulation_template,
             named_variable_set_version_id=named_variable_set_version_id or "",
             name=name,
             description=description,

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/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, List
+from typing import Any, Iterator, Optional, Union, cast, List, Dict
 
 from .file_chunk_stream import FileChunkStream
 from .._proto.api.v0.luminarycloud.common import common_pb2 as commonpb
@@ -139,6 +139,7 @@ def download_solution_physics_ai(
     volume_fields_to_keep: Optional[List[QuantityType]] = None,
     process_volume: bool = False,
     single_precision: bool = False,
+    internal_options: Optional[Dict[str, str]] = None,
 ) -> Optional[FileChunkStream]:
     """
     Returns the download as a file-like object, or None if destination_url is provided.
@@ -184,6 +185,7 @@ def download_solution_physics_ai(
         ),
         process_volume=process_volume,
         single_precision=single_precision,
+        internal_options=internal_options or {},
     )
     response = client.GetSolutionDataPhysicsAI(request)
 

{luminarycloud-0.21.1 → luminarycloud-0.22.0}/luminarycloud/_helpers/proto_decorator.py

@@ -9,6 +9,9 @@ from typing import (
 from google.protobuf.message import Message
 
 from luminarycloud.types import Vector3
+from luminarycloud.types.adfloat import _to_ad_proto
+from luminarycloud.types.vector3 import _to_vector3_ad_proto
+from .._proto.base.base_pb2 import AdFloatType, AdVector3
 
 P = TypeVar("P", bound=Message)
 C = TypeVar("C")
@@ -36,11 +39,16 @@ class proto_decorator(Generic[P]):
                 _type = type_hints.get(field.name, None)
                 if _type:
                     value = getattr(self, field.name)
-                    if issubclass(_type, Vector3):
-                        vector_proto = getattr(proto, field.name)
-                        vector_proto.x = value.x
-                        vector_proto.y = value.y
-                        vector_proto.z = value.z
+                    proto_value = getattr(proto, field.name)
+                    if issubclass(_type, float) and isinstance(proto_value, AdFloatType):
+                        proto_value.CopyFrom(_to_ad_proto(value))
+                    elif issubclass(_type, Vector3):
+                        if isinstance(proto_value, AdVector3):
+                            proto_value.CopyFrom(_to_vector3_ad_proto((value.x, value.y, value.z)))
+                        else:
+                            proto_value.x = value.x
+                            proto_value.y = value.y
+                            proto_value.z = value.z
                     else:
                         setattr(proto, field.name, value)
             return proto

luminarycloud-0.22.0/luminarycloud/_proto/api/v0/luminarycloud/feature_flag/feature_flag_pb2.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: proto/api/v0/luminarycloud/feature_flag/feature_flag.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.api import annotations_pb2 as google_dot_api_dot_annotations__pb2
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:proto/api/v0/luminarycloud/feature_flag/feature_flag.proto\x12\x30luminary.proto.api.v0.luminarycloud.feature_flag\x1a\x1cgoogle/api/annotations.proto\x1a\x1bgoogle/protobuf/empty.proto\"\xd9\x01\n\x17GetFeatureFlagsResponse\x12\x81\x01\n\x15\x65nabled_feature_flags\x18\x01 \x03(\x0b\x32\x62.luminary.proto.api.v0.luminarycloud.feature_flag.GetFeatureFlagsResponse.EnabledFeatureFlagsEntry\x1a:\n\x18\x45nabledFeatureFlagsEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x32\xa6\x01\n\x12\x46\x65\x61tureFlagService\x12\x8f\x01\n\x0fGetFeatureFlags\x12\x16.google.protobuf.Empty\x1aI.luminary.proto.api.v0.luminarycloud.feature_flag.GetFeatureFlagsResponse\"\x19\x82\xd3\xe4\x93\x02\x13\x12\x11/v0/feature_flagsB@Z>luminarycloud.com/core/proto/api/v0/luminarycloud/feature_flagb\x06proto3')
+
+
+
+_GETFEATUREFLAGSRESPONSE = DESCRIPTOR.message_types_by_name['GetFeatureFlagsResponse']
+_GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY = _GETFEATUREFLAGSRESPONSE.nested_types_by_name['EnabledFeatureFlagsEntry']
+GetFeatureFlagsResponse = _reflection.GeneratedProtocolMessageType('GetFeatureFlagsResponse', (_message.Message,), {
+
+  'EnabledFeatureFlagsEntry' : _reflection.GeneratedProtocolMessageType('EnabledFeatureFlagsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY,
+    '__module__' : 'proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2'
+    # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.feature_flag.GetFeatureFlagsResponse.EnabledFeatureFlagsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _GETFEATUREFLAGSRESPONSE,
+  '__module__' : 'proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2'
+  # @@protoc_insertion_point(class_scope:luminary.proto.api.v0.luminarycloud.feature_flag.GetFeatureFlagsResponse)
+  })
+_sym_db.RegisterMessage(GetFeatureFlagsResponse)
+_sym_db.RegisterMessage(GetFeatureFlagsResponse.EnabledFeatureFlagsEntry)
+
+_FEATUREFLAGSERVICE = DESCRIPTOR.services_by_name['FeatureFlagService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z>luminarycloud.com/core/proto/api/v0/luminarycloud/feature_flag'
+  _GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY._options = None
+  _GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY._serialized_options = b'8\001'
+  _FEATUREFLAGSERVICE.methods_by_name['GetFeatureFlags']._options = None
+  _FEATUREFLAGSERVICE.methods_by_name['GetFeatureFlags']._serialized_options = b'\202\323\344\223\002\023\022\021/v0/feature_flags'
+  _GETFEATUREFLAGSRESPONSE._serialized_start=172
+  _GETFEATUREFLAGSRESPONSE._serialized_end=389
+  _GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY._serialized_start=331
+  _GETFEATUREFLAGSRESPONSE_ENABLEDFEATUREFLAGSENTRY._serialized_end=389
+  _FEATUREFLAGSERVICE._serialized_start=392
+  _FEATUREFLAGSERVICE._serialized_end=558
+# @@protoc_insertion_point(module_scope)

luminarycloud-0.22.0/luminarycloud/_proto/api/v0/luminarycloud/feature_flag/feature_flag_pb2.pyi

@@ -0,0 +1,52 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetFeatureFlagsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class EnabledFeatureFlagsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.int
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.int = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    ENABLED_FEATURE_FLAGS_FIELD_NUMBER: builtins.int
+    @property
+    def enabled_feature_flags(self) -> google.protobuf.internal.containers.ScalarMap[builtins.int, builtins.str]:
+        """Map of enabled feature flags where:
+        - Key: experiment ID (uint64)
+        - Value: feature flag name (string)
+        Returns an empty map if no feature flags are enabled for the user.
+        """
+    def __init__(
+        self,
+        *,
+        enabled_feature_flags: collections.abc.Mapping[builtins.int, builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["enabled_feature_flags", b"enabled_feature_flags"]) -> None: ...
+
+global___GetFeatureFlagsResponse = GetFeatureFlagsResponse

luminarycloud-0.22.0/luminarycloud/_proto/api/v0/luminarycloud/feature_flag/feature_flag_pb2_grpc.py

@@ -0,0 +1,72 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+from luminarycloud._proto.api.v0.luminarycloud.feature_flag import feature_flag_pb2 as proto_dot_api_dot_v0_dot_luminarycloud_dot_feature__flag_dot_feature__flag__pb2
+
+
+class FeatureFlagServiceStub(object):
+    """Manages feature flags for internal users.
+    """
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetFeatureFlags = channel.unary_unary(
+                '/luminary.proto.api.v0.luminarycloud.feature_flag.FeatureFlagService/GetFeatureFlags',
+                request_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+                response_deserializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_feature__flag_dot_feature__flag__pb2.GetFeatureFlagsResponse.FromString,
+                )
+
+
+class FeatureFlagServiceServicer(object):
+    """Manages feature flags for internal users.
+    """
+
+    def GetFeatureFlags(self, request, context):
+        """Gets all enabled feature flags for the authenticated user.
+        Feature flags are used to control access to experimental and internal features.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FeatureFlagServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetFeatureFlags': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeatureFlags,
+                    request_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
+                    response_serializer=proto_dot_api_dot_v0_dot_luminarycloud_dot_feature__flag_dot_feature__flag__pb2.GetFeatureFlagsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'luminary.proto.api.v0.luminarycloud.feature_flag.FeatureFlagService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FeatureFlagService(object):
+    """Manages feature flags for internal users.
+    """
+
+    @staticmethod
+    def GetFeatureFlags(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.feature_flag.FeatureFlagService/GetFeatureFlags',
+            google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
+            proto_dot_api_dot_v0_dot_luminarycloud_dot_feature__flag_dot_feature__flag__pb2.GetFeatureFlagsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

luminarycloud-0.22.0/luminarycloud/_proto/api/v0/luminarycloud/feature_flag/feature_flag_pb2_grpc.pyi

@@ -0,0 +1,35 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import abc
+import google.protobuf.empty_pb2
+import grpc
+import luminarycloud._proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2
+
+class FeatureFlagServiceStub:
+    """Manages feature flags for internal users."""
+
+    def __init__(self, channel: grpc.Channel) -> None: ...
+    GetFeatureFlags: grpc.UnaryUnaryMultiCallable[
+        google.protobuf.empty_pb2.Empty,
+        luminarycloud._proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2.GetFeatureFlagsResponse,
+    ]
+    """Gets all enabled feature flags for the authenticated user.
+    Feature flags are used to control access to experimental and internal features.
+    """
+
+class FeatureFlagServiceServicer(metaclass=abc.ABCMeta):
+    """Manages feature flags for internal users."""
+
+    @abc.abstractmethod
+    def GetFeatureFlags(
+        self,
+        request: google.protobuf.empty_pb2.Empty,
+        context: grpc.ServicerContext,
+    ) -> luminarycloud._proto.api.v0.luminarycloud.feature_flag.feature_flag_pb2.GetFeatureFlagsResponse:
+        """Gets all enabled feature flags for the authenticated user.
+        Feature flags are used to control access to experimental and internal features.
+        """
+
+def add_FeatureFlagServiceServicer_to_server(servicer: FeatureFlagServiceServicer, server: grpc.Server) -> None: ...