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

_qwak_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2.pyi

@@ -0,0 +1,279 @@
+"""
+@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.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.timestamp_pb2
+import qwak.offline.serving.v1.feature_values_pb2
+import qwak.offline.serving.v1.options_pb2
+import qwak.offline.serving.v1.population_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _FeatureValuesRequestStatus:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _FeatureValuesRequestStatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FeatureValuesRequestStatus.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    FEATURE_VALUES_REQUEST_STATUS_INVALID: _FeatureValuesRequestStatus.ValueType  # 0
+    """Invalid status"""
+    FEATURE_VALUES_REQUEST_STATUS_SUCCEEDED: _FeatureValuesRequestStatus.ValueType  # 1
+    """Request Success"""
+    FEATURE_VALUES_REQUEST_STATUS_PENDING: _FeatureValuesRequestStatus.ValueType  # 2
+    """Request Pending"""
+    FEATURE_VALUES_REQUEST_STATUS_CANCELLED: _FeatureValuesRequestStatus.ValueType  # 3
+    """Request Canceled"""
+    FEATURE_VALUES_REQUEST_STATUS_FAILED: _FeatureValuesRequestStatus.ValueType  # 4
+    """Request Failed"""
+
+class FeatureValuesRequestStatus(_FeatureValuesRequestStatus, metaclass=_FeatureValuesRequestStatusEnumTypeWrapper): ...
+
+FEATURE_VALUES_REQUEST_STATUS_INVALID: FeatureValuesRequestStatus.ValueType  # 0
+"""Invalid status"""
+FEATURE_VALUES_REQUEST_STATUS_SUCCEEDED: FeatureValuesRequestStatus.ValueType  # 1
+"""Request Success"""
+FEATURE_VALUES_REQUEST_STATUS_PENDING: FeatureValuesRequestStatus.ValueType  # 2
+"""Request Pending"""
+FEATURE_VALUES_REQUEST_STATUS_CANCELLED: FeatureValuesRequestStatus.ValueType  # 3
+"""Request Canceled"""
+FEATURE_VALUES_REQUEST_STATUS_FAILED: FeatureValuesRequestStatus.ValueType  # 4
+"""Request Failed"""
+global___FeatureValuesRequestStatus = FeatureValuesRequestStatus
+
+class _FileFormat:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _FileFormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FileFormat.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    FILE_FORMAT_INVALID: _FileFormat.ValueType  # 0
+    """Invalid format"""
+    FILE_FORMAT_CSV: _FileFormat.ValueType  # 1
+    """Csv format"""
+
+class FileFormat(_FileFormat, metaclass=_FileFormatEnumTypeWrapper): ...
+
+FILE_FORMAT_INVALID: FileFormat.ValueType  # 0
+"""Invalid format"""
+FILE_FORMAT_CSV: FileFormat.ValueType  # 1
+"""Csv format"""
+global___FileFormat = FileFormat
+
+class GetFileUploadUrlRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    POPULATION_FIELD_NUMBER: builtins.int
+    @property
+    def population(self) -> qwak.offline.serving.v1.population_pb2.PopulationFileUploadUrlType:
+        """Used to Get File Upload Url for population dataset"""
+    def __init__(
+        self,
+        *,
+        population: qwak.offline.serving.v1.population_pb2.PopulationFileUploadUrlType | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["population", b"population", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["population", b"population", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["population"] | None: ...
+
+global___GetFileUploadUrlRequest = GetFileUploadUrlRequest
+
+class GetFileUploadUrlResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_UPLOAD_URL_FIELD_NUMBER: builtins.int
+    file_upload_url: builtins.str
+    """The generated url to which the file can be uploaded"""
+    def __init__(
+        self,
+        *,
+        file_upload_url: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file_upload_url", b"file_upload_url"]) -> None: ...
+
+global___GetFileUploadUrlResponse = GetFileUploadUrlResponse
+
+class GetFeatureValuesRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURES_FIELD_NUMBER: builtins.int
+    POPULATION_FIELD_NUMBER: builtins.int
+    RESULT_FILE_FORMAT_FIELD_NUMBER: builtins.int
+    OPTIONS_FIELD_NUMBER: builtins.int
+    @property
+    def features(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.offline.serving.v1.feature_values_pb2.FeaturesetFeatures]:
+        """a List of featureset name and it's feature names"""
+    @property
+    def population(self) -> qwak.offline.serving.v1.population_pb2.TimedPopulation:
+        """The population dataset with timestamp column"""
+    result_file_format: global___FileFormat.ValueType
+    """The format of the result files"""
+    @property
+    def options(self) -> qwak.offline.serving.v1.options_pb2.OfflineServingQueryOptions:
+        """Query Options"""
+    def __init__(
+        self,
+        *,
+        features: collections.abc.Iterable[qwak.offline.serving.v1.feature_values_pb2.FeaturesetFeatures] | None = ...,
+        population: qwak.offline.serving.v1.population_pb2.TimedPopulation | None = ...,
+        result_file_format: global___FileFormat.ValueType = ...,
+        options: qwak.offline.serving.v1.options_pb2.OfflineServingQueryOptions | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["options", b"options", "population", b"population"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["features", b"features", "options", b"options", "population", b"population", "result_file_format", b"result_file_format"]) -> None: ...
+
+global___GetFeatureValuesRequest = GetFeatureValuesRequest
+
+class GetFeatureValuesResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REQUEST_ID_FIELD_NUMBER: builtins.int
+    request_id: builtins.str
+    """The request_id. used as a handler to retrieve results"""
+    def __init__(
+        self,
+        *,
+        request_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+
+global___GetFeatureValuesResponse = GetFeatureValuesResponse
+
+class GetFeatureValuesInRangeRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURES_FIELD_NUMBER: builtins.int
+    LOWER_TIME_BOUND_FIELD_NUMBER: builtins.int
+    UPPER_TIME_BOUND_FIELD_NUMBER: builtins.int
+    POPULATION_FIELD_NUMBER: builtins.int
+    RESULT_FILE_FORMAT_FIELD_NUMBER: builtins.int
+    OPTIONS_FIELD_NUMBER: builtins.int
+    @property
+    def features(self) -> qwak.offline.serving.v1.feature_values_pb2.FeaturesetFeatures:
+        """A single featureset name and it's feature names"""
+    @property
+    def lower_time_bound(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Lower time bound for range query"""
+    @property
+    def upper_time_bound(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        """Upper time bound for range query"""
+    @property
+    def population(self) -> qwak.offline.serving.v1.population_pb2.Population: ...
+    result_file_format: global___FileFormat.ValueType
+    """The format of the result files"""
+    @property
+    def options(self) -> qwak.offline.serving.v1.options_pb2.OfflineServingQueryOptions:
+        """Query Options"""
+    def __init__(
+        self,
+        *,
+        features: qwak.offline.serving.v1.feature_values_pb2.FeaturesetFeatures | None = ...,
+        lower_time_bound: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        upper_time_bound: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        population: qwak.offline.serving.v1.population_pb2.Population | None = ...,
+        result_file_format: global___FileFormat.ValueType = ...,
+        options: qwak.offline.serving.v1.options_pb2.OfflineServingQueryOptions | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["features", b"features", "lower_time_bound", b"lower_time_bound", "optional_population", b"optional_population", "options", b"options", "population", b"population", "upper_time_bound", b"upper_time_bound"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["features", b"features", "lower_time_bound", b"lower_time_bound", "optional_population", b"optional_population", "options", b"options", "population", b"population", "result_file_format", b"result_file_format", "upper_time_bound", b"upper_time_bound"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_population", b"optional_population"]) -> typing_extensions.Literal["population"] | None: ...
+
+global___GetFeatureValuesInRangeRequest = GetFeatureValuesInRangeRequest
+
+class GetFeatureValuesInRangeResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REQUEST_ID_FIELD_NUMBER: builtins.int
+    request_id: builtins.str
+    """The request_id. used as a handler to retrieve results"""
+    def __init__(
+        self,
+        *,
+        request_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+
+global___GetFeatureValuesInRangeResponse = GetFeatureValuesInRangeResponse
+
+class GetFeatureValuesResultRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    REQUEST_ID_FIELD_NUMBER: builtins.int
+    request_id: builtins.str
+    """The request_id handler returned by one of the Get-Feature-Values endpoints"""
+    def __init__(
+        self,
+        *,
+        request_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+
+global___GetFeatureValuesResultRequest = GetFeatureValuesResultRequest
+
+class GetFeatureValuesResultResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    STATUS_FIELD_NUMBER: builtins.int
+    LINK_TO_FILES_FIELD_NUMBER: builtins.int
+    FAILURE_REASON_FIELD_NUMBER: builtins.int
+    status: global___FeatureValuesRequestStatus.ValueType
+    """The status of the request"""
+    @property
+    def link_to_files(self) -> global___LinkToFiles: ...
+    @property
+    def failure_reason(self) -> global___FailureReason: ...
+    def __init__(
+        self,
+        *,
+        status: global___FeatureValuesRequestStatus.ValueType = ...,
+        link_to_files: global___LinkToFiles | None = ...,
+        failure_reason: global___FailureReason | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["failure_reason", b"failure_reason", "link_to_files", b"link_to_files", "payload", b"payload"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["failure_reason", b"failure_reason", "link_to_files", b"link_to_files", "payload", b"payload", "status", b"status"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["payload", b"payload"]) -> typing_extensions.Literal["link_to_files", "failure_reason"] | None: ...
+
+global___GetFeatureValuesResultResponse = GetFeatureValuesResultResponse
+
+class FailureReason(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MESSAGE_FIELD_NUMBER: builtins.int
+    message: builtins.str
+    """Failure reason"""
+    def __init__(
+        self,
+        *,
+        message: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["message", b"message"]) -> None: ...
+
+global___FailureReason = FailureReason
+
+class LinkToFiles(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    LINK_TO_FILES_FIELD_NUMBER: builtins.int
+    @property
+    def link_to_files(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """List of urls to the files"""
+    def __init__(
+        self,
+        *,
+        link_to_files: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["link_to_files", b"link_to_files"]) -> None: ...
+
+global___LinkToFiles = LinkToFiles

_qwak_proto/qwak/offline/serving/v1/offline_serving_async_service_pb2_grpc.py

@@ -0,0 +1,165 @@
+# 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.offline.serving.v1 import offline_serving_async_service_pb2 as qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2
+
+
+class FeatureStoreOfflineServingAsyncServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetFileUploadUrl = channel.unary_unary(
+                '/qwak.feature.store.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFileUploadUrl',
+                request_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlRequest.SerializeToString,
+                response_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlResponse.FromString,
+                )
+        self.GetFeatureValues = channel.unary_unary(
+                '/qwak.feature.store.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValues',
+                request_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesRequest.SerializeToString,
+                response_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResponse.FromString,
+                )
+        self.GetFeatureValuesInRange = channel.unary_unary(
+                '/qwak.feature.store.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValuesInRange',
+                request_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeRequest.SerializeToString,
+                response_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeResponse.FromString,
+                )
+        self.GetFeatureValuesResult = channel.unary_unary(
+                '/qwak.feature.store.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValuesResult',
+                request_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultRequest.SerializeToString,
+                response_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultResponse.FromString,
+                )
+
+
+class FeatureStoreOfflineServingAsyncServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetFileUploadUrl(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 GetFeatureValues(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 GetFeatureValuesInRange(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 GetFeatureValuesResult(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_FeatureStoreOfflineServingAsyncServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetFileUploadUrl': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFileUploadUrl,
+                    request_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlRequest.FromString,
+                    response_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlResponse.SerializeToString,
+            ),
+            'GetFeatureValues': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeatureValues,
+                    request_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesRequest.FromString,
+                    response_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResponse.SerializeToString,
+            ),
+            'GetFeatureValuesInRange': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeatureValuesInRange,
+                    request_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeRequest.FromString,
+                    response_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeResponse.SerializeToString,
+            ),
+            'GetFeatureValuesResult': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetFeatureValuesResult,
+                    request_deserializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultRequest.FromString,
+                    response_serializer=qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.feature.store.offline.serving.v1.FeatureStoreOfflineServingAsyncService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FeatureStoreOfflineServingAsyncService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetFileUploadUrl(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.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFileUploadUrl',
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlRequest.SerializeToString,
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFileUploadUrlResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeatureValues(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.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValues',
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesRequest.SerializeToString,
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeatureValuesInRange(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.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValuesInRange',
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeRequest.SerializeToString,
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesInRangeResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetFeatureValuesResult(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.offline.serving.v1.FeatureStoreOfflineServingAsyncService/GetFeatureValuesResult',
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultRequest.SerializeToString,
+            qwak_dot_offline_dot_serving_dot_v1_dot_offline__serving__async__service__pb2.GetFeatureValuesResultResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_qwak_proto/qwak/offline/serving/v1/options_pb2.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/offline/serving/v1/options.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/offline/serving/v1/options.proto\x12%qwak.feature.store.offline.serving.v1\"G\n\x1aOfflineServingQueryOptions\x12)\n!include_featureset_version_column\x18\x01 \x01(\x08\x42&\n\"com.qwak.ai.offline.serving.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.offline.serving.v1.options_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\"com.qwak.ai.offline.serving.api.v1P\001'
+  _globals['_OFFLINESERVINGQUERYOPTIONS']._serialized_start=80
+  _globals['_OFFLINESERVINGQUERYOPTIONS']._serialized_end=151
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/offline/serving/v1/options_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 OfflineServingQueryOptions(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    INCLUDE_FEATURESET_VERSION_COLUMN_FIELD_NUMBER: builtins.int
+    include_featureset_version_column: builtins.bool
+    """Whether or not to include featureset version as a column in the result set"""
+    def __init__(
+        self,
+        *,
+        include_featureset_version_column: builtins.bool = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["include_featureset_version_column", b"include_featureset_version_column"]) -> None: ...
+
+global___OfflineServingQueryOptions = OfflineServingQueryOptions

_qwak_proto/qwak/offline/serving/v1/options_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/offline/serving/v1/population_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/offline/serving/v1/population.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/offline/serving/v1/population.proto\x12%qwak.feature.store.offline.serving.v1\"\x1d\n\x1bPopulationFileUploadUrlType\"w\n\x0fTimedPopulation\x12\x1d\n\x15timestamp_column_name\x18\x01 \x01(\t\x12\x45\n\npopulation\x18\x02 \x01(\x0b\x32\x31.qwak.feature.store.offline.serving.v1.Population\"f\n\nPopulation\x12P\n\x0fpopulation_file\x18\x01 \x01(\x0b\x32\x35.qwak.feature.store.offline.serving.v1.PopulationFileH\x00\x42\x06\n\x04type\"\x81\x01\n\x0ePopulationFile\x12\x10\n\x08\x66ile_url\x18\x01 \x01(\t\x12S\n\x11\x61rrow_schema_json\x18\x02 \x01(\x0b\x32\x36.qwak.feature.store.offline.serving.v1.ArrowSchemaJsonH\x00\x42\x08\n\x06schema\"!\n\x0f\x41rrowSchemaJson\x12\x0e\n\x06schema\x18\x01 \x01(\tB&\n\"com.qwak.ai.offline.serving.api.v1P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.offline.serving.v1.population_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\"com.qwak.ai.offline.serving.api.v1P\001'
+  _globals['_POPULATIONFILEUPLOADURLTYPE']._serialized_start=83
+  _globals['_POPULATIONFILEUPLOADURLTYPE']._serialized_end=112
+  _globals['_TIMEDPOPULATION']._serialized_start=114
+  _globals['_TIMEDPOPULATION']._serialized_end=233
+  _globals['_POPULATION']._serialized_start=235
+  _globals['_POPULATION']._serialized_end=337
+  _globals['_POPULATIONFILE']._serialized_start=340
+  _globals['_POPULATIONFILE']._serialized_end=469
+  _globals['_ARROWSCHEMAJSON']._serialized_start=471
+  _globals['_ARROWSCHEMAJSON']._serialized_end=504
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/offline/serving/v1/population_pb2.pyi

@@ -0,0 +1,99 @@
+"""
+@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 PopulationFileUploadUrlType(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___PopulationFileUploadUrlType = PopulationFileUploadUrlType
+
+class TimedPopulation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TIMESTAMP_COLUMN_NAME_FIELD_NUMBER: builtins.int
+    POPULATION_FIELD_NUMBER: builtins.int
+    timestamp_column_name: builtins.str
+    """The name of the timestamp column"""
+    @property
+    def population(self) -> global___Population:
+        """The population dataset"""
+    def __init__(
+        self,
+        *,
+        timestamp_column_name: builtins.str = ...,
+        population: global___Population | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["population", b"population"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["population", b"population", "timestamp_column_name", b"timestamp_column_name"]) -> None: ...
+
+global___TimedPopulation = TimedPopulation
+
+class Population(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    POPULATION_FILE_FIELD_NUMBER: builtins.int
+    @property
+    def population_file(self) -> global___PopulationFile:
+        """Reference to population dataset file"""
+    def __init__(
+        self,
+        *,
+        population_file: global___PopulationFile | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["population_file", b"population_file", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["population_file", b"population_file", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["population_file"] | None: ...
+
+global___Population = Population
+
+class PopulationFile(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_URL_FIELD_NUMBER: builtins.int
+    ARROW_SCHEMA_JSON_FIELD_NUMBER: builtins.int
+    file_url: builtins.str
+    """Population dataset file url"""
+    @property
+    def arrow_schema_json(self) -> global___ArrowSchemaJson: ...
+    def __init__(
+        self,
+        *,
+        file_url: builtins.str = ...,
+        arrow_schema_json: global___ArrowSchemaJson | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["arrow_schema_json", b"arrow_schema_json", "schema", b"schema"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["arrow_schema_json", b"arrow_schema_json", "file_url", b"file_url", "schema", b"schema"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["schema", b"schema"]) -> typing_extensions.Literal["arrow_schema_json"] | None: ...
+
+global___PopulationFile = PopulationFile
+
+class ArrowSchemaJson(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SCHEMA_FIELD_NUMBER: builtins.int
+    schema: builtins.str
+    """Schema represented as json with arrow types e.g. {"col1": "double", "clo2": {"list": {"list": "int64"}}, "col3": {"struct": [{"field1": {"struct": [{"field1": "string"}]}}}"""
+    def __init__(
+        self,
+        *,
+        schema: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["schema", b"schema"]) -> None: ...
+
+global___ArrowSchemaJson = ArrowSchemaJson

_qwak_proto/qwak/offline/serving/v1/population_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
+