frogml-core 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

frogml_proto/qwak/deployment/alert_service_pb2_grpc.py

@@ -0,0 +1,237 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml_proto.qwak.deployment import alert_service_pb2 as qwak_dot_deployment_dot_alert__service__pb2
+
+
+class AlertManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetAlertsConfiguration = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/GetAlertsConfiguration',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationResponse.FromString,
+                )
+        self.ApplyAlertsConfiguration = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/ApplyAlertsConfiguration',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationResponse.FromString,
+                )
+        self.GetNotificationChannelsList = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/GetNotificationChannelsList',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListResponse.FromString,
+                )
+        self.ApplyNotificationChannel = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/ApplyNotificationChannel',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelResponse.FromString,
+                )
+        self.DeleteNotificationChannel = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/DeleteNotificationChannel',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelResponse.FromString,
+                )
+        self.GetNotificationChannelDetailsList = channel.unary_unary(
+                '/qwak.deployment.management.AlertManagementService/GetNotificationChannelDetailsList',
+                request_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListRequest.SerializeToString,
+                response_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListResponse.FromString,
+                )
+
+
+class AlertManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetAlertsConfiguration(self, request, context):
+        """Get alerts for deployed model
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ApplyAlertsConfiguration(self, request, context):
+        """Update configuration for alerts in deployment
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetNotificationChannelsList(self, request, context):
+        """Get The Notification Channels list
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ApplyNotificationChannel(self, request, context):
+        """Channels to add or update
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteNotificationChannel(self, request, context):
+        """Delete notification channel
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetNotificationChannelDetailsList(self, request, context):
+        """Get Full Details for channels
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_AlertManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetAlertsConfiguration': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetAlertsConfiguration,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationResponse.SerializeToString,
+            ),
+            'ApplyAlertsConfiguration': grpc.unary_unary_rpc_method_handler(
+                    servicer.ApplyAlertsConfiguration,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationResponse.SerializeToString,
+            ),
+            'GetNotificationChannelsList': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetNotificationChannelsList,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListResponse.SerializeToString,
+            ),
+            'ApplyNotificationChannel': grpc.unary_unary_rpc_method_handler(
+                    servicer.ApplyNotificationChannel,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelResponse.SerializeToString,
+            ),
+            'DeleteNotificationChannel': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteNotificationChannel,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelResponse.SerializeToString,
+            ),
+            'GetNotificationChannelDetailsList': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetNotificationChannelDetailsList,
+                    request_deserializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListRequest.FromString,
+                    response_serializer=qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.deployment.management.AlertManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class AlertManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetAlertsConfiguration(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.deployment.management.AlertManagementService/GetAlertsConfiguration',
+            qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.GetAlertsConfigurationResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ApplyAlertsConfiguration(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.deployment.management.AlertManagementService/ApplyAlertsConfiguration',
+            qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.ApplyAlertsConfigurationResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetNotificationChannelsList(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.deployment.management.AlertManagementService/GetNotificationChannelsList',
+            qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelsListResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ApplyNotificationChannel(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.deployment.management.AlertManagementService/ApplyNotificationChannel',
+            qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.ApplyNotificationChannelResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeleteNotificationChannel(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.deployment.management.AlertManagementService/DeleteNotificationChannel',
+            qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.DeleteNotificationChannelResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetNotificationChannelDetailsList(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.deployment.management.AlertManagementService/GetNotificationChannelDetailsList',
+            qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListRequest.SerializeToString,
+            qwak_dot_deployment_dot_alert__service__pb2.GetNotificationChannelDetailsListResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

frogml_proto/qwak/deployment/deployment_messages_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.deployment/deployment_messages.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 frogml_proto.qwak.deployment import deployment_pb2 as qwak_dot_deployment_dot_deployment__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/deployment/deployment_messages.proto\x12\x1aqwak.deployment.management\x1a qwak/deployment/deployment.proto\"\xb5\x02\n\x1bModelDeploymentNotification\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12\x11\n\tbranch_id\x18\x02 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x43\n\x0fhosting_service\x18\x04 \x01(\x0b\x32*.qwak.deployment.management.HostingService\x12\x41\n\x06status\x18\x05 \x01(\x0e\x32\x31.qwak.deployment.management.ModelDeploymentStatus\x12\x15\n\rdeployment_id\x18\x06 \x01(\t\x12\x1b\n\x13\x66\x61ilure_reason_code\x18\x07 \x01(\t\x12!\n\x19\x66\x61ilure_technical_details\x18\x08 \x01(\tB)\n%com.qwak.ai.management.deployment.apiP\x01\x62\x06proto3')
+
+
+
+_MODELDEPLOYMENTNOTIFICATION = DESCRIPTOR.message_types_by_name['ModelDeploymentNotification']
+ModelDeploymentNotification = _reflection.GeneratedProtocolMessageType('ModelDeploymentNotification', (_message.Message,), {
+  'DESCRIPTOR' : _MODELDEPLOYMENTNOTIFICATION,
+  '__module__' : 'qwak.deployment.deployment_messages_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.deployment.management.ModelDeploymentNotification)
+  })
+_sym_db.RegisterMessage(ModelDeploymentNotification)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001'
+  _MODELDEPLOYMENTNOTIFICATION._serialized_start=108
+  _MODELDEPLOYMENTNOTIFICATION._serialized_end=417
+# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/deployment/deployment_messages_pb2.pyi

@@ -0,0 +1,61 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml_proto.qwak.deployment.deployment_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class ModelDeploymentNotification(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_NAME_FIELD_NUMBER: builtins.int
+    BRANCH_ID_FIELD_NUMBER: builtins.int
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    HOSTING_SERVICE_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_ID_FIELD_NUMBER: builtins.int
+    FAILURE_REASON_CODE_FIELD_NUMBER: builtins.int
+    FAILURE_TECHNICAL_DETAILS_FIELD_NUMBER: builtins.int
+    model_name: builtins.str
+    """Model name, which will be used as a routing key for the model"""
+    branch_id: builtins.str
+    """Branch ID to deploy the model under"""
+    build_id: builtins.str
+    """Build ID (which must be successful) to deploy"""
+    @property
+    def hosting_service(self) -> frogml_proto.qwak.deployment.deployment_pb2.HostingService:
+        """Downstream hosting service that will serve the Qwak model"""
+    status: frogml_proto.qwak.deployment.deployment_pb2.ModelDeploymentStatus.ValueType
+    """The deployment status"""
+    deployment_id: builtins.str
+    """The deployment ID"""
+    failure_reason_code: builtins.str
+    """The failure reason - Machine-readable, UpperCamelCase text"""
+    failure_technical_details: builtins.str
+    """The message describing the reason"""
+    def __init__(
+        self,
+        *,
+        model_name: builtins.str = ...,
+        branch_id: builtins.str = ...,
+        build_id: builtins.str = ...,
+        hosting_service: frogml_proto.qwak.deployment.deployment_pb2.HostingService | None = ...,
+        status: frogml_proto.qwak.deployment.deployment_pb2.ModelDeploymentStatus.ValueType = ...,
+        deployment_id: builtins.str = ...,
+        failure_reason_code: builtins.str = ...,
+        failure_technical_details: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["hosting_service", b"hosting_service"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "build_id", b"build_id", "deployment_id", b"deployment_id", "failure_reason_code", b"failure_reason_code", "failure_technical_details", b"failure_technical_details", "hosting_service", b"hosting_service", "model_name", b"model_name", "status", b"status"]) -> None: ...
+
+global___ModelDeploymentNotification = ModelDeploymentNotification