qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/feature_store/jobs/v1/job_service_pb2_grpc.py

@@ -34,6 +34,11 @@ class JobServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestSuccessfulJobRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestSuccessfulJobResponse.FromString,
                 )
+        self.GetAllLatestSuccessfulJobs = channel.unary_unary(
+                '/qwak.feature.store.jobs.v1.JobService/GetAllLatestSuccessfulJobs',
+                request_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsResponse.FromString,
+                )
         self.GetLatestJob = channel.unary_unary(
                 '/qwak.feature.store.jobs.v1.JobService/GetLatestJob',
                 request_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestJobRequest.SerializeToString,
@@ -49,6 +54,11 @@ class JobServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.DeleteFeaturesetJobByJobIdRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.DeleteFeaturesetJobByJobIdResponse.FromString,
                 )
+        self.GetJobsSummaries = channel.unary_unary(
+                '/qwak.feature.store.jobs.v1.JobService/GetJobsSummaries',
+                request_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesResponse.FromString,
+                )
 
 
 class JobServiceServicer(object):
@@ -82,6 +92,13 @@ class JobServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetAllLatestSuccessfulJobs(self, request, context):
+        """Get all latest successful jobs for each featureset in the environment
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def GetLatestJob(self, request, context):
         """Get latest job of a featureset
         """
@@ -103,6 +120,12 @@ class JobServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetJobsSummaries(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_JobServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -126,6 +149,11 @@ def add_JobServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestSuccessfulJobRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestSuccessfulJobResponse.SerializeToString,
             ),
+            'GetAllLatestSuccessfulJobs': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetAllLatestSuccessfulJobs,
+                    request_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsResponse.SerializeToString,
+            ),
             'GetLatestJob': grpc.unary_unary_rpc_method_handler(
                     servicer.GetLatestJob,
                     request_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetLatestJobRequest.FromString,
@@ -141,6 +169,11 @@ def add_JobServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.DeleteFeaturesetJobByJobIdRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.DeleteFeaturesetJobByJobIdResponse.SerializeToString,
             ),
+            'GetJobsSummaries': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetJobsSummaries,
+                    request_deserializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.feature.store.jobs.v1.JobService', rpc_method_handlers)
@@ -219,6 +252,23 @@ class JobService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def GetAllLatestSuccessfulJobs(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.jobs.v1.JobService/GetAllLatestSuccessfulJobs',
+            qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsRequest.SerializeToString,
+            qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetAllLatestSuccessfulJobsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def GetLatestJob(request,
             target,
@@ -269,3 +319,20 @@ class JobService(object):
             qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.DeleteFeaturesetJobByJobIdResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetJobsSummaries(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.jobs.v1.JobService/GetJobsSummaries',
+            qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesRequest.SerializeToString,
+            qwak_dot_feature__store_dot_jobs_dot_v1_dot_job__service__pb2.GetJobsSummariesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_qwak_proto/qwak/feature_store/platform/platform_details_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/feature_store/platform/platform_details.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 _qwak_proto.qwak.feature_store.repository.common import platform_pb2 as qwak_dot_feature__store_dot_repository_dot_common_dot_platform__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2qwak/feature_store/platform/platform_details.proto\x12\x1bqwak.feature.store.platform\x1a\x33qwak/feature_store/repository/common/platform.proto\"\x9e\x01\n\x0fPlatformDetails\x12Q\n\x11platform_settings\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.repository.common.PlatformSettings\x12\x38\n\tauth_type\x18\x02 \x01(\x0b\x32%.qwak.feature.store.platform.AuthType\"&\n\x08\x41uthType\x12\x12\n\x08token_id\x18\x01 \x01(\tH\x00\x42\x06\n\x04typeB[\n&com.qwak.ai.feature.store.platform.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.platform.platform_details_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.feature.store.platform.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
+  _globals['_PLATFORMDETAILS']._serialized_start=137
+  _globals['_PLATFORMDETAILS']._serialized_end=295
+  _globals['_AUTHTYPE']._serialized_start=297
+  _globals['_AUTHTYPE']._serialized_end=335
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/platform/platform_details_pb2.pyi

@@ -0,0 +1,54 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import qwak.feature_store.repository.common.platform_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class PlatformDetails(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PLATFORM_SETTINGS_FIELD_NUMBER: builtins.int
+    AUTH_TYPE_FIELD_NUMBER: builtins.int
+    @property
+    def platform_settings(self) -> qwak.feature_store.repository.common.platform_pb2.PlatformSettings:
+        """Platform settings"""
+    @property
+    def auth_type(self) -> global___AuthType:
+        """Auth type"""
+    def __init__(
+        self,
+        *,
+        platform_settings: qwak.feature_store.repository.common.platform_pb2.PlatformSettings | None = ...,
+        auth_type: global___AuthType | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "platform_settings", b"platform_settings"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "platform_settings", b"platform_settings"]) -> None: ...
+
+global___PlatformDetails = PlatformDetails
+
+class AuthType(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TOKEN_ID_FIELD_NUMBER: builtins.int
+    token_id: builtins.str
+    def __init__(
+        self,
+        *,
+        token_id: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["token_id", b"token_id", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["token_id", b"token_id", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["token_id"] | None: ...
+
+global___AuthType = AuthType

_qwak_proto/qwak/feature_store/platform/platform_details_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
+

_qwak_proto/qwak/feature_store/reports/report_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/feature_store/reports/report.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
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -18,58 +16,21 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'qwak/feature_store/reports/report.proto\x12\x17qwak.feature.store.jobs\x1a\x1fgoogle/protobuf/timestamp.proto\"\xd4\x01\n\x18\x42\x61tchJobIngestionDetails\x12\x38\n\x14\x62\x61tch_execution_date\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12?\n\x1bingestion_window_start_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12=\n\x19ingestion_window_end_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\x1e\n\x1cStreamingJobIngestionDetails\")\n\'StreamingAggregationJobIngestionDetails\"\xec\x04\n\tJobReport\x12T\n\x17\x62\x61tch_ingestion_details\x18\x01 \x01(\x0b\x32\x31.qwak.feature.store.jobs.BatchJobIngestionDetailsH\x00\x12\\\n\x1bstreaming_ingestion_details\x18\x02 \x01(\x0b\x32\x35.qwak.feature.store.jobs.StreamingJobIngestionDetailsH\x00\x12l\n streaming_aggr_ingestion_details\x18\x03 \x01(\x0b\x32@.qwak.feature.store.jobs.StreamingAggregationJobIngestionDetailsH\x00\x12\x1c\n\x14num_affected_records\x18\x04 \x01(\x03\x12\x32\n\x0erun_start_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x30\n\x0crun_end_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x14\n\x0c\x65xecution_id\x18\x07 \x01(\x03\x12\x16\n\x0e\x65nvironment_id\x18\x08 \x01(\t\x12\x15\n\rfeatureset_id\x18\t \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\n \x01(\t\x12\x32\n\x06status\x18\x0b \x01(\x0e\x32\".qwak.feature.store.jobs.JobStatus\x12\x12\n\ntry_number\x18\x0c \x01(\x05\x42\x13\n\x11ingestion_details*m\n\tJobStatus\x12\x16\n\x12JOB_STATUS_INVALID\x10\x00\x12\x16\n\x12JOB_STATUS_SUCCESS\x10\x01\x12\x16\n\x12JOB_STATUS_FAILURE\x10\x02\x12\x18\n\x14JOB_STATUS_TOO_EARLY\x10\x03\x42%\n!com.qwak.ai.feature.store.reportsP\x01\x62\x06proto3')
 
-_JOBSTATUS = DESCRIPTOR.enum_types_by_name['JobStatus']
-JobStatus = enum_type_wrapper.EnumTypeWrapper(_JOBSTATUS)
-JOB_STATUS_INVALID = 0
-JOB_STATUS_SUCCESS = 1
-JOB_STATUS_FAILURE = 2
-JOB_STATUS_TOO_EARLY = 3
-
-
-_BATCHJOBINGESTIONDETAILS = DESCRIPTOR.message_types_by_name['BatchJobIngestionDetails']
-_STREAMINGJOBINGESTIONDETAILS = DESCRIPTOR.message_types_by_name['StreamingJobIngestionDetails']
-_STREAMINGAGGREGATIONJOBINGESTIONDETAILS = DESCRIPTOR.message_types_by_name['StreamingAggregationJobIngestionDetails']
-_JOBREPORT = DESCRIPTOR.message_types_by_name['JobReport']
-BatchJobIngestionDetails = _reflection.GeneratedProtocolMessageType('BatchJobIngestionDetails', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHJOBINGESTIONDETAILS,
-  '__module__' : 'qwak.feature_store.reports.report_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.jobs.BatchJobIngestionDetails)
-  })
-_sym_db.RegisterMessage(BatchJobIngestionDetails)
-
-StreamingJobIngestionDetails = _reflection.GeneratedProtocolMessageType('StreamingJobIngestionDetails', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGJOBINGESTIONDETAILS,
-  '__module__' : 'qwak.feature_store.reports.report_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.jobs.StreamingJobIngestionDetails)
-  })
-_sym_db.RegisterMessage(StreamingJobIngestionDetails)
-
-StreamingAggregationJobIngestionDetails = _reflection.GeneratedProtocolMessageType('StreamingAggregationJobIngestionDetails', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATIONJOBINGESTIONDETAILS,
-  '__module__' : 'qwak.feature_store.reports.report_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.jobs.StreamingAggregationJobIngestionDetails)
-  })
-_sym_db.RegisterMessage(StreamingAggregationJobIngestionDetails)
-
-JobReport = _reflection.GeneratedProtocolMessageType('JobReport', (_message.Message,), {
-  'DESCRIPTOR' : _JOBREPORT,
-  '__module__' : 'qwak.feature_store.reports.report_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.jobs.JobReport)
-  })
-_sym_db.RegisterMessage(JobReport)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.reports.report_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.feature.store.reportsP\001'
-  _JOBSTATUS._serialized_start=1014
-  _JOBSTATUS._serialized_end=1123
-  _BATCHJOBINGESTIONDETAILS._serialized_start=102
-  _BATCHJOBINGESTIONDETAILS._serialized_end=314
-  _STREAMINGJOBINGESTIONDETAILS._serialized_start=316
-  _STREAMINGJOBINGESTIONDETAILS._serialized_end=346
-  _STREAMINGAGGREGATIONJOBINGESTIONDETAILS._serialized_start=348
-  _STREAMINGAGGREGATIONJOBINGESTIONDETAILS._serialized_end=389
-  _JOBREPORT._serialized_start=392
-  _JOBREPORT._serialized_end=1012
+  _globals['_JOBSTATUS']._serialized_start=1014
+  _globals['_JOBSTATUS']._serialized_end=1123
+  _globals['_BATCHJOBINGESTIONDETAILS']._serialized_start=102
+  _globals['_BATCHJOBINGESTIONDETAILS']._serialized_end=314
+  _globals['_STREAMINGJOBINGESTIONDETAILS']._serialized_start=316
+  _globals['_STREAMINGJOBINGESTIONDETAILS']._serialized_end=346
+  _globals['_STREAMINGAGGREGATIONJOBINGESTIONDETAILS']._serialized_start=348
+  _globals['_STREAMINGAGGREGATIONJOBINGESTIONDETAILS']._serialized_end=389
+  _globals['_JOBREPORT']._serialized_start=392
+  _globals['_JOBREPORT']._serialized_end=1012
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/repository/common/platform_pb2.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: 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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\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_identifierBj\n/com.qwak.ai.feature.store.repository.common.apiP\x01Z5qwak/featurestore/source/repository;source_repositoryb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.repository.common.platform_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.repository.common.apiP\001Z5qwak/featurestore/source/repository;source_repository'
+  _globals['_PLATFORMSETTINGS']._serialized_start=94
+  _globals['_PLATFORMSETTINGS']._serialized_end=276
+  _globals['_REPOSITORYSETTING']._serialized_start=278
+  _globals['_REPOSITORYSETTING']._serialized_end=321
+  _globals['_PROJECTSETTING']._serialized_start=323
+  _globals['_PROJECTSETTING']._serialized_end=384
+# @@protoc_insertion_point(module_scope)

_qwak_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

_qwak_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
+

_qwak_proto/qwak/feature_store/serving/management_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: 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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _qwak_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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.serving.management_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.feature.store.serving.api'
+  _globals['_GETFEATURESETMEMORYUSAGEBYIDREQUEST']._serialized_start=129
+  _globals['_GETFEATURESETMEMORYUSAGEBYIDREQUEST']._serialized_end=189
+  _globals['_GETFEATURESETMEMORYUSAGEBYIDRESPONSE']._serialized_start=191
+  _globals['_GETFEATURESETMEMORYUSAGEBYIDRESPONSE']._serialized_end=316
+  _globals['_GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST']._serialized_start=318
+  _globals['_GETFEATURESETMEMORYUSAGEBYENVIRONMENTREQUEST']._serialized_end=364
+  _globals['_GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE']._serialized_start=367
+  _globals['_GETFEATURESETMEMORYUSAGEBYENVIRONMENTRESPONSE']._serialized_end=507
+  _globals['_FEATURESETMANAGEMENTSERVICE']._serialized_start=510
+  _globals['_FEATURESETMANAGEMENTSERVICE']._serialized_end=938
+# @@protoc_insertion_point(module_scope)

_qwak_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 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) -> qwak.feature_store.serving.metadata_pb2.FeaturesetMemoryUsageStats: ...
+    def __init__(
+        self,
+        *,
+        usage: 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

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