frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/feature_store/reports/report_pb2.pyi

@@ -0,0 +1,159 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_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 BatchJobIngestionDetails(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___BatchJobIngestionDetails = BatchJobIngestionDetails
+
+class StreamingJobIngestionDetails(google.protobuf.message.Message):
+    """row-level streaming"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___StreamingJobIngestionDetails = StreamingJobIngestionDetails
+
+class StreamingAggregationJobIngestionDetails(google.protobuf.message.Message):
+    """streaming aggregation"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___StreamingAggregationJobIngestionDetails = StreamingAggregationJobIngestionDetails
+
+class JobReport(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BATCH_INGESTION_DETAILS_FIELD_NUMBER: builtins.int
+    STREAMING_INGESTION_DETAILS_FIELD_NUMBER: builtins.int
+    STREAMING_AGGR_INGESTION_DETAILS_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
+    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
+    TRY_NUMBER_FIELD_NUMBER: builtins.int
+    @property
+    def batch_ingestion_details(self) -> global___BatchJobIngestionDetails: ...
+    @property
+    def streaming_ingestion_details(self) -> global___StreamingJobIngestionDetails: ...
+    @property
+    def streaming_aggr_ingestion_details(self) -> global___StreamingAggregationJobIngestionDetails: ...
+    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"""
+    execution_id: builtins.int
+    """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"""
+    try_number: builtins.int
+    """try number"""
+    def __init__(
+        self,
+        *,
+        batch_ingestion_details: global___BatchJobIngestionDetails | None = ...,
+        streaming_ingestion_details: global___StreamingJobIngestionDetails | None = ...,
+        streaming_aggr_ingestion_details: global___StreamingAggregationJobIngestionDetails | None = ...,
+        num_affected_records: builtins.int = ...,
+        run_start_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        run_end_time: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        execution_id: builtins.int = ...,
+        environment_id: builtins.str = ...,
+        featureset_id: builtins.str = ...,
+        featureset_name: builtins.str = ...,
+        status: global___JobStatus.ValueType = ...,
+        try_number: builtins.int = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["batch_ingestion_details", b"batch_ingestion_details", "ingestion_details", b"ingestion_details", "run_end_time", b"run_end_time", "run_start_time", b"run_start_time", "streaming_aggr_ingestion_details", b"streaming_aggr_ingestion_details", "streaming_ingestion_details", b"streaming_ingestion_details"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_ingestion_details", b"batch_ingestion_details", "environment_id", b"environment_id", "execution_id", b"execution_id", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "ingestion_details", b"ingestion_details", "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_ingestion_details", b"streaming_aggr_ingestion_details", "streaming_ingestion_details", b"streaming_ingestion_details", "try_number", b"try_number"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["ingestion_details", b"ingestion_details"]) -> typing_extensions.Literal["batch_ingestion_details", "streaming_ingestion_details", "streaming_aggr_ingestion_details"] | None: ...
+
+global___JobReport = JobReport

_frogml_proto/qwak/feature_store/reports/report_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/feature_store/repository/common/platform_pb2.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.feature_store/repository/common/platform.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'\n3qwak/feature_store/repository/common/platform.proto\x12$qwak.feature.store.repository.common\"\xb6\x01\n\x10PlatformSettings\x12S\n\x12repository_setting\x18\x01 \x01(\x0b\x32\x37.qwak.feature.store.repository.common.RepositorySetting\x12M\n\x0fproject_setting\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.repository.common.ProjectSetting\"+\n\x11RepositorySetting\x12\x16\n\x0erepository_key\x18\x01 \x01(\t\"=\n\x0eProjectSetting\x12\x15\n\x0bproject_key\x18\x02 \x01(\tH\x00\x42\x14\n\x12project_identifierBn\n/com.qwak.ai.feature.store.repository.common.apiP\x01Z9qwak/featurestore/featureset/repository;common_repositoryb\x06proto3')
+
+
+
+_PLATFORMSETTINGS = DESCRIPTOR.message_types_by_name['PlatformSettings']
+_REPOSITORYSETTING = DESCRIPTOR.message_types_by_name['RepositorySetting']
+_PROJECTSETTING = DESCRIPTOR.message_types_by_name['ProjectSetting']
+PlatformSettings = _reflection.GeneratedProtocolMessageType('PlatformSettings', (_message.Message,), {
+  'DESCRIPTOR' : _PLATFORMSETTINGS,
+  '__module__' : 'qwak.feature_store.repository.common.platform_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.repository.common.PlatformSettings)
+  })
+_sym_db.RegisterMessage(PlatformSettings)
+
+RepositorySetting = _reflection.GeneratedProtocolMessageType('RepositorySetting', (_message.Message,), {
+  'DESCRIPTOR' : _REPOSITORYSETTING,
+  '__module__' : 'qwak.feature_store.repository.common.platform_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.repository.common.RepositorySetting)
+  })
+_sym_db.RegisterMessage(RepositorySetting)
+
+ProjectSetting = _reflection.GeneratedProtocolMessageType('ProjectSetting', (_message.Message,), {
+  'DESCRIPTOR' : _PROJECTSETTING,
+  '__module__' : 'qwak.feature_store.repository.common.platform_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.repository.common.ProjectSetting)
+  })
+_sym_db.RegisterMessage(ProjectSetting)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.repository.common.apiP\001Z9qwak/featurestore/featureset/repository;common_repository'
+  _PLATFORMSETTINGS._serialized_start=94
+  _PLATFORMSETTINGS._serialized_end=276
+  _REPOSITORYSETTING._serialized_start=278
+  _REPOSITORYSETTING._serialized_end=321
+  _PROJECTSETTING._serialized_start=323
+  _PROJECTSETTING._serialized_end=384
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/feature_store/repository/common/platform_pb2.pyi

@@ -0,0 +1,70 @@
+"""
+@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 PlatformSettings(google.protobuf.message.Message):
+    """Container for all (JFrog) platform setting"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REPOSITORY_SETTING_FIELD_NUMBER: builtins.int
+    PROJECT_SETTING_FIELD_NUMBER: builtins.int
+    @property
+    def repository_setting(self) -> global___RepositorySetting:
+        """Repo related setting"""
+    @property
+    def project_setting(self) -> global___ProjectSetting:
+        """Project related setting"""
+    def __init__(
+        self,
+        *,
+        repository_setting: global___RepositorySetting | None = ...,
+        project_setting: global___ProjectSetting | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["project_setting", b"project_setting", "repository_setting", b"repository_setting"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["project_setting", b"project_setting", "repository_setting", b"repository_setting"]) -> None: ...
+
+global___PlatformSettings = PlatformSettings
+
+class RepositorySetting(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REPOSITORY_KEY_FIELD_NUMBER: builtins.int
+    repository_key: builtins.str
+    def __init__(
+        self,
+        *,
+        repository_key: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["repository_key", b"repository_key"]) -> None: ...
+
+global___RepositorySetting = RepositorySetting
+
+class ProjectSetting(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PROJECT_KEY_FIELD_NUMBER: builtins.int
+    project_key: builtins.str
+    """JFrog project key"""
+    def __init__(
+        self,
+        *,
+        project_key: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["project_identifier", b"project_identifier"]) -> typing_extensions.Literal["project_key"] | None: ...
+
+global___ProjectSetting = ProjectSetting

_frogml_proto/qwak/feature_store/repository/common/platform_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/feature_store/serving/management_pb2.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.feature_store/serving/management.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _frogml_proto.qwak.feature_store.serving import metadata_pb2 as qwak_dot_feature__store_dot_serving_dot_metadata__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+qwak/feature_store/serving/management.proto\x12%com.qwak.ai.feature.store.serving.api\x1a)qwak/feature_store/serving/metadata.proto\"<\n#GetFeaturesetMemoryUsageByIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\"}\n$GetFeaturesetMemoryUsageByIdResponse\x12U\n\x05usage\x18\x01 \x01(\x0b\x32\x46.com.qwak.ai.feature.store.serving.metadata.FeaturesetMemoryUsageStats\".\n,GetFeaturesetMemoryUsageByEnvironmentRequest\"\x8c\x01\n-GetFeaturesetMemoryUsageByEnvironmentResponse\x12[\n\x0busage_stats\x18\x01 \x03(\x0b\x32\x46.com.qwak.ai.feature.store.serving.metadata.FeaturesetMemoryUsageStats2\xac\x03\n\x1b\x46\x65\x61turesetManagementService\x12\xb7\x01\n\x1cGetFeaturesetMemoryUsageById\x12J.com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByIdRequest\x1aK.com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByIdResponse\x12\xd2\x01\n%GetFeaturesetMemoryUsageByEnvironment\x12S.com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByEnvironmentRequest\x1aT.com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByEnvironmentResponseB\'\n%com.qwak.ai.feature.store.serving.apib\x06proto3')
+
+
+
+_GETFEATURESETMEMORYUSAGEBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetFeaturesetMemoryUsageByIdRequest']
+_GETFEATURESETMEMORYUSAGEBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetFeaturesetMemoryUsageByIdResponse']
+_GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST = DESCRIPTOR.message_types_by_name['GetFeaturesetMemoryUsageByEnvironmentRequest']
+_GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE = DESCRIPTOR.message_types_by_name['GetFeaturesetMemoryUsageByEnvironmentResponse']
+GetFeaturesetMemoryUsageByIdRequest = _reflection.GeneratedProtocolMessageType('GetFeaturesetMemoryUsageByIdRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETFEATURESETMEMORYUSAGEBYIDREQUEST,
+  '__module__' : 'qwak.feature_store.serving.management_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByIdRequest)
+  })
+_sym_db.RegisterMessage(GetFeaturesetMemoryUsageByIdRequest)
+
+GetFeaturesetMemoryUsageByIdResponse = _reflection.GeneratedProtocolMessageType('GetFeaturesetMemoryUsageByIdResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETFEATURESETMEMORYUSAGEBYIDRESPONSE,
+  '__module__' : 'qwak.feature_store.serving.management_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByIdResponse)
+  })
+_sym_db.RegisterMessage(GetFeaturesetMemoryUsageByIdResponse)
+
+GetFeaturesetMemoryUsageByEnvironmentRequest = _reflection.GeneratedProtocolMessageType('GetFeaturesetMemoryUsageByEnvironmentRequest', (_message.Message,), {
+  'DESCRIPTOR' : _GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST,
+  '__module__' : 'qwak.feature_store.serving.management_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByEnvironmentRequest)
+  })
+_sym_db.RegisterMessage(GetFeaturesetMemoryUsageByEnvironmentRequest)
+
+GetFeaturesetMemoryUsageByEnvironmentResponse = _reflection.GeneratedProtocolMessageType('GetFeaturesetMemoryUsageByEnvironmentResponse', (_message.Message,), {
+  'DESCRIPTOR' : _GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE,
+  '__module__' : 'qwak.feature_store.serving.management_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.GetFeaturesetMemoryUsageByEnvironmentResponse)
+  })
+_sym_db.RegisterMessage(GetFeaturesetMemoryUsageByEnvironmentResponse)
+
+_FEATURESETMANAGEMENTSERVICE = DESCRIPTOR.services_by_name['FeaturesetManagementService']
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.feature.store.serving.api'
+  _GETFEATURESETMEMORYUSAGEBYIDREQUEST._serialized_start=129
+  _GETFEATURESETMEMORYUSAGEBYIDREQUEST._serialized_end=189
+  _GETFEATURESETMEMORYUSAGEBYIDRESPONSE._serialized_start=191
+  _GETFEATURESETMEMORYUSAGEBYIDRESPONSE._serialized_end=316
+  _GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST._serialized_start=318
+  _GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST._serialized_end=364
+  _GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE._serialized_start=367
+  _GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE._serialized_end=507
+  _FEATURESETMANAGEMENTSERVICE._serialized_start=510
+  _FEATURESETMANAGEMENTSERVICE._serialized_end=938
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/feature_store/serving/management_pb2.pyi

@@ -0,0 +1,72 @@
+"""
+@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.feature_store.serving.metadata_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetFeaturesetMemoryUsageByIdRequest(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___GetFeaturesetMemoryUsageByIdRequest = GetFeaturesetMemoryUsageByIdRequest
+
+class GetFeaturesetMemoryUsageByIdResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    USAGE_FIELD_NUMBER: builtins.int
+    @property
+    def usage(self) -> _frogml_proto.qwak.feature_store.serving.metadata_pb2.FeaturesetMemoryUsageStats: ...
+    def __init__(
+        self,
+        *,
+        usage: _frogml_proto.qwak.feature_store.serving.metadata_pb2.FeaturesetMemoryUsageStats | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["usage", b"usage"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["usage", b"usage"]) -> None: ...
+
+global___GetFeaturesetMemoryUsageByIdResponse = GetFeaturesetMemoryUsageByIdResponse
+
+class GetFeaturesetMemoryUsageByEnvironmentRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___GetFeaturesetMemoryUsageByEnvironmentRequest = GetFeaturesetMemoryUsageByEnvironmentRequest
+
+class GetFeaturesetMemoryUsageByEnvironmentResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    USAGE_STATS_FIELD_NUMBER: builtins.int
+    @property
+    def usage_stats(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.serving.metadata_pb2.FeaturesetMemoryUsageStats]: ...
+    def __init__(
+        self,
+        *,
+        usage_stats: collections.abc.Iterable[qwak.feature_store.serving.metadata_pb2.FeaturesetMemoryUsageStats] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["usage_stats", b"usage_stats"]) -> None: ...
+
+global___GetFeaturesetMemoryUsageByEnvironmentResponse = GetFeaturesetMemoryUsageByEnvironmentResponse

_frogml_proto/qwak/feature_store/serving/management_pb2_grpc.py

@@ -0,0 +1,101 @@
+# 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.feature_store.serving import management_pb2 as qwak_dot_feature__store_dot_serving_dot_management__pb2
+
+
+class FeaturesetManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetFeaturesetMemoryUsageById = channel.unary_unary(
+                '/com.qwak.ai.feature.store.serving.api.FeaturesetManagementService/GetFeaturesetMemoryUsageById',
+                request_serializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdResponse.FromString,
+                )
+        self.GetFeaturesetMemoryUsageByEnvironment = channel.unary_unary(
+                '/com.qwak.ai.feature.store.serving.api.FeaturesetManagementService/GetFeaturesetMemoryUsageByEnvironment',
+                request_serializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentResponse.FromString,
+                )
+
+
+class FeaturesetManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetFeaturesetMemoryUsageById(self, request, context):
+        """get memory usage for a single Featureset
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetFeaturesetMemoryUsageByEnvironment(self, request, context):
+        """get memory usage stats for an entire env
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FeaturesetManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetFeaturesetMemoryUsageById': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeaturesetMemoryUsageById,
+                    request_deserializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdResponse.SerializeToString,
+            ),
+            'GetFeaturesetMemoryUsageByEnvironment': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeaturesetMemoryUsageByEnvironment,
+                    request_deserializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'com.qwak.ai.feature.store.serving.api.FeaturesetManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FeaturesetManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetFeaturesetMemoryUsageById(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, '/com.qwak.ai.feature.store.serving.api.FeaturesetManagementService/GetFeaturesetMemoryUsageById',
+            qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdRequest.SerializeToString,
+            qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByIdResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeaturesetMemoryUsageByEnvironment(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, '/com.qwak.ai.feature.store.serving.api.FeaturesetManagementService/GetFeaturesetMemoryUsageByEnvironment',
+            qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentRequest.SerializeToString,
+            qwak_dot_feature__store_dot_serving_dot_management__pb2.GetFeaturesetMemoryUsageByEnvironmentResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_frogml_proto/qwak/feature_store/serving/metadata_pb2.py

@@ -0,0 +1,67 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.feature_store/serving/metadata.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _frogml_proto.qwak.feature_store.features import feature_set_types_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__types__pb2
+from _frogml_proto.qwak.feature_store.entities import entity_pb2 as qwak_dot_feature__store_dot_entities_dot_entity__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/feature_store/serving/metadata.proto\x12*com.qwak.ai.feature.store.serving.metadata\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\"\xb0\x02\n&StreamingAggregationFeaturesetMetadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x65\n streaming_aggregation_featureset\x18\x04 \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSet\x12=\n\x06\x65ntity\x18\x05 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x18\n\x10\x61vro_schema_json\x18\x06 \x01(\t\"\x8e\x02\n\x19\x42\x61tchFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12K\n\x13\x62\x61tch_featureset_v1\x18\x05 \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\"\x9a\x02\n\x1dStreamingFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12S\n\x17streaming_featureset_v1\x18\x05 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\"\x7f\n\x1a\x46\x65\x61turesetMemoryUsageStats\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x15\n\rfeatureset_id\x18\x02 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x03 \x01(\t\x12\x19\n\x11online_used_bytes\x18\x04 \x01(\x03\x42,\n*com.qwak.ai.feature.store.serving.metadatab\x06proto3')
+
+
+
+_STREAMINGAGGREGATIONFEATURESETMETADATA = DESCRIPTOR.message_types_by_name['StreamingAggregationFeaturesetMetadata']
+_BATCHFEATURESETV1METADATA = DESCRIPTOR.message_types_by_name['BatchFeaturesetV1Metadata']
+_STREAMINGFEATURESETV1METADATA = DESCRIPTOR.message_types_by_name['StreamingFeaturesetV1Metadata']
+_FEATURESETMEMORYUSAGESTATS = DESCRIPTOR.message_types_by_name['FeaturesetMemoryUsageStats']
+StreamingAggregationFeaturesetMetadata = _reflection.GeneratedProtocolMessageType('StreamingAggregationFeaturesetMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _STREAMINGAGGREGATIONFEATURESETMETADATA,
+  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.StreamingAggregationFeaturesetMetadata)
+  })
+_sym_db.RegisterMessage(StreamingAggregationFeaturesetMetadata)
+
+BatchFeaturesetV1Metadata = _reflection.GeneratedProtocolMessageType('BatchFeaturesetV1Metadata', (_message.Message,), {
+  'DESCRIPTOR' : _BATCHFEATURESETV1METADATA,
+  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.BatchFeaturesetV1Metadata)
+  })
+_sym_db.RegisterMessage(BatchFeaturesetV1Metadata)
+
+StreamingFeaturesetV1Metadata = _reflection.GeneratedProtocolMessageType('StreamingFeaturesetV1Metadata', (_message.Message,), {
+  'DESCRIPTOR' : _STREAMINGFEATURESETV1METADATA,
+  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.StreamingFeaturesetV1Metadata)
+  })
+_sym_db.RegisterMessage(StreamingFeaturesetV1Metadata)
+
+FeaturesetMemoryUsageStats = _reflection.GeneratedProtocolMessageType('FeaturesetMemoryUsageStats', (_message.Message,), {
+  'DESCRIPTOR' : _FEATURESETMEMORYUSAGESTATS,
+  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
+  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.FeaturesetMemoryUsageStats)
+  })
+_sym_db.RegisterMessage(FeaturesetMemoryUsageStats)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n*com.qwak.ai.feature.store.serving.metadata'
+  _STREAMINGAGGREGATIONFEATURESETMETADATA._serialized_start=185
+  _STREAMINGAGGREGATIONFEATURESETMETADATA._serialized_end=489
+  _BATCHFEATURESETV1METADATA._serialized_start=492
+  _BATCHFEATURESETV1METADATA._serialized_end=762
+  _STREAMINGFEATURESETV1METADATA._serialized_start=765
+  _STREAMINGFEATURESETV1METADATA._serialized_end=1047
+  _FEATURESETMEMORYUSAGESTATS._serialized_start=1049
+  _FEATURESETMEMORYUSAGESTATS._serialized_end=1176
+# @@protoc_insertion_point(module_scope)