qwak-core 0.4.278__py3-none-any.whl → 0.5.19__py3-none-any.whl

qwak_services_mock/mocks/execution_management_service.py

@@ -1,7 +1,8 @@
-from _qwak_proto.qwak.execution.v1.execution_service_pb2 import TriggerBackfillResponse
+from _qwak_proto.qwak.execution.v1.execution_service_pb2 import (TriggerBackfillResponse, TriggerStreamingAggregationBackfillResponse)
 from _qwak_proto.qwak.execution.v1.execution_service_pb2_grpc import (
     FeatureStoreExecutionServiceServicer,
 )
+
 from grpc import RpcError
 
 
@@ -23,3 +24,10 @@ class ExecutionManagementServiceMock(FeatureStoreExecutionServiceServicer):
         if self._raise_exception_on_request:
             raise RpcError
         return TriggerBackfillResponse(execution_id=self._execution_id)
+
+    def TriggerStreamingAggregationBackfill(self, request, context):
+        if self._raise_exception_on_request:
+            raise RpcError
+        return TriggerStreamingAggregationBackfillResponse(
+            execution_id=self._execution_id
+        )

qwak_services_mock/mocks/feature_store_data_sources_manager_api.py

@@ -1,28 +1,38 @@
 import time
 import uuid
+from typing import Dict, Optional
 
 import grpc
 from _qwak_proto.qwak.feature_store.sources.data_source_pb2 import (
     DataSource,
     DataSourceDefinition,
+    DataSourceMetadata,
+    DataSourceSpec,
+    FeatureSetBrief,
 )
 from _qwak_proto.qwak.feature_store.sources.data_source_service_pb2 import (
     CreateDataSourceResponse,
     CreateDataSourceUploadURLRequest,
     CreateDataSourceUploadURLResponse,
+    DeleteDataSourceResponse,
     GetDataSourceByIdRequest,
     GetDataSourceByIdResponse,
     GetDataSourceByNameResponse,
+    ListDataSourcesResponse,
 )
 from _qwak_proto.qwak.feature_store.sources.data_source_service_pb2_grpc import (
     DataSourceServiceServicer,
 )
+from _qwak_proto.qwak.feature_store.v1.internal.data_source.data_source_service_pb2 import (
+    GetDataSourceSourceCodeUploadResponse,
+)
 
 
 class DataSourceServiceMock(DataSourceServiceServicer):
     def __init__(self):
-        self._data_sources_spec = {}
-        self._data_source_id_to_name = {}
+        self._data_sources_spec: Dict[str, DataSourceSpec] = {}
+        self._data_source_id_to_name: Dict[str, str] = {}
+        self._ds_definition_by_name: Dict[str, DataSourceDefinition] = {}
 
     def CreateDataSource(self, request, context):
         data_source_type = request.data_source_spec.WhichOneof("type")
@@ -33,16 +43,18 @@ class DataSourceServiceMock(DataSourceServiceServicer):
             if data_source_type == "batch_source"
             else request.data_source_spec.stream_source.name
         )
+        data_source_definition = DataSourceDefinition(
+            data_source_id=data_source_id,
+            data_source_spec=request.data_source_spec,
+        )
 
         self._data_sources_spec[ds_name] = request.data_source_spec
         self._data_source_id_to_name[data_source_id] = ds_name
+        self._ds_definition_by_name[ds_name] = data_source_definition
 
         return CreateDataSourceResponse(
             data_source=DataSource(
-                data_source_definition=DataSourceDefinition(
-                    data_source_id=data_source_id,
-                    data_source_spec=request.data_source_spec,
-                ),
+                data_source_definition=data_source_definition,
                 metadata=None,
                 feature_sets=[],
             )
@@ -83,16 +95,46 @@ class DataSourceServiceMock(DataSourceServiceServicer):
             )
         )
 
-    def CreateDataSourceUploadURL(
-        self, request: CreateDataSourceUploadURLRequest, context
-    ):
-        return CreateDataSourceUploadURLResponse(
-            upload_url=f"https://data_source_artifacts.s3.amazonaws.com/{request.data_source_name}_{request.object_name}"
-        )
-
     def CreateDataSourceUploadURLV1(
         self, request: CreateDataSourceUploadURLRequest, context
     ):
         return CreateDataSourceUploadURLResponse(
             upload_url=f"https://data_source_artifacts.s3.amazonaws.com/{request.data_source_name}_{request.object_name}_{int(time.time() * 1000)}"
         )
+
+    def GetDataSourceSourceCodeUploadURL(self, request, context):
+        if request.data_source_name not in self._data_sources_spec:
+            context.set_code(grpc.StatusCode.NOT_FOUND)
+            return GetDataSourceByNameResponse()
+
+        return GetDataSourceSourceCodeUploadResponse(
+            upload_url=f"https://data_source_artifacts.s3.amazonaws.com/{request.data_source_name}_{request.object_name}_{int(time.time() * 1000)}",
+            extra_headers={},
+        )
+
+    def ListDataSources(self, request, context):
+        return ListDataSourcesResponse(
+            data_sources=[
+                DataSource(
+                    data_source_definition=ds_definition,
+                    metadata=DataSourceMetadata(),
+                    feature_sets=[FeatureSetBrief()],
+                )
+                for ds_definition in self._ds_definition_by_name.values()
+            ]
+        )
+
+    def DeleteDataSource(self, request, context):
+        ds_name: Optional[str] = self._data_source_id_to_name.get(
+            request.data_source_id, None
+        )
+        if not ds_name:
+            context.set_details(
+                f"DataSource ID {request.data_source_id} doesn't exist'"
+            )
+            context.set_code(grpc.StatusCode.NOT_FOUND)
+
+        self._data_sources_spec.pop(ds_name)
+        self._data_source_id_to_name.pop(request.data_source_id)
+        self._ds_definition_by_name.pop(ds_name)
+        return DeleteDataSourceResponse()

qwak_services_mock/mocks/feature_store_entities_manager_api.py

@@ -38,49 +38,51 @@ class EntityServiceMock(EntityServiceServicer):
         entity_id: str = request.entity_id
 
         if entity_id not in self._entity_id_name.keys():
+            context.set_details(f"Entity ID {request.entity_id} doesn't exist'")
             context.set_code(grpc.StatusCode.NOT_FOUND)
-        else:
-            entity_name: str = self._entity_id_name[entity_id]
-            del self._entity_id_name[entity_id]
-            del self._entities_spec[entity_name]
+            return
 
-            return DeleteEntityResponse()
+        entity_name: str = self._entity_id_name[entity_id]
+        del self._entity_id_name[entity_id]
+        del self._entities_spec[entity_name]
+
+        return DeleteEntityResponse()
 
     def GetEntityByName(self, request, context):
         if request.entity_name not in self._entities_spec:
             context.set_code(grpc.StatusCode.NOT_FOUND)
             return GetEntityByNameResponse()
-        else:
-            entity_key: str = list(self._entity_id_name.keys())[
-                list(self._entity_id_name.values()).index(request.entity_name)
-            ]
-            return GetEntityByNameResponse(
-                entity=Entity(
-                    entity_definition=EntityDefinition(
-                        entity_id=entity_key,
-                        entity_spec=self._entities_spec[request.entity_name],
-                    ),
-                    metadata=None,
-                    feature_sets=[],
-                )
+
+        entity_key: str = list(self._entity_id_name.keys())[
+            list(self._entity_id_name.values()).index(request.entity_name)
+        ]
+        return GetEntityByNameResponse(
+            entity=Entity(
+                entity_definition=EntityDefinition(
+                    entity_id=entity_key,
+                    entity_spec=self._entities_spec[request.entity_name],
+                ),
+                metadata=None,
+                feature_sets=[],
             )
+        )
 
     def GetEntityById(self, request, context):
         if request.entity_id not in self._entity_id_name:
             context.set_code(grpc.StatusCode.NOT_FOUND)
             return GetEntityByIdResponse()
-        else:
-            entity_name: str = self._entity_id_name[request.entity_id]
-            return GetEntityByIdResponse(
-                entity=Entity(
-                    entity_definition=EntityDefinition(
-                        entity_id=request.entity_id,
-                        entity_spec=self._entities_spec[entity_name],
-                    ),
-                    metadata=None,
-                    feature_sets=[],
-                )
+
+        entity_name: str = self._entity_id_name[request.entity_id]
+        return GetEntityByIdResponse(
+            entity=Entity(
+                entity_definition=EntityDefinition(
+                    entity_id=request.entity_id,
+                    entity_spec=self._entities_spec[entity_name],
+                ),
+                metadata=None,
+                feature_sets=[],
             )
+        )
 
     def ListEntities(self, request, context):
         return ListEntitiesResponse(

qwak_services_mock/mocks/features_operator_v3_service.py

@@ -1,6 +1,7 @@
 import uuid
 from typing import Optional
 
+import grpc
 from _qwak_proto.qwak.features_operator.v3.features_operator_async_service_pb2 import (
     ValidationResponse,
 )
@@ -18,12 +19,31 @@ class FeaturesOperatorV3ServiceMock(FeaturesOperatorAsyncServiceServicer):
         self.response = response
 
     def ValidateDataSource(self, request, context) -> str:
+        if request.num_samples <= 0 or 1_000 < request.num_samples:
+            context.set_details(
+                f"[NumberOfSamples must be between 0 and 1000, got {request.num_samples}]"
+            )
+            context.set_code(grpc.StatusCode.INVALID_ARGUMENT)
+            return
+
         request_id = str(uuid.uuid4())
         return ValidationResponse(request_id=request_id)
 
     def ValidateFeatureSet(self, request, context):
+        if request.num_samples <= 0 or 1_000 < request.num_samples:
+            context.set_details(
+                f"[NumberOfSamples must be between 0 and 1000, got {request.num_samples}]"
+            )
+            context.set_code(grpc.StatusCode.INVALID_ARGUMENT)
+            return
+
         request_id = str(uuid.uuid4())
         return ValidationResponse(request_id=request_id)
 
     def GetValidationResult(self, request, context):
+        if not request.request_id:
+            context.set_details("Request handle id not found")
+            context.set_code(grpc.StatusCode.FAILED_PRECONDITION)
+            return
+
         return self.response

qwak_services_mock/mocks/qwak_mocks.py

@@ -15,7 +15,6 @@ from qwak_services_mock.mocks.automation_management_service import (
 )
 from qwak_services_mock.mocks.autoscaling_service_api import AutoscalingServiceApiMock
 from qwak_services_mock.mocks.batch_job_manager_service import BatchJobManagerService
-from qwak_services_mock.mocks.build_management import BuildsManagementServiceMock
 from qwak_services_mock.mocks.build_orchestrator_build_api import (
     BuildOrchestratorBuildApiMock,
 )
@@ -90,7 +89,6 @@ class QwakMocks:
     build_orchestrator_service_api: BuildOrchestratorServiceApiMock
     build_orchestrator_build_settings_api: BuildOrchestratorBuildSettingsApiMock
     internal_build_orchestrator_service: InternalBuildOrchestratorServiceMock
-    build_management_service_mock: BuildsManagementServiceMock
     alert_manager_service_mock: AlertManagerServiceApiMock
     automation_management_service_mock: AutomationManagementServiceMock
     job_registry_service_mock: JobRegistryServiceApiMock

qwak_services_mock/services_mock.py

@@ -3,6 +3,7 @@ from typing import Any, Generator, List, Tuple
 
 import grpc
 import pytest
+
 from _qwak_proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
     add_AuthenticationServiceServicer_to_server,
 )
@@ -34,9 +35,6 @@ from _qwak_proto.qwak.build_settings.build_settings_api_pb2_grpc import (
 from _qwak_proto.qwak.builds.builds_orchestrator_service_pb2_grpc import (
     add_BuildsOrchestratorServiceServicer_to_server,
 )
-from _qwak_proto.qwak.builds.builds_pb2_grpc import (
-    add_BuildsManagementServiceServicer_to_server,
-)
 from _qwak_proto.qwak.builds.internal_builds_orchestrator_service_pb2_grpc import (
     add_InternalBuildsOrchestratorServiceServicer_to_server,
 )
@@ -133,7 +131,6 @@ from qwak_services_mock.mocks.automation_management_service import (
 )
 from qwak_services_mock.mocks.autoscaling_service_api import AutoscalingServiceApiMock
 from qwak_services_mock.mocks.batch_job_manager_service import BatchJobManagerService
-from qwak_services_mock.mocks.build_management import BuildsManagementServiceMock
 from qwak_services_mock.mocks.build_orchestrator_build_api import (
     BuildOrchestratorBuildApiMock,
 )
@@ -233,7 +230,6 @@ def qwak_container():
         automation_management,
         autoscaling,
         batch_job_management,
-        build_management,
         build_orchestrator,
         data_versioning,
         deployment,
@@ -265,7 +261,6 @@ def qwak_container():
             autoscaling,
             analytics,
             batch_job_management,
-            build_management,
             build_orchestrator,
             data_versioning,
             deployment,
@@ -345,11 +340,6 @@ def attach_servicers(free_port, server):
                 InternalBuildOrchestratorServiceMock,
                 add_InternalBuildsOrchestratorServiceServicer_to_server,
             ),
-            (
-                "build_management_service_mock",
-                BuildsManagementServiceMock,
-                add_BuildsManagementServiceServicer_to_server,
-            ),
             (
                 "alert_manager_service_mock",
                 AlertManagerServiceApiMock,

_qwak_proto/qwak/fitness_service/fitness_service_pb2.py

@@ -1,110 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: qwak/fitness_service/fitness_service.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 _qwak_proto.qwak.fitness_service import fitness_pb2 as qwak_dot_fitness__service_dot_fitness__pb2
-from _qwak_proto.qwak.fitness_service import status_pb2 as qwak_dot_fitness__service_dot_status__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*qwak/fitness_service/fitness_service.proto\x12\x14qwak.fitness.service\x1a\"qwak/fitness_service/fitness.proto\x1a!qwak/fitness_service/status.proto\"H\n\x11\x42uildModelRequest\x12\x33\n\nbuild_spec\x18\x01 \x01(\x0b\x32\x1f.qwak.fitness.service.BuildSpec\"\x14\n\x12\x42uildModelResponse\"+\n\x17\x43\x61ncelBuildModelRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"\x1a\n\x18\x43\x61ncelBuildModelResponse\"2\n\x1eGetRemoteBuildingStatusRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"]\n\x1fGetRemoteBuildingStatusResponse\x12:\n\x06status\x18\x01 \x01(\x0b\x32*.qwak.fitness.service.RemoteBuildingStatus\"4\n CleanRemoteBuildResourcesRequest\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\"#\n!CleanRemoteBuildResourcesResponse2\xfc\x03\n\x0e\x46itnessService\x12_\n\nBuildModel\x12\'.qwak.fitness.service.BuildModelRequest\x1a(.qwak.fitness.service.BuildModelResponse\x12q\n\x10\x43\x61ncelBuildModel\x12-.qwak.fitness.service.CancelBuildModelRequest\x1a..qwak.fitness.service.CancelBuildModelResponse\x12\x86\x01\n\x17GetRemoteBuildingStatus\x12\x34.qwak.fitness.service.GetRemoteBuildingStatusRequest\x1a\x35.qwak.fitness.service.GetRemoteBuildingStatusResponse\x12\x8c\x01\n\x19\x43leanRemoteBuildResources\x12\x36.qwak.fitness.service.CleanRemoteBuildResourcesRequest\x1a\x37.qwak.fitness.service.CleanRemoteBuildResourcesResponseB#\n\x1f\x63om.qwak.ai.fitness.service.apiP\x01\x62\x06proto3')
-
-
-
-_BUILDMODELREQUEST = DESCRIPTOR.message_types_by_name['BuildModelRequest']
-_BUILDMODELRESPONSE = DESCRIPTOR.message_types_by_name['BuildModelResponse']
-_CANCELBUILDMODELREQUEST = DESCRIPTOR.message_types_by_name['CancelBuildModelRequest']
-_CANCELBUILDMODELRESPONSE = DESCRIPTOR.message_types_by_name['CancelBuildModelResponse']
-_GETREMOTEBUILDINGSTATUSREQUEST = DESCRIPTOR.message_types_by_name['GetRemoteBuildingStatusRequest']
-_GETREMOTEBUILDINGSTATUSRESPONSE = DESCRIPTOR.message_types_by_name['GetRemoteBuildingStatusResponse']
-_CLEANREMOTEBUILDRESOURCESREQUEST = DESCRIPTOR.message_types_by_name['CleanRemoteBuildResourcesRequest']
-_CLEANREMOTEBUILDRESOURCESRESPONSE = DESCRIPTOR.message_types_by_name['CleanRemoteBuildResourcesResponse']
-BuildModelRequest = _reflection.GeneratedProtocolMessageType('BuildModelRequest', (_message.Message,), {
-  'DESCRIPTOR' : _BUILDMODELREQUEST,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildModelRequest)
-  })
-_sym_db.RegisterMessage(BuildModelRequest)
-
-BuildModelResponse = _reflection.GeneratedProtocolMessageType('BuildModelResponse', (_message.Message,), {
-  'DESCRIPTOR' : _BUILDMODELRESPONSE,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildModelResponse)
-  })
-_sym_db.RegisterMessage(BuildModelResponse)
-
-CancelBuildModelRequest = _reflection.GeneratedProtocolMessageType('CancelBuildModelRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CANCELBUILDMODELREQUEST,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.CancelBuildModelRequest)
-  })
-_sym_db.RegisterMessage(CancelBuildModelRequest)
-
-CancelBuildModelResponse = _reflection.GeneratedProtocolMessageType('CancelBuildModelResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CANCELBUILDMODELRESPONSE,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.CancelBuildModelResponse)
-  })
-_sym_db.RegisterMessage(CancelBuildModelResponse)
-
-GetRemoteBuildingStatusRequest = _reflection.GeneratedProtocolMessageType('GetRemoteBuildingStatusRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETREMOTEBUILDINGSTATUSREQUEST,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.GetRemoteBuildingStatusRequest)
-  })
-_sym_db.RegisterMessage(GetRemoteBuildingStatusRequest)
-
-GetRemoteBuildingStatusResponse = _reflection.GeneratedProtocolMessageType('GetRemoteBuildingStatusResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETREMOTEBUILDINGSTATUSRESPONSE,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.GetRemoteBuildingStatusResponse)
-  })
-_sym_db.RegisterMessage(GetRemoteBuildingStatusResponse)
-
-CleanRemoteBuildResourcesRequest = _reflection.GeneratedProtocolMessageType('CleanRemoteBuildResourcesRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANREMOTEBUILDRESOURCESREQUEST,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.CleanRemoteBuildResourcesRequest)
-  })
-_sym_db.RegisterMessage(CleanRemoteBuildResourcesRequest)
-
-CleanRemoteBuildResourcesResponse = _reflection.GeneratedProtocolMessageType('CleanRemoteBuildResourcesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CLEANREMOTEBUILDRESOURCESRESPONSE,
-  '__module__' : 'qwak.fitness_service.fitness_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.fitness.service.CleanRemoteBuildResourcesResponse)
-  })
-_sym_db.RegisterMessage(CleanRemoteBuildResourcesResponse)
-
-_FITNESSSERVICE = DESCRIPTOR.services_by_name['FitnessService']
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
-  _BUILDMODELREQUEST._serialized_start=139
-  _BUILDMODELREQUEST._serialized_end=211
-  _BUILDMODELRESPONSE._serialized_start=213
-  _BUILDMODELRESPONSE._serialized_end=233
-  _CANCELBUILDMODELREQUEST._serialized_start=235
-  _CANCELBUILDMODELREQUEST._serialized_end=278
-  _CANCELBUILDMODELRESPONSE._serialized_start=280
-  _CANCELBUILDMODELRESPONSE._serialized_end=306
-  _GETREMOTEBUILDINGSTATUSREQUEST._serialized_start=308
-  _GETREMOTEBUILDINGSTATUSREQUEST._serialized_end=358
-  _GETREMOTEBUILDINGSTATUSRESPONSE._serialized_start=360
-  _GETREMOTEBUILDINGSTATUSRESPONSE._serialized_end=453
-  _CLEANREMOTEBUILDRESOURCESREQUEST._serialized_start=455
-  _CLEANREMOTEBUILDRESOURCESREQUEST._serialized_end=507
-  _CLEANREMOTEBUILDRESOURCESRESPONSE._serialized_start=509
-  _CLEANREMOTEBUILDRESOURCESRESPONSE._serialized_end=544
-  _FITNESSSERVICE._serialized_start=547
-  _FITNESSSERVICE._serialized_end=1055
-# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/fitness_service/fitness_service_pb2.pyi

@@ -1,121 +0,0 @@
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-import builtins
-import google.protobuf.descriptor
-import google.protobuf.message
-import qwak.fitness_service.fitness_pb2
-import qwak.fitness_service.status_pb2
-import sys
-
-if sys.version_info >= (3, 8):
-    import typing as typing_extensions
-else:
-    import typing_extensions
-
-DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
-
-class BuildModelRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_SPEC_FIELD_NUMBER: builtins.int
-    @property
-    def build_spec(self) -> qwak.fitness_service.fitness_pb2.BuildSpec:
-        """Requested build model spec"""
-    def __init__(
-        self,
-        *,
-        build_spec: qwak.fitness_service.fitness_pb2.BuildSpec | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec"]) -> None: ...
-
-global___BuildModelRequest = BuildModelRequest
-
-class BuildModelResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___BuildModelResponse = BuildModelResponse
-
-class CancelBuildModelRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id of the build to cancel"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
-
-global___CancelBuildModelRequest = CancelBuildModelRequest
-
-class CancelBuildModelResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___CancelBuildModelResponse = CancelBuildModelResponse
-
-class GetRemoteBuildingStatusRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id of the build to get status for"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
-
-global___GetRemoteBuildingStatusRequest = GetRemoteBuildingStatusRequest
-
-class GetRemoteBuildingStatusResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    STATUS_FIELD_NUMBER: builtins.int
-    @property
-    def status(self) -> qwak.fitness_service.status_pb2.RemoteBuildingStatus: ...
-    def __init__(
-        self,
-        *,
-        status: qwak.fitness_service.status_pb2.RemoteBuildingStatus | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["status", b"status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["status", b"status"]) -> None: ...
-
-global___GetRemoteBuildingStatusResponse = GetRemoteBuildingStatusResponse
-
-class CleanRemoteBuildResourcesRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
-
-global___CleanRemoteBuildResourcesRequest = CleanRemoteBuildResourcesRequest
-
-class CleanRemoteBuildResourcesResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___CleanRemoteBuildResourcesResponse = CleanRemoteBuildResourcesResponse