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

_qwak_proto/qwak/feature_store/features/feature_set_service_pb2.pyi

@@ -155,6 +155,79 @@ class GetFeatureSetByNameResponse(google.protobuf.message.Message):
 
 global___GetFeatureSetByNameResponse = GetFeatureSetByNameResponse
 
+class ListFeaturesetVersionsByNameRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    """Name of the Featureset to fetch versions for"""
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name"]) -> None: ...
+
+global___ListFeaturesetVersionsByNameRequest = ListFeaturesetVersionsByNameRequest
+
+class ListFeaturesetVersionsByNameResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESETS_FIELD_NUMBER: builtins.int
+    @property
+    def featuresets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.features.feature_set_pb2.FeaturesetInfo]:
+        """Fetched feature set object"""
+    def __init__(
+        self,
+        *,
+        featuresets: collections.abc.Iterable[qwak.feature_store.features.feature_set_pb2.FeaturesetInfo] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featuresets", b"featuresets"]) -> None: ...
+
+global___ListFeaturesetVersionsByNameResponse = ListFeaturesetVersionsByNameResponse
+
+class GetEnvToFeatureSetsMappingRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___GetEnvToFeatureSetsMappingRequest = GetEnvToFeatureSetsMappingRequest
+
+class GetEnvToFeatureSetsMappingResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class EnvToFeatureSetMappingEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        @property
+        def value(self) -> qwak.feature_store.features.feature_set_pb2.FeatureSetList: ...
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: qwak.feature_store.features.feature_set_pb2.FeatureSetList | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
+    ENV_TO_FEATURE_SET_MAPPING_FIELD_NUMBER: builtins.int
+    @property
+    def env_to_feature_set_mapping(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, qwak.feature_store.features.feature_set_pb2.FeatureSetList]:
+        """Mapping of environment to feature sets"""
+    def __init__(
+        self,
+        *,
+        env_to_feature_set_mapping: collections.abc.Mapping[builtins.str, qwak.feature_store.features.feature_set_pb2.FeatureSetList] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["env_to_feature_set_mapping", b"env_to_feature_set_mapping"]) -> None: ...
+
+global___GetEnvToFeatureSetsMappingResponse = GetEnvToFeatureSetsMappingResponse
+
 class DeleteFeatureSetRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -179,6 +252,34 @@ class DeleteFeatureSetResponse(google.protobuf.message.Message):
 
 global___DeleteFeatureSetResponse = DeleteFeatureSetResponse
 
+class DeleteFeaturesetVersionRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    FEATURESET_VERSION_NUMBER_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    """Name of the Featureset"""
+    featureset_version_number: builtins.int
+    """Version number to delete"""
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+        featureset_version_number: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name", "featureset_version_number", b"featureset_version_number"]) -> None: ...
+
+global___DeleteFeaturesetVersionRequest = DeleteFeaturesetVersionRequest
+
+class DeleteFeaturesetVersionResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___DeleteFeaturesetVersionResponse = DeleteFeaturesetVersionResponse
+
 class ListFeatureSetsRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -204,6 +305,31 @@ class ListFeatureSetsResponse(google.protobuf.message.Message):
 
 global___ListFeatureSetsResponse = ListFeatureSetsResponse
 
+class ListFeaturesetsByAccountRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListFeaturesetsByAccountRequest = ListFeaturesetsByAccountRequest
+
+class ListFeaturesetsByAccountResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESETS_FIELD_NUMBER: builtins.int
+    @property
+    def featuresets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.features.feature_set_pb2.FeatureSet]:
+        """List of registered featuresets"""
+    def __init__(
+        self,
+        *,
+        featuresets: collections.abc.Iterable[qwak.feature_store.features.feature_set_pb2.FeatureSet] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featuresets", b"featuresets"]) -> None: ...
+
+global___ListFeaturesetsByAccountResponse = ListFeaturesetsByAccountResponse
+
 class RunBatchFeatureSetRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -222,9 +348,15 @@ global___RunBatchFeatureSetRequest = RunBatchFeatureSetRequest
 class RunBatchFeatureSetResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    EXECUTION_ID_FIELD_NUMBER: builtins.int
+    execution_id: builtins.str
+    """Execution id of the batch featureset run"""
     def __init__(
         self,
+        *,
+        execution_id: builtins.str = ...,
     ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["execution_id", b"execution_id"]) -> None: ...
 
 global___RunBatchFeatureSetResponse = RunBatchFeatureSetResponse
 
@@ -526,3 +658,31 @@ class GetFeaturesetSchedulingStateResponse(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["state", b"state"]) -> None: ...
 
 global___GetFeaturesetSchedulingStateResponse = GetFeaturesetSchedulingStateResponse
+
+class SetActiveFeaturesetVersionRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    FEATURESET_VERSION_NUMBER_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    """Name of the Featureset"""
+    featureset_version_number: builtins.int
+    """Version number to set as active"""
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+        featureset_version_number: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name", "featureset_version_number", b"featureset_version_number"]) -> None: ...
+
+global___SetActiveFeaturesetVersionRequest = SetActiveFeaturesetVersionRequest
+
+class SetActiveFeaturesetVersionResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___SetActiveFeaturesetVersionResponse = SetActiveFeaturesetVersionResponse

_qwak_proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py

@@ -34,16 +34,31 @@ class FeatureSetServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetByNameRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetByNameResponse.FromString,
                 )
+        self.ListFeaturesetVersionsByName = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/ListFeaturesetVersionsByName',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameResponse.FromString,
+                )
         self.DeleteFeatureSet = channel.unary_unary(
                 '/qwak.feature.store.features.FeatureSetService/DeleteFeatureSet',
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeatureSetRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeatureSetResponse.FromString,
                 )
+        self.DeleteFeaturesetVersion = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/DeleteFeaturesetVersion',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionResponse.FromString,
+                )
         self.ListFeatureSets = channel.unary_unary(
                 '/qwak.feature.store.features.FeatureSetService/ListFeatureSets',
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.FromString,
                 )
+        self.ListFeaturesetsByAccount = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/ListFeaturesetsByAccount',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
+                )
         self.RunBatchFeatureSet = channel.unary_unary(
                 '/qwak.feature.store.features.FeatureSetService/RunBatchFeatureSet',
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.SerializeToString,
@@ -79,6 +94,11 @@ class FeatureSetServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByNameRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByNameResponse.FromString,
                 )
+        self.GetEnvToFeatureSetsMapping = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/GetEnvToFeatureSetsMapping',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingResponse.FromString,
+                )
         self.GetFeatureSetStateById = channel.unary_unary(
                 '/qwak.feature.store.features.FeatureSetService/GetFeatureSetStateById',
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByIdRequest.SerializeToString,
@@ -99,6 +119,11 @@ class FeatureSetServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeaturesetSchedulingStateRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeaturesetSchedulingStateResponse.FromString,
                 )
+        self.SetActiveFeaturesetVersion = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/SetActiveFeaturesetVersion',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionResponse.FromString,
+                )
 
 
 class FeatureSetServiceServicer(object):
@@ -132,6 +157,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListFeaturesetVersionsByName(self, request, context):
+        """List Feature Set Versions by name
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def DeleteFeatureSet(self, request, context):
         """Delete a feature from the registry
         """
@@ -139,6 +171,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def DeleteFeaturesetVersion(self, request, context):
+        """Delete a featureset version
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def ListFeatureSets(self, request, context):
         """List all registered features
         """
@@ -146,6 +185,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListFeaturesetsByAccount(self, request, context):
+        """List of registered featuresets
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def RunBatchFeatureSet(self, request, context):
         """Run now feature set
         """
@@ -195,6 +241,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def GetEnvToFeatureSetsMapping(self, request, context):
+        """Retrieve a mapping of environment names to Featuresets
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def GetFeatureSetStateById(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -222,6 +275,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def SetActiveFeaturesetVersion(self, request, context):
+        """Set Active Featureset Version
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_FeatureSetServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -245,16 +305,31 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetByNameRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetByNameResponse.SerializeToString,
             ),
+            'ListFeaturesetVersionsByName': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListFeaturesetVersionsByName,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameResponse.SerializeToString,
+            ),
             'DeleteFeatureSet': grpc.unary_unary_rpc_method_handler(
                     servicer.DeleteFeatureSet,
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeatureSetRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeatureSetResponse.SerializeToString,
             ),
+            'DeleteFeaturesetVersion': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteFeaturesetVersion,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionResponse.SerializeToString,
+            ),
             'ListFeatureSets': grpc.unary_unary_rpc_method_handler(
                     servicer.ListFeatureSets,
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.SerializeToString,
             ),
+            'ListFeaturesetsByAccount': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListFeaturesetsByAccount,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.SerializeToString,
+            ),
             'RunBatchFeatureSet': grpc.unary_unary_rpc_method_handler(
                     servicer.RunBatchFeatureSet,
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.FromString,
@@ -290,6 +365,11 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByNameRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByNameResponse.SerializeToString,
             ),
+            'GetEnvToFeatureSetsMapping': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetEnvToFeatureSetsMapping,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingResponse.SerializeToString,
+            ),
             'GetFeatureSetStateById': grpc.unary_unary_rpc_method_handler(
                     servicer.GetFeatureSetStateById,
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeatureSetStateByIdRequest.FromString,
@@ -310,6 +390,11 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeaturesetSchedulingStateRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeaturesetSchedulingStateResponse.SerializeToString,
             ),
+            'SetActiveFeaturesetVersion': grpc.unary_unary_rpc_method_handler(
+                    servicer.SetActiveFeaturesetVersion,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'qwak.feature.store.features.FeatureSetService', rpc_method_handlers)
@@ -388,6 +473,23 @@ class FeatureSetService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListFeaturesetVersionsByName(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.features.FeatureSetService/ListFeaturesetVersionsByName',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetVersionsByNameResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def DeleteFeatureSet(request,
             target,
@@ -405,6 +507,23 @@ class FeatureSetService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def DeleteFeaturesetVersion(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.features.FeatureSetService/DeleteFeaturesetVersion',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.DeleteFeaturesetVersionResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def ListFeatureSets(request,
             target,
@@ -422,6 +541,23 @@ class FeatureSetService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListFeaturesetsByAccount(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.features.FeatureSetService/ListFeaturesetsByAccount',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def RunBatchFeatureSet(request,
             target,
@@ -541,6 +677,23 @@ class FeatureSetService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def GetEnvToFeatureSetsMapping(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.features.FeatureSetService/GetEnvToFeatureSetsMapping',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetEnvToFeatureSetsMappingResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def GetFeatureSetStateById(request,
             target,
@@ -608,3 +761,20 @@ class FeatureSetService(object):
             qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.GetFeaturesetSchedulingStateResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def SetActiveFeaturesetVersion(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.features.FeatureSetService/SetActiveFeaturesetVersion',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.SetActiveFeaturesetVersionResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)