frogml-core 0.0.115__py3-none-any.whl → 0.0.117__py3-none-any.whl

_proto/qwak/deployment/alert_pb2.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.deployment/alert.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\x1bqwak/deployment/alert.proto\x12\x1aqwak.deployment.management\"\xe2\x01\n\tAlertRule\x12\x39\n\nalert_spec\x18\x01 \x01(\x0b\x32%.qwak.deployment.management.AlertSpec\x12S\n\x16notifications_channels\x18\x02 \x03(\x0b\x32\x33.qwak.deployment.management.NotificationChannelSpec\x12\x10\n\x08\x61lert_id\x18\x03 \x01(\t\x12\x12\n\nis_enabled\x18\x04 \x01(\x08\x12\x1f\n\x17selected_variation_name\x18\x05 \x01(\t\"\xaa\x02\n\tAlertSpec\x12\x45\n\x10reducer_function\x18\x01 \x01(\x0e\x32+.qwak.deployment.management.ReducerFunction\x12\x41\n\x0e\x65valuator_type\x18\x02 \x01(\x0e\x32).qwak.deployment.management.EvaluatorType\x12\x1f\n\x13\x65valuator_threshold\x18\x03 \x01(\x05\x42\x02\x18\x01\x12\x19\n\x11\x65valuation_period\x18\x04 \x01(\x05\x12\x39\n\nalert_type\x18\x05 \x01(\x0e\x32%.qwak.deployment.management.AlertType\x12\x1c\n\x14\x65valuation_threshold\x18\x06 \x01(\x02\"]\n\x17NotificationChannelSpec\x12\x1f\n\x17notification_channel_id\x18\x01 \x01(\t\x12!\n\x19notification_channel_name\x18\x02 \x01(\t\"\xb1\x01\n\x13NotificationChannel\x12\x1f\n\x17notification_channel_id\x18\x01 \x01(\t\x12!\n\x19notification_channel_name\x18\x02 \x01(\t\x12V\n\x15notification_settings\x18\x03 \x01(\x0b\x32\x37.qwak.deployment.management.NotificationChannelSettings\"\xb5\x01\n\x1bNotificationChannelSettings\x12\x43\n\x0eslack_settings\x18\x01 \x01(\x0b\x32).qwak.deployment.management.SlackSettingsH\x00\x12I\n\x11opsgenie_settings\x18\x02 \x01(\x0b\x32,.qwak.deployment.management.OpsgenieSettingsH\x00\x42\x06\n\x04type\"\x1c\n\rSlackSettings\x12\x0b\n\x03url\x18\x01 \x01(\t\"4\n\x10OpsgenieSettings\x12\x0f\n\x07\x61pi_url\x18\x01 \x01(\t\x12\x0f\n\x07\x61pi_key\x18\x02 \x01(\t*\x82\x01\n\tAlertType\x12\x18\n\x14\x41LERT_TYPE_NOT_VALID\x10\x00\x12\x0e\n\nERROR_RATE\x10\x01\x12\x0e\n\nTHROUGHPUT\x10\x02\x12\x0b\n\x07LATENCY\x10\x03\x12\x0e\n\nLATENCY_90\x10\x04\x12\x0e\n\nLATENCY_95\x10\x05\x12\x0e\n\nLATENCY_99\x10\x06*\xad\x01\n\x0fReducerFunction\x12\x1e\n\x1aREDUCER_FUNCTION_NOT_VALID\x10\x00\x12\x07\n\x03MIN\x10\x01\x12\x07\n\x03MAX\x10\x02\x12\x07\n\x03SUM\x10\x03\x12\t\n\x05\x43OUNT\x10\x04\x12\x08\n\x04LAST\x10\x05\x12\n\n\x06MEDIAN\x10\x06\x12\x08\n\x04\x44IFF\x10\x07\x12\x0c\n\x08\x44IFF_ABS\x10\x08\x12\x10\n\x0cPERCENT_DIFF\x10\t\x12\x14\n\x10PERCENT_DIFF_ABS\x10\n*I\n\rEvaluatorType\x12\x1c\n\x18\x45VALUATOR_TYPE_NOT_VALID\x10\x00\x12\x0c\n\x08IS_ABOVE\x10\x01\x12\x0c\n\x08IS_BELOW\x10\x02\x42)\n%com.qwak.ai.management.deployment.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.deployment.alert_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001'
+  _ALERTSPEC.fields_by_name['evaluator_threshold']._options = None
+  _ALERTSPEC.fields_by_name['evaluator_threshold']._serialized_options = b'\030\001'
+  _globals['_ALERTTYPE']._serialized_start=1133
+  _globals['_ALERTTYPE']._serialized_end=1263
+  _globals['_REDUCERFUNCTION']._serialized_start=1266
+  _globals['_REDUCERFUNCTION']._serialized_end=1439
+  _globals['_EVALUATORTYPE']._serialized_start=1441
+  _globals['_EVALUATORTYPE']._serialized_end=1514
+  _globals['_ALERTRULE']._serialized_start=60
+  _globals['_ALERTRULE']._serialized_end=286
+  _globals['_ALERTSPEC']._serialized_start=289
+  _globals['_ALERTSPEC']._serialized_end=587
+  _globals['_NOTIFICATIONCHANNELSPEC']._serialized_start=589
+  _globals['_NOTIFICATIONCHANNELSPEC']._serialized_end=682
+  _globals['_NOTIFICATIONCHANNEL']._serialized_start=685
+  _globals['_NOTIFICATIONCHANNEL']._serialized_end=862
+  _globals['_NOTIFICATIONCHANNELSETTINGS']._serialized_start=865
+  _globals['_NOTIFICATIONCHANNELSETTINGS']._serialized_end=1046
+  _globals['_SLACKSETTINGS']._serialized_start=1048
+  _globals['_SLACKSETTINGS']._serialized_end=1076
+  _globals['_OPSGENIESETTINGS']._serialized_start=1078
+  _globals['_OPSGENIESETTINGS']._serialized_end=1130
+# @@protoc_insertion_point(module_scope)

_proto/qwak/deployment/alert_service_pb2.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.deployment/alert_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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.deployment import alert_pb2 as qwak_dot_deployment_dot_alert__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#qwak/deployment/alert_service.proto\x12\x1aqwak.deployment.management\x1a\x1bqwak/deployment/alert.proto\"`\n\x1dGetAlertsConfigurationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x16\n\x0e\x65nvironment_id\x18\x03 \x01(\t\"W\n\x1eGetAlertsConfigurationResponse\x12\x35\n\x06\x61lerts\x18\x01 \x03(\x0b\x32%.qwak.deployment.management.AlertRule\"\xad\x01\n\x1f\x41pplyAlertsConfigurationRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x35\n\x06\x61lerts\x18\x03 \x03(\x0b\x32%.qwak.deployment.management.AlertRule\x12\x12\n\nmodel_uuid\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"\"\n ApplyAlertsConfigurationResponse\"<\n\"GetNotificationChannelsListRequest\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\"y\n#GetNotificationChannelsListResponse\x12R\n\x15notification_channels\x18\x01 \x03(\x0b\x32\x33.qwak.deployment.management.NotificationChannelSpec\"\x88\x01\n\x1f\x41pplyNotificationChannelRequest\x12M\n\x14notification_channel\x18\x01 \x01(\x0b\x32/.qwak.deployment.management.NotificationChannel\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\"\"\n ApplyNotificationChannelResponse\"[\n DeleteNotificationChannelRequest\x12\x1f\n\x17notification_channel_id\x18\x01 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\"#\n!DeleteNotificationChannelResponse\"B\n(GetNotificationChannelDetailsListRequest\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\"{\n)GetNotificationChannelDetailsListResponse\x12N\n\x15notification_channels\x18\x01 \x03(\x0b\x32/.qwak.deployment.management.NotificationChannel2\xc9\x07\n\x16\x41lertManagementService\x12\x8f\x01\n\x16GetAlertsConfiguration\x12\x39.qwak.deployment.management.GetAlertsConfigurationRequest\x1a:.qwak.deployment.management.GetAlertsConfigurationResponse\x12\x95\x01\n\x18\x41pplyAlertsConfiguration\x12;.qwak.deployment.management.ApplyAlertsConfigurationRequest\x1a<.qwak.deployment.management.ApplyAlertsConfigurationResponse\x12\x9e\x01\n\x1bGetNotificationChannelsList\x12>.qwak.deployment.management.GetNotificationChannelsListRequest\x1a?.qwak.deployment.management.GetNotificationChannelsListResponse\x12\x95\x01\n\x18\x41pplyNotificationChannel\x12;.qwak.deployment.management.ApplyNotificationChannelRequest\x1a<.qwak.deployment.management.ApplyNotificationChannelResponse\x12\x98\x01\n\x19\x44\x65leteNotificationChannel\x12<.qwak.deployment.management.DeleteNotificationChannelRequest\x1a=.qwak.deployment.management.DeleteNotificationChannelResponse\x12\xb0\x01\n!GetNotificationChannelDetailsList\x12\x44.qwak.deployment.management.GetNotificationChannelDetailsListRequest\x1a\x45.qwak.deployment.management.GetNotificationChannelDetailsListResponseB)\n%com.qwak.ai.management.deployment.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.deployment.alert_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001'
+  _GETALERTSCONFIGURATIONREQUEST.fields_by_name['branch_id']._options = None
+  _GETALERTSCONFIGURATIONREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _APPLYALERTSCONFIGURATIONREQUEST.fields_by_name['branch_id']._options = None
+  _APPLYALERTSCONFIGURATIONREQUEST.fields_by_name['branch_id']._serialized_options = b'\030\001'
+  _globals['_GETALERTSCONFIGURATIONREQUEST']._serialized_start=96
+  _globals['_GETALERTSCONFIGURATIONREQUEST']._serialized_end=192
+  _globals['_GETALERTSCONFIGURATIONRESPONSE']._serialized_start=194
+  _globals['_GETALERTSCONFIGURATIONRESPONSE']._serialized_end=281
+  _globals['_APPLYALERTSCONFIGURATIONREQUEST']._serialized_start=284
+  _globals['_APPLYALERTSCONFIGURATIONREQUEST']._serialized_end=457
+  _globals['_APPLYALERTSCONFIGURATIONRESPONSE']._serialized_start=459
+  _globals['_APPLYALERTSCONFIGURATIONRESPONSE']._serialized_end=493
+  _globals['_GETNOTIFICATIONCHANNELSLISTREQUEST']._serialized_start=495
+  _globals['_GETNOTIFICATIONCHANNELSLISTREQUEST']._serialized_end=555
+  _globals['_GETNOTIFICATIONCHANNELSLISTRESPONSE']._serialized_start=557
+  _globals['_GETNOTIFICATIONCHANNELSLISTRESPONSE']._serialized_end=678
+  _globals['_APPLYNOTIFICATIONCHANNELREQUEST']._serialized_start=681
+  _globals['_APPLYNOTIFICATIONCHANNELREQUEST']._serialized_end=817
+  _globals['_APPLYNOTIFICATIONCHANNELRESPONSE']._serialized_start=819
+  _globals['_APPLYNOTIFICATIONCHANNELRESPONSE']._serialized_end=853
+  _globals['_DELETENOTIFICATIONCHANNELREQUEST']._serialized_start=855
+  _globals['_DELETENOTIFICATIONCHANNELREQUEST']._serialized_end=946
+  _globals['_DELETENOTIFICATIONCHANNELRESPONSE']._serialized_start=948
+  _globals['_DELETENOTIFICATIONCHANNELRESPONSE']._serialized_end=983
+  _globals['_GETNOTIFICATIONCHANNELDETAILSLISTREQUEST']._serialized_start=985
+  _globals['_GETNOTIFICATIONCHANNELDETAILSLISTREQUEST']._serialized_end=1051
+  _globals['_GETNOTIFICATIONCHANNELDETAILSLISTRESPONSE']._serialized_start=1053
+  _globals['_GETNOTIFICATIONCHANNELDETAILSLISTRESPONSE']._serialized_end=1176
+  _globals['_ALERTMANAGEMENTSERVICE']._serialized_start=1179
+  _globals['_ALERTMANAGEMENTSERVICE']._serialized_end=2148
+# @@protoc_insertion_point(module_scope)

_proto/qwak/deployment/alert_service_pb2.pyi

@@ -0,0 +1,218 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import frogml._proto.qwak.deployment.alert_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetAlertsConfigurationRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BRANCH_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """The model id get alerts for"""
+    branch_id: builtins.str
+    """The branch id get alerts for"""
+    environment_id: builtins.str
+    """The environment id to save alerts."""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        branch_id: builtins.str = ...,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "environment_id", b"environment_id", "model_id", b"model_id"]) -> None: ...
+
+global___GetAlertsConfigurationRequest = GetAlertsConfigurationRequest
+
+class GetAlertsConfigurationResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ALERTS_FIELD_NUMBER: builtins.int
+    @property
+    def alerts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.deployment.alert_pb2.AlertRule]:
+        """The saved alerts for the model"""
+    def __init__(
+        self,
+        *,
+        alerts: collections.abc.Iterable[qwak.deployment.alert_pb2.AlertRule] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["alerts", b"alerts"]) -> None: ...
+
+global___GetAlertsConfigurationResponse = GetAlertsConfigurationResponse
+
+class ApplyAlertsConfigurationRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BRANCH_ID_FIELD_NUMBER: builtins.int
+    ALERTS_FIELD_NUMBER: builtins.int
+    MODEL_UUID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """Model named ID, which will be used as a routing key for the model"""
+    branch_id: builtins.str
+    """Branch ID to deploy the model under"""
+    @property
+    def alerts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.deployment.alert_pb2.AlertRule]:
+        """The alerts list for model"""
+    model_uuid: builtins.str
+    """Model UUID uses to show deployments of model."""
+    environment_id: builtins.str
+    """Environment to apply alerts"""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        branch_id: builtins.str = ...,
+        alerts: collections.abc.Iterable[qwak.deployment.alert_pb2.AlertRule] | None = ...,
+        model_uuid: builtins.str = ...,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["alerts", b"alerts", "branch_id", b"branch_id", "environment_id", b"environment_id", "model_id", b"model_id", "model_uuid", b"model_uuid"]) -> None: ...
+
+global___ApplyAlertsConfigurationRequest = ApplyAlertsConfigurationRequest
+
+class ApplyAlertsConfigurationResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ApplyAlertsConfigurationResponse = ApplyAlertsConfigurationResponse
+
+class GetNotificationChannelsListRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    environment_id: builtins.str
+    """Environment to get notification for"""
+    def __init__(
+        self,
+        *,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id"]) -> None: ...
+
+global___GetNotificationChannelsListRequest = GetNotificationChannelsListRequest
+
+class GetNotificationChannelsListResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NOTIFICATION_CHANNELS_FIELD_NUMBER: builtins.int
+    @property
+    def notification_channels(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.deployment.alert_pb2.NotificationChannelSpec]:
+        """The Notification Channels exist"""
+    def __init__(
+        self,
+        *,
+        notification_channels: collections.abc.Iterable[qwak.deployment.alert_pb2.NotificationChannelSpec] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["notification_channels", b"notification_channels"]) -> None: ...
+
+global___GetNotificationChannelsListResponse = GetNotificationChannelsListResponse
+
+class ApplyNotificationChannelRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NOTIFICATION_CHANNEL_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    @property
+    def notification_channel(self) -> frogml._proto.qwak.deployment.alert_pb2.NotificationChannel:
+        """The Notification channel to add"""
+    environment_id: builtins.str
+    """Environment to apply notification channels"""
+    def __init__(
+        self,
+        *,
+        notification_channel: frogml._proto.qwak.deployment.alert_pb2.NotificationChannel | None = ...,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["notification_channel", b"notification_channel"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id", "notification_channel", b"notification_channel"]) -> None: ...
+
+global___ApplyNotificationChannelRequest = ApplyNotificationChannelRequest
+
+class ApplyNotificationChannelResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ApplyNotificationChannelResponse = ApplyNotificationChannelResponse
+
+class DeleteNotificationChannelRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NOTIFICATION_CHANNEL_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    notification_channel_id: builtins.str
+    """Notification channel id to delete"""
+    environment_id: builtins.str
+    """Environment to delete it from"""
+    def __init__(
+        self,
+        *,
+        notification_channel_id: builtins.str = ...,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id", "notification_channel_id", b"notification_channel_id"]) -> None: ...
+
+global___DeleteNotificationChannelRequest = DeleteNotificationChannelRequest
+
+class DeleteNotificationChannelResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DeleteNotificationChannelResponse = DeleteNotificationChannelResponse
+
+class GetNotificationChannelDetailsListRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    environment_id: builtins.str
+    """Environment to get notification details list for"""
+    def __init__(
+        self,
+        *,
+        environment_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id"]) -> None: ...
+
+global___GetNotificationChannelDetailsListRequest = GetNotificationChannelDetailsListRequest
+
+class GetNotificationChannelDetailsListResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NOTIFICATION_CHANNELS_FIELD_NUMBER: builtins.int
+    @property
+    def notification_channels(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.deployment.alert_pb2.NotificationChannel]:
+        """The channels saved"""
+    def __init__(
+        self,
+        *,
+        notification_channels: collections.abc.Iterable[qwak.deployment.alert_pb2.NotificationChannel] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["notification_channels", b"notification_channels"]) -> None: ...
+
+global___GetNotificationChannelDetailsListResponse = GetNotificationChannelDetailsListResponse

_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)

_proto/qwak/deployment/deployment_messages_pb2.py

@@ -0,0 +1,28 @@
+# -*- 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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.deployment.deployment_messages_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.management.deployment.apiP\001'
+  _globals['_MODELDEPLOYMENTNOTIFICATION']._serialized_start=108
+  _globals['_MODELDEPLOYMENTNOTIFICATION']._serialized_end=417
+# @@protoc_insertion_point(module_scope)

_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