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

luminarycloud/_proto/quantity/quantity_pb2.pyi

@@ -1,7 +1,7 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
-Copyright 2020-2025 Luminary Cloud, Inc. All Rights Reserved.
+Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
 Generated by quantities.py. DO NOT EDIT
 """
 import builtins
@@ -34,15 +34,18 @@ class _QuantityTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._E
     VOLUME: _QuantityType.ValueType  # 30891
     TIME: _QuantityType.ValueType  # 20394
     ENERGY: _QuantityType.ValueType  # 30269
-    SPECIFIC_ENTHALPY: _QuantityType.ValueType  # 15033
+    ENTHALPY: _QuantityType.ValueType  # 15033
+    TOTAL_ENTHALPY: _QuantityType.ValueType  # 55281
+    ISENTROPIC_ENTHALPY: _QuantityType.ValueType  # 53299
+    ISENTROPIC_TOTAL_ENTHALPY: _QuantityType.ValueType  # 39573
     RADIANS: _QuantityType.ValueType  # 61564
     DEGREE: _QuantityType.ValueType  # 42822
     POWER: _QuantityType.ValueType  # 64408
     POWER_PER_UNIT_VOLUME: _QuantityType.ValueType  # 51
     THERMAL_EXPANSION_COEFFICIENT: _QuantityType.ValueType  # 16951
     TEMPERATURE: _QuantityType.ValueType  # 3
-    DENSITY: _QuantityType.ValueType  # 1
     PRESSURE: _QuantityType.ValueType  # 2
+    DENSITY: _QuantityType.ValueType  # 1
     VELOCITY: _QuantityType.ValueType  # 5
     VELOCITY_MAGNITUDE: _QuantityType.ValueType  # 30
     EDDY_VISCOSITY: _QuantityType.ValueType  # 4
@@ -199,15 +202,18 @@ AREA: QuantityType.ValueType  # 10
 VOLUME: QuantityType.ValueType  # 30891
 TIME: QuantityType.ValueType  # 20394
 ENERGY: QuantityType.ValueType  # 30269
-SPECIFIC_ENTHALPY: QuantityType.ValueType  # 15033
+ENTHALPY: QuantityType.ValueType  # 15033
+TOTAL_ENTHALPY: QuantityType.ValueType  # 55281
+ISENTROPIC_ENTHALPY: QuantityType.ValueType  # 53299
+ISENTROPIC_TOTAL_ENTHALPY: QuantityType.ValueType  # 39573
 RADIANS: QuantityType.ValueType  # 61564
 DEGREE: QuantityType.ValueType  # 42822
 POWER: QuantityType.ValueType  # 64408
 POWER_PER_UNIT_VOLUME: QuantityType.ValueType  # 51
 THERMAL_EXPANSION_COEFFICIENT: QuantityType.ValueType  # 16951
 TEMPERATURE: QuantityType.ValueType  # 3
-DENSITY: QuantityType.ValueType  # 1
 PRESSURE: QuantityType.ValueType  # 2
+DENSITY: QuantityType.ValueType  # 1
 VELOCITY: QuantityType.ValueType  # 5
 VELOCITY_MAGNITUDE: QuantityType.ValueType  # 30
 EDDY_VISCOSITY: QuantityType.ValueType  # 4

luminarycloud/enum/__init__.py

@@ -23,6 +23,7 @@ from .vis_enums import (
     CameraProjection,
     ColorMapPreset,
     EntityType,
+    FieldAssociation,
     FieldComponent,
     RenderStatusType,
     ExtractStatusType,

luminarycloud/enum/gpu_type.py

@@ -10,3 +10,5 @@ class GPUType(IntEnum):
     UNSPECIFIED = simulationpb.SimulationOptions.GPU_TYPE_UNSPECIFIED
     V100 = simulationpb.SimulationOptions.GPU_TYPE_V100
     A100 = simulationpb.SimulationOptions.GPU_TYPE_A100
+    T4 = simulationpb.SimulationOptions.GPU_TYPE_T4
+    H100 = simulationpb.SimulationOptions.GPU_TYPE_H100

luminarycloud/enum/quantity_type.py

@@ -60,6 +60,10 @@ class QuantityType(IntEnum):
     TOTAL_MOMENT_COEFFICIENT
     TOTAL_PRESSURE
     TOTAL_TEMPERATURE
+    ENTHALPY
+    TOTAL_ENTHALPY
+    ISENTROPIC_ENTHALPY
+    ISENTROPIC_TOTAL_ENTHALPY
     VELOCITY
     VELOCITY_MAGNITUDE
     VELOCITY_TIME_AVERAGE
@@ -116,6 +120,10 @@ class QuantityType(IntEnum):
     TEMPERATURE = quantitypb.TEMPERATURE
     TOTAL_PRESSURE = quantitypb.TOTAL_PRESSURE
     TOTAL_TEMPERATURE = quantitypb.TOTAL_TEMPERATURE
+    ENTHALPY = quantitypb.ENTHALPY
+    TOTAL_ENTHALPY = quantitypb.TOTAL_ENTHALPY
+    ISENTROPIC_ENTHALPY = quantitypb.ISENTROPIC_ENTHALPY
+    ISENTROPIC_TOTAL_ENTHALPY = quantitypb.ISENTROPIC_TOTAL_ENTHALPY
     VELOCITY = quantitypb.VELOCITY
     VELOCITY_MAGNITUDE = quantitypb.VELOCITY_MAGNITUDE
     Y_PLUS = quantitypb.Y_PLUS
@@ -173,6 +181,7 @@ class QuantityType(IntEnum):
     SENSITIVITY = quantitypb.SENSITIVITY
     NORMAL_SENSITIVITY = quantitypb.NORMAL_SENSITIVITY
     SMOOTHED_NORMAL_SENSITIVITY = quantitypb.SMOOTHED_NORMAL_SENSITIVITY
+    ENTROPY = quantitypb.ENTROPY
 
     # Quantities needed for table upload
     LENGTH = quantitypb.LENGTH

luminarycloud/enum/vis_enums.py

@@ -15,11 +15,8 @@ class VisQuantity(IntEnum):
     """
 
     NONE = quantitypb.INVALID_QUANTITY_TYPE
-    ABSOLUTE_PRESSURE = quantitypb.ABSOLUTE_PRESSURE
     PRESSURE = quantitypb.PRESSURE
-    DENSITY = quantitypb.DENSITY
     TEMPERATURE = quantitypb.TEMPERATURE
-    MACH = quantitypb.MACH
     Q_CRITERION = quantitypb.Q_CRITERION
     VELOCITY = quantitypb.VELOCITY
     WALL_SHEAR_STRESS = quantitypb.WALL_SHEAR_STRESS
@@ -39,6 +36,10 @@ class VisQuantity(IntEnum):
     HEAT_TRANSFER_COEFFICIENT = quantitypb.HEAT_TRANSFER_COEFFICIENT
     GRID_VELOCITY = quantitypb.GRID_VELOCITY
     # Derived fields
+    ABSOLUTE_PRESSURE = quantitypb.ABSOLUTE_PRESSURE
+    MACH = quantitypb.MACH
+    DENSITY = quantitypb.DENSITY
+    ENTROPY = quantitypb.ENTROPY
     RELATIVE_VELOCITY = quantitypb.RELATIVE_VELOCITY
     RELATIVE_MACH = quantitypb.RELATIVE_MACH
     PRESSURE_COEFFICIENT = quantitypb.PRESSURE_COEFFICIENT
@@ -71,6 +72,7 @@ class VisQuantity(IntEnum):
     MASS_FLUX_TIME_AVERAGE = quantitypb.MASS_FLUX_TIME_AVERAGE
     Q_CRITERION_TIME_AVERAGE = quantitypb.Q_CRITERION_TIME_AVERAGE
     HEAT_FLUX_TIME_AVERAGE = quantitypb.HEAT_FLUX_TIME_AVERAGE
+    DEBUG_QUANTITY = quantitypb.DEBUG_QUANTITY
 
 
 # Return the text name for the VisQuantity including the units, as it appears in the UI
@@ -306,3 +308,21 @@ class SceneMode(str, Enum):
     INLINE = "inline"
     SIDE_PANEL = "side_panel"
     FULLSCREEN = "fullscreen"
+
+
+class FieldAssociation(IntEnum):
+    """
+    An enum to specifiy the association of a field used in the range query.
+
+    .. warning:: This feature is experimental and may change or be removed in the future.
+
+    Attributes
+    ----------
+    POINTS
+        Field values are associated with the points of the mesh.
+    CELLS
+        Field values are associated with the cells of the mesh.
+    """
+
+    POINTS = 0
+    CELLS = 1

luminarycloud/exceptions.py

@@ -22,6 +22,12 @@ class SDKException(Exception):
         return [self.message]
 
 
+class Timeout(SDKException):
+    """Raised when some long-running operation doesn't finish before its deadline."""
+
+    pass
+
+
 class RpcError(SDKException):
     """Raised when an RPC error occurs."""
 
@@ -70,6 +76,6 @@ class FailedPreconditionError(RpcError):
 
 
 class DeadlineExceededError(RpcError):
-    """Raised when the deadline expired before the operation could complete.  I.e. it timed out."""
+    """Raised when the gRPC deadline expired before the operation could complete.  I.e. it timed out."""
 
     pass

luminarycloud/feature_modification.py

@@ -1,12 +1,12 @@
 # Copyright 2025 Luminary Cloud, Inc. All Rights Reserved.
 from enum import Enum, auto
-from typing import Dict, Iterable, List, Optional, cast
+from typing import Dict, Iterable, List, Optional
 from copy import deepcopy
 
-from luminarycloud.types.adfloat import _to_ad_proto, _from_ad_proto
+from luminarycloud.types.adfloat import _to_ad_proto
 from ._proto.geometry import geometry_pb2 as gpb
 from .types import Vector3Like
-from .types.vector3 import _to_vector3
+from .types.vector3 import _to_vector3_ad_proto
 from .params.geometry import Shape, Sphere, Cube, Cylinder, Torus, Cone, HalfSphere, Volume
 from google.protobuf.internal.containers import RepeatedScalarFieldContainer
 
@@ -265,6 +265,7 @@ def modify_delete(
 def modify_union(
     feature: gpb.Feature,
     volumes: Optional[List[Volume | int]] = None,
+    keep: Optional[bool] = None,
 ) -> gpb.Modification:
     """
     Modify a boolean union feature with optional new body IDs.
@@ -273,6 +274,7 @@ def modify_union(
     Args:
         feature: A gpb.Feature object
         volumes: List of volumes or volume IDs to union
+        keep: Whether to keep the original bodies
 
     Returns:
         A gpb.Modification object
@@ -287,6 +289,9 @@ def modify_union(
         vol_ids = _volumes_to_int_list(volumes)
         _update_repeated_field(boolean_op.reg_union.bodies, vol_ids)
 
+    if keep is not None:
+        boolean_op.reg_union.keep_source_bodies = keep
+
     return gpb.Modification(
         mod_type=gpb.Modification.ModificationType.MODIFICATION_TYPE_UPDATE_FEATURE,
         feature=feature_copy,
@@ -297,6 +302,8 @@ def modify_subtraction(
     feature: gpb.Feature,
     volumes: Optional[List[Volume | int]] = None,
     tool_volumes: Optional[List[Volume | int]] = None,
+    keep_source_bodies: Optional[bool] = None,
+    keep_tool_bodies: Optional[bool] = None,
     propagate_tool_tags: Optional[bool] = None,
 ) -> gpb.Modification:
     """
@@ -307,6 +314,8 @@ def modify_subtraction(
         feature: A gpb.Feature object
         volumes: List of volumes or volume IDs to subtract from
         tool_volumes: List of volumes or volume IDs to use for subtraction
+        keep_source_bodies: Whether to keep the original bodies
+        keep_tool_bodies: Whether to keep the tool bodies
         propagate_tool_tags: Whether to propagate tool tags
 
     Returns:
@@ -325,6 +334,12 @@ def modify_subtraction(
         tool_ids = _volumes_to_int_list(tool_volumes)
         _update_repeated_field(boolean_op.reg_subtraction.tools, tool_ids)
 
+    if keep_source_bodies is not None:
+        boolean_op.reg_subtraction.keep_source_bodies = keep_source_bodies
+
+    if keep_tool_bodies is not None:
+        boolean_op.reg_subtraction.keep_tool_bodies = keep_tool_bodies
+
     if propagate_tool_tags is not None:
         boolean_op.reg_subtraction.propagate_tool_tags = propagate_tool_tags
 
@@ -335,7 +350,9 @@ def modify_subtraction(
 
 
 def modify_intersection(
-    feature: gpb.Feature, volumes: Optional[List[Volume | int]] = None
+    feature: gpb.Feature,
+    volumes: Optional[List[Volume | int]] = None,
+    keep: Optional[bool] = None,
 ) -> gpb.Modification:
     """
     Modify a boolean intersection feature with optional new volumes.
@@ -344,6 +361,7 @@ def modify_intersection(
     Args:
         feature: A gpb.Feature object
         volumes: List of volumes or volume IDs to intersect
+        keep: Whether to keep the original bodies
 
     Returns:
         A gpb.Modification object
@@ -358,6 +376,9 @@ def modify_intersection(
         vol_ids = _volumes_to_int_list(volumes)
         _update_repeated_field(boolean_op.reg_intersection.bodies, vol_ids)
 
+    if keep is not None:
+        boolean_op.reg_intersection.keep_source_bodies = keep
+
     return gpb.Modification(
         mod_type=gpb.Modification.ModificationType.MODIFICATION_TYPE_UPDATE_FEATURE,
         feature=feature_copy,
@@ -368,6 +389,8 @@ def modify_chop(
     feature: gpb.Feature,
     volumes: Optional[List[Volume | int]] = None,
     tool_volumes: Optional[List[Volume | int]] = None,
+    keep_source_bodies: Optional[bool] = None,
+    keep_tool_bodies: Optional[bool] = None,
     propagate_tool_tags: Optional[bool] = None,
 ) -> gpb.Modification:
     """
@@ -378,6 +401,8 @@ def modify_chop(
         feature: A gpb.Feature object
         volumes: List of volumes or volume IDs to chop
         tool_volumes: List of volumes or volume IDs to use for chopping
+        keep_source_bodies: Whether to keep the original bodies
+        keep_tool_bodies: Whether to keep the tool bodies
         propagate_tool_tags: Whether to propagate tool tags
 
     Returns:
@@ -397,6 +422,12 @@ def modify_chop(
         tool_ids = _volumes_to_int_list(tool_volumes)
         _update_repeated_field(boolean_op.reg_chop.tools, tool_ids)
 
+    if keep_source_bodies is not None:
+        boolean_op.reg_chop.keep_source_bodies = keep_source_bodies
+
+    if keep_tool_bodies is not None:
+        boolean_op.reg_chop.keep_tool_bodies = keep_tool_bodies
+
     if propagate_tool_tags is not None:
         boolean_op.reg_chop.propagate_tool_tags = propagate_tool_tags
 
@@ -483,10 +514,7 @@ def modify_translate(
         _update_repeated_field(transform_op.body, vol_ids)
 
     if displacement is not None:
-        vec = _to_vector3(displacement)
-        transform_op.translation.vector.x = vec.x
-        transform_op.translation.vector.y = vec.y
-        transform_op.translation.vector.z = vec.z
+        transform_op.translation.vector.CopyFrom(_to_vector3_ad_proto(displacement))
 
     if keep is not None:
         transform_op.keep = keep
@@ -532,19 +560,13 @@ def modify_rotate(
 
     # Update existing rotation
     if angle is not None:
-        transform_op.rotation.angle = angle
+        transform_op.rotation.angle.CopyFrom(_to_ad_proto(angle))
 
     if axis is not None:
-        axis_vec = _to_vector3(axis)
-        transform_op.rotation.arbitrary.direction.x = axis_vec.x
-        transform_op.rotation.arbitrary.direction.y = axis_vec.y
-        transform_op.rotation.arbitrary.direction.z = axis_vec.z
+        transform_op.rotation.arbitrary.direction.CopyFrom(_to_vector3_ad_proto(axis))
 
     if origin is not None:
-        origin_vec = _to_vector3(origin)
-        transform_op.rotation.arbitrary.origin.x = origin_vec.x
-        transform_op.rotation.arbitrary.origin.y = origin_vec.y
-        transform_op.rotation.arbitrary.origin.z = origin_vec.z
+        transform_op.rotation.arbitrary.origin.CopyFrom(_to_vector3_ad_proto(origin))
 
     if keep is not None:
         transform_op.keep = keep
@@ -587,13 +609,10 @@ def modify_scale(
         _update_repeated_field(transform_op.body, vol_ids)
 
     if scale_factor is not None:
-        transform_op.scaling.isotropic = scale_factor
+        transform_op.scaling.isotropic.CopyFrom(_to_ad_proto(scale_factor))
 
     if origin is not None:
-        origin_vec = _to_vector3(origin)
-        transform_op.scaling.arbitrary.x = origin_vec.x
-        transform_op.scaling.arbitrary.y = origin_vec.y
-        transform_op.scaling.arbitrary.z = origin_vec.z
+        transform_op.scaling.arbitrary.CopyFrom(_to_vector3_ad_proto(origin))
 
     if keep is not None:
         transform_op.keep = keep
@@ -769,10 +788,7 @@ def modify_linear_pattern(
 
     if direction is not None:
         # Update existing linear pattern direction
-        dir_vec = _to_vector3(direction)
-        pattern_op.direction.linear_spacing.vector.x = dir_vec.x
-        pattern_op.direction.linear_spacing.vector.y = dir_vec.y
-        pattern_op.direction.linear_spacing.vector.z = dir_vec.z
+        pattern_op.direction.linear_spacing.vector.CopyFrom(_to_vector3_ad_proto(direction))
 
     if quantity is not None:
         pattern_op.direction.quantity = quantity
@@ -832,19 +848,13 @@ def modify_circular_pattern(
     circular = pattern_op.direction.circular_distribution
 
     if angle is not None:
-        circular.rotation.angle = angle
+        circular.rotation.angle.CopyFrom(_to_ad_proto(angle))
 
     if axis is not None:
-        axis_vec = _to_vector3(axis)
-        circular.rotation.arbitrary.direction.x = axis_vec.x
-        circular.rotation.arbitrary.direction.y = axis_vec.y
-        circular.rotation.arbitrary.direction.z = axis_vec.z
+        circular.rotation.arbitrary.direction.CopyFrom(_to_vector3_ad_proto(axis))
 
     if origin is not None:
-        origin_vec = _to_vector3(origin)
-        circular.rotation.arbitrary.origin.x = origin_vec.x
-        circular.rotation.arbitrary.origin.y = origin_vec.y
-        circular.rotation.arbitrary.origin.z = origin_vec.z
+        circular.rotation.arbitrary.origin.CopyFrom(_to_vector3_ad_proto(origin))
 
     if full_rotation is not None:
         circular.full = full_rotation

luminarycloud/geometry.py

@@ -29,7 +29,7 @@ from .named_variable_set import NamedVariableSet, get_named_variable_set
 
 if TYPE_CHECKING:
     from .project import Project
-    from .geometry_version import GeometryVersion
+    from .geometry_version import GeometryVersion, GeometryVersionIterator
 
 
 @ProtoWrapper(geometrypb.Geometry)
@@ -42,6 +42,8 @@ class Geometry(ProtoWrapperBase):
     "Geometry name."
     status: GeometryStatus
     "The status of the geometry."
+    project_id: ProjectID
+    "The ID of the project this geometry belongs to."
 
     _proto: geometrypb.Geometry
 
@@ -63,7 +65,7 @@ class Geometry(ProtoWrapperBase):
         """
         Get the project this geometry belongs to.
         """
-        return lc.get_project(ProjectID(self._proto.project_id))
+        return lc.get_project(self.project_id)
 
     def update(self, *, name: str | None = None) -> None:
         """
@@ -82,6 +84,18 @@ class Geometry(ProtoWrapperBase):
             res = get_default_client().UpdateGeometry(req)
             self._proto = res.geometry
 
+    def delete(self) -> None:
+        """
+        Delete the geometry.
+
+        This operation cannot be reverted and all the geometry data will be deleted as part of
+        this request.
+        """
+        req = geometrypb.DeleteGeometryRequest(
+            geometry_id=self.id,
+        )
+        get_default_client().DeleteGeometry(req)
+
     def copy(self, name: str = "") -> "Geometry":
         """
         Copy the geometry.
@@ -290,6 +304,46 @@ class Geometry(ProtoWrapperBase):
             )
         )
 
+    def versions(self, unfiltered: bool = False, page_size: int = 50) -> "GeometryVersionIterator":
+        """
+        List the geometry versions for this geometry in chronological order, oldest first.
+
+        By default, this only returns versions that are named OR have an associated Mesh OR are the
+        latest version of the geometry. If `unfiltered` is true, this returns all versions.
+
+        The geometry versions are fetched lazily in batches using pagination to optimize memory
+        usage and API calls.
+
+        Parameters
+        ----------
+        unfiltered : bool, optional
+            If True, returns all versions. If False, returns only versions that are named OR have an
+            associated Mesh OR are the latest version of the geometry. Defaults to False.
+        page_size : int, optional
+            Number of geometry versions to fetch per page. Defaults to 50, max is 500.
+
+        Returns
+        -------
+        GeometryVersionIterator
+            An iterator that yields GeometryVersion objects one at a time.
+
+        Examples
+        --------
+        Fetch the versions of this geometry with default filtering applied.
+        >>> for version in geometry.versions():
+        ...     print(version.id, version.name)
+
+        Fetch all versions of the geometry.
+        >>> for version in geometry.versions(unfiltered=True):
+        ...     print(version.id, version.name)
+
+        Build a list of versions of this geometry that have the name "So Important".
+        >>> important_versions = [ver for ver in geometry.versions() if ver.name == "So Important"]
+        """
+        from .geometry_version import GeometryVersionIterator
+
+        return GeometryVersionIterator(self.id, unfiltered, page_size)
+
     def latest_version(self) -> GeometryVersion:
         """
         Get the latest version of the geometry.
@@ -344,7 +398,7 @@ class Geometry(ProtoWrapperBase):
             jitter = random.uniform(0.5, 1.5)
             time.sleep(2 + jitter)
 
-    def create_tag(self, name: str, entities: Sequence[Volume | Surface]) -> None:
+    def create_tag(self, name: str, entities: Sequence[Volume | Surface]) -> Tag:
         """
         Create a tag in the geometry.
 
@@ -354,6 +408,11 @@ class Geometry(ProtoWrapperBase):
             The name of the tag to create.
         entities : list of Volumes or Surfaces
             The Volumes and Surfaces to tag.
+
+        Returns
+        -------
+        Tag
+            The tag that was created.
         """
         volume_ids = []
         surface_ids = []
@@ -365,8 +424,7 @@ class Geometry(ProtoWrapperBase):
             else:
                 raise TypeError("entities must be of type Volume or Surface")
 
-        req = geometrypb.ModifyGeometryRequest(
-            geometry_id=self.id,
+        self._modify(
             modification=gpb.Modification(
                 mod_type=gpb.Modification.MODIFICATION_TYPE_CREATE_TAG,
                 create_or_update_tag=gpb.CreateOrUpdateTag(
@@ -376,9 +434,9 @@ class Geometry(ProtoWrapperBase):
                 ),
             ),
         )
-        get_default_client().ModifyGeometry(req)
+        return self._get_tag_by_name(name)
 
-    def rename_tag(self, old_name: str, new_name: str) -> None:
+    def rename_tag(self, old_name: str, new_name: str) -> Tag:
         """
         Rename a tag in the geometry.
 
@@ -388,6 +446,11 @@ class Geometry(ProtoWrapperBase):
             The name of the tag to rename.
         new_name : str
             The new name for the tag.
+
+        Returns
+        -------
+        Tag
+            The updated tag.
         """
         self._modify(
             modification=gpb.Modification(
@@ -398,8 +461,9 @@ class Geometry(ProtoWrapperBase):
                 ),
             ),
         )
+        return self._get_tag_by_name(new_name)
 
-    def untag_entities(self, name: str, entities: Sequence[Volume | Surface]) -> None:
+    def untag_entities(self, name: str, entities: Sequence[Volume | Surface]) -> Tag | None:
         """
         Untag entities from a tag in the geometry.
 
@@ -410,6 +474,11 @@ class Geometry(ProtoWrapperBase):
         entities : list of Volumes or Surfaces
             The Volumes and Surfaces to untag. If empty, all entities with the
             tag will be untagged.
+
+        Returns
+        -------
+        Tag
+            The updated tag.
         """
         volume_ids = []
         surface_ids = []
@@ -431,8 +500,12 @@ class Geometry(ProtoWrapperBase):
                 ),
             ),
         )
+        try:
+            return self._get_tag_by_name(name)
+        except ValueError:
+            return None
 
-    def update_tag(self, name: str, entities: Sequence[Volume | Surface]) -> None:
+    def update_tag(self, name: str, entities: Sequence[Volume | Surface]) -> Tag:
         """
         Adds entities to a tag in the geometry.
 
@@ -441,6 +514,11 @@ class Geometry(ProtoWrapperBase):
         name : str
             The name of the tag to update.
         entities : list of Volumes or Surfaces
+
+        Returns
+        -------
+        Tag
+            The updated tag.
         """
         volume_ids = []
         surface_ids = []
@@ -462,6 +540,7 @@ class Geometry(ProtoWrapperBase):
                 ),
             ),
         )
+        return self._get_tag_by_name(name)
 
     def list_tags(self) -> list[Tag]:
         """
@@ -477,6 +556,25 @@ class Geometry(ProtoWrapperBase):
         res: geometrypb.ListTagsResponse = get_default_client().ListTags(req)
         return [Tag(t) for t in res.tags]
 
+    def _get_tag_by_name(self, name: str) -> Tag:
+        """
+        Get a specific tag from the geometry.
+
+        Parameters
+        ----------
+        name : str
+            The name of the tag.
+
+        Returns
+        -------
+        Tag
+            The tag with the specified name.
+        """
+        try:
+            return next(filter(lambda t: t.name == name, self.list_tags()))
+        except StopIteration:
+            raise ValueError(f"Tag '{name}' not found in geometry {self.id}")
+
     def list_entities(self) -> tuple[list[Surface], list[Volume]]:
         """
         List all the entities in the geometry.