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

qwak_services_mock/services_mock.py

@@ -3,9 +3,13 @@ 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,
 )
+from _qwak_proto.qwak.admiral.secret.v0.system_secret_service_pb2_grpc import (
+    add_SystemSecretServiceServicer_to_server,
+)
 from _qwak_proto.qwak.admiral.user_application_instance.v0.user_application_instance_service_pb2_grpc import (
     add_UserApplicationInstanceServiceServicer_to_server,
 )
@@ -25,12 +29,12 @@ from _qwak_proto.qwak.batch_job.v1.batch_job_service_pb2_grpc import (
     add_BatchJobManagementServiceServicer_to_server,
 )
 from _qwak_proto.qwak.build.v1.build_api_pb2_grpc import add_BuildAPIServicer_to_server
+from _qwak_proto.qwak.build_settings.build_settings_api_pb2_grpc import (
+    add_BuildSettingsApiServicer_to_server,
+)
 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,
 )
@@ -70,6 +74,9 @@ from _qwak_proto.qwak.file_versioning.file_versioning_service_pb2_grpc import (
 from _qwak_proto.qwak.instance_template.instance_template_service_pb2_grpc import (
     add_InstanceTemplateManagementServiceServicer_to_server,
 )
+from _qwak_proto.qwak.integration.integration_service_pb2_grpc import (
+    add_IntegrationManagementServiceServicer_to_server,
+)
 from _qwak_proto.qwak.kube_deployment_captain.kube_deployment_captain_service_pb2_grpc import (
     add_KubeDeploymentCaptainServicer_to_server,
 )
@@ -82,15 +89,24 @@ from _qwak_proto.qwak.models.models_pb2_grpc import (
 from _qwak_proto.qwak.monitoring.v0.alerting_channel_management_service_pb2_grpc import (
     add_AlertingChannelManagementServiceServicer_to_server,
 )
+from _qwak_proto.qwak.offline.serving.v1.offline_serving_async_service_pb2_grpc import (
+    add_FeatureStoreOfflineServingAsyncServiceServicer_to_server,
+)
 from _qwak_proto.qwak.projects.projects_pb2_grpc import (
     add_ProjectsManagementServiceServicer_to_server,
 )
+from _qwak_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
+    add_PromptManagerServiceServicer_to_server,
+)
 from _qwak_proto.qwak.secret_service.secret_service_pb2_grpc import (
     add_SecretServiceServicer_to_server,
 )
 from _qwak_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
     add_UserServiceServicer_to_server,
 )
+from _qwak_proto.qwak.service_discovery.service_discovery_location_service_pb2_grpc import (
+    add_LocationDiscoveryServiceServicer_to_server,
+)
 from _qwak_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
     add_VectorCollectionServiceServicer_to_server,
 )
@@ -115,10 +131,12 @@ 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,
 )
+from qwak_services_mock.mocks.build_orchestrator_build_settings_api import (
+    BuildOrchestratorBuildSettingsApiMock,
+)
 from qwak_services_mock.mocks.build_orchestrator_service_api import (
     BuildOrchestratorServiceApiMock,
 )
@@ -143,24 +161,35 @@ from qwak_services_mock.mocks.features_operator_v3_service import (
     FeaturesOperatorV3ServiceMock,
 )
 from qwak_services_mock.mocks.file_versioning_service import FileVersioningServiceMock
+from qwak_services_mock.mocks.fs_offline_serving_service import (
+    FsOfflineServingServiceMock,
+)
 from qwak_services_mock.mocks.instance_template_management_service import (
     InstanceTemplateManagementServiceMock,
 )
+from qwak_services_mock.mocks.integration_management_service import (
+    IntegrationManagementServiceMock,
+)
 from qwak_services_mock.mocks.internal_build_orchestrator_service import (
     InternalBuildOrchestratorServiceMock,
 )
 from qwak_services_mock.mocks.job_registry_service_api import JobRegistryServiceApiMock
 from qwak_services_mock.mocks.kube_captain_service_api import KubeCaptainServiceApiMock
+from qwak_services_mock.mocks.location_discovery_service_api import (
+    LocationDiscoveryServiceApiMock,
+)
 from qwak_services_mock.mocks.logging_service import LoggingServiceApiMock
 from qwak_services_mock.mocks.model_management_service import (
     ModelsManagementServiceMock,
 )
 from qwak_services_mock.mocks.project_manager_service import ProjectManagerServiceMock
+from qwak_services_mock.mocks.prompt_manager_service import PromptManagerServiceMock
 from qwak_services_mock.mocks.qwak_mocks import QwakMocks
 from qwak_services_mock.mocks.secret_service import SecretServiceMock
 from qwak_services_mock.mocks.self_service_user_service import (
     SelfServiceUserServiceMock,
 )
+from qwak_services_mock.mocks.system_secret_service import SystemSecretServiceMock
 from qwak_services_mock.mocks.user_application_instance_service_api import (
     UserApplicationInstanceServiceApiMock,
 )
@@ -187,6 +216,9 @@ def qwak_container():
                     "internal_address": f"localhost:{free_port}",
                     "enable_ssl": False,
                 },
+                "authentication": {
+                    "enable_ssl": False,
+                },
             },
         },
     )
@@ -198,7 +230,6 @@ def qwak_container():
         automation_management,
         autoscaling,
         batch_job_management,
-        build_management,
         build_orchestrator,
         data_versioning,
         deployment,
@@ -206,6 +237,7 @@ def qwak_container():
         file_versioning,
         instance_template,
         kube_deployment_captain,
+        location_discovery,
         logging_client,
         model_management,
         project,
@@ -214,17 +246,21 @@ def qwak_container():
         vector_store,
         workspace_manager,
     )
-    from qwak.clients.administration import eco_system, self_service
+    from qwak.clients.administration import authentication, eco_system, self_service
+    from qwak.clients.integration_management import integration_manager_client
+    from qwak.clients.prompt_manager import prompt_manager_client
+    from qwak.clients.system_secret import system_secret_client
+    from qwak.vector_store.utils import upsert_utils
 
     container.wire(
         packages=[
+            authentication,
             alert_management,
             audience,
             automation_management,
             autoscaling,
             analytics,
             batch_job_management,
-            build_management,
             build_orchestrator,
             data_versioning,
             deployment,
@@ -242,6 +278,11 @@ def qwak_container():
             alerts_registry,
             workspace_manager,
             vector_store,
+            upsert_utils,
+            integration_manager_client,
+            system_secret_client,
+            prompt_manager_client,
+            location_discovery,
         ]
     )
 
@@ -264,6 +305,16 @@ def attach_servicers(free_port, server):
     qwak_mocks = qwak_service_mock_creator(
         server,
         [
+            (
+                "integration_management_service",
+                IntegrationManagementServiceMock,
+                add_IntegrationManagementServiceServicer_to_server,
+            ),
+            (
+                "system_secret_service",
+                SystemSecretServiceMock,
+                add_SystemSecretServiceServicer_to_server,
+            ),
             (
                 "autoscaling_service_mock",
                 AutoscalingServiceApiMock,
@@ -279,16 +330,16 @@ def attach_servicers(free_port, server):
                 BuildOrchestratorServiceApiMock,
                 add_BuildsOrchestratorServiceServicer_to_server,
             ),
+            (
+                "build_orchestrator_build_settings_api",
+                BuildOrchestratorBuildSettingsApiMock,
+                add_BuildSettingsApiServicer_to_server,
+            ),
             (
                 "internal_build_orchestrator_service",
                 InternalBuildOrchestratorServiceMock,
                 add_InternalBuildsOrchestratorServiceServicer_to_server,
             ),
-            (
-                "build_management_service_mock",
-                BuildsManagementServiceMock,
-                add_BuildsManagementServiceServicer_to_server,
-            ),
             (
                 "alert_manager_service_mock",
                 AlertManagerServiceApiMock,
@@ -399,6 +450,11 @@ def attach_servicers(free_port, server):
                 FeaturesOperatorV3ServiceMock,
                 add_FeaturesOperatorAsyncServiceServicer_to_server,
             ),
+            (
+                "fs_offline_serving_service",
+                FsOfflineServingServiceMock,
+                add_FeatureStoreOfflineServingAsyncServiceServicer_to_server,
+            ),
             (
                 "instance_templates_service",
                 InstanceTemplateManagementServiceMock,
@@ -429,6 +485,16 @@ def attach_servicers(free_port, server):
                 ExecutionManagementServiceMock,
                 add_FeatureStoreExecutionServiceServicer_to_server,
             ),
+            (
+                "prompt_manager_service",
+                PromptManagerServiceMock,
+                add_PromptManagerServiceServicer_to_server,
+            ),
+            (
+                "location_discovery_service",
+                LocationDiscoveryServiceApiMock,
+                add_LocationDiscoveryServiceServicer_to_server,
+            ),
             ("port", free_port, None),
         ],
     )

_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

_qwak_proto/qwak/fitness_service/fitness_service_pb2_grpc.py

@@ -1,169 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from _qwak_proto.qwak.fitness_service import fitness_service_pb2 as qwak_dot_fitness__service_dot_fitness__service__pb2
-
-
-class FitnessServiceStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.BuildModel = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/BuildModel',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.FromString,
-                )
-        self.CancelBuildModel = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/CancelBuildModel',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.FromString,
-                )
-        self.GetRemoteBuildingStatus = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/GetRemoteBuildingStatus',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.FromString,
-                )
-        self.CleanRemoteBuildResources = channel.unary_unary(
-                '/qwak.fitness.service.FitnessService/CleanRemoteBuildResources',
-                request_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.SerializeToString,
-                response_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.FromString,
-                )
-
-
-class FitnessServiceServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def BuildModel(self, request, context):
-        """Build a serving model image
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def CancelBuildModel(self, request, context):
-        """Cancel an ongoing build
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetRemoteBuildingStatus(self, request, context):
-        """Get the status of remote build
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def CleanRemoteBuildResources(self, request, context):
-        """Clean the remote build resources
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_FitnessServiceServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'BuildModel': grpc.unary_unary_rpc_method_handler(
-                    servicer.BuildModel,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.SerializeToString,
-            ),
-            'CancelBuildModel': grpc.unary_unary_rpc_method_handler(
-                    servicer.CancelBuildModel,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.SerializeToString,
-            ),
-            'GetRemoteBuildingStatus': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetRemoteBuildingStatus,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.SerializeToString,
-            ),
-            'CleanRemoteBuildResources': grpc.unary_unary_rpc_method_handler(
-                    servicer.CleanRemoteBuildResources,
-                    request_deserializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.FromString,
-                    response_serializer=qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'qwak.fitness.service.FitnessService', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class FitnessService(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def BuildModel(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, '/qwak.fitness.service.FitnessService/BuildModel',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.BuildModelResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def CancelBuildModel(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, '/qwak.fitness.service.FitnessService/CancelBuildModel',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CancelBuildModelResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetRemoteBuildingStatus(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, '/qwak.fitness.service.FitnessService/GetRemoteBuildingStatus',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.GetRemoteBuildingStatusResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def CleanRemoteBuildResources(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, '/qwak.fitness.service.FitnessService/CleanRemoteBuildResources',
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesRequest.SerializeToString,
-            qwak_dot_fitness__service_dot_fitness__service__pb2.CleanRemoteBuildResourcesResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

qwak/clients/build_management/__init__.py

@@ -1 +0,0 @@
-from .client import BuildsManagementClient