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

_qwak_proto/qwak/feature_store/sources/streaming_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/feature_store/sources/streaming.proto
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -17,119 +15,9 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*qwak/feature_store/sources/streaming.proto\x12\x1aqwak.feature.store.sources\"\xc4\x01\n\x0fStreamingSource\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x42\n\x0bkafkaSource\x18\x03 \x01(\x0b\x32\'.qwak.feature.store.sources.KafkaSourceB\x02\x18\x01H\x00\x12\x42\n\rkafkaSourceV1\x18\x04 \x01(\x0b\x32).qwak.feature.store.sources.KafkaSourceV1H\x00\x42\x06\n\x04type\"\x9f\x01\n\x0bKafkaSource\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x01(\t\x12\x16\n\x0e\x63onsumer_topic\x18\x02 \x01(\t\x12\x1c\n\x14username_secret_name\x18\x03 \x01(\t\x12\x1c\n\x14password_secret_name\x18\x04 \x01(\t\x12\x1e\n\x16\x63onsumer_configuration\x18\x05 \x01(\t:\x02\x18\x01\"\xbd\x04\n\rKafkaSourceV1\x12\x19\n\x11\x62ootstrap_servers\x18\x01 \x01(\t\x12\x10\n\x06\x61ssign\x18\x02 \x01(\tH\x00\x12\x13\n\tsubscribe\x18\x03 \x01(\tH\x00\x12\x1b\n\x11subscribe_pattern\x18\x04 \x01(\tH\x00\x12T\n\x0esecret_configs\x18\x07 \x03(\x0b\x32<.qwak.feature.store.sources.KafkaSourceV1.SecretConfigsEntry\x12^\n\x13passthrough_configs\x18\x08 \x03(\x0b\x32\x41.qwak.feature.store.sources.KafkaSourceV1.PassthroughConfigsEntry\x12\x44\n\x0f\x64\x65serialization\x18\t \x01(\x0b\x32+.qwak.feature.store.sources.Deserialization\x12I\n\x15\x61uthentication_method\x18\n \x01(\x0b\x32*.qwak.feature.store.sources.Authentication\x1a\x34\n\x12SecretConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x39\n\x17PassthroughConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x15\n\x13topic_configuration\"\xd8\x01\n\x0e\x41uthentication\x12>\n\x12sasl_configuration\x18\x01 \x01(\x0b\x32 .qwak.feature.store.sources.SaslH\x00\x12@\n\x13plain_configuration\x18\x02 \x01(\x0b\x32!.qwak.feature.store.sources.PlainH\x00\x12<\n\x11ssl_configuration\x18\x03 \x01(\x0b\x32\x1f.qwak.feature.store.sources.SslH\x00\x42\x06\n\x04type\"\x05\n\x03Ssl\"\x07\n\x05Plain\"\xc4\x01\n\x04Sasl\x12\x17\n\x0fusername_secret\x18\x01 \x01(\t\x12\x17\n\x0fpassword_secret\x18\x02 \x01(\t\x12\x41\n\x0esasl_mechanism\x18\x03 \x01(\x0e\x32).qwak.feature.store.sources.SaslMechanism\x12G\n\x11security_protocol\x18\x04 \x01(\x0e\x32,.qwak.feature.store.sources.SecurityProtocol\"\xb9\x01\n\x0f\x44\x65serialization\x12M\n\x13\x63ustom_deserializer\x18\x01 \x01(\x0b\x32..qwak.feature.store.sources.CustomDeserializerH\x00\x12O\n\x14generic_deserializer\x18\x02 \x01(\x0b\x32/.qwak.feature.store.sources.GenericDeserializerH\x00\x42\x06\n\x04type\"B\n\x12\x43ustomDeserializer\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x15\n\rartifact_path\x18\x02 \x01(\t\"m\n\x13GenericDeserializer\x12\x46\n\x13\x64\x65serializer_format\x18\x01 \x01(\x0e\x32).qwak.feature.store.sources.MessageFormat\x12\x0e\n\x06schema\x18\x02 \x01(\t*<\n\rSaslMechanism\x12\x0f\n\x0bSCRAMSHA256\x10\x00\x12\x0f\n\x0bSCRAMSHA512\x10\x01\x12\t\n\x05PLAIN\x10\x02* \n\x10SecurityProtocol\x12\x0c\n\x08SASL_SSL\x10\x00*0\n\rMessageFormat\x12\x0b\n\x07INVALID\x10\x00\x12\x08\n\x04JSON\x10\x01\x12\x08\n\x04\x41VRO\x10\x02\x42X\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
-_SASLMECHANISM = DESCRIPTOR.enum_types_by_name['SaslMechanism']
-SaslMechanism = enum_type_wrapper.EnumTypeWrapper(_SASLMECHANISM)
-_SECURITYPROTOCOL = DESCRIPTOR.enum_types_by_name['SecurityProtocol']
-SecurityProtocol = enum_type_wrapper.EnumTypeWrapper(_SECURITYPROTOCOL)
-_MESSAGEFORMAT = DESCRIPTOR.enum_types_by_name['MessageFormat']
-MessageFormat = enum_type_wrapper.EnumTypeWrapper(_MESSAGEFORMAT)
-SCRAMSHA256 = 0
-SCRAMSHA512 = 1
-PLAIN = 2
-SASL_SSL = 0
-INVALID = 0
-JSON = 1
-AVRO = 2
-
-
-_STREAMINGSOURCE = DESCRIPTOR.message_types_by_name['StreamingSource']
-_KAFKASOURCE = DESCRIPTOR.message_types_by_name['KafkaSource']
-_KAFKASOURCEV1 = DESCRIPTOR.message_types_by_name['KafkaSourceV1']
-_KAFKASOURCEV1_SECRETCONFIGSENTRY = _KAFKASOURCEV1.nested_types_by_name['SecretConfigsEntry']
-_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY = _KAFKASOURCEV1.nested_types_by_name['PassthroughConfigsEntry']
-_AUTHENTICATION = DESCRIPTOR.message_types_by_name['Authentication']
-_SSL = DESCRIPTOR.message_types_by_name['Ssl']
-_PLAIN = DESCRIPTOR.message_types_by_name['Plain']
-_SASL = DESCRIPTOR.message_types_by_name['Sasl']
-_DESERIALIZATION = DESCRIPTOR.message_types_by_name['Deserialization']
-_CUSTOMDESERIALIZER = DESCRIPTOR.message_types_by_name['CustomDeserializer']
-_GENERICDESERIALIZER = DESCRIPTOR.message_types_by_name['GenericDeserializer']
-StreamingSource = _reflection.GeneratedProtocolMessageType('StreamingSource', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGSOURCE,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.StreamingSource)
-  })
-_sym_db.RegisterMessage(StreamingSource)
-
-KafkaSource = _reflection.GeneratedProtocolMessageType('KafkaSource', (_message.Message,), {
-  'DESCRIPTOR' : _KAFKASOURCE,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSource)
-  })
-_sym_db.RegisterMessage(KafkaSource)
-
-KafkaSourceV1 = _reflection.GeneratedProtocolMessageType('KafkaSourceV1', (_message.Message,), {
-
-  'SecretConfigsEntry' : _reflection.GeneratedProtocolMessageType('SecretConfigsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKASOURCEV1_SECRETCONFIGSENTRY,
-    '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1.SecretConfigsEntry)
-    })
-  ,
-
-  'PassthroughConfigsEntry' : _reflection.GeneratedProtocolMessageType('PassthroughConfigsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY,
-    '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1.PassthroughConfigsEntry)
-    })
-  ,
-  'DESCRIPTOR' : _KAFKASOURCEV1,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1)
-  })
-_sym_db.RegisterMessage(KafkaSourceV1)
-_sym_db.RegisterMessage(KafkaSourceV1.SecretConfigsEntry)
-_sym_db.RegisterMessage(KafkaSourceV1.PassthroughConfigsEntry)
-
-Authentication = _reflection.GeneratedProtocolMessageType('Authentication', (_message.Message,), {
-  'DESCRIPTOR' : _AUTHENTICATION,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Authentication)
-  })
-_sym_db.RegisterMessage(Authentication)
-
-Ssl = _reflection.GeneratedProtocolMessageType('Ssl', (_message.Message,), {
-  'DESCRIPTOR' : _SSL,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Ssl)
-  })
-_sym_db.RegisterMessage(Ssl)
-
-Plain = _reflection.GeneratedProtocolMessageType('Plain', (_message.Message,), {
-  'DESCRIPTOR' : _PLAIN,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Plain)
-  })
-_sym_db.RegisterMessage(Plain)
-
-Sasl = _reflection.GeneratedProtocolMessageType('Sasl', (_message.Message,), {
-  'DESCRIPTOR' : _SASL,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Sasl)
-  })
-_sym_db.RegisterMessage(Sasl)
-
-Deserialization = _reflection.GeneratedProtocolMessageType('Deserialization', (_message.Message,), {
-  'DESCRIPTOR' : _DESERIALIZATION,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Deserialization)
-  })
-_sym_db.RegisterMessage(Deserialization)
-
-CustomDeserializer = _reflection.GeneratedProtocolMessageType('CustomDeserializer', (_message.Message,), {
-  'DESCRIPTOR' : _CUSTOMDESERIALIZER,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CustomDeserializer)
-  })
-_sym_db.RegisterMessage(CustomDeserializer)
-
-GenericDeserializer = _reflection.GeneratedProtocolMessageType('GenericDeserializer', (_message.Message,), {
-  'DESCRIPTOR' : _GENERICDESERIALIZER,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GenericDeserializer)
-  })
-_sym_db.RegisterMessage(GenericDeserializer)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sources.streaming_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -142,34 +30,34 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_options = b'8\001'
   _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._options = None
   _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_options = b'8\001'
-  _SASLMECHANISM._serialized_start=1812
-  _SASLMECHANISM._serialized_end=1872
-  _SECURITYPROTOCOL._serialized_start=1874
-  _SECURITYPROTOCOL._serialized_end=1906
-  _MESSAGEFORMAT._serialized_start=1908
-  _MESSAGEFORMAT._serialized_end=1956
-  _STREAMINGSOURCE._serialized_start=75
-  _STREAMINGSOURCE._serialized_end=271
-  _KAFKASOURCE._serialized_start=274
-  _KAFKASOURCE._serialized_end=433
-  _KAFKASOURCEV1._serialized_start=436
-  _KAFKASOURCEV1._serialized_end=1009
-  _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_start=875
-  _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_end=927
-  _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_start=929
-  _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_end=986
-  _AUTHENTICATION._serialized_start=1012
-  _AUTHENTICATION._serialized_end=1228
-  _SSL._serialized_start=1230
-  _SSL._serialized_end=1235
-  _PLAIN._serialized_start=1237
-  _PLAIN._serialized_end=1244
-  _SASL._serialized_start=1247
-  _SASL._serialized_end=1443
-  _DESERIALIZATION._serialized_start=1446
-  _DESERIALIZATION._serialized_end=1631
-  _CUSTOMDESERIALIZER._serialized_start=1633
-  _CUSTOMDESERIALIZER._serialized_end=1699
-  _GENERICDESERIALIZER._serialized_start=1701
-  _GENERICDESERIALIZER._serialized_end=1810
+  _globals['_SASLMECHANISM']._serialized_start=1812
+  _globals['_SASLMECHANISM']._serialized_end=1872
+  _globals['_SECURITYPROTOCOL']._serialized_start=1874
+  _globals['_SECURITYPROTOCOL']._serialized_end=1906
+  _globals['_MESSAGEFORMAT']._serialized_start=1908
+  _globals['_MESSAGEFORMAT']._serialized_end=1956
+  _globals['_STREAMINGSOURCE']._serialized_start=75
+  _globals['_STREAMINGSOURCE']._serialized_end=271
+  _globals['_KAFKASOURCE']._serialized_start=274
+  _globals['_KAFKASOURCE']._serialized_end=433
+  _globals['_KAFKASOURCEV1']._serialized_start=436
+  _globals['_KAFKASOURCEV1']._serialized_end=1009
+  _globals['_KAFKASOURCEV1_SECRETCONFIGSENTRY']._serialized_start=875
+  _globals['_KAFKASOURCEV1_SECRETCONFIGSENTRY']._serialized_end=927
+  _globals['_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY']._serialized_start=929
+  _globals['_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY']._serialized_end=986
+  _globals['_AUTHENTICATION']._serialized_start=1012
+  _globals['_AUTHENTICATION']._serialized_end=1228
+  _globals['_SSL']._serialized_start=1230
+  _globals['_SSL']._serialized_end=1235
+  _globals['_PLAIN']._serialized_start=1237
+  _globals['_PLAIN']._serialized_end=1244
+  _globals['_SASL']._serialized_start=1247
+  _globals['_SASL']._serialized_end=1443
+  _globals['_DESERIALIZATION']._serialized_start=1446
+  _globals['_DESERIALIZATION']._serialized_end=1631
+  _globals['_CUSTOMDESERIALIZER']._serialized_start=1633
+  _globals['_CUSTOMDESERIALIZER']._serialized_end=1699
+  _globals['_GENERICDESERIALIZER']._serialized_start=1701
+  _globals['_GENERICDESERIALIZER']._serialized_end=1810
 # @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: 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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n@qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto\x12+qwak.feature.store.v1.common.jfrog_artifact\"\x1d\n\rJfrogArtifact\x12\x0c\n\x04path\x18\x01 \x01(\tBp\n2com.qwak.ai.feature.store.v1.common.jfrog_artifactP\x01Z8qwak/featurestore/v1/common/jfrog_artifact;jfrogartifactb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2', _globals)
+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'
+  _globals['_JFROGARTIFACT']._serialized_start=113
+  _globals['_JFROGARTIFACT']._serialized_end=142
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,30 @@
+"""
+@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
+    path: builtins.str
+    """artifact path"""
+    def __init__(
+        self,
+        *,
+        path: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["path", b"path"]) -> None: ...
+
+global___JfrogArtifact = JfrogArtifact

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

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

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/feature_store/v1/common/source_code/source_code.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _qwak_proto.qwak.feature_store.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\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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.common.source_code.source_code_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.v1.common.source_codeP\001Z2qwak/featurestore/v1/common/source_code;sourcecode'
+  _globals['_SOURCECODESPEC']._serialized_start=170
+  _globals['_SOURCECODESPEC']._serialized_end=266
+  _globals['_SOURCECODEARTIFACT']._serialized_start=268
+  _globals['_SOURCECODEARTIFACT']._serialized_end=375
+  _globals['_ZIPARTIFACT']._serialized_start=378
+  _globals['_ZIPARTIFACT']._serialized_end=530
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,77 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class SourceCodeSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ARTIFACT_FIELD_NUMBER: builtins.int
+    @property
+    def artifact(self) -> global___SourceCodeArtifact: ...
+    def __init__(
+        self,
+        *,
+        artifact: global___SourceCodeArtifact | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["artifact", b"artifact"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["artifact", b"artifact"]) -> None: ...
+
+global___SourceCodeSpec = SourceCodeSpec
+
+class SourceCodeArtifact(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ZIP_ARTIFACT_FIELD_NUMBER: builtins.int
+    @property
+    def zip_artifact(self) -> global___ZipArtifact: ...
+    def __init__(
+        self,
+        *,
+        zip_artifact: global___ZipArtifact | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["type", b"type", "zip_artifact", b"zip_artifact"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["type", b"type", "zip_artifact", b"zip_artifact"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["zip_artifact"] | None: ...
+
+global___SourceCodeArtifact = SourceCodeArtifact
+
+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, used by Qwak users e.g: s3://some-bucket/file.zip"""
+    @property
+    def jfrog_artifact_path(self) -> 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
+    """
+    def __init__(
+        self,
+        *,
+        path: builtins.str = ...,
+        jfrog_artifact_path: qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2.JfrogArtifact | None = ...,
+        main_file: builtins.str = ...,
+    ) -> 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

_qwak_proto/qwak/feature_store/v1/common/source_code/source_code_pb2_grpc.py

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

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

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/feature_store/v1/internal/data_source/data_source_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+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\"q\n$GetDataSourceSourceCodeUploadRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x19\n\x0frepository_name\x18\x02 \x01(\tH\x00\x42\x14\n\x12repository_details\"\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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2', _globals)
+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'
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLREQUEST']._serialized_start=123
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLREQUEST']._serialized_end=192
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE']._serialized_start=194
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE']._serialized_end=260
+  _globals['_GETDATASOURCESOURCECODEUPLOADREQUEST']._serialized_start=262
+  _globals['_GETDATASOURCESOURCECODEUPLOADREQUEST']._serialized_end=375
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE']._serialized_start=378
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE']._serialized_end=622
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY']._serialized_start=571
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY']._serialized_end=622
+  _globals['_DATASOURCESERVICE']._serialized_start=625
+  _globals['_DATASOURCESERVICE']._serialized_end=1088
+# @@protoc_insertion_point(module_scope)

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

@@ -0,0 +1,99 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class GetDataSourceSourceCodeDownloadURLRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_SOURCE_NAME_FIELD_NUMBER: builtins.int
+    data_source_name: builtins.str
+    def __init__(
+        self,
+        *,
+        data_source_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_source_name", b"data_source_name"]) -> None: ...
+
+global___GetDataSourceSourceCodeDownloadURLRequest = GetDataSourceSourceCodeDownloadURLRequest
+
+class GetDataSourceSourceCodeDownloadURLResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOWNLOAD_URL_FIELD_NUMBER: builtins.int
+    download_url: builtins.str
+    def __init__(
+        self,
+        *,
+        download_url: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["download_url", b"download_url"]) -> None: ...
+
+global___GetDataSourceSourceCodeDownloadURLResponse = GetDataSourceSourceCodeDownloadURLResponse
+
+class GetDataSourceSourceCodeUploadRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_SOURCE_NAME_FIELD_NUMBER: builtins.int
+    REPOSITORY_NAME_FIELD_NUMBER: builtins.int
+    data_source_name: builtins.str
+    repository_name: builtins.str
+    def __init__(
+        self,
+        *,
+        data_source_name: builtins.str = ...,
+        repository_name: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["repository_details", b"repository_details", "repository_name", b"repository_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_source_name", b"data_source_name", "repository_details", b"repository_details", "repository_name", b"repository_name"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["repository_details", b"repository_details"]) -> typing_extensions.Literal["repository_name"] | None: ...
+
+global___GetDataSourceSourceCodeUploadRequest = GetDataSourceSourceCodeUploadRequest
+
+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["extra_headers", b"extra_headers", "upload_url", b"upload_url"]) -> None: ...
+
+global___GetDataSourceSourceCodeUploadResponse = GetDataSourceSourceCodeUploadResponse

_qwak_proto/qwak/feature_store/v1/internal/data_source/data_source_service_pb2_grpc.py

@@ -0,0 +1,99 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from _qwak_proto.qwak.feature_store.v1.internal.data_source import data_source_service_pb2 as qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2
+
+
+class DataSourceServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetDataSourceSourceCodeUploadURL = channel.unary_unary(
+                '/qwak.feature.store.management.v1.internal.sources.DataSourceService/GetDataSourceSourceCodeUploadURL',
+                request_serializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadResponse.FromString,
+                )
+        self.GetDataSourceSourceCodeDownloadURL = channel.unary_unary(
+                '/qwak.feature.store.management.v1.internal.sources.DataSourceService/GetDataSourceSourceCodeDownloadURL',
+                request_serializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLResponse.FromString,
+                )
+
+
+class DataSourceServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetDataSourceSourceCodeUploadURL(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetDataSourceSourceCodeDownloadURL(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_DataSourceServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetDataSourceSourceCodeUploadURL': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetDataSourceSourceCodeUploadURL,
+                    request_deserializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadResponse.SerializeToString,
+            ),
+            'GetDataSourceSourceCodeDownloadURL': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetDataSourceSourceCodeDownloadURL,
+                    request_deserializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.feature.store.management.v1.internal.sources.DataSourceService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class DataSourceService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetDataSourceSourceCodeUploadURL(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.feature.store.management.v1.internal.sources.DataSourceService/GetDataSourceSourceCodeUploadURL',
+            qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadRequest.SerializeToString,
+            qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeUploadResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetDataSourceSourceCodeDownloadURL(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.feature.store.management.v1.internal.sources.DataSourceService/GetDataSourceSourceCodeDownloadURL',
+            qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLRequest.SerializeToString,
+            qwak_dot_feature__store_dot_v1_dot_internal_dot_data__source_dot_data__source__service__pb2.GetDataSourceSourceCodeDownloadURLResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)