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

_proto/qwak/execution/v1/jobs/reports/report_pb2.pyi

@@ -0,0 +1,266 @@
+"""
+@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.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import frogml._proto.qwak.execution.v1.backfill_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _JobStatus:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _JobStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_JobStatus.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    JOB_STATUS_INVALID: _JobStatus.ValueType  # 0
+    """reserved"""
+    JOB_STATUS_SUCCESS: _JobStatus.ValueType  # 1
+    """job successful"""
+    JOB_STATUS_FAILURE: _JobStatus.ValueType  # 2
+    """job failed"""
+    JOB_STATUS_TOO_EARLY: _JobStatus.ValueType  # 3
+    """job is too early"""
+
+class JobStatus(_JobStatus, metaclass=_JobStatusEnumTypeWrapper): ...
+
+JOB_STATUS_INVALID: JobStatus.ValueType  # 0
+"""reserved"""
+JOB_STATUS_SUCCESS: JobStatus.ValueType  # 1
+"""job successful"""
+JOB_STATUS_FAILURE: JobStatus.ValueType  # 2
+"""job failed"""
+JOB_STATUS_TOO_EARLY: JobStatus.ValueType  # 3
+"""job is too early"""
+global___JobStatus = JobStatus
+
+class FailureInformation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ERROR_MESSAGE_FIELD_NUMBER: builtins.int
+    error_message: builtins.str
+    """holds the full stacktrace of the exception"""
+    def __init__(
+        self,
+        *,
+        error_message: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["error_message", b"error_message"]) -> None: ...
+
+global___FailureInformation = FailureInformation
+
+class BatchIngestionJobMetadata(google.protobuf.message.Message):
+    """batchV1 job ingestion details"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BATCH_EXECUTION_DATE_FIELD_NUMBER: builtins.int
+    INGESTION_WINDOW_START_TIME_FIELD_NUMBER: builtins.int
+    INGESTION_WINDOW_END_TIME_FIELD_NUMBER: builtins.int
+    @property
+    def batch_execution_date(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """LOGICAL batch execution date (cron tick)"""
+    @property
+    def ingestion_window_start_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """LOGICAL ingestion window start time"""
+    @property
+    def ingestion_window_end_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """LOGICAL ingestion window end time"""
+    def __init__(
+        self,
+        *,
+        batch_execution_date: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        ingestion_window_start_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        ingestion_window_end_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["batch_execution_date", b"batch_execution_date", "ingestion_window_end_time", b"ingestion_window_end_time", "ingestion_window_start_time", b"ingestion_window_start_time"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_execution_date", b"batch_execution_date", "ingestion_window_end_time", b"ingestion_window_end_time", "ingestion_window_start_time", b"ingestion_window_start_time"]) -> None: ...
+
+global___BatchIngestionJobMetadata = BatchIngestionJobMetadata
+
+class BatchBackfillJobMetadata(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NUM_INTERVAL_ROWS_FIELD_NUMBER: builtins.int
+    BACKFILL_SPEC_FIELD_NUMBER: builtins.int
+    OFFLINE_PROCESSING_START_TIME_FIELD_NUMBER: builtins.int
+    OFFLINE_PROCESSING_END_TIME_FIELD_NUMBER: builtins.int
+    ONLINE_PROCESSING_START_TIME_FIELD_NUMBER: builtins.int
+    ONLINE_PROCESSING_END_TIME_FIELD_NUMBER: builtins.int
+    LAST_BATCH_LOGICAL_DATE_FIELD_NUMBER: builtins.int
+    num_interval_rows: builtins.int
+    """number of incoming rows in the interval"""
+    @property
+    def backfill_spec(self) -> frogml._proto.qwak.execution.v1.backfill_pb2.BackfillSpec:
+        """number of rows affected by the backfill"""
+    @property
+    def offline_processing_start_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM time where offline processing started"""
+    @property
+    def offline_processing_end_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM time where offline processing ended"""
+    @property
+    def online_processing_start_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM time when online processing started"""
+    @property
+    def online_processing_end_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM time when online processing ended"""
+    @property
+    def last_batch_logical_date(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Cron tick of last ingestion (window end time)"""
+    def __init__(
+        self,
+        *,
+        num_interval_rows: builtins.int = ...,
+        backfill_spec: frogml._proto.qwak.execution.v1.backfill_pb2.BackfillSpec | None = ...,
+        offline_processing_start_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        offline_processing_end_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        online_processing_start_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        online_processing_end_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        last_batch_logical_date: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["backfill_spec", b"backfill_spec", "last_batch_logical_date", b"last_batch_logical_date", "offline_processing_end_time", b"offline_processing_end_time", "offline_processing_start_time", b"offline_processing_start_time", "online_processing_end_time", b"online_processing_end_time", "online_processing_start_time", b"online_processing_start_time"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["backfill_spec", b"backfill_spec", "last_batch_logical_date", b"last_batch_logical_date", "num_interval_rows", b"num_interval_rows", "offline_processing_end_time", b"offline_processing_end_time", "offline_processing_start_time", b"offline_processing_start_time", "online_processing_end_time", b"online_processing_end_time", "online_processing_start_time", b"online_processing_start_time"]) -> None: ...
+
+global___BatchBackfillJobMetadata = BatchBackfillJobMetadata
+
+class StreamingJobMetadata(google.protobuf.message.Message):
+    """row-level streaming"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BATCH_ID_FIELD_NUMBER: builtins.int
+    batch_id: builtins.int
+    """Offline feature-store batch id"""
+    def __init__(
+        self,
+        *,
+        batch_id: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_id", b"batch_id"]) -> None: ...
+
+global___StreamingJobMetadata = StreamingJobMetadata
+
+class StreamingAggregationJobMetadata(google.protobuf.message.Message):
+    """streaming aggregation"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___StreamingAggregationJobMetadata = StreamingAggregationJobMetadata
+
+class JobReport(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class JobLabelsEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    BATCH_INGESTION_METADATA_FIELD_NUMBER: builtins.int
+    STREAMING_JOB_METADATA_FIELD_NUMBER: builtins.int
+    STREAMING_AGGR_JOB_METADATA_FIELD_NUMBER: builtins.int
+    BATCH_BACKFILL_JOB_METADATA_FIELD_NUMBER: builtins.int
+    NUM_AFFECTED_RECORDS_FIELD_NUMBER: builtins.int
+    RUN_START_TIME_FIELD_NUMBER: builtins.int
+    RUN_END_TIME_FIELD_NUMBER: builtins.int
+    LEGACY_RUN_ID_FIELD_NUMBER: builtins.int
+    EXECUTION_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    JOB_LABELS_FIELD_NUMBER: builtins.int
+    JOB_ID_FIELD_NUMBER: builtins.int
+    INFO_FIELD_NUMBER: builtins.int
+    @property
+    def batch_ingestion_metadata(self) -> global___BatchIngestionJobMetadata: ...
+    @property
+    def streaming_job_metadata(self) -> global___StreamingJobMetadata: ...
+    @property
+    def streaming_aggr_job_metadata(self) -> global___StreamingAggregationJobMetadata: ...
+    @property
+    def batch_backfill_job_metadata(self) -> global___BatchBackfillJobMetadata: ...
+    num_affected_records: builtins.int
+    """number of affected records"""
+    @property
+    def run_start_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM start time of the job"""
+    @property
+    def run_end_time(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """SYSTEM end time of the job"""
+    legacy_run_id: builtins.int
+    """legacy run id (what used to be called execution_id in the old report"""
+    execution_id: builtins.str
+    """Execution Manager Execution ID"""
+    environment_id: builtins.str
+    """environment_id"""
+    featureset_id: builtins.str
+    """featureset id"""
+    featureset_name: builtins.str
+    """featureset name"""
+    status: global___JobStatus.ValueType
+    """job status"""
+    @property
+    def job_labels(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """general tags/labels"""
+    job_id: builtins.str
+    """Job ID"""
+    @property
+    def info(self) -> global___FailureInformation: ...
+    def __init__(
+        self,
+        *,
+        batch_ingestion_metadata: global___BatchIngestionJobMetadata | None = ...,
+        streaming_job_metadata: global___StreamingJobMetadata | None = ...,
+        streaming_aggr_job_metadata: global___StreamingAggregationJobMetadata | None = ...,
+        batch_backfill_job_metadata: global___BatchBackfillJobMetadata | None = ...,
+        num_affected_records: builtins.int = ...,
+        run_start_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        run_end_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        legacy_run_id: builtins.int = ...,
+        execution_id: builtins.str = ...,
+        environment_id: builtins.str = ...,
+        featureset_id: builtins.str = ...,
+        featureset_name: builtins.str = ...,
+        status: global___JobStatus.ValueType = ...,
+        job_labels: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        job_id: builtins.str = ...,
+        info: global___FailureInformation | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["batch_backfill_job_metadata", b"batch_backfill_job_metadata", "batch_ingestion_metadata", b"batch_ingestion_metadata", "execution_id", b"execution_id", "execution_identifier", b"execution_identifier", "failure_info", b"failure_info", "info", b"info", "job_metadata_type", b"job_metadata_type", "legacy_run_id", b"legacy_run_id", "run_end_time", b"run_end_time", "run_start_time", b"run_start_time", "streaming_aggr_job_metadata", b"streaming_aggr_job_metadata", "streaming_job_metadata", b"streaming_job_metadata"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_backfill_job_metadata", b"batch_backfill_job_metadata", "batch_ingestion_metadata", b"batch_ingestion_metadata", "environment_id", b"environment_id", "execution_id", b"execution_id", "execution_identifier", b"execution_identifier", "failure_info", b"failure_info", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "info", b"info", "job_id", b"job_id", "job_labels", b"job_labels", "job_metadata_type", b"job_metadata_type", "legacy_run_id", b"legacy_run_id", "num_affected_records", b"num_affected_records", "run_end_time", b"run_end_time", "run_start_time", b"run_start_time", "status", b"status", "streaming_aggr_job_metadata", b"streaming_aggr_job_metadata", "streaming_job_metadata", b"streaming_job_metadata"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["execution_identifier", b"execution_identifier"]) -> typing_extensions.Literal["legacy_run_id", "execution_id"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["failure_info", b"failure_info"]) -> typing_extensions.Literal["info"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["job_metadata_type", b"job_metadata_type"]) -> typing_extensions.Literal["batch_ingestion_metadata", "streaming_job_metadata", "streaming_aggr_job_metadata", "batch_backfill_job_metadata"] | None: ...
+
+global___JobReport = JobReport

_proto/qwak/execution/v1/state/execution_state_pb2.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.execution/v1/state/execution_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 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.execution.v1.state import spark_execution_state_pb2 as qwak_dot_execution_dot_v1_dot_state_dot_spark__execution__state__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/execution/v1/state/execution_state.proto\x12%qwak.feature.store.execution.v1.state\x1a\x33qwak/execution/v1/state/spark_execution_state.proto\"\xba\x01\n\x0e\x45xecutionState\x12\x14\n\x0c\x65xecution_id\x18\x01 \x01(\t\x12\x15\n\rfeatureset_id\x18\x02 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x03 \x01(\t\x12U\n\x0fspark_app_state\x18\x04 \x01(\x0b\x32:.qwak.feature.store.execution.v1.state.SparkExecutionStateH\x00\x42\x0c\n\nexec_stateB`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.state.execution_state_pb2', _globals)
+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'
+  _globals['_EXECUTIONSTATE']._serialized_start=142
+  _globals['_EXECUTIONSTATE']._serialized_end=328
+# @@protoc_insertion_point(module_scope)

_proto/qwak/execution/v1/state/execution_state_pb2.pyi

@@ -0,0 +1,45 @@
+"""
+@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.execution.v1.state.spark_execution_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 ExecutionState(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    EXECUTION_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    SPARK_APP_STATE_FIELD_NUMBER: builtins.int
+    execution_id: builtins.str
+    """Execution Id"""
+    featureset_id: builtins.str
+    """featureset id this execution belongs to (featureset:execution is 1:many relation)"""
+    environment_id: builtins.str
+    """environmentId of the featureset"""
+    @property
+    def spark_app_state(self) -> frogml._proto.qwak.execution.v1.state.spark_execution_state_pb2.SparkExecutionState: ...
+    def __init__(
+        self,
+        *,
+        execution_id: builtins.str = ...,
+        featureset_id: builtins.str = ...,
+        environment_id: builtins.str = ...,
+        spark_app_state: frogml._proto.qwak.execution.v1.state.spark_execution_state_pb2.SparkExecutionState | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["exec_state", b"exec_state", "spark_app_state", b"spark_app_state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id", "exec_state", b"exec_state", "execution_id", b"execution_id", "featureset_id", b"featureset_id", "spark_app_state", b"spark_app_state"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["exec_state", b"exec_state"]) -> typing_extensions.Literal["spark_app_state"] | None: ...
+
+global___ExecutionState = ExecutionState

_proto/qwak/execution/v1/state/execution_state_service_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.execution/v1/state/execution_state_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.execution.v1.state import execution_state_pb2 as qwak_dot_execution_dot_v1_dot_state_dot_execution__state__pb2
+from frogml._proto.qwak.execution.v1.state import featureset_state_pb2 as qwak_dot_execution_dot_v1_dot_state_dot_featureset__state__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5qwak/execution/v1/state/execution_state_service.proto\x12%qwak.feature.store.execution.v1.state\x1a-qwak/execution/v1/state/execution_state.proto\x1a.qwak/execution/v1/state/featureset_state.proto\"6\n\x1dGetFeaturesetStateByIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\"r\n\x1eGetFeaturesetStateByIdResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\":\n\x1fGetFeaturesetStateByNameRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"t\n GetFeaturesetStateByNameResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"=\n#GetMultiFeaturesetStateByIdsRequest\x12\x16\n\x0e\x66\x65\x61tureset_ids\x18\x01 \x03(\t\"y\n$GetMultiFeaturesetStateByIdsResponse\x12Q\n\x11\x66\x65\x61tureset_states\x18\x01 \x03(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"?\n$GetMultiFeaturesetStateByNameRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x03(\t\"y\n%GetMultiFeaturesetStateByNameResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x03(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"r\n UpdateMultiExecutionStateRequest\x12N\n\x0f\x65xecution_state\x18\x01 \x03(\x0b\x32\x35.qwak.feature.store.execution.v1.state.ExecutionState\"#\n!UpdateMultiExecutionStateResponse\"9\n DeleteFeaturesetStateByIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\"@\n!DeleteFeaturesetStateByIdResponse\x12\x1b\n\x13num_deleted_records\x18\x01 \x01(\x03\x32\xd2\x08\n!FeatureStoreExecutionStateService\x12\xa5\x01\n\x16GetFeaturesetStateById\x12\x44.qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdRequest\x1a\x45.qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdResponse\x12\xab\x01\n\x18GetFeaturesetStateByName\x12\x46.qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameRequest\x1aG.qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameResponse\x12\xb7\x01\n\x1cGetMultiFeaturesetStateByIds\x12J.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsRequest\x1aK.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsResponse\x12\xba\x01\n\x1dGetMultiFeaturesetStateByName\x12K.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameRequest\x1aL.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameResponse\x12\xae\x01\n\x19UpdateMultiExecutionState\x12G.qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateRequest\x1aH.qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateResponse\x12\xae\x01\n\x19\x44\x65leteFeaturesetStateById\x12G.qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdRequest\x1aH.qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdResponseB`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.state.execution_state_service_pb2', _globals)
+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'
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_start=191
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_end=245
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_start=247
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_end=361
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_start=363
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_end=421
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_start=423
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_end=539
+  _globals['_GETMULTIFEATURESETSTATEBYIDSREQUEST']._serialized_start=541
+  _globals['_GETMULTIFEATURESETSTATEBYIDSREQUEST']._serialized_end=602
+  _globals['_GETMULTIFEATURESETSTATEBYIDSRESPONSE']._serialized_start=604
+  _globals['_GETMULTIFEATURESETSTATEBYIDSRESPONSE']._serialized_end=725
+  _globals['_GETMULTIFEATURESETSTATEBYNAMEREQUEST']._serialized_start=727
+  _globals['_GETMULTIFEATURESETSTATEBYNAMEREQUEST']._serialized_end=790
+  _globals['_GETMULTIFEATURESETSTATEBYNAMERESPONSE']._serialized_start=792
+  _globals['_GETMULTIFEATURESETSTATEBYNAMERESPONSE']._serialized_end=913
+  _globals['_UPDATEMULTIEXECUTIONSTATEREQUEST']._serialized_start=915
+  _globals['_UPDATEMULTIEXECUTIONSTATEREQUEST']._serialized_end=1029
+  _globals['_UPDATEMULTIEXECUTIONSTATERESPONSE']._serialized_start=1031
+  _globals['_UPDATEMULTIEXECUTIONSTATERESPONSE']._serialized_end=1066
+  _globals['_DELETEFEATURESETSTATEBYIDREQUEST']._serialized_start=1068
+  _globals['_DELETEFEATURESETSTATEBYIDREQUEST']._serialized_end=1125
+  _globals['_DELETEFEATURESETSTATEBYIDRESPONSE']._serialized_start=1127
+  _globals['_DELETEFEATURESETSTATEBYIDRESPONSE']._serialized_end=1191
+  _globals['_FEATURESTOREEXECUTIONSTATESERVICE']._serialized_start=1194
+  _globals['_FEATURESTOREEXECUTIONSTATESERVICE']._serialized_end=2300
+# @@protoc_insertion_point(module_scope)

_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