frogml-core 0.0.59__py3-none-any.whl → 0.0.61__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.59"
+__version__ = "0.0.61"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

{frogml_core-0.0.59.dist-info → frogml_core-0.0.61.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.59
+Version: 0.0.61
 Summary: frogml Core 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.59.dist-info → frogml_core-0.0.61.dist-info}/RECORD

@@ -1,4 +1,4 @@
-frogml_core/__init__.py,sha256=332MK42v0tjbNrCIM6ab7ZqBLzR03NsBL7ohUOtPVK8,777
+frogml_core/__init__.py,sha256=KCxPyv8TJpydgYwbTHocA-vftJl07CYrqDRQlprF3To,777
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
 frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
@@ -773,13 +773,13 @@ frogml_proto/qwak/feature_store/jobs/v1/job_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7
 frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2.py,sha256=FtWY33ulu5tbuXvCysgZZwkO9ChFygVCDxPO-VTbmPU,15330
 frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2.pyi,sha256=xv5A4PDffjBdcWokNO7lxTLfWW5G-YqUXPBsDSVR5nM,14178
 frogml_proto/qwak/feature_store/jobs/v1/job_service_pb2_grpc.py,sha256=z9JmMsr6730bYRpKcyaHuJGfwddDIPgoYaxVzmlC8iU,18466
-frogml_proto/qwak/feature_store/platform/platform_details_pb2.py,sha256=FstgtfkXllZH922gcvISajWPFfCKcJDyYZ9HbosZv7Q,2487
+frogml_proto/qwak/feature_store/platform/platform_details_pb2.py,sha256=FTA-FEKIGDWcmheWsW4nomMjQZTlEYyUoUQaa8f9qGE,2487
 frogml_proto/qwak/feature_store/platform/platform_details_pb2.pyi,sha256=k1L4pBAFrxRt2PbbF_Im2dyVhFPkrYpmdtOdiyPQhQs,2108
 frogml_proto/qwak/feature_store/platform/platform_details_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/feature_store/reports/report_pb2.py,sha256=wWu_eT1lAsmKDLBCYiqxI76Yy6-uBs19lQ4zNoGC-vw,5171
 frogml_proto/qwak/feature_store/reports/report_pb2.pyi,sha256=qcHoPr3wNQ5-dyCSwuJ5VdpzlzLKGJysY_K3appd1WM,7436
 frogml_proto/qwak/feature_store/reports/report_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/feature_store/repository/common/platform_pb2.py,sha256=YFseV8wzQzd4kl5Sn1dIyReakumGZBlyD1QjTpTqLro,3042
+frogml_proto/qwak/feature_store/repository/common/platform_pb2.py,sha256=BELZ7dq_raPi4zOMBOIj23KURPZKS0JfTvEB47H4OPA,3034
 frogml_proto/qwak/feature_store/repository/common/platform_pb2.pyi,sha256=FpO12GJAtApZfCHaASsmjQOx-IEP14xiO-yUMb87kUk,2682
 frogml_proto/qwak/feature_store/repository/common/platform_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
 frogml_proto/qwak/feature_store/serving/management_pb2.py,sha256=GUrb0rvF0kYFjQrU1OfZLlqBNmzaLMPh2vBVTBXeIn0,5125
@@ -812,11 +812,14 @@ frogml_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py,sha256=L
 frogml_proto/qwak/feature_store/sources/streaming_pb2.py,sha256=TiTVdKAErRoS1l1MRGUk4nNueTLKGlCEAI3x0900Yug,11062
 frogml_proto/qwak/feature_store/sources/streaming_pb2.pyi,sha256=Gl9XE_awS5aLpFEW9m-sxwWxvcI3l_6EIq8OjBRzqnU,15942
 frogml_proto/qwak/feature_store/sources/streaming_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.py,sha256=Yth8wPB6lip3cOz7iYjWqpNPIpO75YVHKuqXWjW2cBw,2979
-frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.pyi,sha256=jimDXVSe8Ird1aB7DKNRb-Feb6IXhXjC3Ttu_u1KAC4,2402
+frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.py,sha256=77wnfMiz82D7xTLNf2BZ3aEOOD_KehUPyv9SWXL3enI,1746
+frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.pyi,sha256=HpqD0UT8PLOOCPUldO1K3SxXTetx5jLyvhAgcU_bVKY,910
+frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
+frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.py,sha256=NhMPJGEz_RJsTKszNb5bo4PL_eDOwmpofLnq_j4CaOY,3391
+frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.pyi,sha256=YiyanY3mYTkXp92oG0HbLxhCZVX_KUSKntjPPmpCFH0,3300
 frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.py,sha256=1bn6yrmQ1Q_ltQrhCJAqMWZijLTcnuekCaPYIAP2I6c,5195
-frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.pyi,sha256=DSChqVqQkLYLbR6whgnMtyi2r_buDDEzcfG-fKUqHTM,2427
+frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.py,sha256=VeOjoQXLm7-Jr5aot4cdDLwX1-mLpqpjm3R1XGvoJPA,6459
+frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.pyi,sha256=RuuyLe7eAmaLzj2zs0tSsB5pIBfHJqor5K_-IxZ_bcM,3430
 frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2_grpc.py,sha256=vDItq1YEaxnErMyswclxAKrEJ_MssqDqU6Ncj-bW_QU,5909
 frogml_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.py,sha256=y7VI72SZGsKIjVzV-xbrVlhovvU82t5RhWTUd85RtLQ,3181
 frogml_proto/qwak/feature_store/v1/internal/featureset/featureset_token_service_pb2.pyi,sha256=jmrIw0qDsqqu_kwLkBoszvfUgaZlu6iIqao-kXEsX5U,1467
@@ -1079,6 +1082,6 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=_34z6rFCHFwcSni-9eyJlrH264xsL_QDNmG-EBv0zMc,20281
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_core-0.0.59.dist-info/METADATA,sha256=2jY8VbbTprGW7V9Op4iANVxp_19gebi7IW-Ji5pnOhc,2004
-frogml_core-0.0.59.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-frogml_core-0.0.59.dist-info/RECORD,,
+frogml_core-0.0.61.dist-info/METADATA,sha256=VY52kUn65_5nYwYpZNJXhS_dn4czRzaSVDBbvchRn3M,2004
+frogml_core-0.0.61.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+frogml_core-0.0.61.dist-info/RECORD,,

frogml_proto/qwak/feature_store/platform/platform_details_pb2.py

@@ -15,7 +15,7 @@ _sym_db = _symbol_database.Default()
 from frogml_proto.qwak.feature_store.repository.common import platform_pb2 as qwak_dot_feature__store_dot_repository_dot_common_dot_platform__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2qwak/feature_store/platform/platform_details.proto\x12\x1bqwak.feature.store.platform\x1a\x33qwak/feature_store/repository/common/platform.proto\"\x9e\x01\n\x0fPlatformDetails\x12Q\n\x11platform_settings\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.repository.common.PlatformSettings\x12\x38\n\tauth_type\x18\x02 \x01(\x0b\x32%.qwak.feature.store.platform.AuthType\"&\n\x08\x41uthType\x12\x12\n\x08token_id\x18\x01 \x01(\tH\x00\x42\x06\n\x04typeB[\n&com.qwak.ai.feature.store.platform.apiP\x01Z/qwak/featurestore/features;featurestoreplatformb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2qwak/feature_store/platform/platform_details.proto\x12\x1bqwak.feature.store.platform\x1a\x33qwak/feature_store/repository/common/platform.proto\"\x9e\x01\n\x0fPlatformDetails\x12Q\n\x11platform_settings\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.repository.common.PlatformSettings\x12\x38\n\tauth_type\x18\x02 \x01(\x0b\x32%.qwak.feature.store.platform.AuthType\"&\n\x08\x41uthType\x12\x12\n\x08token_id\x18\x01 \x01(\tH\x00\x42\x06\n\x04typeB[\n&com.qwak.ai.feature.store.platform.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
 
 
@@ -38,7 +38,7 @@ _sym_db.RegisterMessage(AuthType)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.feature.store.platform.apiP\001Z/qwak/featurestore/features;featurestoreplatform'
+  DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.feature.store.platform.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
   _PLATFORMDETAILS._serialized_start=137
   _PLATFORMDETAILS._serialized_end=295
   _AUTHTYPE._serialized_start=297

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

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/feature_store/repository/common/platform.proto\x12$qwak.feature.store.repository.common\"\xb6\x01\n\x10PlatformSettings\x12S\n\x12repository_setting\x18\x01 \x01(\x0b\x32\x37.qwak.feature.store.repository.common.RepositorySetting\x12M\n\x0fproject_setting\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.repository.common.ProjectSetting\"+\n\x11RepositorySetting\x12\x16\n\x0erepository_key\x18\x01 \x01(\t\"=\n\x0eProjectSetting\x12\x15\n\x0bproject_key\x18\x02 \x01(\tH\x00\x42\x14\n\x12project_identifierBn\n/com.qwak.ai.feature.store.repository.common.apiP\x01Z9qwak/featurestore/featureset/repository;common_repositoryb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/feature_store/repository/common/platform.proto\x12$qwak.feature.store.repository.common\"\xb6\x01\n\x10PlatformSettings\x12S\n\x12repository_setting\x18\x01 \x01(\x0b\x32\x37.qwak.feature.store.repository.common.RepositorySetting\x12M\n\x0fproject_setting\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.repository.common.ProjectSetting\"+\n\x11RepositorySetting\x12\x16\n\x0erepository_key\x18\x01 \x01(\t\"=\n\x0eProjectSetting\x12\x15\n\x0bproject_key\x18\x02 \x01(\tH\x00\x42\x14\n\x12project_identifierBj\n/com.qwak.ai.feature.store.repository.common.apiP\x01Z5qwak/featurestore/source/repository;source_repositoryb\x06proto3')
 
 
 
@@ -45,7 +45,7 @@ _sym_db.RegisterMessage(ProjectSetting)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.repository.common.apiP\001Z9qwak/featurestore/featureset/repository;common_repository'
+  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.repository.common.apiP\001Z5qwak/featurestore/source/repository;source_repository'
   _PLATFORMSETTINGS._serialized_start=94
   _PLATFORMSETTINGS._serialized_end=276
   _REPOSITORYSETTING._serialized_start=278

frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n@qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto\x12+qwak.feature.store.v1.common.jfrog_artifact\"/\n\rJfrogArtifact\x12\x0c\n\x04path\x18\x01 \x01(\t\x12\x10\n\x08\x63hecksum\x18\x02 \x01(\tBp\n2com.qwak.ai.feature.store.v1.common.jfrog_artifactP\x01Z8qwak/featurestore/v1/common/jfrog_artifact;jfrogartifactb\x06proto3')
+
+
+
+_JFROGARTIFACT = DESCRIPTOR.message_types_by_name['JfrogArtifact']
+JfrogArtifact = _reflection.GeneratedProtocolMessageType('JfrogArtifact', (_message.Message,), {
+  'DESCRIPTOR' : _JFROGARTIFACT,
+  '__module__' : 'qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.feature.store.v1.common.jfrog_artifact.JfrogArtifact)
+  })
+_sym_db.RegisterMessage(JfrogArtifact)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n2com.qwak.ai.feature.store.v1.common.jfrog_artifactP\001Z8qwak/featurestore/v1/common/jfrog_artifact;jfrogartifact'
+  _JFROGARTIFACT._serialized_start=113
+  _JFROGARTIFACT._serialized_end=160
+# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2.pyi

@@ -0,0 +1,34 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class JfrogArtifact(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PATH_FIELD_NUMBER: builtins.int
+    CHECKSUM_FIELD_NUMBER: builtins.int
+    path: builtins.str
+    """artifact path"""
+    checksum: builtins.str
+    """artifact checksum"""
+    def __init__(
+        self,
+        *,
+        path: builtins.str = ...,
+        checksum: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["checksum", b"checksum", "path", b"path"]) -> None: ...
+
+global___JfrogArtifact = JfrogArtifact

frogml_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.py

@@ -12,9 +12,10 @@ from google.protobuf import symbol_database as _symbol_database
 _sym_db = _symbol_database.Default()
 
 
+from frogml_proto.qwak.feature_store.v1.common.jfrog_artifact import jfrog_artifact_pb2 as qwak_dot_feature__store_dot_v1_dot_common_dot_jfrog__artifact_dot_jfrog__artifact__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:qwak/feature_store/v1/common/source_code/source_code.proto\x12(qwak.feature.store.v1.common.source_code\"`\n\x0eSourceCodeSpec\x12N\n\x08\x61rtifact\x18\x01 \x01(\x0b\x32<.qwak.feature.store.v1.common.source_code.SourceCodeArtifact\"k\n\x12SourceCodeArtifact\x12M\n\x0czip_artifact\x18\x01 \x01(\x0b\x32\x35.qwak.feature.store.v1.common.source_code.ZipArtifactH\x00\x42\x06\n\x04type\".\n\x0bZipArtifact\x12\x0c\n\x04path\x18\x01 \x01(\t\x12\x11\n\tmain_file\x18\x02 \x01(\tB\\\n/com.qwak.ai.feature.store.v1.common.source_codeP\x01Z\'qwak/featurestore/v1/common;source_codeb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:qwak/feature_store/v1/common/source_code/source_code.proto\x12(qwak.feature.store.v1.common.source_code\x1a@qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto\"`\n\x0eSourceCodeSpec\x12N\n\x08\x61rtifact\x18\x01 \x01(\x0b\x32<.qwak.feature.store.v1.common.source_code.SourceCodeArtifact\"k\n\x12SourceCodeArtifact\x12M\n\x0czip_artifact\x18\x01 \x01(\x0b\x32\x35.qwak.feature.store.v1.common.source_code.ZipArtifactH\x00\x42\x06\n\x04type\"\x98\x01\n\x0bZipArtifact\x12\x0e\n\x04path\x18\x01 \x01(\tH\x00\x12Y\n\x13jfrog_artifact_path\x18\x03 \x01(\x0b\x32:.qwak.feature.store.v1.common.jfrog_artifact.JfrogArtifactH\x00\x12\x11\n\tmain_file\x18\x02 \x01(\tB\x0b\n\tpath_typeBg\n/com.qwak.ai.feature.store.v1.common.source_codeP\x01Z2qwak/featurestore/v1/common/source_code;sourcecodeb\x06proto3')
 
 
 
@@ -45,11 +46,11 @@ _sym_db.RegisterMessage(ZipArtifact)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.v1.common.source_codeP\001Z\'qwak/featurestore/v1/common;source_code'
-  _SOURCECODESPEC._serialized_start=104
-  _SOURCECODESPEC._serialized_end=200
-  _SOURCECODEARTIFACT._serialized_start=202
-  _SOURCECODEARTIFACT._serialized_end=309
-  _ZIPARTIFACT._serialized_start=311
-  _ZIPARTIFACT._serialized_end=357
+  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.v1.common.source_codeP\001Z2qwak/featurestore/v1/common/source_code;sourcecode'
+  _SOURCECODESPEC._serialized_start=170
+  _SOURCECODESPEC._serialized_end=266
+  _SOURCECODEARTIFACT._serialized_start=268
+  _SOURCECODEARTIFACT._serialized_end=375
+  _ZIPARTIFACT._serialized_start=378
+  _ZIPARTIFACT._serialized_end=530
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/feature_store/v1/common/source_code/source_code_pb2.pyi

@@ -5,6 +5,7 @@ isort:skip_file
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.message
+import frogml_proto.qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2
 import sys
 
 if sys.version_info >= (3, 8):
@@ -51,9 +52,13 @@ class ZipArtifact(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     PATH_FIELD_NUMBER: builtins.int
+    JFROG_ARTIFACT_PATH_FIELD_NUMBER: builtins.int
     MAIN_FILE_FIELD_NUMBER: builtins.int
     path: builtins.str
-    """Path to artifact, e.g: s3://some-bucket/file.zip"""
+    """Path to artifact, used by Qwak users e.g: s3://some-bucket/file.zip"""
+    @property
+    def jfrog_artifact_path(self) -> frogml_proto.qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2.JfrogArtifact:
+        """Path to the artifact stored in Artifactory"""
     main_file: builtins.str
     """Relative path inside the zip, e.g:
     the file the featureset is defined
@@ -62,8 +67,11 @@ class ZipArtifact(google.protobuf.message.Message):
         self,
         *,
         path: builtins.str = ...,
+        jfrog_artifact_path: frogml_proto.qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2.JfrogArtifact | None = ...,
         main_file: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["main_file", b"main_file", "path", b"path"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["jfrog_artifact_path", b"jfrog_artifact_path", "path", b"path", "path_type", b"path_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["jfrog_artifact_path", b"jfrog_artifact_path", "main_file", b"main_file", "path", b"path", "path_type", b"path_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["path_type", b"path_type"]) -> typing_extensions.Literal["path", "jfrog_artifact_path"] | None: ...
 
 global___ZipArtifact = ZipArtifact

frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nDqwak/feature_store/v1/internal/data_source/data_source_service.proto\x12\x31qwak.feature.store.management.v1.internal.sources\"E\n)GetDataSourceSourceCodeDownloadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"B\n*GetDataSourceSourceCodeDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\"@\n$GetDataSourceSourceCodeUploadRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\";\n%GetDataSourceSourceCodeUploadResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xcf\x03\n\x11\x44\x61taSourceService\x12\xd5\x01\n GetDataSourceSourceCodeUploadURL\x12W.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadRequest\x1aX.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse\x12\xe1\x01\n\"GetDataSourceSourceCodeDownloadURL\x12\\.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLRequest\x1a].qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLResponseBw\n<com.qwak.ai.feature.store.management.v1.api.internal.sourcesP\x01Z5qwak/featurestore/sources;featurestoreinternalsourcesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nDqwak/feature_store/v1/internal/data_source/data_source_service.proto\x12\x31qwak.feature.store.management.v1.internal.sources\"E\n)GetDataSourceSourceCodeDownloadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"B\n*GetDataSourceSourceCodeDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\"@\n$GetDataSourceSourceCodeUploadRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"\xf4\x01\n%GetDataSourceSourceCodeUploadResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x81\x01\n\rextra_headers\x18\x02 \x03(\x0b\x32j.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry\x1a\x33\n\x11\x45xtraHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x32\xcf\x03\n\x11\x44\x61taSourceService\x12\xd5\x01\n GetDataSourceSourceCodeUploadURL\x12W.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadRequest\x1aX.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse\x12\xe1\x01\n\"GetDataSourceSourceCodeDownloadURL\x12\\.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLRequest\x1a].qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLResponseBw\n<com.qwak.ai.feature.store.management.v1.api.internal.sourcesP\x01Z5qwak/featurestore/sources;featurestoreinternalsourcesb\x06proto3')
 
 
 
@@ -22,6 +22,7 @@ _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST = DESCRIPTOR.message_types_by_name['G
 _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeDownloadURLResponse']
 _GETDATASOURCESOURCECODEUPLOADREQUEST = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeUploadRequest']
 _GETDATASOURCESOURCECODEUPLOADRESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeUploadResponse']
+_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY = _GETDATASOURCESOURCECODEUPLOADRESPONSE.nested_types_by_name['ExtraHeadersEntry']
 GetDataSourceSourceCodeDownloadURLRequest = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeDownloadURLRequest', (_message.Message,), {
   'DESCRIPTOR' : _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST,
   '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
@@ -44,25 +45,37 @@ GetDataSourceSourceCodeUploadRequest = _reflection.GeneratedProtocolMessageType(
 _sym_db.RegisterMessage(GetDataSourceSourceCodeUploadRequest)
 
 GetDataSourceSourceCodeUploadResponse = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeUploadResponse', (_message.Message,), {
+
+  'ExtraHeadersEntry' : _reflection.GeneratedProtocolMessageType('ExtraHeadersEntry', (_message.Message,), {
+    'DESCRIPTOR' : _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY,
+    '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
+    # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry)
+    })
+  ,
   'DESCRIPTOR' : _GETDATASOURCESOURCECODEUPLOADRESPONSE,
   '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
   # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse)
   })
 _sym_db.RegisterMessage(GetDataSourceSourceCodeUploadResponse)
+_sym_db.RegisterMessage(GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry)
 
 _DATASOURCESERVICE = DESCRIPTOR.services_by_name['DataSourceService']
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n<com.qwak.ai.feature.store.management.v1.api.internal.sourcesP\001Z5qwak/featurestore/sources;featurestoreinternalsources'
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._options = None
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_options = b'8\001'
   _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST._serialized_start=123
   _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST._serialized_end=192
   _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE._serialized_start=194
   _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE._serialized_end=260
   _GETDATASOURCESOURCECODEUPLOADREQUEST._serialized_start=262
   _GETDATASOURCESOURCECODEUPLOADREQUEST._serialized_end=326
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_start=328
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_end=387
-  _DATASOURCESERVICE._serialized_start=390
-  _DATASOURCESERVICE._serialized_end=853
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_start=329
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_end=573
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_start=522
+  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_end=573
+  _DATASOURCESERVICE._serialized_start=576
+  _DATASOURCESERVICE._serialized_end=1039
 # @@protoc_insertion_point(module_scope)

frogml_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2.pyi

@@ -3,7 +3,9 @@
 isort:skip_file
 """
 import builtins
+import collections.abc
 import google.protobuf.descriptor
+import google.protobuf.internal.containers
 import google.protobuf.message
 import sys
 
@@ -59,13 +61,34 @@ global___GetDataSourceSourceCodeUploadRequest = GetDataSourceSourceCodeUploadReq
 class GetDataSourceSourceCodeUploadResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    class ExtraHeadersEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
+            *,
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+
     UPLOAD_URL_FIELD_NUMBER: builtins.int
+    EXTRA_HEADERS_FIELD_NUMBER: builtins.int
     upload_url: builtins.str
+    """Upload url intended for the data source source code"""
+    @property
+    def extra_headers(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
+        """Extra headers that are required for the upload"""
     def __init__(
         self,
         *,
         upload_url: builtins.str = ...,
+        extra_headers: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["upload_url", b"upload_url"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["extra_headers", b"extra_headers", "upload_url", b"upload_url"]) -> None: ...
 
 global___GetDataSourceSourceCodeUploadResponse = GetDataSourceSourceCodeUploadResponse