frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/user_application/common/v0/resources_pb2.pyi

@@ -0,0 +1,333 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+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 _MemoryUnit:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MemoryUnit.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_MEMORY_UNIT: _MemoryUnit.ValueType  # 0
+    MIB: _MemoryUnit.ValueType  # 1
+    GIB: _MemoryUnit.ValueType  # 2
+
+class MemoryUnit(_MemoryUnit, metaclass=_MemoryUnitEnumTypeWrapper): ...
+
+INVALID_MEMORY_UNIT: MemoryUnit.ValueType  # 0
+MIB: MemoryUnit.ValueType  # 1
+GIB: MemoryUnit.ValueType  # 2
+global___MemoryUnit = MemoryUnit
+
+class _GpuType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_GpuType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_GPU_TYPE: _GpuType.ValueType  # 0
+    NVIDIA_K80: _GpuType.ValueType  # 1
+    NVIDIA_V100: _GpuType.ValueType  # 2
+    NVIDIA_A100: _GpuType.ValueType  # 3
+    NVIDIA_T4: _GpuType.ValueType  # 4
+    NVIDIA_A10G: _GpuType.ValueType  # 5
+    NVIDIA_L4: _GpuType.ValueType  # 6
+    NVIDIA_T4_1_4_15: _GpuType.ValueType  # 7
+    """More specific node types
+    t4.xl
+    """
+    NVIDIA_T4_1_8_30: _GpuType.ValueType  # 8
+    """t4.2xl"""
+    NVIDIA_T4_1_16_60: _GpuType.ValueType  # 9
+    """t4.4xl"""
+    NVIDIA_A100_80GB_8_96_1360: _GpuType.ValueType  # 10
+    """a100.8xl"""
+    NVIDIA_V100_1_8_52: _GpuType.ValueType  # 11
+    """a100.xl"""
+    NVIDIA_V100_4_32_208: _GpuType.ValueType  # 12
+    """v100.4xl"""
+
+class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
+
+INVALID_GPU_TYPE: GpuType.ValueType  # 0
+NVIDIA_K80: GpuType.ValueType  # 1
+NVIDIA_V100: GpuType.ValueType  # 2
+NVIDIA_A100: GpuType.ValueType  # 3
+NVIDIA_T4: GpuType.ValueType  # 4
+NVIDIA_A10G: GpuType.ValueType  # 5
+NVIDIA_L4: GpuType.ValueType  # 6
+NVIDIA_T4_1_4_15: GpuType.ValueType  # 7
+"""More specific node types
+t4.xl
+"""
+NVIDIA_T4_1_8_30: GpuType.ValueType  # 8
+"""t4.2xl"""
+NVIDIA_T4_1_16_60: GpuType.ValueType  # 9
+"""t4.4xl"""
+NVIDIA_A100_80GB_8_96_1360: GpuType.ValueType  # 10
+"""a100.8xl"""
+NVIDIA_V100_1_8_52: GpuType.ValueType  # 11
+"""a100.xl"""
+NVIDIA_V100_4_32_208: GpuType.ValueType  # 12
+"""v100.4xl"""
+global___GpuType = GpuType
+
+class _SparkClusterResourceTemplate:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _SparkClusterResourceTemplateEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SparkClusterResourceTemplate.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    SPARK_CLUSTER_RESOURCE_INVALID: _SparkClusterResourceTemplate.ValueType  # 0
+    SPARK_CLUSTER_RESOURCE_SMALL: _SparkClusterResourceTemplate.ValueType  # 1
+    SPARK_CLUSTER_RESOURCE_MEDIUM: _SparkClusterResourceTemplate.ValueType  # 2
+    SPARK_CLUSTER_RESOURCE_LARGE: _SparkClusterResourceTemplate.ValueType  # 3
+    SPARK_CLUSTER_RESOURCE_XL: _SparkClusterResourceTemplate.ValueType  # 4
+    SPARK_CLUSTER_RESOURCE_2XL: _SparkClusterResourceTemplate.ValueType  # 5
+    SPARK_CLUSTER_RESOURCE_3XL: _SparkClusterResourceTemplate.ValueType  # 6
+
+class SparkClusterResourceTemplate(_SparkClusterResourceTemplate, metaclass=_SparkClusterResourceTemplateEnumTypeWrapper):
+    """Templates for clusters the user may choose"""
+
+SPARK_CLUSTER_RESOURCE_INVALID: SparkClusterResourceTemplate.ValueType  # 0
+SPARK_CLUSTER_RESOURCE_SMALL: SparkClusterResourceTemplate.ValueType  # 1
+SPARK_CLUSTER_RESOURCE_MEDIUM: SparkClusterResourceTemplate.ValueType  # 2
+SPARK_CLUSTER_RESOURCE_LARGE: SparkClusterResourceTemplate.ValueType  # 3
+SPARK_CLUSTER_RESOURCE_XL: SparkClusterResourceTemplate.ValueType  # 4
+SPARK_CLUSTER_RESOURCE_2XL: SparkClusterResourceTemplate.ValueType  # 5
+SPARK_CLUSTER_RESOURCE_3XL: SparkClusterResourceTemplate.ValueType  # 6
+global___SparkClusterResourceTemplate = SparkClusterResourceTemplate
+
+class _NodeOptimizationType:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _NodeOptimizationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_NodeOptimizationType.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    NODE_OPTIMIZATION_INVALID: _NodeOptimizationType.ValueType  # 0
+    NODE_OPTIMIZATION_NONE: _NodeOptimizationType.ValueType  # 1
+    NODE_OPTIMIZATION_CPU: _NodeOptimizationType.ValueType  # 2
+    NODE_OPTIMIZATION_MEMORY: _NodeOptimizationType.ValueType  # 3
+
+class NodeOptimizationType(_NodeOptimizationType, metaclass=_NodeOptimizationTypeEnumTypeWrapper): ...
+
+NODE_OPTIMIZATION_INVALID: NodeOptimizationType.ValueType  # 0
+NODE_OPTIMIZATION_NONE: NodeOptimizationType.ValueType  # 1
+NODE_OPTIMIZATION_CPU: NodeOptimizationType.ValueType  # 2
+NODE_OPTIMIZATION_MEMORY: NodeOptimizationType.ValueType  # 3
+global___NodeOptimizationType = NodeOptimizationType
+
+class CpuResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CPU_FIELD_NUMBER: builtins.int
+    MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
+    MEMORY_UNITS_FIELD_NUMBER: builtins.int
+    cpu: builtins.float
+    """Amount of CPU cores"""
+    memory_amount: builtins.int
+    """Amount of memory"""
+    memory_units: global___MemoryUnit.ValueType
+    """Units type of memory"""
+    def __init__(
+        self,
+        *,
+        cpu: builtins.float = ...,
+        memory_amount: builtins.int = ...,
+        memory_units: global___MemoryUnit.ValueType = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cpu", b"cpu", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
+
+global___CpuResources = CpuResources
+
+class GpuResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    GPU_TYPE_FIELD_NUMBER: builtins.int
+    GPU_AMOUNT_FIELD_NUMBER: builtins.int
+    gpu_type: global___GpuType.ValueType
+    """The type of the GPU"""
+    gpu_amount: builtins.int
+    """Amount of GPUs"""
+    def __init__(
+        self,
+        *,
+        gpu_type: global___GpuType.ValueType = ...,
+        gpu_amount: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type"]) -> None: ...
+
+global___GpuResources = GpuResources
+
+class SparkClusterResourceTemplateSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TEMPLATE_FIELD_NUMBER: builtins.int
+    template: global___SparkClusterResourceTemplate.ValueType
+    def __init__(
+        self,
+        *,
+        template: global___SparkClusterResourceTemplate.ValueType = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["template", b"template"]) -> None: ...
+
+global___SparkClusterResourceTemplateSpec = SparkClusterResourceTemplateSpec
+
+class SparkResourceConfiguration(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DRIVER_MEMORY_FIELD_NUMBER: builtins.int
+    DRIVER_CORES_FIELD_NUMBER: builtins.int
+    INITIAL_EXECUTORS_FIELD_NUMBER: builtins.int
+    MIN_EXECUTORS_FIELD_NUMBER: builtins.int
+    MAX_EXECUTORS_FIELD_NUMBER: builtins.int
+    EXECUTOR_MEMORY_FIELD_NUMBER: builtins.int
+    EXECUTOR_CORES_FIELD_NUMBER: builtins.int
+    driver_memory: builtins.str
+    """driver memory, corresponds to "spark.driver.memory" and has the same format"""
+    driver_cores: builtins.int
+    """driver cores, corresponds to "spark.driver.cores" """
+    initial_executors: builtins.int
+    """initial executor count, corresponds to "spark.dynamicAllocation.initialExecutors" """
+    min_executors: builtins.int
+    """min number of executors, corresponds to "spark.dynamicAllocation.minExecutors" """
+    max_executors: builtins.int
+    """max number of executors, corresponds to "spark.dynamicAllocation.maxExecutors" """
+    executor_memory: builtins.str
+    """executor memory, corresponds to (and has the same format of) "spark.executor.memory" """
+    executor_cores: builtins.int
+    """executor cores, corresponds to "spark.executor.cores" """
+    def __init__(
+        self,
+        *,
+        driver_memory: builtins.str = ...,
+        driver_cores: builtins.int = ...,
+        initial_executors: builtins.int = ...,
+        min_executors: builtins.int = ...,
+        max_executors: builtins.int = ...,
+        executor_memory: builtins.str = ...,
+        executor_cores: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["driver_cores", b"driver_cores", "driver_memory", b"driver_memory", "executor_cores", b"executor_cores", "executor_memory", b"executor_memory", "initial_executors", b"initial_executors", "max_executors", b"max_executors", "min_executors", b"min_executors"]) -> None: ...
+
+global___SparkResourceConfiguration = SparkResourceConfiguration
+
+class PodComputeResourceTemplateSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TEMPLATE_ID_FIELD_NUMBER: builtins.int
+    template_id: builtins.str
+    def __init__(
+        self,
+        *,
+        template_id: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["template_id", b"template_id"]) -> None: ...
+
+global___PodComputeResourceTemplateSpec = PodComputeResourceTemplateSpec
+
+class PodComputeResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OPTIMIZATION_TYPE_FIELD_NUMBER: builtins.int
+    CPU_RESOURCES_FIELD_NUMBER: builtins.int
+    GPU_RESOURCES_FIELD_NUMBER: builtins.int
+    TEMPLATE_SPEC_FIELD_NUMBER: builtins.int
+    optimization_type: global___NodeOptimizationType.ValueType
+    @property
+    def cpu_resources(self) -> global___CpuResources: ...
+    @property
+    def gpu_resources(self) -> global___GpuResources: ...
+    @property
+    def template_spec(self) -> global___PodComputeResourceTemplateSpec: ...
+    def __init__(
+        self,
+        *,
+        optimization_type: global___NodeOptimizationType.ValueType = ...,
+        cpu_resources: global___CpuResources | None = ...,
+        gpu_resources: global___GpuResources | None = ...,
+        template_spec: global___PodComputeResourceTemplateSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "template_spec", b"template_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "optimization_type", b"optimization_type", "template_spec", b"template_spec"]) -> None: ...
+
+global___PodComputeResources = PodComputeResources
+
+class SparkComputeResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    RESOURCE_CONFIGURATION_FIELD_NUMBER: builtins.int
+    @property
+    def resource_configuration(self) -> global___SparkResourceConfiguration: ...
+    def __init__(
+        self,
+        *,
+        resource_configuration: global___SparkResourceConfiguration | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["resource_configuration", b"resource_configuration"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["resource_configuration", b"resource_configuration"]) -> None: ...
+
+global___SparkComputeResources = SparkComputeResources
+
+class ClientSparkComputeResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TEMPLATE_SPEC_FIELD_NUMBER: builtins.int
+    RESOURCE_CONFIGURATION_FIELD_NUMBER: builtins.int
+    @property
+    def template_spec(self) -> global___SparkClusterResourceTemplateSpec: ...
+    @property
+    def resource_configuration(self) -> global___SparkResourceConfiguration: ...
+    def __init__(
+        self,
+        *,
+        template_spec: global___SparkClusterResourceTemplateSpec | None = ...,
+        resource_configuration: global___SparkResourceConfiguration | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["resource_configuration", b"resource_configuration", "resources", b"resources", "template_spec", b"template_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["resource_configuration", b"resource_configuration", "resources", b"resources", "template_spec", b"template_spec"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["resources", b"resources"]) -> typing_extensions.Literal["template_spec", "resource_configuration"] | None: ...
+
+global___ClientSparkComputeResources = ClientSparkComputeResources
+
+class ClientPodComputeResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NODE_OPTIMIZATION_TYPE_FIELD_NUMBER: builtins.int
+    TEMPLATE_SPEC_FIELD_NUMBER: builtins.int
+    CPU_RESOURCES_FIELD_NUMBER: builtins.int
+    GPU_RESOURCES_FIELD_NUMBER: builtins.int
+    node_optimization_type: global___NodeOptimizationType.ValueType
+    @property
+    def template_spec(self) -> global___PodComputeResourceTemplateSpec: ...
+    @property
+    def cpu_resources(self) -> global___CpuResources: ...
+    @property
+    def gpu_resources(self) -> global___GpuResources: ...
+    def __init__(
+        self,
+        *,
+        node_optimization_type: global___NodeOptimizationType.ValueType = ...,
+        template_spec: global___PodComputeResourceTemplateSpec | None = ...,
+        cpu_resources: global___CpuResources | None = ...,
+        gpu_resources: global___GpuResources | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "resources", b"resources", "template_spec", b"template_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "node_optimization_type", b"node_optimization_type", "resources", b"resources", "template_spec", b"template_spec"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["resources", b"resources"]) -> typing_extensions.Literal["template_spec", "cpu_resources", "gpu_resources"] | None: ...
+
+global___ClientPodComputeResources = ClientPodComputeResources

_frogml_proto/qwak/user_application/common/v0/resources_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/user_application/v0/user_application_pb2.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.user_application/v0/user_application.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
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n/qwak/user_application/v0/user_application.proto\x12\x18qwak.user_application.v0\x1a\x1cgoogle/protobuf/struct.proto\".\n\x04Spec\x12&\n\x05value\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct\"0\n\x06Status\x12&\n\x05value\x18\x01 \x01(\x0b\x32\x17.google.protobuf.Struct*\xf5\x02\n\x04Type\x12!\n\x1dINVALID_USER_APPLICATION_TYPE\x10\x00\x12\x10\n\x0cREMOTE_BUILD\x10\x01\x12\x1f\n\x17\x42\x41TCH_FEATURE_PROCESSOR\x10\x02\x1a\x02\x08\x01\x12#\n\x1bSTREAMING_FEATURE_PROCESSOR\x10\x03\x1a\x02\x08\x01\x12\x18\n\x14\x42\x41TCH_INFERENCE_TASK\x10\x04\x12\x14\n\x10MODEL_DEPLOYMENT\x10\x05\x12\x11\n\rMODEL_MONITOR\x10\x06\x12\x1b\n\x17\x44\x45PLOYMENT_MICROSERVICE\x10\x07\x12\x0f\n\x0b\x41PPLICATION\x10\x08\x12\x14\n\x10\x42\x41TCH_FEATURESET\x10\t\x12\x18\n\x14\x43USTOMER_ALERT_STACK\x10\n\x12\x11\n\rMODEL_TRAFFIC\x10\x0b\x12\x18\n\x14STREAMING_FEATURESET\x10\x0c\x12$\n STREAMING_AGGREGATION_FEATURESET\x10\rB\xaa\x01\n\x1f\x63om.qwak.ai.user_application.v0P\x01Z\x84\x01github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application/v0;user_application_v0b\x06proto3')
+
+_TYPE = DESCRIPTOR.enum_types_by_name['Type']
+Type = enum_type_wrapper.EnumTypeWrapper(_TYPE)
+INVALID_USER_APPLICATION_TYPE = 0
+REMOTE_BUILD = 1
+BATCH_FEATURE_PROCESSOR = 2
+STREAMING_FEATURE_PROCESSOR = 3
+BATCH_INFERENCE_TASK = 4
+MODEL_DEPLOYMENT = 5
+MODEL_MONITOR = 6
+DEPLOYMENT_MICROSERVICE = 7
+APPLICATION = 8
+BATCH_FEATURESET = 9
+CUSTOMER_ALERT_STACK = 10
+MODEL_TRAFFIC = 11
+STREAMING_FEATURESET = 12
+STREAMING_AGGREGATION_FEATURESET = 13
+
+
+_SPEC = DESCRIPTOR.message_types_by_name['Spec']
+_STATUS = DESCRIPTOR.message_types_by_name['Status']
+Spec = _reflection.GeneratedProtocolMessageType('Spec', (_message.Message,), {
+  'DESCRIPTOR' : _SPEC,
+  '__module__' : 'qwak.user_application.v0.user_application_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.user_application.v0.Spec)
+  })
+_sym_db.RegisterMessage(Spec)
+
+Status = _reflection.GeneratedProtocolMessageType('Status', (_message.Message,), {
+  'DESCRIPTOR' : _STATUS,
+  '__module__' : 'qwak.user_application.v0.user_application_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.user_application.v0.Status)
+  })
+_sym_db.RegisterMessage(Status)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.user_application.v0P\001Z\204\001github.com/qwak-ai/qwak-platform/services/core/go/admiral/user-application-value-api/pb/qwak/user_application/v0;user_application_v0'
+  _TYPE.values_by_name["BATCH_FEATURE_PROCESSOR"]._options = None
+  _TYPE.values_by_name["BATCH_FEATURE_PROCESSOR"]._serialized_options = b'\010\001'
+  _TYPE.values_by_name["STREAMING_FEATURE_PROCESSOR"]._options = None
+  _TYPE.values_by_name["STREAMING_FEATURE_PROCESSOR"]._serialized_options = b'\010\001'
+  _TYPE._serialized_start=206
+  _TYPE._serialized_end=579
+  _SPEC._serialized_start=107
+  _SPEC._serialized_end=153
+  _STATUS._serialized_start=155
+  _STATUS._serialized_end=203
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/user_application/v0/user_application_pb2.pyi

@@ -0,0 +1,89 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.struct_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 _Type:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_Type.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_USER_APPLICATION_TYPE: _Type.ValueType  # 0
+    REMOTE_BUILD: _Type.ValueType  # 1
+    BATCH_FEATURE_PROCESSOR: _Type.ValueType  # 2
+    STREAMING_FEATURE_PROCESSOR: _Type.ValueType  # 3
+    BATCH_INFERENCE_TASK: _Type.ValueType  # 4
+    MODEL_DEPLOYMENT: _Type.ValueType  # 5
+    MODEL_MONITOR: _Type.ValueType  # 6
+    DEPLOYMENT_MICROSERVICE: _Type.ValueType  # 7
+    APPLICATION: _Type.ValueType  # 8
+    BATCH_FEATURESET: _Type.ValueType  # 9
+    CUSTOMER_ALERT_STACK: _Type.ValueType  # 10
+    MODEL_TRAFFIC: _Type.ValueType  # 11
+    STREAMING_FEATURESET: _Type.ValueType  # 12
+    STREAMING_AGGREGATION_FEATURESET: _Type.ValueType  # 13
+
+class Type(_Type, metaclass=_TypeEnumTypeWrapper): ...
+
+INVALID_USER_APPLICATION_TYPE: Type.ValueType  # 0
+REMOTE_BUILD: Type.ValueType  # 1
+BATCH_FEATURE_PROCESSOR: Type.ValueType  # 2
+STREAMING_FEATURE_PROCESSOR: Type.ValueType  # 3
+BATCH_INFERENCE_TASK: Type.ValueType  # 4
+MODEL_DEPLOYMENT: Type.ValueType  # 5
+MODEL_MONITOR: Type.ValueType  # 6
+DEPLOYMENT_MICROSERVICE: Type.ValueType  # 7
+APPLICATION: Type.ValueType  # 8
+BATCH_FEATURESET: Type.ValueType  # 9
+CUSTOMER_ALERT_STACK: Type.ValueType  # 10
+MODEL_TRAFFIC: Type.ValueType  # 11
+STREAMING_FEATURESET: Type.ValueType  # 12
+STREAMING_AGGREGATION_FEATURESET: Type.ValueType  # 13
+global___Type = Type
+
+class Spec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VALUE_FIELD_NUMBER: builtins.int
+    @property
+    def value(self) -> google.protobuf.struct_pb2.Struct: ...
+    def __init__(
+        self,
+        *,
+        value: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["value", b"value"]) -> None: ...
+
+global___Spec = Spec
+
+class Status(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    VALUE_FIELD_NUMBER: builtins.int
+    @property
+    def value(self) -> google.protobuf.struct_pb2.Struct: ...
+    def __init__(
+        self,
+        *,
+        value: google.protobuf.struct_pb2.Struct | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["value", b"value"]) -> None: ...
+
+global___Status = Status

_frogml_proto/qwak/user_application/v0/user_application_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/vectors/v1/collection/collection_pb2.py

@@ -0,0 +1,128 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.vectors/v1/collection/collection.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
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+qwak/vectors/v1/collection/collection.proto\x12\x1fqwak.vector.store.v1.collection\x1a\x1fgoogle/protobuf/timestamp.proto\"\xa2\x01\n\x0cQwakMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\"?\n\x1aVectorCollectionVectorizer\x12\x19\n\x0fqwak_model_name\x18\x01 \x01(\tH\x00\x42\x06\n\x04type\"\x85\x02\n\x14VectorCollectionSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12O\n\nvectorizer\x18\x03 \x01(\x0b\x32;.qwak.vector.store.v1.collection.VectorCollectionVectorizer\x12G\n\x06metric\x18\x04 \x01(\x0e\x32\x37.qwak.vector.store.v1.collection.VectorCollectionMetric\x12\x11\n\tdimension\x18\x05 \x01(\x05\x12\x1d\n\x15multi_tenancy_enabled\x18\x06 \x01(\x08\"x\n\x1aVectorCollectionDefinition\x12\n\n\x02id\x18\x01 \x01(\t\x12N\n\x0f\x63ollection_spec\x18\x02 \x01(\x0b\x32\x35.qwak.vector.store.v1.collection.VectorCollectionSpec\"\xed\x01\n\x10VectorCollection\x12?\n\x08metadata\x18\x01 \x01(\x0b\x32-.qwak.vector.store.v1.collection.QwakMetadata\x12O\n\ndefinition\x18\x02 \x01(\x0b\x32;.qwak.vector.store.v1.collection.VectorCollectionDefinition\x12G\n\x06status\x18\x03 \x01(\x0e\x32\x37.qwak.vector.store.v1.collection.VectorCollectionStatus\"Z\n\x08Property\x12\x0c\n\x04name\x18\x01 \x01(\t\x12@\n\tdata_type\x18\x02 \x01(\x0e\x32-.qwak.vector.store.v1.collection.PropertyType\"h\n\x12\x43ollectionMetadata\x12\x13\n\x0bnum_vectors\x18\x01 \x01(\x03\x12=\n\nproperties\x18\x02 \x03(\x0b\x32).qwak.vector.store.v1.collection.Property*\x8f\x02\n\x16VectorCollectionStatus\x12\x1d\n\x19\x43OLLECTION_STATUS_INVALID\x10\x00\x12&\n\"COLLECTION_STATUS_CREATE_REQUESTED\x10\x01\x12\x1d\n\x19\x43OLLECTION_STATUS_CREATED\x10\x02\x12#\n\x1f\x43OLLECTION_STATUS_CREATE_FAILED\x10\x03\x12&\n\"COLLECTION_STATUS_DELETE_REQUESTED\x10\x04\x12#\n\x1f\x43OLLECTION_STATUS_DELETE_FAILED\x10\x05\x12\x1d\n\x19\x43OLLECTION_STATUS_DELETED\x10\x06*\xd3\x01\n\x16VectorCollectionMetric\x12\x1d\n\x19\x43OLLECTION_METRIC_INVALID\x10\x00\x12 \n\x1c\x43OLLECTION_METRIC_L2_SQUARED\x10\x01\x12\x1c\n\x18\x43OLLECTION_METRIC_COSINE\x10\x02\x12!\n\x1d\x43OLLECTION_METRIC_DOT_PRODUCT\x10\x03\x12\x18\n\x14\x43OLLECTION_METRIC_L1\x10\x04\x12\x1d\n\x19\x43OLLECTION_METRIC_HAMMING\x10\x05*\xac\x01\n\x0cPropertyType\x12\x19\n\x15PROPERTY_TYPE_INVALID\x10\x00\x12\x18\n\x14PROPERTY_TYPE_STRING\x10\x01\x12\x15\n\x11PROPERTY_TYPE_INT\x10\x02\x12\x18\n\x14PROPERTY_TYPE_DOUBLE\x10\x03\x12\x19\n\x15PROPERTY_TYPE_BOOLEAN\x10\x04\x12\x1b\n\x17PROPERTY_TYPE_TIMESTAMP\x10\x05\x42K\n!com.qwak.ai.vectors.v1.collectionP\x01Z$qwak/vector_management/v1;collectionb\x06proto3')
+
+_VECTORCOLLECTIONSTATUS = DESCRIPTOR.enum_types_by_name['VectorCollectionStatus']
+VectorCollectionStatus = enum_type_wrapper.EnumTypeWrapper(_VECTORCOLLECTIONSTATUS)
+_VECTORCOLLECTIONMETRIC = DESCRIPTOR.enum_types_by_name['VectorCollectionMetric']
+VectorCollectionMetric = enum_type_wrapper.EnumTypeWrapper(_VECTORCOLLECTIONMETRIC)
+_PROPERTYTYPE = DESCRIPTOR.enum_types_by_name['PropertyType']
+PropertyType = enum_type_wrapper.EnumTypeWrapper(_PROPERTYTYPE)
+COLLECTION_STATUS_INVALID = 0
+COLLECTION_STATUS_CREATE_REQUESTED = 1
+COLLECTION_STATUS_CREATED = 2
+COLLECTION_STATUS_CREATE_FAILED = 3
+COLLECTION_STATUS_DELETE_REQUESTED = 4
+COLLECTION_STATUS_DELETE_FAILED = 5
+COLLECTION_STATUS_DELETED = 6
+COLLECTION_METRIC_INVALID = 0
+COLLECTION_METRIC_L2_SQUARED = 1
+COLLECTION_METRIC_COSINE = 2
+COLLECTION_METRIC_DOT_PRODUCT = 3
+COLLECTION_METRIC_L1 = 4
+COLLECTION_METRIC_HAMMING = 5
+PROPERTY_TYPE_INVALID = 0
+PROPERTY_TYPE_STRING = 1
+PROPERTY_TYPE_INT = 2
+PROPERTY_TYPE_DOUBLE = 3
+PROPERTY_TYPE_BOOLEAN = 4
+PROPERTY_TYPE_TIMESTAMP = 5
+
+
+_QWAKMETADATA = DESCRIPTOR.message_types_by_name['QwakMetadata']
+_VECTORCOLLECTIONVECTORIZER = DESCRIPTOR.message_types_by_name['VectorCollectionVectorizer']
+_VECTORCOLLECTIONSPEC = DESCRIPTOR.message_types_by_name['VectorCollectionSpec']
+_VECTORCOLLECTIONDEFINITION = DESCRIPTOR.message_types_by_name['VectorCollectionDefinition']
+_VECTORCOLLECTION = DESCRIPTOR.message_types_by_name['VectorCollection']
+_PROPERTY = DESCRIPTOR.message_types_by_name['Property']
+_COLLECTIONMETADATA = DESCRIPTOR.message_types_by_name['CollectionMetadata']
+QwakMetadata = _reflection.GeneratedProtocolMessageType('QwakMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _QWAKMETADATA,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.QwakMetadata)
+  })
+_sym_db.RegisterMessage(QwakMetadata)
+
+VectorCollectionVectorizer = _reflection.GeneratedProtocolMessageType('VectorCollectionVectorizer', (_message.Message,), {
+  'DESCRIPTOR' : _VECTORCOLLECTIONVECTORIZER,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.VectorCollectionVectorizer)
+  })
+_sym_db.RegisterMessage(VectorCollectionVectorizer)
+
+VectorCollectionSpec = _reflection.GeneratedProtocolMessageType('VectorCollectionSpec', (_message.Message,), {
+  'DESCRIPTOR' : _VECTORCOLLECTIONSPEC,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.VectorCollectionSpec)
+  })
+_sym_db.RegisterMessage(VectorCollectionSpec)
+
+VectorCollectionDefinition = _reflection.GeneratedProtocolMessageType('VectorCollectionDefinition', (_message.Message,), {
+  'DESCRIPTOR' : _VECTORCOLLECTIONDEFINITION,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.VectorCollectionDefinition)
+  })
+_sym_db.RegisterMessage(VectorCollectionDefinition)
+
+VectorCollection = _reflection.GeneratedProtocolMessageType('VectorCollection', (_message.Message,), {
+  'DESCRIPTOR' : _VECTORCOLLECTION,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.VectorCollection)
+  })
+_sym_db.RegisterMessage(VectorCollection)
+
+Property = _reflection.GeneratedProtocolMessageType('Property', (_message.Message,), {
+  'DESCRIPTOR' : _PROPERTY,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.Property)
+  })
+_sym_db.RegisterMessage(Property)
+
+CollectionMetadata = _reflection.GeneratedProtocolMessageType('CollectionMetadata', (_message.Message,), {
+  'DESCRIPTOR' : _COLLECTIONMETADATA,
+  '__module__' : 'qwak.vectors.v1.collection.collection_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.vector.store.v1.collection.CollectionMetadata)
+  })
+_sym_db.RegisterMessage(CollectionMetadata)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n!com.qwak.ai.vectors.v1.collectionP\001Z$qwak/vector_management/v1;collection'
+  _VECTORCOLLECTIONSTATUS._serialized_start=1168
+  _VECTORCOLLECTIONSTATUS._serialized_end=1439
+  _VECTORCOLLECTIONMETRIC._serialized_start=1442
+  _VECTORCOLLECTIONMETRIC._serialized_end=1653
+  _PROPERTYTYPE._serialized_start=1656
+  _PROPERTYTYPE._serialized_end=1828
+  _QWAKMETADATA._serialized_start=114
+  _QWAKMETADATA._serialized_end=276
+  _VECTORCOLLECTIONVECTORIZER._serialized_start=278
+  _VECTORCOLLECTIONVECTORIZER._serialized_end=341
+  _VECTORCOLLECTIONSPEC._serialized_start=344
+  _VECTORCOLLECTIONSPEC._serialized_end=605
+  _VECTORCOLLECTIONDEFINITION._serialized_start=607
+  _VECTORCOLLECTIONDEFINITION._serialized_end=727
+  _VECTORCOLLECTION._serialized_start=730
+  _VECTORCOLLECTION._serialized_end=967
+  _PROPERTY._serialized_start=969
+  _PROPERTY._serialized_end=1059
+  _COLLECTIONMETADATA._serialized_start=1061
+  _COLLECTIONMETADATA._serialized_end=1165
+# @@protoc_insertion_point(module_scope)