frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/execution/v1/state/execution_state_service_pb2.pyi

@@ -0,0 +1,192 @@
+"""
+@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.execution.v1.state.execution_state_pb2
+import _frogml_proto.qwak.execution.v1.state.featureset_state_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetFeaturesetStateByIdRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    featureset_id: builtins.str
+    def __init__(
+        self,
+        *,
+        featureset_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_id", b"featureset_id"]) -> None: ...
+
+global___GetFeaturesetStateByIdRequest = GetFeaturesetStateByIdRequest
+
+class GetFeaturesetStateByIdResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_STATE_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_state(self) -> _frogml_proto.qwak.execution.v1.state.featureset_state_pb2.FeaturesetState: ...
+    def __init__(
+        self,
+        *,
+        featureset_state: _frogml_proto.qwak.execution.v1.state.featureset_state_pb2.FeaturesetState | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["featureset_state", b"featureset_state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_state", b"featureset_state"]) -> None: ...
+
+global___GetFeaturesetStateByIdResponse = GetFeaturesetStateByIdResponse
+
+class GetFeaturesetStateByNameRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name"]) -> None: ...
+
+global___GetFeaturesetStateByNameRequest = GetFeaturesetStateByNameRequest
+
+class GetFeaturesetStateByNameResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_STATE_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_state(self) -> _frogml_proto.qwak.execution.v1.state.featureset_state_pb2.FeaturesetState: ...
+    def __init__(
+        self,
+        *,
+        featureset_state: _frogml_proto.qwak.execution.v1.state.featureset_state_pb2.FeaturesetState | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["featureset_state", b"featureset_state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_state", b"featureset_state"]) -> None: ...
+
+global___GetFeaturesetStateByNameResponse = GetFeaturesetStateByNameResponse
+
+class GetMultiFeaturesetStateByIdsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_IDS_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        featureset_ids: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_ids", b"featureset_ids"]) -> None: ...
+
+global___GetMultiFeaturesetStateByIdsRequest = GetMultiFeaturesetStateByIdsRequest
+
+class GetMultiFeaturesetStateByIdsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_STATES_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_states(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.execution.v1.state.featureset_state_pb2.FeaturesetState]: ...
+    def __init__(
+        self,
+        *,
+        featureset_states: collections.abc.Iterable[qwak.execution.v1.state.featureset_state_pb2.FeaturesetState] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_states", b"featureset_states"]) -> None: ...
+
+global___GetMultiFeaturesetStateByIdsResponse = GetMultiFeaturesetStateByIdsResponse
+
+class GetMultiFeaturesetStateByNameRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_name(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
+        *,
+        featureset_name: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name"]) -> None: ...
+
+global___GetMultiFeaturesetStateByNameRequest = GetMultiFeaturesetStateByNameRequest
+
+class GetMultiFeaturesetStateByNameResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_STATE_FIELD_NUMBER: builtins.int
+    @property
+    def featureset_state(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.execution.v1.state.featureset_state_pb2.FeaturesetState]: ...
+    def __init__(
+        self,
+        *,
+        featureset_state: collections.abc.Iterable[qwak.execution.v1.state.featureset_state_pb2.FeaturesetState] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_state", b"featureset_state"]) -> None: ...
+
+global___GetMultiFeaturesetStateByNameResponse = GetMultiFeaturesetStateByNameResponse
+
+class UpdateMultiExecutionStateRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    EXECUTION_STATE_FIELD_NUMBER: builtins.int
+    @property
+    def execution_state(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.execution.v1.state.execution_state_pb2.ExecutionState]: ...
+    def __init__(
+        self,
+        *,
+        execution_state: collections.abc.Iterable[qwak.execution.v1.state.execution_state_pb2.ExecutionState] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["execution_state", b"execution_state"]) -> None: ...
+
+global___UpdateMultiExecutionStateRequest = UpdateMultiExecutionStateRequest
+
+class UpdateMultiExecutionStateResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___UpdateMultiExecutionStateResponse = UpdateMultiExecutionStateResponse
+
+class DeleteFeaturesetStateByIdRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    featureset_id: builtins.str
+    def __init__(
+        self,
+        *,
+        featureset_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_id", b"featureset_id"]) -> None: ...
+
+global___DeleteFeaturesetStateByIdRequest = DeleteFeaturesetStateByIdRequest
+
+class DeleteFeaturesetStateByIdResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUM_DELETED_RECORDS_FIELD_NUMBER: builtins.int
+    num_deleted_records: builtins.int
+    """number of deleted records"""
+    def __init__(
+        self,
+        *,
+        num_deleted_records: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["num_deleted_records", b"num_deleted_records"]) -> None: ...
+
+global___DeleteFeaturesetStateByIdResponse = DeleteFeaturesetStateByIdResponse

_frogml_proto/qwak/execution/v1/state/execution_state_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.execution.v1.state import execution_state_service_pb2 as qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2
+
+
+class FeatureStoreExecutionStateServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetFeaturesetStateById = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetFeaturesetStateById',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdResponse.FromString,
+                )
+        self.GetFeaturesetStateByName = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetFeaturesetStateByName',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameResponse.FromString,
+                )
+        self.GetMultiFeaturesetStateByIds = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetMultiFeaturesetStateByIds',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsResponse.FromString,
+                )
+        self.GetMultiFeaturesetStateByName = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetMultiFeaturesetStateByName',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameResponse.FromString,
+                )
+        self.UpdateMultiExecutionState = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/UpdateMultiExecutionState',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateResponse.FromString,
+                )
+        self.DeleteFeaturesetStateById = channel.unary_unary(
+                '/qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService/DeleteFeaturesetStateById',
+                request_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdRequest.SerializeToString,
+                response_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdResponse.FromString,
+                )
+
+
+class FeatureStoreExecutionStateServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetFeaturesetStateById(self, request, context):
+        """Get feature set state by feature set id
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetFeaturesetStateByName(self, request, context):
+        """Get feature set state by feature set name
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetMultiFeaturesetStateByIds(self, request, context):
+        """Get state of multiple feature sets by a list of IDs
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetMultiFeaturesetStateByName(self, request, context):
+        """Get state of multiple feature sets by a list of names
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateMultiExecutionState(self, request, context):
+        """Update state multiple executions - called by the edge state publisher
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteFeaturesetStateById(self, request, context):
+        """Delete feature set state
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FeatureStoreExecutionStateServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetFeaturesetStateById': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeaturesetStateById,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdResponse.SerializeToString,
+            ),
+            'GetFeaturesetStateByName': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeaturesetStateByName,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameResponse.SerializeToString,
+            ),
+            'GetMultiFeaturesetStateByIds': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetMultiFeaturesetStateByIds,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsResponse.SerializeToString,
+            ),
+            'GetMultiFeaturesetStateByName': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetMultiFeaturesetStateByName,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameResponse.SerializeToString,
+            ),
+            'UpdateMultiExecutionState': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateMultiExecutionState,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateResponse.SerializeToString,
+            ),
+            'DeleteFeaturesetStateById': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteFeaturesetStateById,
+                    request_deserializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdRequest.FromString,
+                    response_serializer=qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.feature.store.execution.v1.state.FeatureStoreExecutionStateService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FeatureStoreExecutionStateService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetFeaturesetStateById(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetFeaturesetStateById',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByIdResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeaturesetStateByName(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetFeaturesetStateByName',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetFeaturesetStateByNameResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetMultiFeaturesetStateByIds(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetMultiFeaturesetStateByIds',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByIdsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetMultiFeaturesetStateByName(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/GetMultiFeaturesetStateByName',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.GetMultiFeaturesetStateByNameResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def UpdateMultiExecutionState(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/UpdateMultiExecutionState',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.UpdateMultiExecutionStateResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeleteFeaturesetStateById(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.feature.store.execution.v1.state.FeatureStoreExecutionStateService/DeleteFeaturesetStateById',
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdRequest.SerializeToString,
+            qwak_dot_execution_dot_v1_dot_state_dot_execution__state__service__pb2.DeleteFeaturesetStateByIdResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_frogml_proto/qwak/execution/v1/state/featureset_state_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.execution/v1/state/featureset_state.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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.qwak/execution/v1/state/featureset_state.proto\x12%qwak.feature.store.execution.v1.state\"A\n\x0f\x46\x65\x61turesetState\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\tB`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
+
+
+
+_FEATURESETSTATE = DESCRIPTOR.message_types_by_name['FeaturesetState']
+FeaturesetState = _reflection.GeneratedProtocolMessageType('FeaturesetState', (_message.Message,), {
+  'DESCRIPTOR' : _FEATURESETSTATE,
+  '__module__' : 'qwak.execution.v1.state.featureset_state_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.FeaturesetState)
+  })
+_sym_db.RegisterMessage(FeaturesetState)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.features.execution.api.v1.stateP\001Z/qwak/execution_management/state;execution_state'
+  _FEATURESETSTATE._serialized_start=89
+  _FEATURESETSTATE._serialized_end=154
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/execution/v1/state/featureset_state_pb2.pyi

@@ -0,0 +1,37 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class FeaturesetState(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    featureset_id: builtins.str
+    featureset_name: builtins.str
+    """TODO - FILL THIS WITH THE ACTUAL STATE REPRESENTATION
+    we need to find an elegant way of representing a state that is not just an enum value -
+    for example, in streaming aggr, if compaction failed but row-level is running.
+    another example is batchV1 where batch ingestion worked but realtime is failing
+    """
+    def __init__(
+        self,
+        *,
+        featureset_id: builtins.str = ...,
+        featureset_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_id", b"featureset_id", "featureset_name", b"featureset_name"]) -> None: ...
+
+global___FeaturesetState = FeaturesetState

_frogml_proto/qwak/execution/v1/state/featureset_state_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/execution/v1/state/spark_execution_state_pb2.py

@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.execution/v1/state/spark_execution_state.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+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()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/execution/v1/state/spark_execution_state.proto\x12%qwak.feature.store.execution.v1.state\"\x8f\x03\n\x13SparkExecutionState\x12\x16\n\x0e\x61pplication_id\x18\x01 \x01(\t\x12S\n\x16spark_application_type\x18\x02 \x01(\x0e\x32\x33.qwak.feature.store.execution.v1.state.SparkAppType\x12O\n\x0fspark_app_state\x18\x03 \x01(\x0b\x32\x34.qwak.feature.store.execution.v1.state.SparkAppStateH\x00\x12l\n\x11\x61pplicationLabels\x18\x04 \x03(\x0b\x32Q.qwak.feature.store.execution.v1.state.SparkExecutionState.ApplicationLabelsEntry\x1a\x38\n\x16\x41pplicationLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x12\n\x10state_descriptor\"t\n\rSparkAppState\x12L\n\nstate_type\x18\x01 \x01(\x0e\x32\x38.qwak.feature.store.execution.v1.state.SparkAppStateType\x12\x15\n\rerror_message\x18\x02 \x01(\t*\xdf\x03\n\x0cSparkAppType\x12\x1a\n\x16SPARK_APP_TYPE_INVALID\x10\x00\x12\"\n\x1eSPARK_APP_TYPE_BATCH_INGESTION\x10\x01\x12!\n\x1dSPARK_APP_TYPE_BATCH_BACKFILL\x10\x02\x12!\n\x1dSPARK_APP_TYPE_BATCH_DELETION\x10\x03\x12#\n\x1fSPARK_APP_TYPE_STREAMING_ONLINE\x10\x04\x12$\n SPARK_APP_TYPE_STREAMING_OFFLINE\x10\x05\x12%\n!SPARK_APP_TYPE_STREAMING_BACKFILL\x10\x06\x12%\n!SPARK_APP_TYPE_STREAMING_DELETION\x10\x07\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_ROWLEVEL\x10\x08\x12,\n(SPARK_APP_TYPE_STREAMING_AGGR_COMPACTION\x10\t\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_BACKFILL\x10\n\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_DELETION\x10\x0b*\x88\x03\n\x11SparkAppStateType\x12\x1b\n\x17SPARK_APP_STATE_INVALID\x10\x00\x12\x1d\n\x19SPARK_APP_STATE_COMPLETED\x10\x01\x12\x1a\n\x16SPARK_APP_STATE_FAILED\x10\x02\x12%\n!SPARK_APP_STATE_SUBMISSION_FAILED\x10\x03\x12\x1b\n\x17SPARK_APP_STATE_FAILING\x10\x04\x12 \n\x1cSPARK_APP_STATE_INVALIDATING\x10\x05\x12\x19\n\x15SPARK_APP_STATE_EMPTY\x10\x06\x12!\n\x1dSPARK_APP_STATE_PENDING_RERUN\x10\x07\x12\x1b\n\x17SPARK_APP_STATE_RUNNING\x10\x08\x12\x1d\n\x19SPARK_APP_STATE_SUBMITTED\x10\t\x12\x1e\n\x1aSPARK_APP_STATE_SUCCEEDING\x10\n\x12\x1b\n\x17SPARK_APP_STATE_UNKNOWN\x10\x0b\x42`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
+
+_SPARKAPPTYPE = DESCRIPTOR.enum_types_by_name['SparkAppType']
+SparkAppType = enum_type_wrapper.EnumTypeWrapper(_SPARKAPPTYPE)
+_SPARKAPPSTATETYPE = DESCRIPTOR.enum_types_by_name['SparkAppStateType']
+SparkAppStateType = enum_type_wrapper.EnumTypeWrapper(_SPARKAPPSTATETYPE)
+SPARK_APP_TYPE_INVALID = 0
+SPARK_APP_TYPE_BATCH_INGESTION = 1
+SPARK_APP_TYPE_BATCH_BACKFILL = 2
+SPARK_APP_TYPE_BATCH_DELETION = 3
+SPARK_APP_TYPE_STREAMING_ONLINE = 4
+SPARK_APP_TYPE_STREAMING_OFFLINE = 5
+SPARK_APP_TYPE_STREAMING_BACKFILL = 6
+SPARK_APP_TYPE_STREAMING_DELETION = 7
+SPARK_APP_TYPE_STREAMING_AGGR_ROWLEVEL = 8
+SPARK_APP_TYPE_STREAMING_AGGR_COMPACTION = 9
+SPARK_APP_TYPE_STREAMING_AGGR_BACKFILL = 10
+SPARK_APP_TYPE_STREAMING_AGGR_DELETION = 11
+SPARK_APP_STATE_INVALID = 0
+SPARK_APP_STATE_COMPLETED = 1
+SPARK_APP_STATE_FAILED = 2
+SPARK_APP_STATE_SUBMISSION_FAILED = 3
+SPARK_APP_STATE_FAILING = 4
+SPARK_APP_STATE_INVALIDATING = 5
+SPARK_APP_STATE_EMPTY = 6
+SPARK_APP_STATE_PENDING_RERUN = 7
+SPARK_APP_STATE_RUNNING = 8
+SPARK_APP_STATE_SUBMITTED = 9
+SPARK_APP_STATE_SUCCEEDING = 10
+SPARK_APP_STATE_UNKNOWN = 11
+
+
+_SPARKEXECUTIONSTATE = DESCRIPTOR.message_types_by_name['SparkExecutionState']
+_SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY = _SPARKEXECUTIONSTATE.nested_types_by_name['ApplicationLabelsEntry']
+_SPARKAPPSTATE = DESCRIPTOR.message_types_by_name['SparkAppState']
+SparkExecutionState = _reflection.GeneratedProtocolMessageType('SparkExecutionState', (_message.Message,), {
+
+  'ApplicationLabelsEntry' : _reflection.GeneratedProtocolMessageType('ApplicationLabelsEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY,
+    '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
+    # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkExecutionState.ApplicationLabelsEntry)
+    })
+  ,
+  'DESCRIPTOR' : _SPARKEXECUTIONSTATE,
+  '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkExecutionState)
+  })
+_sym_db.RegisterMessage(SparkExecutionState)
+_sym_db.RegisterMessage(SparkExecutionState.ApplicationLabelsEntry)
+
+SparkAppState = _reflection.GeneratedProtocolMessageType('SparkAppState', (_message.Message,), {
+  'DESCRIPTOR' : _SPARKAPPSTATE,
+  '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkAppState)
+  })
+_sym_db.RegisterMessage(SparkAppState)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.features.execution.api.v1.stateP\001Z/qwak/execution_management/state;execution_state'
+  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._options = None
+  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_options = b'8\001'
+  _SPARKAPPTYPE._serialized_start=615
+  _SPARKAPPTYPE._serialized_end=1094
+  _SPARKAPPSTATETYPE._serialized_start=1097
+  _SPARKAPPSTATETYPE._serialized_end=1489
+  _SPARKEXECUTIONSTATE._serialized_start=95
+  _SPARKEXECUTIONSTATE._serialized_end=494
+  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_start=418
+  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_end=474
+  _SPARKAPPSTATE._serialized_start=496
+  _SPARKAPPSTATE._serialized_end=612
+# @@protoc_insertion_point(module_scope)