frogml-core 0.0.119__tar.gz → 0.0.121__tar.gz

{frogml_core-0.0.119 → frogml_core-0.0.121}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.119
+Version: 0.0.121
 Summary: frogml contains the necessary objects and communication tools for using the JFrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/features/feature_set_pb2.py

@@ -21,7 +21,7 @@ from frogml._proto.qwak.feature_store.repository.common import platform_pb2 as q
 from frogml._proto.qwak.feature_store.platform import platform_details_pb2 as qwak_dot_feature__store_dot_platform_dot_platform__details__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/feature_store/features/feature_set.proto\x12\x1bqwak.feature.store.features\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\x1a\x33qwak/feature_store/features/feature_set_state.proto\x1a\x37qwak/feature_store/features/feature_set_attribute.proto\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\x1a\x33qwak/feature_store/repository/common/platform.proto\x1a\x32qwak/feature_store/platform/platform_details.proto\"\xfc\x01\n\nFeatureSet\x12Q\n\x16\x66\x65\x61ture_set_definition\x18\x01 \x01(\x0b\x32\x31.qwak.feature.store.features.FeatureSetDefinition\x12\x41\n\x08metadata\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.FeatureSetMetadata\x12X\n\x1b\x64\x65ployed_models_in_use_link\x18\x03 \x03(\x0b\x32\x33.qwak.feature.store.features.DeployedModelInUseLink\"\xa5\x01\n\x0e\x46\x65\x61turesetInfo\x12P\n\x15\x66\x65\x61tureset_definition\x18\x01 \x01(\x0b\x32\x31.qwak.feature.store.features.FeatureSetDefinition\x12\x41\n\x08metadata\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.FeatureSetMetadata\"x\n\x11\x46\x65\x61turesetSummary\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12I\n\x10\x66\x65\x61ture_set_type\x18\x03 \x01(\x0e\x32/.qwak.feature.store.features.FeatureSetTypeView\"\xf6\x04\n\x14\x46\x65\x61tureSetDefinition\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\x12:\n\x06status\x18\x03 \x01(\x0e\x32*.qwak.feature.store.features.FeatureStatus\x12g\n feature_set_last_execution_state\x18\x04 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionStateB\x02\x18\x01\x12U\n\x0e\x66\x65\x61ture_health\x18\x05 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionStateB\x02\x18\x01\x12K\n\x11\x66\x65\x61ture_set_state\x18\x06 \x03(\x0b\x32,.qwak.feature.store.features.FeatureSetStateB\x02\x18\x01\x12*\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x42\x02\x18\x01\x12!\n\x19\x66\x65\x61tureset_version_number\x18\x08 \x01(\x05\x12H\n\x10platform_details\x18\t \x01(\x0b\x32,.qwak.feature.store.platform.PlatformDetailsH\x00\x42\x1d\n\x1b\x66\x65\x61tureset_platform_details\"\xdb\x04\n\x0e\x46\x65\x61tureSetSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x33.qwak.feature.store.features.FeatureSetUserMetadata\x12\x16\n\ngit_commit\x18\x03 \x01(\tB\x02\x18\x01\x12=\n\x06\x65ntity\x18\x04 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x45\n\x10\x66\x65\x61ture_set_type\x18\x05 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetType\x12\x36\n\x08\x66\x65\x61tures\x18\x06 \x03(\x0b\x32$.qwak.feature.store.features.Feature\x12*\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x42\x02\x18\x01\x12V\n\x10source_code_spec\x18\x08 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecB\x02\x18\x01\x12P\n\x15\x66\x65\x61tureset_attributes\x18\t \x01(\x0b\x32\x31.qwak.feature.store.features.FeaturesetAttributes\x12$\n\x1a\x66\x65\x61tureset_repository_name\x18\n \x01(\tH\x00\x42\"\n featureset_repository_identifier\"\xb8\x01\n\x16\x46\x65\x61tureSetUserMetadata\x12\r\n\x05owner\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\x12\x64\n featureset_version_user_metadata\x18\x04 \x01(\x0b\x32:.qwak.feature.store.features.FeaturesetVersionUserMetadata\"0\n\x1d\x46\x65\x61turesetVersionUserMetadata\x12\x0f\n\x07\x63omment\x18\x01 \x01(\t\"\x85\x02\n\x12\x46\x65\x61tureSetMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\x12[\n\x1b\x66\x65\x61tureset_version_metadata\x18\x05 \x01(\x0b\x32\x36.qwak.feature.store.features.FeaturesetVersionMetadata\"_\n\x19\x46\x65\x61turesetVersionMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\">\n\x16\x44\x65ployedModelInUseLink\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nmodel_name\x18\x02 \x01(\t\"5\n\x07\x46\x65\x61ture\x12\x14\n\x0c\x66\x65\x61ture_name\x18\x01 \x01(\t\x12\x14\n\x0c\x66\x65\x61ture_type\x18\x02 \x01(\t\"\xa7\x01\n\x16\x44\x65ployedFeatureSetSpec\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x01 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x02 \x01(\t\x12\x12\n\naccount_id\x18\x03 \x01(\t\x12\x1a\n\x12\x66\x65\x61tureset_version\x18\x04 \x01(\x05\"O\n\x0e\x46\x65\x61tureSetList\x12=\n\x0c\x66\x65\x61ture_sets\x18\x01 \x03(\x0b\x32\'.qwak.feature.store.features.FeatureSet*\xb4\x01\n\rFeatureStatus\x12\x0b\n\x07INVALID\x10\x00\x12\t\n\x05VALID\x10\x01\x12\x0b\n\x07\x44\x45LETED\x10\x02\x12\x17\n\x13REGISTRATION_FAILED\x10\x03\x12\x1c\n\x18REGISTRATION_IN_PROGRESS\x10\x04\x12\x18\n\x14\x44\x45LETION_IN_PROGRESS\x10\x05\x12\x13\n\x0f\x44\x45LETION_FAILED\x10\x06\x12\x18\n\x14\x42\x41\x43KFILL_IN_PROGRESS\x10\x07*\x92\x01\n\x19\x46\x65\x61turesetSchedulingState\x12\x1c\n\x18SCHEDULING_STATE_INVALID\x10\x00\x12\x1c\n\x18SCHEDULING_STATE_ENABLED\x10\x01\x12\x1b\n\x17SCHEDULING_STATE_PAUSED\x10\x02\x12\x1c\n\x18SCHEDULING_STATE_UNKNOWN\x10\x03\x42[\n&com.qwak.ai.feature.store.features.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/feature_store/features/feature_set.proto\x12\x1bqwak.feature.store.features\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\x1a\x33qwak/feature_store/features/feature_set_state.proto\x1a\x37qwak/feature_store/features/feature_set_attribute.proto\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\x1a\x33qwak/feature_store/repository/common/platform.proto\x1a\x32qwak/feature_store/platform/platform_details.proto\"\xfc\x01\n\nFeatureSet\x12Q\n\x16\x66\x65\x61ture_set_definition\x18\x01 \x01(\x0b\x32\x31.qwak.feature.store.features.FeatureSetDefinition\x12\x41\n\x08metadata\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.FeatureSetMetadata\x12X\n\x1b\x64\x65ployed_models_in_use_link\x18\x03 \x03(\x0b\x32\x33.qwak.feature.store.features.DeployedModelInUseLink\"\xa5\x01\n\x0e\x46\x65\x61turesetInfo\x12P\n\x15\x66\x65\x61tureset_definition\x18\x01 \x01(\x0b\x32\x31.qwak.feature.store.features.FeatureSetDefinition\x12\x41\n\x08metadata\x18\x02 \x01(\x0b\x32/.qwak.feature.store.features.FeatureSetMetadata\"x\n\x11\x46\x65\x61turesetSummary\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12I\n\x10\x66\x65\x61ture_set_type\x18\x03 \x01(\x0e\x32/.qwak.feature.store.features.FeatureSetTypeView\"\xf6\x04\n\x14\x46\x65\x61tureSetDefinition\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\x12:\n\x06status\x18\x03 \x01(\x0e\x32*.qwak.feature.store.features.FeatureStatus\x12g\n feature_set_last_execution_state\x18\x04 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionStateB\x02\x18\x01\x12U\n\x0e\x66\x65\x61ture_health\x18\x05 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionStateB\x02\x18\x01\x12K\n\x11\x66\x65\x61ture_set_state\x18\x06 \x03(\x0b\x32,.qwak.feature.store.features.FeatureSetStateB\x02\x18\x01\x12*\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x42\x02\x18\x01\x12!\n\x19\x66\x65\x61tureset_version_number\x18\x08 \x01(\x05\x12H\n\x10platform_details\x18\t \x01(\x0b\x32,.qwak.feature.store.platform.PlatformDetailsH\x00\x42\x1d\n\x1b\x66\x65\x61tureset_platform_details\"\xdb\x04\n\x0e\x46\x65\x61tureSetSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x33.qwak.feature.store.features.FeatureSetUserMetadata\x12\x16\n\ngit_commit\x18\x03 \x01(\tB\x02\x18\x01\x12=\n\x06\x65ntity\x18\x04 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x45\n\x10\x66\x65\x61ture_set_type\x18\x05 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetType\x12\x36\n\x08\x66\x65\x61tures\x18\x06 \x03(\x0b\x32$.qwak.feature.store.features.Feature\x12*\n\x1eqwak_internal_protocol_version\x18\x07 \x01(\x05\x42\x02\x18\x01\x12V\n\x10source_code_spec\x18\x08 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecB\x02\x18\x01\x12P\n\x15\x66\x65\x61tureset_attributes\x18\t \x01(\x0b\x32\x31.qwak.feature.store.features.FeaturesetAttributes\x12$\n\x1a\x66\x65\x61tureset_repository_name\x18\n \x01(\tH\x00\x42\"\n featureset_repository_identifier\"\xb8\x01\n\x16\x46\x65\x61tureSetUserMetadata\x12\r\n\x05owner\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x14\n\x0c\x64isplay_name\x18\x03 \x01(\t\x12\x64\n featureset_version_user_metadata\x18\x04 \x01(\x0b\x32:.qwak.feature.store.features.FeaturesetVersionUserMetadata\"0\n\x1d\x46\x65\x61turesetVersionUserMetadata\x12\x0f\n\x07\x63omment\x18\x01 \x01(\t\"\x9d\x02\n\x12\x46\x65\x61tureSetMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\x12[\n\x1b\x66\x65\x61tureset_version_metadata\x18\x05 \x01(\x0b\x32\x36.qwak.feature.store.features.FeaturesetVersionMetadata\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\t\"_\n\x19\x46\x65\x61turesetVersionMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\">\n\x16\x44\x65ployedModelInUseLink\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x12\n\nmodel_name\x18\x02 \x01(\t\"5\n\x07\x46\x65\x61ture\x12\x14\n\x0c\x66\x65\x61ture_name\x18\x01 \x01(\t\x12\x14\n\x0c\x66\x65\x61ture_type\x18\x02 \x01(\t\"\xa7\x01\n\x16\x44\x65ployedFeatureSetSpec\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x01 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x02 \x01(\t\x12\x12\n\naccount_id\x18\x03 \x01(\t\x12\x1a\n\x12\x66\x65\x61tureset_version\x18\x04 \x01(\x05\"O\n\x0e\x46\x65\x61tureSetList\x12=\n\x0c\x66\x65\x61ture_sets\x18\x01 \x03(\x0b\x32\'.qwak.feature.store.features.FeatureSet*\xb4\x01\n\rFeatureStatus\x12\x0b\n\x07INVALID\x10\x00\x12\t\n\x05VALID\x10\x01\x12\x0b\n\x07\x44\x45LETED\x10\x02\x12\x17\n\x13REGISTRATION_FAILED\x10\x03\x12\x1c\n\x18REGISTRATION_IN_PROGRESS\x10\x04\x12\x18\n\x14\x44\x45LETION_IN_PROGRESS\x10\x05\x12\x13\n\x0f\x44\x45LETION_FAILED\x10\x06\x12\x18\n\x14\x42\x41\x43KFILL_IN_PROGRESS\x10\x07*\x92\x01\n\x19\x46\x65\x61turesetSchedulingState\x12\x1c\n\x18SCHEDULING_STATE_INVALID\x10\x00\x12\x1c\n\x18SCHEDULING_STATE_ENABLED\x10\x01\x12\x1b\n\x17SCHEDULING_STATE_PAUSED\x10\x02\x12\x1c\n\x18SCHEDULING_STATE_UNKNOWN\x10\x03\x42[\n&com.qwak.ai.feature.store.features.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -44,10 +44,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _FEATURESETSPEC.fields_by_name['qwak_internal_protocol_version']._serialized_options = b'\030\001'
   _FEATURESETSPEC.fields_by_name['source_code_spec']._options = None
   _FEATURESETSPEC.fields_by_name['source_code_spec']._serialized_options = b'\030\001'
-  _globals['_FEATURESTATUS']._serialized_start=3234
-  _globals['_FEATURESTATUS']._serialized_end=3414
-  _globals['_FEATURESETSCHEDULINGSTATE']._serialized_start=3417
-  _globals['_FEATURESETSCHEDULINGSTATE']._serialized_end=3563
+  _globals['_FEATURESTATUS']._serialized_start=3258
+  _globals['_FEATURESTATUS']._serialized_end=3438
+  _globals['_FEATURESETSCHEDULINGSTATE']._serialized_start=3441
+  _globals['_FEATURESETSCHEDULINGSTATE']._serialized_end=3587
   _globals['_FEATURESET']._serialized_start=482
   _globals['_FEATURESET']._serialized_end=734
   _globals['_FEATURESETINFO']._serialized_start=737
@@ -63,15 +63,15 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_FEATURESETVERSIONUSERMETADATA']._serialized_start=2452
   _globals['_FEATURESETVERSIONUSERMETADATA']._serialized_end=2500
   _globals['_FEATURESETMETADATA']._serialized_start=2503
-  _globals['_FEATURESETMETADATA']._serialized_end=2764
-  _globals['_FEATURESETVERSIONMETADATA']._serialized_start=2766
-  _globals['_FEATURESETVERSIONMETADATA']._serialized_end=2861
-  _globals['_DEPLOYEDMODELINUSELINK']._serialized_start=2863
-  _globals['_DEPLOYEDMODELINUSELINK']._serialized_end=2925
-  _globals['_FEATURE']._serialized_start=2927
-  _globals['_FEATURE']._serialized_end=2980
-  _globals['_DEPLOYEDFEATURESETSPEC']._serialized_start=2983
-  _globals['_DEPLOYEDFEATURESETSPEC']._serialized_end=3150
-  _globals['_FEATURESETLIST']._serialized_start=3152
-  _globals['_FEATURESETLIST']._serialized_end=3231
+  _globals['_FEATURESETMETADATA']._serialized_end=2788
+  _globals['_FEATURESETVERSIONMETADATA']._serialized_start=2790
+  _globals['_FEATURESETVERSIONMETADATA']._serialized_end=2885
+  _globals['_DEPLOYEDMODELINUSELINK']._serialized_start=2887
+  _globals['_DEPLOYEDMODELINUSELINK']._serialized_end=2949
+  _globals['_FEATURE']._serialized_start=2951
+  _globals['_FEATURE']._serialized_end=3004
+  _globals['_DEPLOYEDFEATURESETSPEC']._serialized_start=3007
+  _globals['_DEPLOYEDFEATURESETSPEC']._serialized_end=3174
+  _globals['_FEATURESETLIST']._serialized_start=3176
+  _globals['_FEATURESETLIST']._serialized_end=3255
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/features/feature_set_pb2.pyi

@@ -312,6 +312,7 @@ class FeatureSetMetadata(google.protobuf.message.Message):
     LAST_MODIFIED_AT_FIELD_NUMBER: builtins.int
     LAST_MODIFIED_BY_FIELD_NUMBER: builtins.int
     FEATURESET_VERSION_METADATA_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
     @property
     def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """When the feature set was created"""
@@ -325,6 +326,8 @@ class FeatureSetMetadata(google.protobuf.message.Message):
     @property
     def featureset_version_metadata(self) -> global___FeaturesetVersionMetadata:
         """Featureset version metadata"""
+    environment_id: builtins.str
+    """Environment id"""
     def __init__(
         self,
         *,
@@ -333,9 +336,10 @@ class FeatureSetMetadata(google.protobuf.message.Message):
         last_modified_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         last_modified_by: builtins.str = ...,
         featureset_version_metadata: global___FeaturesetVersionMetadata | None = ...,
+        environment_id: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "featureset_version_metadata", b"featureset_version_metadata", "last_modified_at", b"last_modified_at"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "featureset_version_metadata", b"featureset_version_metadata", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "environment_id", b"environment_id", "featureset_version_metadata", b"featureset_version_metadata", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
 
 global___FeatureSetMetadata = FeatureSetMetadata
 

frogml_core-0.0.121/frogml/_proto/qwak/feature_store/features/feature_set_service_pb2.py

@@ -0,0 +1,126 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.feature_store/features/feature_set_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.feature_store.features import feature_set_state_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__state__pb2
+from frogml._proto.qwak.feature_store.features import feature_set_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__pb2
+from frogml._proto.qwak.feature_store.features import feature_set_types_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__types__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5qwak/feature_store/features/feature_set_service.proto\x12\x1bqwak.feature.store.features\x1a\x33qwak/feature_store/features/feature_set_state.proto\x1a-qwak/feature_store/features/feature_set.proto\x1a\x33qwak/feature_store/features/feature_set_types.proto\"b\n\x19RegisterFeatureSetRequest\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x01 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\"Z\n\x1aRegisterFeatureSetResponse\x12<\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"x\n\x17UpdateFeatureSetRequest\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x45\n\x10\x66\x65\x61ture_set_spec\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetSpec\"X\n\x18UpdateFeatureSetResponse\x12<\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"2\n\x18GetFeatureSetByIdRequest\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\"Y\n\x19GetFeatureSetByIdResponse\x12<\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"6\n\x1aGetFeatureSetByNameRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\"[\n\x1bGetFeatureSetByNameResponse\x12<\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.features.FeatureSet\">\n#ListFeaturesetVersionsByNameRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"h\n$ListFeaturesetVersionsByNameResponse\x12@\n\x0b\x66\x65\x61turesets\x18\x01 \x03(\x0b\x32+.qwak.feature.store.features.FeaturesetInfo\"#\n!GetEnvToFeatureSetsMappingRequest\"\x91\x02\n\"GetEnvToFeatureSetsMappingResponse\x12\x7f\n\x1a\x65nv_to_feature_set_mapping\x18\x01 \x03(\x0b\x32[.qwak.feature.store.features.GetEnvToFeatureSetsMappingResponse.EnvToFeatureSetMappingEntry\x1aj\n\x1b\x45nvToFeatureSetMappingEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12:\n\x05value\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.FeatureSetList:\x02\x38\x01\"1\n\x17\x44\x65leteFeatureSetRequest\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\"\x1a\n\x18\x44\x65leteFeatureSetResponse\"\\\n\x1e\x44\x65leteFeaturesetVersionRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12!\n\x19\x66\x65\x61tureset_version_number\x18\x02 \x01(\x05\"!\n\x1f\x44\x65leteFeaturesetVersionResponse\"\x18\n\x16ListFeatureSetsRequest\"\\\n\x17ListFeatureSetsResponse\x12\x41\n\x10\x66\x65\x61ture_families\x18\x01 \x03(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"!\n\x1fListFeaturesetsByAccountRequest\"`\n ListFeaturesetsByAccountResponse\x12<\n\x0b\x66\x65\x61turesets\x18\x01 \x03(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"5\n\x19RunBatchFeatureSetRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\"2\n\x1aRunBatchFeatureSetResponse\x12\x14\n\x0c\x65xecution_id\x18\x01 \x01(\t\"p\n RunParquetBatchFeatureSetRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\x12\x18\n\x10parquet_location\x18\x02 \x01(\t\x12\x18\n\x10timestamp_column\x18\x03 \x01(\t\"#\n!RunParquetBatchFeatureSetResponse\"2\n\x16\x43reateUploadURLRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\"x\n\x18\x43reateUploadURLRequestV1\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12\x43\n\x0etransformation\x18\x02 \x01(\x0b\x32+.qwak.feature.store.features.Transformation\"I\n.CreateRealTimeFeatureExtractorUploadURLRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"-\n\x17\x43reateUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\"\x89\x02\n\x1cUpdateFeatureSetStateRequest\x12m\n\x13\x66\x65\x61ture_names_state\x18\x01 \x03(\x0b\x32P.qwak.feature.store.features.UpdateFeatureSetStateRequest.FeatureNamesStateEntry\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\x1a\x62\n\x16\x46\x65\x61tureNamesStateEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32(.qwak.feature.store.features.StateUpdate:\x02\x38\x01\"e\n\x0bStateUpdate\x12V\n\x13\x66\x65\x61ture_names_state\x18\x01 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionState\"\x1f\n\x1dUpdateFeatureSetStateResponse\"K\n\x1fGetFeatureSetStateByNameRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\x03\"x\n GetFeatureSetStateByNameResponse\x12T\n\x11\x66\x65\x61ture_set_state\x18\x01 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionState\"9\n\x1dGetFeatureSetStateByIdRequest\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\"^\n\x1eGetFeatureSetStateByIdResponse\x12<\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.features.FeatureSet\"1\n\x16PauseFeaturesetRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\x19\n\x17PauseFeaturesetResponse\"2\n\x17ResumeFeaturesetRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\x1a\n\x18ResumeFeaturesetResponse\">\n#GetFeaturesetSchedulingStateRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"m\n$GetFeaturesetSchedulingStateResponse\x12\x45\n\x05state\x18\x01 \x01(\x0e\x32\x36.qwak.feature.store.features.FeaturesetSchedulingState\"_\n!SetActiveFeaturesetVersionRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12!\n\x19\x66\x65\x61tureset_version_number\x18\x02 \x01(\x05\"$\n\"SetActiveFeaturesetVersionResponse2\x81\x19\n\x11\x46\x65\x61tureSetService\x12\x85\x01\n\x12RegisterFeatureSet\x12\x36.qwak.feature.store.features.RegisterFeatureSetRequest\x1a\x37.qwak.feature.store.features.RegisterFeatureSetResponse\x12\x7f\n\x10UpdateFeatureSet\x12\x34.qwak.feature.store.features.UpdateFeatureSetRequest\x1a\x35.qwak.feature.store.features.UpdateFeatureSetResponse\x12\x82\x01\n\x11GetFeatureSetById\x12\x35.qwak.feature.store.features.GetFeatureSetByIdRequest\x1a\x36.qwak.feature.store.features.GetFeatureSetByIdResponse\x12\x88\x01\n\x13GetFeatureSetByName\x12\x37.qwak.feature.store.features.GetFeatureSetByNameRequest\x1a\x38.qwak.feature.store.features.GetFeatureSetByNameResponse\x12\xa3\x01\n\x1cListFeaturesetVersionsByName\x12@.qwak.feature.store.features.ListFeaturesetVersionsByNameRequest\x1a\x41.qwak.feature.store.features.ListFeaturesetVersionsByNameResponse\x12\x7f\n\x10\x44\x65leteFeatureSet\x12\x34.qwak.feature.store.features.DeleteFeatureSetRequest\x1a\x35.qwak.feature.store.features.DeleteFeatureSetResponse\x12\x94\x01\n\x17\x44\x65leteFeaturesetVersion\x12;.qwak.feature.store.features.DeleteFeaturesetVersionRequest\x1a<.qwak.feature.store.features.DeleteFeaturesetVersionResponse\x12|\n\x0fListFeatureSets\x12\x33.qwak.feature.store.features.ListFeatureSetsRequest\x1a\x34.qwak.feature.store.features.ListFeatureSetsResponse\x12\x97\x01\n\x18ListFeaturesetsByAccount\x12<.qwak.feature.store.features.ListFeaturesetsByAccountRequest\x1a=.qwak.feature.store.features.ListFeaturesetsByAccountResponse\x12\x85\x01\n\x12RunBatchFeatureSet\x12\x36.qwak.feature.store.features.RunBatchFeatureSetRequest\x1a\x37.qwak.feature.store.features.RunBatchFeatureSetResponse\x12\x9a\x01\n\x19RunParquetBatchFeatureSet\x12=.qwak.feature.store.features.RunParquetBatchFeatureSetRequest\x1a>.qwak.feature.store.features.RunParquetBatchFeatureSetResponse\x12\x81\x01\n\x0f\x43reateUploadURL\x12\x33.qwak.feature.store.features.CreateUploadURLRequest\x1a\x34.qwak.feature.store.features.CreateUploadURLResponse\"\x03\x88\x02\x01\x12\x80\x01\n\x11\x43reateUploadURLV1\x12\x35.qwak.feature.store.features.CreateUploadURLRequestV1\x1a\x34.qwak.feature.store.features.CreateUploadURLResponse\x12\xac\x01\n\'CreateRealTimeFeatureExtractorUploadURL\x12K.qwak.feature.store.features.CreateRealTimeFeatureExtractorUploadURLRequest\x1a\x34.qwak.feature.store.features.CreateUploadURLResponse\x12\x8e\x01\n\x15UpdateFeatureSetState\x12\x39.qwak.feature.store.features.UpdateFeatureSetStateRequest\x1a:.qwak.feature.store.features.UpdateFeatureSetStateResponse\x12\x97\x01\n\x18GetFeatureSetStateByName\x12<.qwak.feature.store.features.GetFeatureSetStateByNameRequest\x1a=.qwak.feature.store.features.GetFeatureSetStateByNameResponse\x12\x9d\x01\n\x1aGetEnvToFeatureSetsMapping\x12>.qwak.feature.store.features.GetEnvToFeatureSetsMappingRequest\x1a?.qwak.feature.store.features.GetEnvToFeatureSetsMappingResponse\x12\x91\x01\n\x16GetFeatureSetStateById\x12:.qwak.feature.store.features.GetFeatureSetStateByIdRequest\x1a;.qwak.feature.store.features.GetFeatureSetStateByIdResponse\x12|\n\x0fPauseFeatureset\x12\x33.qwak.feature.store.features.PauseFeaturesetRequest\x1a\x34.qwak.feature.store.features.PauseFeaturesetResponse\x12\x7f\n\x10ResumeFeatureset\x12\x34.qwak.feature.store.features.ResumeFeaturesetRequest\x1a\x35.qwak.feature.store.features.ResumeFeaturesetResponse\x12\xa3\x01\n\x1cGetFeaturesetSchedulingState\x12@.qwak.feature.store.features.GetFeaturesetSchedulingStateRequest\x1a\x41.qwak.feature.store.features.GetFeaturesetSchedulingStateResponse\x12\x9d\x01\n\x1aSetActiveFeaturesetVersion\x12>.qwak.feature.store.features.SetActiveFeaturesetVersionRequest\x1a?.qwak.feature.store.features.SetActiveFeaturesetVersionResponseB]\n(com.qwak.ai.feature.store.management.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.features.feature_set_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n(com.qwak.ai.feature.store.management.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
+  _GETENVTOFEATURESETSMAPPINGRESPONSE_ENVTOFEATURESETMAPPINGENTRY._options = None
+  _GETENVTOFEATURESETSMAPPINGRESPONSE_ENVTOFEATURESETMAPPINGENTRY._serialized_options = b'8\001'
+  _UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY._options = None
+  _UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY._serialized_options = b'8\001'
+  _FEATURESETSERVICE.methods_by_name['CreateUploadURL']._options = None
+  _FEATURESETSERVICE.methods_by_name['CreateUploadURL']._serialized_options = b'\210\002\001'
+  _globals['_REGISTERFEATURESETREQUEST']._serialized_start=239
+  _globals['_REGISTERFEATURESETREQUEST']._serialized_end=337
+  _globals['_REGISTERFEATURESETRESPONSE']._serialized_start=339
+  _globals['_REGISTERFEATURESETRESPONSE']._serialized_end=429
+  _globals['_UPDATEFEATURESETREQUEST']._serialized_start=431
+  _globals['_UPDATEFEATURESETREQUEST']._serialized_end=551
+  _globals['_UPDATEFEATURESETRESPONSE']._serialized_start=553
+  _globals['_UPDATEFEATURESETRESPONSE']._serialized_end=641
+  _globals['_GETFEATURESETBYIDREQUEST']._serialized_start=643
+  _globals['_GETFEATURESETBYIDREQUEST']._serialized_end=693
+  _globals['_GETFEATURESETBYIDRESPONSE']._serialized_start=695
+  _globals['_GETFEATURESETBYIDRESPONSE']._serialized_end=784
+  _globals['_GETFEATURESETBYNAMEREQUEST']._serialized_start=786
+  _globals['_GETFEATURESETBYNAMEREQUEST']._serialized_end=840
+  _globals['_GETFEATURESETBYNAMERESPONSE']._serialized_start=842
+  _globals['_GETFEATURESETBYNAMERESPONSE']._serialized_end=933
+  _globals['_LISTFEATURESETVERSIONSBYNAMEREQUEST']._serialized_start=935
+  _globals['_LISTFEATURESETVERSIONSBYNAMEREQUEST']._serialized_end=997
+  _globals['_LISTFEATURESETVERSIONSBYNAMERESPONSE']._serialized_start=999
+  _globals['_LISTFEATURESETVERSIONSBYNAMERESPONSE']._serialized_end=1103
+  _globals['_GETENVTOFEATURESETSMAPPINGREQUEST']._serialized_start=1105
+  _globals['_GETENVTOFEATURESETSMAPPINGREQUEST']._serialized_end=1140
+  _globals['_GETENVTOFEATURESETSMAPPINGRESPONSE']._serialized_start=1143
+  _globals['_GETENVTOFEATURESETSMAPPINGRESPONSE']._serialized_end=1416
+  _globals['_GETENVTOFEATURESETSMAPPINGRESPONSE_ENVTOFEATURESETMAPPINGENTRY']._serialized_start=1310
+  _globals['_GETENVTOFEATURESETSMAPPINGRESPONSE_ENVTOFEATURESETMAPPINGENTRY']._serialized_end=1416
+  _globals['_DELETEFEATURESETREQUEST']._serialized_start=1418
+  _globals['_DELETEFEATURESETREQUEST']._serialized_end=1467
+  _globals['_DELETEFEATURESETRESPONSE']._serialized_start=1469
+  _globals['_DELETEFEATURESETRESPONSE']._serialized_end=1495
+  _globals['_DELETEFEATURESETVERSIONREQUEST']._serialized_start=1497
+  _globals['_DELETEFEATURESETVERSIONREQUEST']._serialized_end=1589
+  _globals['_DELETEFEATURESETVERSIONRESPONSE']._serialized_start=1591
+  _globals['_DELETEFEATURESETVERSIONRESPONSE']._serialized_end=1624
+  _globals['_LISTFEATURESETSREQUEST']._serialized_start=1626
+  _globals['_LISTFEATURESETSREQUEST']._serialized_end=1650
+  _globals['_LISTFEATURESETSRESPONSE']._serialized_start=1652
+  _globals['_LISTFEATURESETSRESPONSE']._serialized_end=1744
+  _globals['_LISTFEATURESETSBYACCOUNTREQUEST']._serialized_start=1746
+  _globals['_LISTFEATURESETSBYACCOUNTREQUEST']._serialized_end=1779
+  _globals['_LISTFEATURESETSBYACCOUNTRESPONSE']._serialized_start=1781
+  _globals['_LISTFEATURESETSBYACCOUNTRESPONSE']._serialized_end=1877
+  _globals['_RUNBATCHFEATURESETREQUEST']._serialized_start=1879
+  _globals['_RUNBATCHFEATURESETREQUEST']._serialized_end=1932
+  _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_start=1934
+  _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_end=1984
+  _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_start=1986
+  _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_end=2098
+  _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_start=2100
+  _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_end=2135
+  _globals['_CREATEUPLOADURLREQUEST']._serialized_start=2137
+  _globals['_CREATEUPLOADURLREQUEST']._serialized_end=2187
+  _globals['_CREATEUPLOADURLREQUESTV1']._serialized_start=2189
+  _globals['_CREATEUPLOADURLREQUESTV1']._serialized_end=2309
+  _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_start=2311
+  _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_end=2384
+  _globals['_CREATEUPLOADURLRESPONSE']._serialized_start=2386
+  _globals['_CREATEUPLOADURLRESPONSE']._serialized_end=2431
+  _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_start=2434
+  _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_end=2699
+  _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_start=2601
+  _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_end=2699
+  _globals['_STATEUPDATE']._serialized_start=2701
+  _globals['_STATEUPDATE']._serialized_end=2802
+  _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_start=2804
+  _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_end=2835
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_start=2837
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_end=2912
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_start=2914
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_end=3034
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_start=3036
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_end=3093
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_start=3095
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_end=3189
+  _globals['_PAUSEFEATURESETREQUEST']._serialized_start=3191
+  _globals['_PAUSEFEATURESETREQUEST']._serialized_end=3240
+  _globals['_PAUSEFEATURESETRESPONSE']._serialized_start=3242
+  _globals['_PAUSEFEATURESETRESPONSE']._serialized_end=3267
+  _globals['_RESUMEFEATURESETREQUEST']._serialized_start=3269
+  _globals['_RESUMEFEATURESETREQUEST']._serialized_end=3319
+  _globals['_RESUMEFEATURESETRESPONSE']._serialized_start=3321
+  _globals['_RESUMEFEATURESETRESPONSE']._serialized_end=3347
+  _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_start=3349
+  _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_end=3411
+  _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_start=3413
+  _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_end=3522
+  _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_start=3524
+  _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_end=3619
+  _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_start=3621
+  _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_end=3657
+  _globals['_FEATURESETSERVICE']._serialized_start=3660
+  _globals['_FEATURESETSERVICE']._serialized_end=6861
+# @@protoc_insertion_point(module_scope)

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/features/feature_set_service_pb2.pyi

@@ -305,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
 

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py

@@ -54,6 +54,11 @@ class FeatureSetServiceStub(object):
                 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,
@@ -180,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
         """
@@ -313,6 +325,11 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
                     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,
@@ -524,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,

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/sources/data_source_pb2.py

@@ -18,7 +18,7 @@ from frogml._proto.qwak.feature_store.sources import data_source_attribute_pb2 a
 from frogml._proto.qwak.feature_store.v1.common.source_code import source_code_pb2 as qwak_dot_feature__store_dot_v1_dot_common_dot_source__code_dot_source__code__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/feature_store/sources/data_source.proto\x12\x1aqwak.feature.store.sources\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\x1a\x36qwak/feature_store/sources/data_source_attribute.proto\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\"\xe3\x01\n\nDataSource\x12P\n\x16\x64\x61ta_source_definition\x18\x01 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceDefinition\x12@\n\x08metadata\x18\x02 \x01(\x0b\x32..qwak.feature.store.sources.DataSourceMetadata\x12\x41\n\x0c\x66\x65\x61ture_sets\x18\x03 \x03(\x0b\x32+.qwak.feature.store.sources.FeatureSetBrief\"t\n\x14\x44\x61taSourceDefinition\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"\xa8\x01\n\x12\x44\x61taSourceMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\"\xe3\x03\n\x0e\x44\x61taSourceSpec\x12?\n\x0c\x62\x61tch_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSourceH\x00\x12\x44\n\rstream_source\x18\x02 \x01(\x0b\x32+.qwak.feature.store.sources.StreamingSourceH\x00\x12V\n\x10source_code_spec\x18\x03 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecB\x02\x18\x01\x12P\n\x16\x64\x61ta_source_attributes\x18\x04 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceAttributes\x12(\n\x1a\x66\x65\x61tureset_repository_name\x18\x05 \x01(\tB\x02\x18\x01H\x01\x12%\n\x1b\x64\x61ta_source_repository_name\x18\x06 \x01(\tH\x02\x42\x06\n\x04typeB\"\n featureset_repository_identifierB#\n!data_source_repository_identifier\"]\n\x0f\x46\x65\x61tureSetBrief\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_type\x18\x03 \x01(\x05\x42X\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/feature_store/sources/data_source.proto\x12\x1aqwak.feature.store.sources\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\x1a\x36qwak/feature_store/sources/data_source_attribute.proto\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\"\xe3\x01\n\nDataSource\x12P\n\x16\x64\x61ta_source_definition\x18\x01 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceDefinition\x12@\n\x08metadata\x18\x02 \x01(\x0b\x32..qwak.feature.store.sources.DataSourceMetadata\x12\x41\n\x0c\x66\x65\x61ture_sets\x18\x03 \x03(\x0b\x32+.qwak.feature.store.sources.FeatureSetBrief\"t\n\x14\x44\x61taSourceDefinition\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"\xc0\x01\n\x12\x44\x61taSourceMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"\xe3\x03\n\x0e\x44\x61taSourceSpec\x12?\n\x0c\x62\x61tch_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSourceH\x00\x12\x44\n\rstream_source\x18\x02 \x01(\x0b\x32+.qwak.feature.store.sources.StreamingSourceH\x00\x12V\n\x10source_code_spec\x18\x03 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecB\x02\x18\x01\x12P\n\x16\x64\x61ta_source_attributes\x18\x04 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceAttributes\x12(\n\x1a\x66\x65\x61tureset_repository_name\x18\x05 \x01(\tB\x02\x18\x01H\x01\x12%\n\x1b\x64\x61ta_source_repository_name\x18\x06 \x01(\tH\x02\x42\x06\n\x04typeB\"\n featureset_repository_identifierB#\n!data_source_repository_identifier\"]\n\x0f\x46\x65\x61tureSetBrief\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_type\x18\x03 \x01(\x05\x42X\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -36,9 +36,9 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_DATASOURCEDEFINITION']._serialized_start=539
   _globals['_DATASOURCEDEFINITION']._serialized_end=655
   _globals['_DATASOURCEMETADATA']._serialized_start=658
-  _globals['_DATASOURCEMETADATA']._serialized_end=826
-  _globals['_DATASOURCESPEC']._serialized_start=829
-  _globals['_DATASOURCESPEC']._serialized_end=1312
-  _globals['_FEATURESETBRIEF']._serialized_start=1314
-  _globals['_FEATURESETBRIEF']._serialized_end=1407
+  _globals['_DATASOURCEMETADATA']._serialized_end=850
+  _globals['_DATASOURCESPEC']._serialized_start=853
+  _globals['_DATASOURCESPEC']._serialized_end=1336
+  _globals['_FEATURESETBRIEF']._serialized_start=1338
+  _globals['_FEATURESETBRIEF']._serialized_end=1431
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/sources/data_source_pb2.pyi

@@ -77,6 +77,7 @@ class DataSourceMetadata(google.protobuf.message.Message):
     CREATED_BY_FIELD_NUMBER: builtins.int
     LAST_MODIFIED_AT_FIELD_NUMBER: builtins.int
     LAST_MODIFIED_BY_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
     @property
     def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """When the data source was created"""
@@ -87,6 +88,8 @@ class DataSourceMetadata(google.protobuf.message.Message):
         """Last modified"""
     last_modified_by: builtins.str
     """Last modified by"""
+    environment_id: builtins.str
+    """Environment id"""
     def __init__(
         self,
         *,
@@ -94,9 +97,10 @@ class DataSourceMetadata(google.protobuf.message.Message):
         created_by: builtins.str = ...,
         last_modified_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         last_modified_by: builtins.str = ...,
+        environment_id: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "last_modified_at", b"last_modified_at"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "environment_id", b"environment_id", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
 
 global___DataSourceMetadata = DataSourceMetadata
 

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/sources/data_source_service_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 from frogml._proto.qwak.feature_store.sources import data_source_pb2 as qwak_dot_feature__store_dot_sources_dot_data__source__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4qwak/feature_store/sources/data_source_service.proto\x12\x1aqwak.feature.store.sources\x1a,qwak/feature_store/sources/data_source.proto\"_\n\x17\x43reateDataSourceRequest\x12\x44\n\x10\x64\x61ta_source_spec\x18\x01 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18\x43reateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"w\n\x17UpdateDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18UpdateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"1\n\x17\x44\x65leteDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"\x1a\n\x18\x44\x65leteDataSourceResponse\"2\n\x18GetDataSourceByIdRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"X\n\x19GetDataSourceByIdResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"6\n\x1aGetDataSourceByNameRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"Z\n\x1bGetDataSourceByNameResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"\x18\n\x16ListDataSourcesRequest\"W\n\x17ListDataSourcesResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"Q\n CreateDataSourceUploadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x13\n\x0bobject_name\x18\x02 \x01(\t\"7\n!CreateDataSourceUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xd5\x08\n\x11\x44\x61taSourceService\x12}\n\x10\x43reateDataSource\x12\x33.qwak.feature.store.sources.CreateDataSourceRequest\x1a\x34.qwak.feature.store.sources.CreateDataSourceResponse\x12}\n\x10UpdateDataSource\x12\x33.qwak.feature.store.sources.UpdateDataSourceRequest\x1a\x34.qwak.feature.store.sources.UpdateDataSourceResponse\x12}\n\x10\x44\x65leteDataSource\x12\x33.qwak.feature.store.sources.DeleteDataSourceRequest\x1a\x34.qwak.feature.store.sources.DeleteDataSourceResponse\x12\x80\x01\n\x11GetDataSourceById\x12\x34.qwak.feature.store.sources.GetDataSourceByIdRequest\x1a\x35.qwak.feature.store.sources.GetDataSourceByIdResponse\x12\x86\x01\n\x13GetDataSourceByName\x12\x36.qwak.feature.store.sources.GetDataSourceByNameRequest\x1a\x37.qwak.feature.store.sources.GetDataSourceByNameResponse\x12z\n\x0fListDataSources\x12\x32.qwak.feature.store.sources.ListDataSourcesRequest\x1a\x33.qwak.feature.store.sources.ListDataSourcesResponse\x12\x9d\x01\n\x19\x43reateDataSourceUploadURL\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1b\x43reateDataSourceUploadURLV1\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponseB[\n(com.qwak.ai.feature.store.management.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4qwak/feature_store/sources/data_source_service.proto\x12\x1aqwak.feature.store.sources\x1a,qwak/feature_store/sources/data_source.proto\"_\n\x17\x43reateDataSourceRequest\x12\x44\n\x10\x64\x61ta_source_spec\x18\x01 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18\x43reateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"w\n\x17UpdateDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18UpdateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"1\n\x17\x44\x65leteDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"\x1a\n\x18\x44\x65leteDataSourceResponse\"2\n\x18GetDataSourceByIdRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"X\n\x19GetDataSourceByIdResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"6\n\x1aGetDataSourceByNameRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"Z\n\x1bGetDataSourceByNameResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"\x18\n\x16ListDataSourcesRequest\"W\n\x17ListDataSourcesResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"!\n\x1fListDataSourcesByAccountRequest\"`\n ListDataSourcesByAccountResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"Q\n CreateDataSourceUploadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x13\n\x0bobject_name\x18\x02 \x01(\t\"7\n!CreateDataSourceUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xed\t\n\x11\x44\x61taSourceService\x12}\n\x10\x43reateDataSource\x12\x33.qwak.feature.store.sources.CreateDataSourceRequest\x1a\x34.qwak.feature.store.sources.CreateDataSourceResponse\x12}\n\x10UpdateDataSource\x12\x33.qwak.feature.store.sources.UpdateDataSourceRequest\x1a\x34.qwak.feature.store.sources.UpdateDataSourceResponse\x12}\n\x10\x44\x65leteDataSource\x12\x33.qwak.feature.store.sources.DeleteDataSourceRequest\x1a\x34.qwak.feature.store.sources.DeleteDataSourceResponse\x12\x80\x01\n\x11GetDataSourceById\x12\x34.qwak.feature.store.sources.GetDataSourceByIdRequest\x1a\x35.qwak.feature.store.sources.GetDataSourceByIdResponse\x12\x86\x01\n\x13GetDataSourceByName\x12\x36.qwak.feature.store.sources.GetDataSourceByNameRequest\x1a\x37.qwak.feature.store.sources.GetDataSourceByNameResponse\x12z\n\x0fListDataSources\x12\x32.qwak.feature.store.sources.ListDataSourcesRequest\x1a\x33.qwak.feature.store.sources.ListDataSourcesResponse\x12\x95\x01\n\x18ListDataSourcesByAccount\x12;.qwak.feature.store.sources.ListDataSourcesByAccountRequest\x1a<.qwak.feature.store.sources.ListDataSourcesByAccountResponse\x12\x9d\x01\n\x19\x43reateDataSourceUploadURL\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1b\x43reateDataSourceUploadURLV1\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponseB[\n(com.qwak.ai.feature.store.management.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -49,10 +49,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_LISTDATASOURCESREQUEST']._serialized_end=919
   _globals['_LISTDATASOURCESRESPONSE']._serialized_start=921
   _globals['_LISTDATASOURCESRESPONSE']._serialized_end=1008
-  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_start=1010
-  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_end=1091
-  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_start=1093
-  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_end=1148
-  _globals['_DATASOURCESERVICE']._serialized_start=1151
-  _globals['_DATASOURCESERVICE']._serialized_end=2260
+  _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_start=1010
+  _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_end=1043
+  _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_start=1045
+  _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_end=1141
+  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_start=1143
+  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_end=1224
+  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_start=1226
+  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_end=1281
+  _globals['_DATASOURCESERVICE']._serialized_start=1284
+  _globals['_DATASOURCESERVICE']._serialized_end=2545
 # @@protoc_insertion_point(module_scope)

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/sources/data_source_service_pb2.pyi

@@ -202,6 +202,31 @@ class ListDataSourcesResponse(google.protobuf.message.Message):
 
 global___ListDataSourcesResponse = ListDataSourcesResponse
 
+class ListDataSourcesByAccountRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListDataSourcesByAccountRequest = ListDataSourcesByAccountRequest
+
+class ListDataSourcesByAccountResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_SOURCES_FIELD_NUMBER: builtins.int
+    @property
+    def data_sources(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.sources.data_source_pb2.DataSource]:
+        """List of registered data sources"""
+    def __init__(
+        self,
+        *,
+        data_sources: collections.abc.Iterable[qwak.feature_store.sources.data_source_pb2.DataSource] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_sources", b"data_sources"]) -> None: ...
+
+global___ListDataSourcesByAccountResponse = ListDataSourcesByAccountResponse
+
 class CreateDataSourceUploadURLRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py

@@ -44,6 +44,11 @@ class DataSourceServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.FromString,
                 )
+        self.ListDataSourcesByAccount = channel.unary_unary(
+                '/qwak.feature.store.sources.DataSourceService/ListDataSourcesByAccount',
+                request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
+                )
         self.CreateDataSourceUploadURL = channel.unary_unary(
                 '/qwak.feature.store.sources.DataSourceService/CreateDataSourceUploadURL',
                 request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.SerializeToString,
@@ -101,6 +106,13 @@ class DataSourceServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListDataSourcesByAccount(self, request, context):
+        """List of registered data sources
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def CreateDataSourceUploadURL(self, request, context):
         """Create a url for uploading data source configuration
         """
@@ -148,6 +160,11 @@ def add_DataSourceServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.SerializeToString,
             ),
+            'ListDataSourcesByAccount': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListDataSourcesByAccount,
+                    request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.SerializeToString,
+            ),
             'CreateDataSourceUploadURL': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateDataSourceUploadURL,
                     request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.FromString,
@@ -270,6 +287,23 @@ class DataSourceService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListDataSourcesByAccount(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.sources.DataSourceService/ListDataSourcesByAccount',
+            qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
+            qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateDataSourceUploadURL(request,
             target,

{frogml_core-0.0.119 → frogml_core-0.0.121}/frogml/core/clients/logging_client/client.py

@@ -1,7 +1,5 @@
 from typing import Optional
 
-from grpc import RpcError
-
 from frogml._proto.qwak.logging.log_filter_pb2 import LogText, SearchFilter
 from frogml._proto.qwak.logging.log_reader_service_pb2 import (
     ReadLogsRequest,
@@ -17,6 +15,7 @@ from frogml._proto.qwak.logging.log_source_pb2 import (
 from frogml.core.clients.administration.eco_system.client import EcosystemClient
 from frogml.core.exceptions import FrogmlException
 from frogml.core.inner.tool.grpc.grpc_tools import create_grpc_channel
+from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
 
 
 class LoggingClient:
@@ -120,6 +119,7 @@ class LoggingClient:
         except FrogmlException as e:
             raise FrogmlException(f"Failed to fetch execution logs, error is [{e}]")
 
+    @grpc_try_catch_wrapper("Failed to read logs request")
     def read_logs(
         self,
         source,
@@ -127,22 +127,16 @@ class LoggingClient:
         after_offset,
         max_number_of_results,
         log_text_filter,
-    ):
-        try:
-            response = self._logging_service.ReadLogs(
-                ReadLogsRequest(
-                    source=source,
-                    before_offset=before_offset,
-                    after_offset=after_offset,
-                    search_filter=SearchFilter(
-                        log_text_filter=LogText(contains=log_text_filter)
-                    ),
-                    max_number_of_results=max_number_of_results,
-                )
-            )
-            return response
-        except RpcError as e:
-            raise FrogmlException(
-                f"Failed grpc read logs request, grpc error is "
-                f"[{e.details() if e.details() else e.code()}]"
+    ) -> ReadLogsResponse:
+        response: ReadLogsResponse = self._logging_service.ReadLogs(
+            ReadLogsRequest(
+                source=source,
+                before_offset=before_offset,
+                after_offset=after_offset,
+                search_filter=SearchFilter(
+                    log_text_filter=LogText(contains=log_text_filter)
+                ),
+                max_number_of_results=max_number_of_results,
             )
+        )
+        return response