Update to latest protobuf messages

Author: lukasbindreiter

tilebox-datasets/tilebox/datasets/tilebox/v1/query_pb2.py

@@ -1,13 +1,23 @@
 from tilebox.datasets.buf.validate import validate_pb2 as _validate_pb2
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from tilebox.datasets.tilebox.v1 import id_pb2 as _id_pb2
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from collections.abc import Mapping as _Mapping
 from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
+class SortDirection(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    SORT_DIRECTION_UNSPECIFIED: _ClassVar[SortDirection]
+    SORT_DIRECTION_ASCENDING: _ClassVar[SortDirection]
+    SORT_DIRECTION_DESCENDING: _ClassVar[SortDirection]
+SORT_DIRECTION_UNSPECIFIED: SortDirection
+SORT_DIRECTION_ASCENDING: SortDirection
+SORT_DIRECTION_DESCENDING: SortDirection
+
 class TimeInterval(_message.Message):
     __slots__ = ("start_time", "end_time", "start_exclusive", "end_inclusive")
     START_TIME_FIELD_NUMBER: _ClassVar[int]

tilebox-datasets/tilebox/datasets/tilebox/v1/query_pb2.pyi

@@ -10,7 +10,9 @@
 from tilebox.workflows.data import (
     Cluster,
 )
-from tilebox.workflows.workflows.v1.core_pb2 import Cluster as ClusterMessage
+from tilebox.workflows.workflows.v1.workflows_pb2 import (
+    Cluster as ClusterMessage,
+)
 from tilebox.workflows.workflows.v1.workflows_pb2 import (
     CreateClusterRequest,
     DeleteClusterRequest,

tilebox-workflows/tests/clusters/test_client.py

@@ -39,7 +39,7 @@
 from tilebox.datasets.sync.client import Client as DatasetsClient
 from tilebox.workflows.observability.tracing import NoopWorkflowTracer, WorkflowTracer
 from tilebox.workflows.workflows.v1 import automation_pb2 as automation_pb
-from tilebox.workflows.workflows.v1 import core_pb2, job_pb2, task_pb2
+from tilebox.workflows.workflows.v1 import core_pb2, job_pb2, task_pb2, workflows_pb2
 
 _VERSION_PATTERN = re.compile(r"^v(\d+)\.(\d+)$")  # matches a version string in the format "v3.2"
 
@@ -323,13 +323,13 @@ class Cluster:
     deletable: bool
 
     @classmethod  # lets use typing.Self once we require python >= 3.11
-    def from_message(cls, cluster: core_pb2.Cluster) -> "Cluster":
+    def from_message(cls, cluster: workflows_pb2.Cluster) -> "Cluster":
         """Convert a Cluster protobuf message to a Cluster object."""
         return cls(slug=cluster.slug, display_name=cluster.display_name, deletable=cluster.deletable)
 
-    def to_message(self) -> core_pb2.Cluster:
+    def to_message(self) -> workflows_pb2.Cluster:
         """Convert a Cluster object to a Cluster protobuf message."""
-        return core_pb2.Cluster(slug=self.slug, display_name=self.display_name, deletable=self.deletable)
+        return workflows_pb2.Cluster(slug=self.slug, display_name=self.display_name, deletable=self.deletable)
 
 
 @dataclass