frogml-core 0.0.116__py3-none-any.whl → 0.0.118__py3-none-any.whl

frogml_proto/qwak/user_application/common/v0/resources_pb2.py

@@ -1,51 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: frogml_proto.qwak.user_application/common/v0/resources.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 symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/qwak/user_application/common/v0/resources.proto\x12\x1fqwak.user_application.common.v0\"u\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12\x41\n\x0cmemory_units\x18\x03 \x01(\x0e\x32+.qwak.user_application.common.v0.MemoryUnit\"^\n\x0cGpuResources\x12:\n\x08gpu_type\x18\x01 \x01(\x0e\x32(.qwak.user_application.common.v0.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\"s\n SparkClusterResourceTemplateSpec\x12O\n\x08template\x18\x01 \x01(\x0e\x32=.qwak.user_application.common.v0.SparkClusterResourceTemplate\"\xc3\x01\n\x1aSparkResourceConfiguration\x12\x15\n\rdriver_memory\x18\x01 \x01(\t\x12\x14\n\x0c\x64river_cores\x18\x02 \x01(\x05\x12\x19\n\x11initial_executors\x18\x03 \x01(\x05\x12\x15\n\rmin_executors\x18\x04 \x01(\x05\x12\x15\n\rmax_executors\x18\x05 \x01(\x05\x12\x17\n\x0f\x65xecutor_memory\x18\x06 \x01(\t\x12\x16\n\x0e\x65xecutor_cores\x18\x07 \x01(\x05\"5\n\x1ePodComputeResourceTemplateSpec\x12\x13\n\x0btemplate_id\x18\x01 \x01(\t\"\xcb\x02\n\x13PodComputeResources\x12P\n\x11optimization_type\x18\x01 \x01(\x0e\x32\x35.qwak.user_application.common.v0.NodeOptimizationType\x12\x44\n\rcpu_resources\x18\x02 \x01(\x0b\x32-.qwak.user_application.common.v0.CpuResources\x12\x44\n\rgpu_resources\x18\x03 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResources\x12V\n\rtemplate_spec\x18\x04 \x01(\x0b\x32?.qwak.user_application.common.v0.PodComputeResourceTemplateSpec\"t\n\x15SparkComputeResources\x12[\n\x16resource_configuration\x18\x01 \x01(\x0b\x32;.qwak.user_application.common.v0.SparkResourceConfiguration\"\xe5\x01\n\x1b\x43lientSparkComputeResources\x12Z\n\rtemplate_spec\x18\x01 \x01(\x0b\x32\x41.qwak.user_application.common.v0.SparkClusterResourceTemplateSpecH\x00\x12]\n\x16resource_configuration\x18\x02 \x01(\x0b\x32;.qwak.user_application.common.v0.SparkResourceConfigurationH\x00\x42\x0b\n\tresources\"\xe9\x02\n\x19\x43lientPodComputeResources\x12U\n\x16node_optimization_type\x18\x01 \x01(\x0e\x32\x35.qwak.user_application.common.v0.NodeOptimizationType\x12X\n\rtemplate_spec\x18\x02 \x01(\x0b\x32?.qwak.user_application.common.v0.PodComputeResourceTemplateSpecH\x00\x12\x46\n\rcpu_resources\x18\x03 \x01(\x0b\x32-.qwak.user_application.common.v0.CpuResourcesH\x00\x12\x46\n\rgpu_resources\x18\x04 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesH\x00\x42\x0b\n\tresources*7\n\nMemoryUnit\x12\x17\n\x13INVALID_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x95\x02\n\x07GpuType\x12\x14\n\x10INVALID_GPU_TYPE\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c*\x88\x02\n\x1cSparkClusterResourceTemplate\x12\"\n\x1eSPARK_CLUSTER_RESOURCE_INVALID\x10\x00\x12 \n\x1cSPARK_CLUSTER_RESOURCE_SMALL\x10\x01\x12!\n\x1dSPARK_CLUSTER_RESOURCE_MEDIUM\x10\x02\x12 \n\x1cSPARK_CLUSTER_RESOURCE_LARGE\x10\x03\x12\x1d\n\x19SPARK_CLUSTER_RESOURCE_XL\x10\x04\x12\x1e\n\x1aSPARK_CLUSTER_RESOURCE_2XL\x10\x05\x12\x1e\n\x1aSPARK_CLUSTER_RESOURCE_3XL\x10\x06*\x8a\x01\n\x14NodeOptimizationType\x12\x1d\n\x19NODE_OPTIMIZATION_INVALID\x10\x00\x12\x1a\n\x16NODE_OPTIMIZATION_NONE\x10\x01\x12\x19\n\x15NODE_OPTIMIZATION_CPU\x10\x02\x12\x1c\n\x18NODE_OPTIMIZATION_MEMORY\x10\x03\x42\xc5\x01\n&com.qwak.ai.user_application.common.v0P\x01Z\x98\x01github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application_value/common/v0;user_application_common_v0b\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.user_application.common.v0.resources_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.user_application.common.v0P\001Z\230\001github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application_value/common/v0;user_application_common_v0'
-  _globals['_MEMORYUNIT']._serialized_start=1717
-  _globals['_MEMORYUNIT']._serialized_end=1772
-  _globals['_GPUTYPE']._serialized_start=1775
-  _globals['_GPUTYPE']._serialized_end=2052
-  _globals['_SPARKCLUSTERRESOURCETEMPLATE']._serialized_start=2055
-  _globals['_SPARKCLUSTERRESOURCETEMPLATE']._serialized_end=2319
-  _globals['_NODEOPTIMIZATIONTYPE']._serialized_start=2322
-  _globals['_NODEOPTIMIZATIONTYPE']._serialized_end=2460
-  _globals['_CPURESOURCES']._serialized_start=84
-  _globals['_CPURESOURCES']._serialized_end=201
-  _globals['_GPURESOURCES']._serialized_start=203
-  _globals['_GPURESOURCES']._serialized_end=297
-  _globals['_SPARKCLUSTERRESOURCETEMPLATESPEC']._serialized_start=299
-  _globals['_SPARKCLUSTERRESOURCETEMPLATESPEC']._serialized_end=414
-  _globals['_SPARKRESOURCECONFIGURATION']._serialized_start=417
-  _globals['_SPARKRESOURCECONFIGURATION']._serialized_end=612
-  _globals['_PODCOMPUTERESOURCETEMPLATESPEC']._serialized_start=614
-  _globals['_PODCOMPUTERESOURCETEMPLATESPEC']._serialized_end=667
-  _globals['_PODCOMPUTERESOURCES']._serialized_start=670
-  _globals['_PODCOMPUTERESOURCES']._serialized_end=1001
-  _globals['_SPARKCOMPUTERESOURCES']._serialized_start=1003
-  _globals['_SPARKCOMPUTERESOURCES']._serialized_end=1119
-  _globals['_CLIENTSPARKCOMPUTERESOURCES']._serialized_start=1122
-  _globals['_CLIENTSPARKCOMPUTERESOURCES']._serialized_end=1351
-  _globals['_CLIENTPODCOMPUTERESOURCES']._serialized_start=1354
-  _globals['_CLIENTPODCOMPUTERESOURCES']._serialized_end=1715
-# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/user_application/v0/user_application_pb2.py

@@ -1,36 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: frogml_proto.qwak.user_application/v0/user_application.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 symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/qwak/user_application/v0/user_application.proto\x12\x18qwak.user_application.v0\x1a\x1cgoogle/protobuf/struct.proto\".\n\x04Spec\x12&\n\x05value\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"0\n\x06Status\x12&\n\x05value\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct*\xf5\x02\n\x04Type\x12!\n\x1dINVALID_USER_APPLICATION_TYPE\x10\x00\x12\x10\n\x0cREMOTE_BUILD\x10\x01\x12\x1f\n\x17\x42\x41TCH_FEATURE_PROCESSOR\x10\x02\x1a\x02\x08\x01\x12#\n\x1bSTREAMING_FEATURE_PROCESSOR\x10\x03\x1a\x02\x08\x01\x12\x18\n\x14\x42\x41TCH_INFERENCE_TASK\x10\x04\x12\x14\n\x10MODEL_DEPLOYMENT\x10\x05\x12\x11\n\rMODEL_MONITOR\x10\x06\x12\x1b\n\x17\x44\x45PLOYMENT_MICROSERVICE\x10\x07\x12\x0f\n\x0b\x41PPLICATION\x10\x08\x12\x14\n\x10\x42\x41TCH_FEATURESET\x10\t\x12\x18\n\x14\x43USTOMER_ALERT_STACK\x10\n\x12\x11\n\rMODEL_TRAFFIC\x10\x0b\x12\x18\n\x14STREAMING_FEATURESET\x10\x0c\x12$\n STREAMING_AGGREGATION_FEATURESET\x10\rB\xaa\x01\n\x1f\x63om.qwak.ai.user_application.v0P\x01Z\x84\x01github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application/v0;user_application_v0b\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.user_application.v0.user_application_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.user_application.v0P\001Z\204\001github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application/v0;user_application_v0'
-  _TYPE.values_by_name["BATCH_FEATURE_PROCESSOR"]._options = None
-  _TYPE.values_by_name["BATCH_FEATURE_PROCESSOR"]._serialized_options = b'\010\001'
-  _TYPE.values_by_name["STREAMING_FEATURE_PROCESSOR"]._options = None
-  _TYPE.values_by_name["STREAMING_FEATURE_PROCESSOR"]._serialized_options = b'\010\001'
-  _globals['_TYPE']._serialized_start=206
-  _globals['_TYPE']._serialized_end=579
-  _globals['_SPEC']._serialized_start=107
-  _globals['_SPEC']._serialized_end=153
-  _globals['_STATUS']._serialized_start=155
-  _globals['_STATUS']._serialized_end=203
-# @@protoc_insertion_point(module_scope)

frogml_services_mock/__init__.py

@@ -1 +0,0 @@
-from .services_mock import frogml_services_mock

frogml_services_mock/mocks/alert_manager_service_api.py

@@ -1,56 +0,0 @@
-import uuid
-
-from frogml_proto.qwak.deployment.alert_service_pb2 import (
-    ApplyNotificationChannelRequest,
-    ApplyNotificationChannelResponse,
-    DeleteNotificationChannelRequest,
-    DeleteNotificationChannelResponse,
-    GetNotificationChannelDetailsListRequest,
-    GetNotificationChannelDetailsListResponse,
-)
-from frogml_proto.qwak.deployment.alert_service_pb2_grpc import (
-    AlertManagementServiceServicer,
-)
-from frogml_services_mock.mocks.utils.exception_handlers import (
-    raise_internal_grpc_error,
-)
-
-
-class AlertManagerServiceApiMock(AlertManagementServiceServicer):
-    def __init__(self):
-        super(AlertManagerServiceApiMock, self).__init__()
-        self.notification_channels = dict()
-
-    def ApplyNotificationChannel(
-        self, request: ApplyNotificationChannelRequest, context
-    ) -> ApplyNotificationChannelResponse:
-        """Channels to add or update"""
-        try:
-            request.notification_channel.notification_channel_id = str(uuid.uuid4())
-            self.notification_channels[
-                request.notification_channel.notification_channel_id
-            ] = request.notification_channel
-            return ApplyNotificationChannelResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def DeleteNotificationChannel(
-        self, request: DeleteNotificationChannelRequest, context
-    ) -> DeleteNotificationChannelResponse:
-        """Delete notification channel"""
-        try:
-            del self.notification_channels[request.notification_channel_id]
-            return DeleteNotificationChannelResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def GetNotificationChannelDetailsList(
-        self, request: GetNotificationChannelDetailsListRequest, context
-    ) -> GetNotificationChannelDetailsListResponse:
-        """Get Full Details for channels"""
-        try:
-            return GetNotificationChannelDetailsListResponse(
-                notification_channels=list(self.notification_channels.values())
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)

frogml_services_mock/mocks/alert_registry_service_api.py

@@ -1,62 +0,0 @@
-import uuid
-
-from frogml_proto.qwak.monitoring.v0.alerting_channel_management_service_pb2 import (
-    CreateAlertingChannelRequest,
-    CreateAlertingChannelResponse,
-    DeleteAlertingChannelRequest,
-    DeleteAlertingChannelResponse,
-    GetAlertingChannelRequestByName,
-    GetAlertingChannelResponseByName,
-    ListAlertingChannelRequest,
-    ListAlertingChannelResponse,
-)
-from frogml_proto.qwak.monitoring.v0.alerting_channel_management_service_pb2_grpc import (
-    AlertingChannelManagementServiceServicer,
-)
-from frogml_proto.qwak.monitoring.v0.alerting_channel_pb2 import (
-    AlertingChannelDescription,
-    AlertingChannelMetadata,
-)
-from frogml_services_mock.mocks.utils.exception_handlers import (
-    raise_internal_grpc_error,
-)
-
-
-class AlertsRegistryServiceApiMock(AlertingChannelManagementServiceServicer):
-    def __init__(self):
-        super(AlertsRegistryServiceApiMock, self).__init__()
-        self.alerts_channels = dict()
-
-    def CreateAlertingChannel(self, request: CreateAlertingChannelRequest, context):
-        metadata = AlertingChannelMetadata(
-            id=str(uuid.uuid4()), name=request.options.name
-        )
-        self.alerts_channels[request.options.name] = AlertingChannelDescription(
-            spec=request.options.spec, metadata=metadata
-        )
-        return CreateAlertingChannelResponse()
-
-    def GetAlertingChannelByName(
-        self, request: GetAlertingChannelRequestByName, context
-    ):
-        try:
-            return GetAlertingChannelResponseByName(
-                description=self.alerts_channels[request.name]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def ListAlertingChannel(self, request: ListAlertingChannelRequest, context):
-        try:
-            return ListAlertingChannelResponse(
-                description=[desc for name, desc in self.alerts_channels.items()]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def DeleteAlertingChannel(self, request: DeleteAlertingChannelRequest, context):
-        try:
-            del self.alerts_channels[request.name]
-            return DeleteAlertingChannelResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)

frogml_services_mock/mocks/analytics_api.py

@@ -1,54 +0,0 @@
-from frogml_proto.qwak.analytics.analytics_pb2 import QueryStatus
-from frogml_proto.qwak.analytics.analytics_service_pb2 import (
-    GetQueryResultDownloadURLResponse,
-    GetQueryResultsPageResponse,
-    QueryResponse,
-)
-from frogml_proto.qwak.analytics.analytics_service_pb2_grpc import (
-    AnalyticsQueryServiceServicer,
-)
-
-SUCCESSFUL_QUERY = "select * from successful"
-FAILED_QUERY = "select * from failed"
-TIMEOUT_QUERY = "select * from timeout"
-CANCELLED_QUERY = "select * from cancelled"
-SUCCESSFUL_QUERY_ID: str = "successful"
-FAILED_QUERY_ID: str = "failed"
-TIMEOUT_QUERY_ID: str = "timeout"
-CANCELLED_QUERY_ID: str = "cancelled"
-FAILURE_REASON: str = "failure reason"
-DOWNLOAD_URL: str = "https://jfrog.com/download"
-
-
-class AnalyticsApiMock(AnalyticsQueryServiceServicer):
-    def __init__(self):
-        super().__init__()
-
-    def Query(self, request, context):
-        query = request.query
-        if query == SUCCESSFUL_QUERY:
-            return QueryResponse(query_id=SUCCESSFUL_QUERY_ID)
-        elif query == FAILED_QUERY:
-            return QueryResponse(query_id=FAILED_QUERY_ID)
-        elif query == TIMEOUT_QUERY:
-            return QueryResponse(query_id=TIMEOUT_QUERY_ID)
-        elif query == CANCELLED_QUERY:
-            return QueryResponse(query_id=CANCELLED_QUERY_ID)
-        else:
-            return QueryResponse(query_id="unknown")
-
-    def GetQueryResultsPage(self, request, context):
-        query_id = request.query_id
-        if query_id == SUCCESSFUL_QUERY_ID:
-            return GetQueryResultsPageResponse(status=QueryStatus.SUCCESS, data=None)
-        elif query_id == FAILED_QUERY_ID:
-            return GetQueryResultsPageResponse(
-                status=QueryStatus.FAILED, failure_reason=FAILURE_REASON
-            )
-        elif query_id == TIMEOUT_QUERY_ID:
-            return GetQueryResultsPageResponse(status=QueryStatus.PENDING, data=None)
-        elif query_id == CANCELLED_QUERY_ID:
-            return GetQueryResultsPageResponse(status=QueryStatus.CANCELED, data=None)
-
-    def GetQueryResultDownloadURL(self, request, context):
-        return GetQueryResultDownloadURLResponse(download_url=DOWNLOAD_URL)

frogml_services_mock/mocks/audience_service_api.py

@@ -1,79 +0,0 @@
-import uuid
-
-from frogml_proto.qwak.audience.v1.audience_api_pb2 import (
-    CreateAudienceRequest,
-    CreateAudienceResponse,
-    DeleteAudienceRequest,
-    DeleteAudienceResponse,
-    GetAudienceRequest,
-    GetAudienceResponse,
-    ListAudienceRequest,
-    ListAudienceResponse,
-    SyncAudiencesRequest,
-    SyncAudiencesResponse,
-    UpdateAudienceRequest,
-    UpdateAudienceResponse,
-)
-from frogml_proto.qwak.audience.v1.audience_api_pb2_grpc import AudienceAPIServicer
-from frogml_proto.qwak.audience.v1.audience_pb2 import AudienceEntry
-from frogml_services_mock.mocks.utils.exception_handlers import (
-    raise_internal_grpc_error,
-)
-
-
-class AudienceServiceApiMock(AudienceAPIServicer):
-    def __init__(self):
-        super(AudienceServiceApiMock, self).__init__()
-        self.audiences = dict()
-
-    def CreateAudience(
-        self, request: CreateAudienceRequest, context
-    ) -> CreateAudienceResponse:
-        try:
-            audience_id = str(uuid.uuid4())
-            self.audiences[audience_id] = request.audience
-            return CreateAudienceResponse(audience_id=audience_id)
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def UpdateAudience(
-        self, request: UpdateAudienceRequest, context
-    ) -> UpdateAudienceResponse:
-        try:
-            self.audiences[request.id] = request.audience
-            return UpdateAudienceResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def GetAudience(self, request: GetAudienceRequest, context) -> GetAudienceResponse:
-        try:
-            return GetAudienceResponse(audience=self.audiences[request.audience_id])
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def ListAudience(
-        self, request: ListAudienceRequest, context
-    ) -> ListAudienceResponse:
-        try:
-            return ListAudienceResponse(
-                audience_entries=[
-                    AudienceEntry(id=audience_id, audience=audience)
-                    for audience_id, audience in self.audiences.items()
-                ]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def DeleteAudience(
-        self, request: DeleteAudienceRequest, context
-    ) -> DeleteAudienceResponse:
-        try:
-            del self.audiences[request.audience_id]
-            return DeleteAudienceResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def SyncAudiences(
-        self, request: SyncAudiencesRequest, context
-    ) -> SyncAudiencesResponse:
-        return SyncAudiencesResponse()

frogml_services_mock/mocks/authentication_service.py

@@ -1,33 +0,0 @@
-from datetime import datetime, timedelta
-
-from jose import jwt
-
-from frogml_proto.qwak.administration.v0.authentication.authentication_service_pb2 import (
-    AuthenticateRequest,
-    AuthenticateResponse,
-)
-from frogml_proto.qwak.administration.v0.authentication.authentication_service_pb2_grpc import (
-    AuthenticationServiceServicer,
-)
-
-epoch = datetime.timestamp(datetime.utcnow() + timedelta(days=1))
-
-_payload = {
-    "https://auth-token.qwak.ai/qwak-partner-id": "9",
-    "https://auth-token.qwak.ai/qwak-user-id": "Test User",
-    "https://auth-token.qwak.ai/mlflow-server": "http://localhost",
-    "https://auth-token.qwak.ai/data-api": "http://localhost",
-    "https://auth-token.qwak.ai/grpc-api": "localhost",
-    "https://auth-token.qwak.ai/models-api": "localhost",
-    "https://auth-token.qwak.ai/qwak-bucket": "frogml_core.bucket",
-    "exp": epoch,
-}
-
-mock_jwt_token = jwt.encode(_payload, key="secret", algorithm="HS256")
-
-
-class AuthenticationServiceMock(AuthenticationServiceServicer):
-    def Authenticate(
-        self, request: AuthenticateRequest, context
-    ) -> AuthenticateResponse:
-        return AuthenticateResponse(access_token=mock_jwt_token)

frogml_services_mock/mocks/automation_management_service.py

@@ -1,228 +0,0 @@
-import itertools
-import uuid
-from collections import defaultdict
-from datetime import datetime
-
-from frogml_proto.qwak.automation.v1.automation_management_service_pb2 import (
-    CreateAutomationRequest,
-    CreateAutomationResponse,
-    DeleteAutomationRequest,
-    DeleteAutomationResponse,
-    GetAutomationByNameRequest,
-    GetAutomationByNameResponse,
-    GetAutomationRequest,
-    GetAutomationResponse,
-    ListAutomationExecutionsRequest,
-    ListAutomationExecutionsResponse,
-    ListAutomationsRequest,
-    ListAutomationsResponse,
-    RegisterAutomationExecutionRequest,
-    RegisterAutomationExecutionResponse,
-    RunAutomationRequest,
-    RunAutomationResponse,
-    ToggleAutomationActivityRequest,
-    ToggleAutomationActivityResponse,
-    UpdateAutomationExecutionRequest,
-    UpdateAutomationExecutionResponse,
-    UpdateAutomationRequest,
-    UpdateAutomationResponse,
-)
-from frogml_proto.qwak.automation.v1.automation_management_service_pb2_grpc import (
-    AutomationManagementServiceServicer,
-)
-from frogml_proto.qwak.automation.v1.automation_pb2 import Automation
-from frogml_core.automations.automation_executions import (
-    AutomationExecution,
-    ExecutionRunDetails,
-    ExecutionStatus,
-)
-from frogml_services_mock.mocks.utils.exception_handlers import (
-    raise_internal_grpc_error,
-)
-
-
-class AutomationManagementServiceMock(AutomationManagementServiceServicer):
-    FROGML_ENVIRONMENT = "test_environment"
-
-    def __init__(self):
-        self.automations = dict()
-        self.automation_executions = defaultdict(list)
-
-    def CreateAutomation(
-        self, request: CreateAutomationRequest, context
-    ) -> CreateAutomationResponse:
-        """Create an automation"""
-        try:
-            automation_id = str(uuid.uuid4())
-            automation = Automation(
-                automation_id=automation_id,
-                automation_spec=request.automation_spec,
-                qwak_environment_id=self.FROGML_ENVIRONMENT,
-            )
-            self.automations[automation_id] = automation
-            return CreateAutomationResponse(automation_id=automation_id)
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def UpdateAutomation(
-        self, request: UpdateAutomationRequest, context
-    ) -> UpdateAutomationResponse:
-        """Update an automation"""
-        try:
-            automation = Automation(
-                automation_id=request.automation_id,
-                automation_spec=request.automation_spec,
-                qwak_environment_id=self.FROGML_ENVIRONMENT,
-            )
-            self.automations[request.automation_id] = automation
-            return UpdateAutomationResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def DeleteAutomation(
-        self, request: DeleteAutomationRequest, context
-    ) -> DeleteAutomationResponse:
-        """Delete an automation"""
-        try:
-            del self.automations[request.automation_id]
-            return DeleteAutomationResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def GetAutomation(
-        self, request: GetAutomationRequest, context
-    ) -> GetAutomationResponse:
-        """Get an automation By ID"""
-        try:
-            return GetAutomationResponse(
-                automation=self.automations[request.automation_id]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def GetAutomationByName(
-        self, request: GetAutomationByNameRequest, context
-    ) -> GetAutomationByNameResponse:
-        """Get an automation by name"""
-        try:
-            return GetAutomationByNameResponse(
-                automation=list(
-                    filter(
-                        lambda a: a.automation_spec.automation_name
-                        == request.automation_name,
-                        self.automations.values(),
-                    )
-                )[0]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def ListAutomations(
-        self, request: ListAutomationsRequest, context
-    ) -> ListAutomationsResponse:
-        """List all automation"""
-        try:
-            return ListAutomationsResponse(automations=list(self.automations.values()))
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def ListAutomationExecutions(
-        self, request: ListAutomationExecutionsRequest, context
-    ) -> ListAutomationExecutionsResponse:
-        """List automation's executions"""
-        try:
-            return ListAutomationExecutionsResponse(
-                automation_executions=self.automation_executions[request.automation_id]
-            )
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def RegisterAutomationExecution(
-        self, request: RegisterAutomationExecutionRequest, context
-    ) -> RegisterAutomationExecutionResponse:
-        """Register an execution"""
-        try:
-            run_details = ExecutionRunDetails(
-                start_time=datetime.now(), status=ExecutionStatus.RUNNING
-            )
-            execution = AutomationExecution(
-                execution_id=request.execution_id, run_details=run_details
-            )
-            self.automation_executions[request.automation_id].append(execution)
-            return RegisterAutomationExecutionResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def UpdateAutomationExecution(
-        self, request: UpdateAutomationExecutionRequest, context
-    ) -> UpdateAutomationExecutionResponse:
-        """Update an execution"""
-        try:
-            existing_executions = list(
-                itertools.chain(*self.automation_executions.values())
-            )
-            relevant_execution = list(
-                filter(
-                    lambda ex: ex.execution_id == request.execution_id,
-                    existing_executions,
-                )
-            )[0]
-            self.__merge_run_details(
-                relevant_execution.run_details, request.run_details
-            )
-            return UpdateAutomationExecutionResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def ToggleAutomationActivity(
-        self, request: ToggleAutomationActivityRequest, context
-    ) -> ToggleAutomationActivityResponse:
-        """Toggle whether an automation is enabled or disabled"""
-        try:
-            self.automations[request.automation_id].automation_spec = request.is_enabled
-            return ToggleAutomationActivityResponse()
-        except Exception as e:
-            raise_internal_grpc_error(context, e)
-
-    def RunAutomation(
-        self, request: RunAutomationRequest, context
-    ) -> RunAutomationResponse:
-        """Run an automation"""
-        if self.automations[request.automation_id]:
-            return RunAutomationResponse()
-        else:
-            raise_internal_grpc_error(
-                context,
-                ValueError(f"Automation with ID {request.automation_id} not found"),
-            )
-
-    @staticmethod
-    def __merge_run_details(
-        run_details_1: ExecutionRunDetails, run_details_2: ExecutionRunDetails
-    ):
-        run_details_1.start_time = (
-            run_details_2.start_time
-            if run_details_2.start_time
-            else run_details_1.start_time
-        )
-        run_details_1.end_time = (
-            run_details_2.end_time if run_details_2.end_time else run_details_1.end_time
-        )
-        run_details_1.error_details = (
-            run_details_2.error_details
-            if run_details_2.error_details
-            else run_details_1.error_details
-        )
-        run_details_1.status = (
-            run_details_2.status if run_details_2.status else run_details_1.status
-        )
-        run_details_1.task = (
-            run_details_2.task if run_details_2.task else run_details_1.task
-        )
-        run_details_1.finish_cause = (
-            run_details_2.finish_cause
-            if run_details_2.finish_cause
-            else run_details_1.finish_cause
-        )
-
-        return run_details_1

frogml_services_mock/mocks/autoscaling_service_api.py

@@ -1,31 +0,0 @@
-import uuid
-
-from frogml_proto.qwak.auto_scaling.v1.auto_scaling_service_pb2 import (
-    AttachAutoScalingRequest,
-    AttachAutoScalingResponse,
-)
-from frogml_proto.qwak.auto_scaling.v1.auto_scaling_service_pb2_grpc import (
-    AutoScalingServiceServicer,
-)
-from frogml_services_mock.mocks.utils.exception_handlers import (
-    raise_internal_grpc_error,
-)
-
-
-class AutoscalingServiceApiMock(AutoScalingServiceServicer):
-    def __init__(self):
-        super(AutoscalingServiceApiMock, self).__init__()
-        self.autoscaling_policies = dict()
-
-    def AttachAutoScaling(
-        self, request: AttachAutoScalingRequest, context
-    ) -> AttachAutoScalingResponse:
-        try:
-            autoscale_id = str(uuid.uuid4())
-            self.autoscaling_policies[request.model_id] = (
-                autoscale_id,
-                request.auto_scaling_config,
-            )
-            return AttachAutoScalingResponse(auto_scaling_id=autoscale_id)
-        except Exception as e:
-            raise_internal_grpc_error(context, e)