frogml-core 0.0.119__py3-none-any.whl → 0.0.121__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -21,7 +21,7 @@ from frogml._proto.qwak.feature_store.repository.common import platform_pb2 as q
21
21
  from frogml._proto.qwak.feature_store.platform import platform_details_pb2 as qwak_dot_feature__store_dot_platform_dot_platform__details__pb2
22
22
 
23
23
 
24
- 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')
24
+ 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')
25
25
 
26
26
  _globals = globals()
27
27
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -44,10 +44,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
44
44
  _FEATURESETSPEC.fields_by_name['qwak_internal_protocol_version']._serialized_options = b'\030\001'
45
45
  _FEATURESETSPEC.fields_by_name['source_code_spec']._options = None
46
46
  _FEATURESETSPEC.fields_by_name['source_code_spec']._serialized_options = b'\030\001'
47
- _globals['_FEATURESTATUS']._serialized_start=3234
48
- _globals['_FEATURESTATUS']._serialized_end=3414
49
- _globals['_FEATURESETSCHEDULINGSTATE']._serialized_start=3417
50
- _globals['_FEATURESETSCHEDULINGSTATE']._serialized_end=3563
47
+ _globals['_FEATURESTATUS']._serialized_start=3258
48
+ _globals['_FEATURESTATUS']._serialized_end=3438
49
+ _globals['_FEATURESETSCHEDULINGSTATE']._serialized_start=3441
50
+ _globals['_FEATURESETSCHEDULINGSTATE']._serialized_end=3587
51
51
  _globals['_FEATURESET']._serialized_start=482
52
52
  _globals['_FEATURESET']._serialized_end=734
53
53
  _globals['_FEATURESETINFO']._serialized_start=737
@@ -63,15 +63,15 @@ if _descriptor._USE_C_DESCRIPTORS == False:
63
63
  _globals['_FEATURESETVERSIONUSERMETADATA']._serialized_start=2452
64
64
  _globals['_FEATURESETVERSIONUSERMETADATA']._serialized_end=2500
65
65
  _globals['_FEATURESETMETADATA']._serialized_start=2503
66
- _globals['_FEATURESETMETADATA']._serialized_end=2764
67
- _globals['_FEATURESETVERSIONMETADATA']._serialized_start=2766
68
- _globals['_FEATURESETVERSIONMETADATA']._serialized_end=2861
69
- _globals['_DEPLOYEDMODELINUSELINK']._serialized_start=2863
70
- _globals['_DEPLOYEDMODELINUSELINK']._serialized_end=2925
71
- _globals['_FEATURE']._serialized_start=2927
72
- _globals['_FEATURE']._serialized_end=2980
73
- _globals['_DEPLOYEDFEATURESETSPEC']._serialized_start=2983
74
- _globals['_DEPLOYEDFEATURESETSPEC']._serialized_end=3150
75
- _globals['_FEATURESETLIST']._serialized_start=3152
76
- _globals['_FEATURESETLIST']._serialized_end=3231
66
+ _globals['_FEATURESETMETADATA']._serialized_end=2788
67
+ _globals['_FEATURESETVERSIONMETADATA']._serialized_start=2790
68
+ _globals['_FEATURESETVERSIONMETADATA']._serialized_end=2885
69
+ _globals['_DEPLOYEDMODELINUSELINK']._serialized_start=2887
70
+ _globals['_DEPLOYEDMODELINUSELINK']._serialized_end=2949
71
+ _globals['_FEATURE']._serialized_start=2951
72
+ _globals['_FEATURE']._serialized_end=3004
73
+ _globals['_DEPLOYEDFEATURESETSPEC']._serialized_start=3007
74
+ _globals['_DEPLOYEDFEATURESETSPEC']._serialized_end=3174
75
+ _globals['_FEATURESETLIST']._serialized_start=3176
76
+ _globals['_FEATURESETLIST']._serialized_end=3255
77
77
  # @@protoc_insertion_point(module_scope)
@@ -312,6 +312,7 @@ class FeatureSetMetadata(google.protobuf.message.Message):
312
312
  LAST_MODIFIED_AT_FIELD_NUMBER: builtins.int
313
313
  LAST_MODIFIED_BY_FIELD_NUMBER: builtins.int
314
314
  FEATURESET_VERSION_METADATA_FIELD_NUMBER: builtins.int
315
+ ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
315
316
  @property
316
317
  def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
317
318
  """When the feature set was created"""
@@ -325,6 +326,8 @@ class FeatureSetMetadata(google.protobuf.message.Message):
325
326
  @property
326
327
  def featureset_version_metadata(self) -> global___FeaturesetVersionMetadata:
327
328
  """Featureset version metadata"""
329
+ environment_id: builtins.str
330
+ """Environment id"""
328
331
  def __init__(
329
332
  self,
330
333
  *,
@@ -333,9 +336,10 @@ class FeatureSetMetadata(google.protobuf.message.Message):
333
336
  last_modified_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
334
337
  last_modified_by: builtins.str = ...,
335
338
  featureset_version_metadata: global___FeaturesetVersionMetadata | None = ...,
339
+ environment_id: builtins.str = ...,
336
340
  ) -> None: ...
337
341
  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: ...
338
- 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: ...
342
+ 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: ...
339
343
 
340
344
  global___FeatureSetMetadata = FeatureSetMetadata
341
345
 
@@ -16,7 +16,7 @@ from frogml._proto.qwak.feature_store.features import feature_set_pb2 as qwak_do
16
16
  from frogml._proto.qwak.feature_store.features import feature_set_types_pb2 as qwak_dot_feature__store_dot_features_dot_feature__set__types__pb2
17
17
 
18
18
 
19
- 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\"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\xe7\x17\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\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')
19
+ 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')
20
20
 
21
21
  _globals = globals()
22
22
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -69,54 +69,58 @@ if _descriptor._USE_C_DESCRIPTORS == False:
69
69
  _globals['_LISTFEATURESETSREQUEST']._serialized_end=1650
70
70
  _globals['_LISTFEATURESETSRESPONSE']._serialized_start=1652
71
71
  _globals['_LISTFEATURESETSRESPONSE']._serialized_end=1744
72
- _globals['_RUNBATCHFEATURESETREQUEST']._serialized_start=1746
73
- _globals['_RUNBATCHFEATURESETREQUEST']._serialized_end=1799
74
- _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_start=1801
75
- _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_end=1851
76
- _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_start=1853
77
- _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_end=1965
78
- _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_start=1967
79
- _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_end=2002
80
- _globals['_CREATEUPLOADURLREQUEST']._serialized_start=2004
81
- _globals['_CREATEUPLOADURLREQUEST']._serialized_end=2054
82
- _globals['_CREATEUPLOADURLREQUESTV1']._serialized_start=2056
83
- _globals['_CREATEUPLOADURLREQUESTV1']._serialized_end=2176
84
- _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_start=2178
85
- _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_end=2251
86
- _globals['_CREATEUPLOADURLRESPONSE']._serialized_start=2253
87
- _globals['_CREATEUPLOADURLRESPONSE']._serialized_end=2298
88
- _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_start=2301
89
- _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_end=2566
90
- _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_start=2468
91
- _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_end=2566
92
- _globals['_STATEUPDATE']._serialized_start=2568
93
- _globals['_STATEUPDATE']._serialized_end=2669
94
- _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_start=2671
95
- _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_end=2702
96
- _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_start=2704
97
- _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_end=2779
98
- _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_start=2781
99
- _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_end=2901
100
- _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_start=2903
101
- _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_end=2960
102
- _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_start=2962
103
- _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_end=3056
104
- _globals['_PAUSEFEATURESETREQUEST']._serialized_start=3058
105
- _globals['_PAUSEFEATURESETREQUEST']._serialized_end=3107
106
- _globals['_PAUSEFEATURESETRESPONSE']._serialized_start=3109
107
- _globals['_PAUSEFEATURESETRESPONSE']._serialized_end=3134
108
- _globals['_RESUMEFEATURESETREQUEST']._serialized_start=3136
109
- _globals['_RESUMEFEATURESETREQUEST']._serialized_end=3186
110
- _globals['_RESUMEFEATURESETRESPONSE']._serialized_start=3188
111
- _globals['_RESUMEFEATURESETRESPONSE']._serialized_end=3214
112
- _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_start=3216
113
- _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_end=3278
114
- _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_start=3280
115
- _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_end=3389
116
- _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_start=3391
117
- _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_end=3486
118
- _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_start=3488
119
- _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_end=3524
120
- _globals['_FEATURESETSERVICE']._serialized_start=3527
121
- _globals['_FEATURESETSERVICE']._serialized_end=6574
72
+ _globals['_LISTFEATURESETSBYACCOUNTREQUEST']._serialized_start=1746
73
+ _globals['_LISTFEATURESETSBYACCOUNTREQUEST']._serialized_end=1779
74
+ _globals['_LISTFEATURESETSBYACCOUNTRESPONSE']._serialized_start=1781
75
+ _globals['_LISTFEATURESETSBYACCOUNTRESPONSE']._serialized_end=1877
76
+ _globals['_RUNBATCHFEATURESETREQUEST']._serialized_start=1879
77
+ _globals['_RUNBATCHFEATURESETREQUEST']._serialized_end=1932
78
+ _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_start=1934
79
+ _globals['_RUNBATCHFEATURESETRESPONSE']._serialized_end=1984
80
+ _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_start=1986
81
+ _globals['_RUNPARQUETBATCHFEATURESETREQUEST']._serialized_end=2098
82
+ _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_start=2100
83
+ _globals['_RUNPARQUETBATCHFEATURESETRESPONSE']._serialized_end=2135
84
+ _globals['_CREATEUPLOADURLREQUEST']._serialized_start=2137
85
+ _globals['_CREATEUPLOADURLREQUEST']._serialized_end=2187
86
+ _globals['_CREATEUPLOADURLREQUESTV1']._serialized_start=2189
87
+ _globals['_CREATEUPLOADURLREQUESTV1']._serialized_end=2309
88
+ _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_start=2311
89
+ _globals['_CREATEREALTIMEFEATUREEXTRACTORUPLOADURLREQUEST']._serialized_end=2384
90
+ _globals['_CREATEUPLOADURLRESPONSE']._serialized_start=2386
91
+ _globals['_CREATEUPLOADURLRESPONSE']._serialized_end=2431
92
+ _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_start=2434
93
+ _globals['_UPDATEFEATURESETSTATEREQUEST']._serialized_end=2699
94
+ _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_start=2601
95
+ _globals['_UPDATEFEATURESETSTATEREQUEST_FEATURENAMESSTATEENTRY']._serialized_end=2699
96
+ _globals['_STATEUPDATE']._serialized_start=2701
97
+ _globals['_STATEUPDATE']._serialized_end=2802
98
+ _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_start=2804
99
+ _globals['_UPDATEFEATURESETSTATERESPONSE']._serialized_end=2835
100
+ _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_start=2837
101
+ _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_end=2912
102
+ _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_start=2914
103
+ _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_end=3034
104
+ _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_start=3036
105
+ _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_end=3093
106
+ _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_start=3095
107
+ _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_end=3189
108
+ _globals['_PAUSEFEATURESETREQUEST']._serialized_start=3191
109
+ _globals['_PAUSEFEATURESETREQUEST']._serialized_end=3240
110
+ _globals['_PAUSEFEATURESETRESPONSE']._serialized_start=3242
111
+ _globals['_PAUSEFEATURESETRESPONSE']._serialized_end=3267
112
+ _globals['_RESUMEFEATURESETREQUEST']._serialized_start=3269
113
+ _globals['_RESUMEFEATURESETREQUEST']._serialized_end=3319
114
+ _globals['_RESUMEFEATURESETRESPONSE']._serialized_start=3321
115
+ _globals['_RESUMEFEATURESETRESPONSE']._serialized_end=3347
116
+ _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_start=3349
117
+ _globals['_GETFEATURESETSCHEDULINGSTATEREQUEST']._serialized_end=3411
118
+ _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_start=3413
119
+ _globals['_GETFEATURESETSCHEDULINGSTATERESPONSE']._serialized_end=3522
120
+ _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_start=3524
121
+ _globals['_SETACTIVEFEATURESETVERSIONREQUEST']._serialized_end=3619
122
+ _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_start=3621
123
+ _globals['_SETACTIVEFEATURESETVERSIONRESPONSE']._serialized_end=3657
124
+ _globals['_FEATURESETSERVICE']._serialized_start=3660
125
+ _globals['_FEATURESETSERVICE']._serialized_end=6861
122
126
  # @@protoc_insertion_point(module_scope)
@@ -305,6 +305,31 @@ class ListFeatureSetsResponse(google.protobuf.message.Message):
305
305
 
306
306
  global___ListFeatureSetsResponse = ListFeatureSetsResponse
307
307
 
308
+ class ListFeaturesetsByAccountRequest(google.protobuf.message.Message):
309
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
310
+
311
+ def __init__(
312
+ self,
313
+ ) -> None: ...
314
+
315
+ global___ListFeaturesetsByAccountRequest = ListFeaturesetsByAccountRequest
316
+
317
+ class ListFeaturesetsByAccountResponse(google.protobuf.message.Message):
318
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
319
+
320
+ FEATURESETS_FIELD_NUMBER: builtins.int
321
+ @property
322
+ def featuresets(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.features.feature_set_pb2.FeatureSet]:
323
+ """List of registered featuresets"""
324
+ def __init__(
325
+ self,
326
+ *,
327
+ featuresets: collections.abc.Iterable[qwak.feature_store.features.feature_set_pb2.FeatureSet] | None = ...,
328
+ ) -> None: ...
329
+ def ClearField(self, field_name: typing_extensions.Literal["featuresets", b"featuresets"]) -> None: ...
330
+
331
+ global___ListFeaturesetsByAccountResponse = ListFeaturesetsByAccountResponse
332
+
308
333
  class RunBatchFeatureSetRequest(google.protobuf.message.Message):
309
334
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
310
335
 
@@ -54,6 +54,11 @@ class FeatureSetServiceStub(object):
54
54
  request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.SerializeToString,
55
55
  response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.FromString,
56
56
  )
57
+ self.ListFeaturesetsByAccount = channel.unary_unary(
58
+ '/qwak.feature.store.features.FeatureSetService/ListFeaturesetsByAccount',
59
+ request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
60
+ response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
61
+ )
57
62
  self.RunBatchFeatureSet = channel.unary_unary(
58
63
  '/qwak.feature.store.features.FeatureSetService/RunBatchFeatureSet',
59
64
  request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.SerializeToString,
@@ -180,6 +185,13 @@ class FeatureSetServiceServicer(object):
180
185
  context.set_details('Method not implemented!')
181
186
  raise NotImplementedError('Method not implemented!')
182
187
 
188
+ def ListFeaturesetsByAccount(self, request, context):
189
+ """List of registered featuresets
190
+ """
191
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
192
+ context.set_details('Method not implemented!')
193
+ raise NotImplementedError('Method not implemented!')
194
+
183
195
  def RunBatchFeatureSet(self, request, context):
184
196
  """Run now feature set
185
197
  """
@@ -313,6 +325,11 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
313
325
  request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.FromString,
314
326
  response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.SerializeToString,
315
327
  ),
328
+ 'ListFeaturesetsByAccount': grpc.unary_unary_rpc_method_handler(
329
+ servicer.ListFeaturesetsByAccount,
330
+ request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.FromString,
331
+ response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.SerializeToString,
332
+ ),
316
333
  'RunBatchFeatureSet': grpc.unary_unary_rpc_method_handler(
317
334
  servicer.RunBatchFeatureSet,
318
335
  request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.FromString,
@@ -524,6 +541,23 @@ class FeatureSetService(object):
524
541
  options, channel_credentials,
525
542
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
526
543
 
544
+ @staticmethod
545
+ def ListFeaturesetsByAccount(request,
546
+ target,
547
+ options=(),
548
+ channel_credentials=None,
549
+ call_credentials=None,
550
+ insecure=False,
551
+ compression=None,
552
+ wait_for_ready=None,
553
+ timeout=None,
554
+ metadata=None):
555
+ return grpc.experimental.unary_unary(request, target, '/qwak.feature.store.features.FeatureSetService/ListFeaturesetsByAccount',
556
+ qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
557
+ qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
558
+ options, channel_credentials,
559
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
560
+
527
561
  @staticmethod
528
562
  def RunBatchFeatureSet(request,
529
563
  target,
@@ -18,7 +18,7 @@ from frogml._proto.qwak.feature_store.sources import data_source_attribute_pb2 a
18
18
  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
19
19
 
20
20
 
21
- 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')
21
+ 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')
22
22
 
23
23
  _globals = globals()
24
24
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -36,9 +36,9 @@ if _descriptor._USE_C_DESCRIPTORS == False:
36
36
  _globals['_DATASOURCEDEFINITION']._serialized_start=539
37
37
  _globals['_DATASOURCEDEFINITION']._serialized_end=655
38
38
  _globals['_DATASOURCEMETADATA']._serialized_start=658
39
- _globals['_DATASOURCEMETADATA']._serialized_end=826
40
- _globals['_DATASOURCESPEC']._serialized_start=829
41
- _globals['_DATASOURCESPEC']._serialized_end=1312
42
- _globals['_FEATURESETBRIEF']._serialized_start=1314
43
- _globals['_FEATURESETBRIEF']._serialized_end=1407
39
+ _globals['_DATASOURCEMETADATA']._serialized_end=850
40
+ _globals['_DATASOURCESPEC']._serialized_start=853
41
+ _globals['_DATASOURCESPEC']._serialized_end=1336
42
+ _globals['_FEATURESETBRIEF']._serialized_start=1338
43
+ _globals['_FEATURESETBRIEF']._serialized_end=1431
44
44
  # @@protoc_insertion_point(module_scope)
@@ -77,6 +77,7 @@ class DataSourceMetadata(google.protobuf.message.Message):
77
77
  CREATED_BY_FIELD_NUMBER: builtins.int
78
78
  LAST_MODIFIED_AT_FIELD_NUMBER: builtins.int
79
79
  LAST_MODIFIED_BY_FIELD_NUMBER: builtins.int
80
+ ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
80
81
  @property
81
82
  def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
82
83
  """When the data source was created"""
@@ -87,6 +88,8 @@ class DataSourceMetadata(google.protobuf.message.Message):
87
88
  """Last modified"""
88
89
  last_modified_by: builtins.str
89
90
  """Last modified by"""
91
+ environment_id: builtins.str
92
+ """Environment id"""
90
93
  def __init__(
91
94
  self,
92
95
  *,
@@ -94,9 +97,10 @@ class DataSourceMetadata(google.protobuf.message.Message):
94
97
  created_by: builtins.str = ...,
95
98
  last_modified_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
96
99
  last_modified_by: builtins.str = ...,
100
+ environment_id: builtins.str = ...,
97
101
  ) -> None: ...
98
102
  def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "last_modified_at", b"last_modified_at"]) -> builtins.bool: ...
99
- 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: ...
103
+ 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: ...
100
104
 
101
105
  global___DataSourceMetadata = DataSourceMetadata
102
106
 
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
  from frogml._proto.qwak.feature_store.sources import data_source_pb2 as qwak_dot_feature__store_dot_sources_dot_data__source__pb2
15
15
 
16
16
 
17
- 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')
17
+ 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')
18
18
 
19
19
  _globals = globals()
20
20
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -49,10 +49,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
49
49
  _globals['_LISTDATASOURCESREQUEST']._serialized_end=919
50
50
  _globals['_LISTDATASOURCESRESPONSE']._serialized_start=921
51
51
  _globals['_LISTDATASOURCESRESPONSE']._serialized_end=1008
52
- _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_start=1010
53
- _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_end=1091
54
- _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_start=1093
55
- _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_end=1148
56
- _globals['_DATASOURCESERVICE']._serialized_start=1151
57
- _globals['_DATASOURCESERVICE']._serialized_end=2260
52
+ _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_start=1010
53
+ _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_end=1043
54
+ _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_start=1045
55
+ _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_end=1141
56
+ _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_start=1143
57
+ _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_end=1224
58
+ _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_start=1226
59
+ _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_end=1281
60
+ _globals['_DATASOURCESERVICE']._serialized_start=1284
61
+ _globals['_DATASOURCESERVICE']._serialized_end=2545
58
62
  # @@protoc_insertion_point(module_scope)
@@ -202,6 +202,31 @@ class ListDataSourcesResponse(google.protobuf.message.Message):
202
202
 
203
203
  global___ListDataSourcesResponse = ListDataSourcesResponse
204
204
 
205
+ class ListDataSourcesByAccountRequest(google.protobuf.message.Message):
206
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
207
+
208
+ def __init__(
209
+ self,
210
+ ) -> None: ...
211
+
212
+ global___ListDataSourcesByAccountRequest = ListDataSourcesByAccountRequest
213
+
214
+ class ListDataSourcesByAccountResponse(google.protobuf.message.Message):
215
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
216
+
217
+ DATA_SOURCES_FIELD_NUMBER: builtins.int
218
+ @property
219
+ def data_sources(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.sources.data_source_pb2.DataSource]:
220
+ """List of registered data sources"""
221
+ def __init__(
222
+ self,
223
+ *,
224
+ data_sources: collections.abc.Iterable[qwak.feature_store.sources.data_source_pb2.DataSource] | None = ...,
225
+ ) -> None: ...
226
+ def ClearField(self, field_name: typing_extensions.Literal["data_sources", b"data_sources"]) -> None: ...
227
+
228
+ global___ListDataSourcesByAccountResponse = ListDataSourcesByAccountResponse
229
+
205
230
  class CreateDataSourceUploadURLRequest(google.protobuf.message.Message):
206
231
  DESCRIPTOR: google.protobuf.descriptor.Descriptor
207
232
 
@@ -44,6 +44,11 @@ class DataSourceServiceStub(object):
44
44
  request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.SerializeToString,
45
45
  response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.FromString,
46
46
  )
47
+ self.ListDataSourcesByAccount = channel.unary_unary(
48
+ '/qwak.feature.store.sources.DataSourceService/ListDataSourcesByAccount',
49
+ request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
50
+ response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
51
+ )
47
52
  self.CreateDataSourceUploadURL = channel.unary_unary(
48
53
  '/qwak.feature.store.sources.DataSourceService/CreateDataSourceUploadURL',
49
54
  request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.SerializeToString,
@@ -101,6 +106,13 @@ class DataSourceServiceServicer(object):
101
106
  context.set_details('Method not implemented!')
102
107
  raise NotImplementedError('Method not implemented!')
103
108
 
109
+ def ListDataSourcesByAccount(self, request, context):
110
+ """List of registered data sources
111
+ """
112
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
113
+ context.set_details('Method not implemented!')
114
+ raise NotImplementedError('Method not implemented!')
115
+
104
116
  def CreateDataSourceUploadURL(self, request, context):
105
117
  """Create a url for uploading data source configuration
106
118
  """
@@ -148,6 +160,11 @@ def add_DataSourceServiceServicer_to_server(servicer, server):
148
160
  request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.FromString,
149
161
  response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.SerializeToString,
150
162
  ),
163
+ 'ListDataSourcesByAccount': grpc.unary_unary_rpc_method_handler(
164
+ servicer.ListDataSourcesByAccount,
165
+ request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.FromString,
166
+ response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.SerializeToString,
167
+ ),
151
168
  'CreateDataSourceUploadURL': grpc.unary_unary_rpc_method_handler(
152
169
  servicer.CreateDataSourceUploadURL,
153
170
  request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.FromString,
@@ -270,6 +287,23 @@ class DataSourceService(object):
270
287
  options, channel_credentials,
271
288
  insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
272
289
 
290
+ @staticmethod
291
+ def ListDataSourcesByAccount(request,
292
+ target,
293
+ options=(),
294
+ channel_credentials=None,
295
+ call_credentials=None,
296
+ insecure=False,
297
+ compression=None,
298
+ wait_for_ready=None,
299
+ timeout=None,
300
+ metadata=None):
301
+ return grpc.experimental.unary_unary(request, target, '/qwak.feature.store.sources.DataSourceService/ListDataSourcesByAccount',
302
+ qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
303
+ qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
304
+ options, channel_credentials,
305
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
306
+
273
307
  @staticmethod
274
308
  def CreateDataSourceUploadURL(request,
275
309
  target,
@@ -1,7 +1,5 @@
1
1
  from typing import Optional
2
2
 
3
- from grpc import RpcError
4
-
5
3
  from frogml._proto.qwak.logging.log_filter_pb2 import LogText, SearchFilter
6
4
  from frogml._proto.qwak.logging.log_reader_service_pb2 import (
7
5
  ReadLogsRequest,
@@ -17,6 +15,7 @@ from frogml._proto.qwak.logging.log_source_pb2 import (
17
15
  from frogml.core.clients.administration.eco_system.client import EcosystemClient
18
16
  from frogml.core.exceptions import FrogmlException
19
17
  from frogml.core.inner.tool.grpc.grpc_tools import create_grpc_channel
18
+ from frogml.core.inner.tool.grpc.grpc_try_wrapping import grpc_try_catch_wrapper
20
19
 
21
20
 
22
21
  class LoggingClient:
@@ -120,6 +119,7 @@ class LoggingClient:
120
119
  except FrogmlException as e:
121
120
  raise FrogmlException(f"Failed to fetch execution logs, error is [{e}]")
122
121
 
122
+ @grpc_try_catch_wrapper("Failed to read logs request")
123
123
  def read_logs(
124
124
  self,
125
125
  source,
@@ -127,22 +127,16 @@ class LoggingClient:
127
127
  after_offset,
128
128
  max_number_of_results,
129
129
  log_text_filter,
130
- ):
131
- try:
132
- response = self._logging_service.ReadLogs(
133
- ReadLogsRequest(
134
- source=source,
135
- before_offset=before_offset,
136
- after_offset=after_offset,
137
- search_filter=SearchFilter(
138
- log_text_filter=LogText(contains=log_text_filter)
139
- ),
140
- max_number_of_results=max_number_of_results,
141
- )
142
- )
143
- return response
144
- except RpcError as e:
145
- raise FrogmlException(
146
- f"Failed grpc read logs request, grpc error is "
147
- f"[{e.details() if e.details() else e.code()}]"
130
+ ) -> ReadLogsResponse:
131
+ response: ReadLogsResponse = self._logging_service.ReadLogs(
132
+ ReadLogsRequest(
133
+ source=source,
134
+ before_offset=before_offset,
135
+ after_offset=after_offset,
136
+ search_filter=SearchFilter(
137
+ log_text_filter=LogText(contains=log_text_filter)
138
+ ),
139
+ max_number_of_results=max_number_of_results,
148
140
  )
141
+ )
142
+ return response
@@ -10,6 +10,7 @@ from frogml.core.clients.instance_template.client import (
10
10
  )
11
11
  from frogml.core.clients.model_management import ModelsManagementClient
12
12
  from frogml.core.inner.build_logic.dependency_manager_type import DependencyManagerType
13
+ from frogml.core.inner.di_configuration import UserAccountConfiguration
13
14
 
14
15
 
15
16
  @dataclass
@@ -56,3 +57,7 @@ class Context(metaclass=ABCMeta):
56
57
  # Upload Custom Wheels
57
58
  custom_runtime_wheel: Optional[Path] = field(default=None)
58
59
  custom_core_wheel: Optional[Path] = field(default=None)
60
+
61
+ platform_url: str = field(
62
+ default_factory=UserAccountConfiguration().retrieve_platform_url
63
+ )
@@ -104,3 +104,11 @@ class UserAccountConfiguration:
104
104
  return getattr(auth, "token")
105
105
  except AttributeError as e:
106
106
  raise FrogmlLoginException("Token is not configured") from e
107
+
108
+ @staticmethod
109
+ def retrieve_platform_url() -> str:
110
+ """Returns the platform URL with the '/ui/ml' suffix based on the authentication client configuration."""
111
+ try:
112
+ return f"{FrogMLAuthClient().get_base_url()}/ui/ml"
113
+ except ValueError:
114
+ return ""
core/inner/tool/auth.py CHANGED
@@ -23,6 +23,10 @@ class FrogMLAuthClient:
23
23
 
24
24
  return cast(str, self._token)
25
25
 
26
+ def get_base_url(self) -> str:
27
+ artifactory_url, _ = get_credentials(self.auth_config)
28
+ return self.__format_artifactory_url(artifactory_url)
29
+
26
30
  def get_tenant_id(self: Self) -> str:
27
31
  if not self._tenant_id:
28
32
  self.login()
@@ -46,12 +50,7 @@ class FrogMLAuthClient:
46
50
  "Failed to authenticate with JFrog. Please check your credentials"
47
51
  )
48
52
 
49
- # Remove '/artifactory/' from the URL
50
- if "/artifactory" in artifactory_url:
51
- base_url = artifactory_url.replace("/artifactory", "/ui")
52
- else:
53
- # Remove trailing slash if it exists and append /ui
54
- base_url = artifactory_url.rstrip("/") + "/ui"
53
+ base_url: str = self.__format_artifactory_url(artifactory_url)
55
54
 
56
55
  url = f"{base_url}/api/v1/system/auth/screen/footer"
57
56
 
@@ -73,3 +72,10 @@ class FrogMLAuthClient:
73
72
  "Authentication with JFrog failed: Only JWT Access Tokens are supported. "
74
73
  "Please ensure you are using a valid JWT Access Token."
75
74
  )
75
+
76
+ @staticmethod
77
+ def __format_artifactory_url(artifactory_url: str) -> str:
78
+ # Remove '/artifactory' from the URL
79
+ base_url: str = artifactory_url.replace("/artifactory", "")
80
+ # Remove trailing slash if exists
81
+ return base_url.rstrip("/")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: frogml-core
3
- Version: 0.0.119
3
+ Version: 0.0.121
4
4
  Summary: frogml contains the necessary objects and communication tools for using the JFrog ml Platform
5
5
  License: Apache-2.0
6
6
  Keywords: mlops,ml,deployment,serving,model
@@ -325,12 +325,12 @@ _proto/qwak/feature_store/features/execution_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw
325
325
  _proto/qwak/feature_store/features/feature_set_attribute_pb2.py,sha256=lpzsSq-ql-affwFjQWdxAr8_heF_MK41yXUfJJcYcos,1725
326
326
  _proto/qwak/feature_store/features/feature_set_attribute_pb2.pyi,sha256=ZXkJZVES1V56ofWT4iYeMqi0Zvrf8ne7G88chvaCEN4,1229
327
327
  _proto/qwak/feature_store/features/feature_set_attribute_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
328
- _proto/qwak/feature_store/features/feature_set_pb2.py,sha256=ioKi4yB8b6hc7U9DyJKHSH5DdW4GheU9Y8CWros0FJM,10515
329
- _proto/qwak/feature_store/features/feature_set_pb2.pyi,sha256=ugvLzzf1UL0hIcNpDh9sCD94qlpDtlbAJnn8_70cTZo,22322
328
+ _proto/qwak/feature_store/features/feature_set_pb2.py,sha256=MvwP2Ft1aOcxCiEqeCghvnTZF-dR3USgAKrFZFGNkm8,10562
329
+ _proto/qwak/feature_store/features/feature_set_pb2.pyi,sha256=rZNYUjKN6EAuApqXWbIo9_499FdwU31zGJ8HLAVRO_o,22507
330
330
  _proto/qwak/feature_store/features/feature_set_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
331
- _proto/qwak/feature_store/features/feature_set_service_pb2.py,sha256=YDKU-NVhgzYjhw8hoIc2hWqtjP5O0Uy0v97wb4cZGgY,16550
332
- _proto/qwak/feature_store/features/feature_set_service_pb2.pyi,sha256=ziAzRBx2eNtRmsgtfDz5uFL62Em4kcaFaQye_9YMubo,26672
333
- _proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py,sha256=TvOzhqFYrnqFv7WREdr32E-a6gXEH2liI8e2KwkhZ9k,43687
331
+ _proto/qwak/feature_store/features/feature_set_service_pb2.py,sha256=SaZg-vzkzIu9R8Q4sZro8VDFVFqMNwln1Y_pePpJ72s,17173
332
+ _proto/qwak/feature_store/features/feature_set_service_pb2.pyi,sha256=N-TBJHNff4qNuab6jFXfi1craBomX_BUNZpKaeRM9Fo,27676
333
+ _proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py,sha256=8wV6mMgKhPgmk_3anKo6SQ-6veROxmD2CDCfGAg5xtk,45753
334
334
  _proto/qwak/feature_store/features/feature_set_state_pb2.py,sha256=gP7MjV-nvxFnebfh1JALQscbUmG6GhJ4VKrY-1bH0oI,6221
335
335
  _proto/qwak/feature_store/features/feature_set_state_pb2.pyi,sha256=8XYM1VvPDpl71qJH5BVjwkRvv4tVYq3EzscAUl81LDQ,14071
336
336
  _proto/qwak/feature_store/features/feature_set_state_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -388,12 +388,12 @@ _proto/qwak/feature_store/sources/batch_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8
388
388
  _proto/qwak/feature_store/sources/data_source_attribute_pb2.py,sha256=0L7Hc0LxQYcT8vXYKJDGybDo_RKN3wX7tazpYRiojm4,1712
389
389
  _proto/qwak/feature_store/sources/data_source_attribute_pb2.pyi,sha256=E-YJTCTHI6VJVAjhQB2SrsZ5vOSb-wxZAvEDbQXu2z8,1232
390
390
  _proto/qwak/feature_store/sources/data_source_attribute_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
391
- _proto/qwak/feature_store/sources/data_source_pb2.py,sha256=_1rQD7lBX8dFWBLXBNxIojeEIaNALXi3vFLVpuRh6JY,4644
392
- _proto/qwak/feature_store/sources/data_source_pb2.pyi,sha256=l6URzbtAsaCe5-e2LuUC7wqi2g3ITyZlAR6Zs14C8iM,8785
391
+ _proto/qwak/feature_store/sources/data_source_pb2.py,sha256=m2xW0FNCnmMfhy0NUcNF12RS2vaPyli7ITtFHSRCicw,4691
392
+ _proto/qwak/feature_store/sources/data_source_pb2.pyi,sha256=l2IIdc3DMZ0wJ_Pak7pVKdd2u8URhNLbhn7BKznlwD4,8970
393
393
  _proto/qwak/feature_store/sources/data_source_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
394
- _proto/qwak/feature_store/sources/data_source_service_pb2.py,sha256=JEsnAMfsN36rieUOGRGPbZn2ipIWmGcIwfNYu6G7PFg,6244
395
- _proto/qwak/feature_store/sources/data_source_service_pb2.pyi,sha256=d4dUfwndYChXzh5C6Pu5jXNsmFxe0OEdcywZs2NNC04,9426
396
- _proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py,sha256=qQb5qOhPsVsaqnpnLtW_Cp--UbvqgIbvGNrePGB_cuw,17073
394
+ _proto/qwak/feature_store/sources/data_source_service_pb2.py,sha256=rxfe2OPCPDWI6hNqkbFcZDvMs44385IUev7Ayvr0ZHk,6859
395
+ _proto/qwak/feature_store/sources/data_source_service_pb2.pyi,sha256=NwrpI2yy6p0qhATlxKsD2V5YnE256LlGfK9WbGT_A6w,10434
396
+ _proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py,sha256=V26Auqs9Z7kpRTqa86Oi3AInsL6jRi_oiGF5RX_Fo_g,19132
397
397
  _proto/qwak/feature_store/sources/streaming_pb2.py,sha256=9S7ajwaXgPXx84hPcX0OE-V3eDt_bvGLvU0smOpARxc,6239
398
398
  _proto/qwak/feature_store/sources/streaming_pb2.pyi,sha256=Gl9XE_awS5aLpFEW9m-sxwWxvcI3l_6EIq8OjBRzqnU,15942
399
399
  _proto/qwak/feature_store/sources/streaming_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
@@ -674,7 +674,7 @@ core/clients/kube_deployment_captain/client.py,sha256=la4V7WsVL1z4zuLQtPZET1kjvG
674
674
  core/clients/location_discovery/__init__.py,sha256=sqGQ75YHFE6nvOcir38fykUUmAa6cFEIze8PJYgYWRc,44
675
675
  core/clients/location_discovery/client.py,sha256=ZcBbbLqzZ9E7oLyamBTLTpp611LldhfFOSkzWroyIQg,2895
676
676
  core/clients/logging_client/__init__.py,sha256=1OCHnigQBYThBwGbxCreYA0BgP0HcuLFzNEWd3Yxh-c,34
677
- core/clients/logging_client/client.py,sha256=qTI4X9be4iK6v6G17-xiITJD8EQwxmXLqC3Iiec-6cE,4954
677
+ core/clients/logging_client/client.py,sha256=vq2Fv7UAEnrG910Npp1COxxWQ2_NWofbAStPkLhiz1w,4836
678
678
  core/clients/model_management/__init__.py,sha256=vjWVP8MjmK4_A70WOgJqa6x24AeLK-ABjGJtogGzw9w,43
679
679
  core/clients/model_management/client.py,sha256=bDBHrK4rssA67-cPPHCe4wA6jkphIjaxaMwxBYPfyIU,4918
680
680
  core/clients/model_version_manager/__init__.py,sha256=4Pnfzj4Egps48__dProdbSKQl5pNip0hGJd75w67BfY,46
@@ -765,7 +765,7 @@ core/inner/build_logic/execute_build_pipeline.py,sha256=n607h5ozBFcFexJ6E9FHwgrl
765
765
  core/inner/build_logic/interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
766
766
  core/inner/build_logic/interface/build_logger_interface.py,sha256=JUxldJuhMVHoXk9nZc9tz7-LPOMC99Z32oQYHSHpffY,528
767
767
  core/inner/build_logic/interface/build_phase.py,sha256=DNOcg11RVCSPpnGcAk2xDYEXwQ6UNWFoiGMpjgSFPt8,675
768
- core/inner/build_logic/interface/context_interface.py,sha256=R4HMaoTg3x8prLExrje8JGbFovCScjYcR81VJ1hukRs,2120
768
+ core/inner/build_logic/interface/context_interface.py,sha256=97RU5gucN6CSSXud4rqKdlGhF73eroPup4OlIWAWgh4,2303
769
769
  core/inner/build_logic/interface/phase_run_handler.py,sha256=bTeKMgbG85SuL7MfDJpzsO8HlN_H6F8Q6Wnt8wJYXGM,1744
770
770
  core/inner/build_logic/interface/step_inteface.py,sha256=vRQRk5Z5n7JuA3GNgVJOIc62i8iG7WSvsG1f2nOOVDQ,746
771
771
  core/inner/build_logic/interface/time_source.py,sha256=njwil9A1sxP4SoB8LxtNhWhlV2-w1XOpoF4_wccNVYU,585
@@ -801,7 +801,7 @@ core/inner/build_logic/tools/text.py,sha256=tH-v19Mt8l90sMVxku5XRtrderT0qdRqJ-jL
801
801
  core/inner/build_logic/trigger_build_context.py,sha256=eWR35HQQKKNaPOfl64IB8xvCzl2vU6hgSR55qf9l7Lw,207
802
802
  core/inner/const.py,sha256=eP8Naruk3EnRWDdS9gwpozUYA9Nxznybw7Eiv4k5l38,126
803
803
  core/inner/di_configuration/__init__.py,sha256=7TE1MMfkFvBzHIzhAZ1L-vRgbScCIIMd-aXU9IFcs98,2130
804
- core/inner/di_configuration/account.py,sha256=InonyAqGXY-46olKsWYp362J8DeDu7koVmk61WBACZE,2822
804
+ core/inner/di_configuration/account.py,sha256=oREHMbzSEgRSR6V9cQRjDYefygq9wu5_Q_nEC1ws4X4,3121
805
805
  core/inner/di_configuration/config.yml,sha256=GUvaZMWIDIR_d7hFcPVG_kHdCwpERKH1AFDakG3vqI4,242
806
806
  core/inner/di_configuration/containers.py,sha256=E8ebGj0d1_8iMOn_fbcfKXwbXgSbLBQCKgA9_JV38E4,1148
807
807
  core/inner/instance_template/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -810,7 +810,7 @@ core/inner/model_loggers_utils.py,sha256=mGBZQ1CbiEePaoftFEH_yfLlsxP8fcPcu83Cb8U
810
810
  core/inner/provider.py,sha256=3evQnyp0v0enpvGGDyaZziusO4BGi-U9j1sno8DAHo4,70
811
811
  core/inner/singleton_meta.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
812
812
  core/inner/tool/__init__.py,sha256=rmOSE-ejnzDG_H7kbikPQxEO4TFIkhBWjOXhTIrldiU,35
813
- core/inner/tool/auth.py,sha256=ZQdb-vgRNwBz1yshdPofcH5Qfc5YAkAJG55bbp95sVU,2652
813
+ core/inner/tool/auth.py,sha256=Ya55AgSY_feZLtaIje91z1rSG5JsLC-Jh6NdomsOTcs,2858
814
814
  core/inner/tool/grpc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
815
815
  core/inner/tool/grpc/grpc_auth.py,sha256=8lGC1B3IgLuVzY1X6aEozGI3gnzLlA5WDZ7VWn4qNY0,1421
816
816
  core/inner/tool/grpc/grpc_tools.py,sha256=g49mtjSblXdSrFBSNPmsgCgDXHEzliqQ4uYDWRs527Q,7247
@@ -1011,6 +1011,6 @@ storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbxCdk,299
1011
1011
  storage/utils/_input_checks_utility.py,sha256=TgQ7Ycdkdu8bdDPX-jOvzo2QazccsRqPv4Qm3c_U4HU,3221
1012
1012
  storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
1013
1013
  storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
1014
- frogml_core-0.0.119.dist-info/METADATA,sha256=J1hUaLbRW1EOi-TCqJ_G6GRKrdzVY2n8b1cauLkS61s,14893
1015
- frogml_core-0.0.119.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1016
- frogml_core-0.0.119.dist-info/RECORD,,
1014
+ frogml_core-0.0.121.dist-info/METADATA,sha256=S8rJjrsTtvUqacp-lOam7MnxtxQClofCVISy2785iHo,14893
1015
+ frogml_core-0.0.121.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
1016
+ frogml_core-0.0.121.dist-info/RECORD,,