chalkpy 2.93.4__py3-none-any.whl → 2.93.6__py3-none-any.whl

chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi

@@ -1,7 +1,14 @@
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
@@ -23,25 +30,206 @@ class KubernetesScaledObjectTargetRef(_message.Message):
         env_source_container_name: _Optional[str] = ...,
     ) -> None: ...
 
+class KubernetesGroupVersionKindResource(_message.Message):
+    __slots__ = ("group", "version", "kind", "resource")
+    GROUP_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_FIELD_NUMBER: _ClassVar[int]
+    group: str
+    version: str
+    kind: str
+    resource: str
+    def __init__(
+        self,
+        group: _Optional[str] = ...,
+        version: _Optional[str] = ...,
+        kind: _Optional[str] = ...,
+        resource: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectCondition(_message.Message):
+    __slots__ = ("type", "status", "reason", "message")
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    REASON_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    type: str
+    status: str
+    reason: str
+    message: str
+    def __init__(
+        self,
+        type: _Optional[str] = ...,
+        status: _Optional[str] = ...,
+        reason: _Optional[str] = ...,
+        message: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectHealthStatus(_message.Message):
+    __slots__ = ("number_of_failures", "status")
+    NUMBER_OF_FAILURES_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    number_of_failures: int
+    status: str
+    def __init__(self, number_of_failures: _Optional[int] = ..., status: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectAuthenticationRef(_message.Message):
+    __slots__ = ("name", "kind")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    kind: str
+    def __init__(self, name: _Optional[str] = ..., kind: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectTrigger(_message.Message):
+    __slots__ = ("type", "name", "use_cached_metrics", "metadata", "authentication_ref", "metric_type")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    USE_CACHED_METRICS_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    AUTHENTICATION_REF_FIELD_NUMBER: _ClassVar[int]
+    METRIC_TYPE_FIELD_NUMBER: _ClassVar[int]
+    type: str
+    name: str
+    use_cached_metrics: bool
+    metadata: _containers.ScalarMap[str, str]
+    authentication_ref: KubernetesScaledObjectAuthenticationRef
+    metric_type: str
+    def __init__(
+        self,
+        type: _Optional[str] = ...,
+        name: _Optional[str] = ...,
+        use_cached_metrics: bool = ...,
+        metadata: _Optional[_Mapping[str, str]] = ...,
+        authentication_ref: _Optional[_Union[KubernetesScaledObjectAuthenticationRef, _Mapping]] = ...,
+        metric_type: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectFallback(_message.Message):
+    __slots__ = ("failure_threshold", "replicas")
+    FAILURE_THRESHOLD_FIELD_NUMBER: _ClassVar[int]
+    REPLICAS_FIELD_NUMBER: _ClassVar[int]
+    failure_threshold: int
+    replicas: int
+    def __init__(self, failure_threshold: _Optional[int] = ..., replicas: _Optional[int] = ...) -> None: ...
+
+class KubernetesScaledObjectScalingModifiers(_message.Message):
+    __slots__ = ("formula", "target", "activation_target", "metric_type")
+    FORMULA_FIELD_NUMBER: _ClassVar[int]
+    TARGET_FIELD_NUMBER: _ClassVar[int]
+    ACTIVATION_TARGET_FIELD_NUMBER: _ClassVar[int]
+    METRIC_TYPE_FIELD_NUMBER: _ClassVar[int]
+    formula: str
+    target: str
+    activation_target: str
+    metric_type: str
+    def __init__(
+        self,
+        formula: _Optional[str] = ...,
+        target: _Optional[str] = ...,
+        activation_target: _Optional[str] = ...,
+        metric_type: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectHPAConfig(_message.Message):
+    __slots__ = ("name",)
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    def __init__(self, name: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectAdvancedConfig(_message.Message):
+    __slots__ = ("horizontal_pod_autoscaler_config", "restore_to_original_replica_count", "scaling_modifiers")
+    HORIZONTAL_POD_AUTOSCALER_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    RESTORE_TO_ORIGINAL_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
+    SCALING_MODIFIERS_FIELD_NUMBER: _ClassVar[int]
+    horizontal_pod_autoscaler_config: KubernetesScaledObjectHPAConfig
+    restore_to_original_replica_count: bool
+    scaling_modifiers: KubernetesScaledObjectScalingModifiers
+    def __init__(
+        self,
+        horizontal_pod_autoscaler_config: _Optional[_Union[KubernetesScaledObjectHPAConfig, _Mapping]] = ...,
+        restore_to_original_replica_count: bool = ...,
+        scaling_modifiers: _Optional[_Union[KubernetesScaledObjectScalingModifiers, _Mapping]] = ...,
+    ) -> None: ...
+
 class KubernetesScaledObjectStatus(_message.Message):
-    __slots__ = ("scale_target_kind", "original_replica_count", "last_active_time", "paused_replica_count", "hpa_name")
+    __slots__ = (
+        "scale_target_kind",
+        "scale_target_gvkr",
+        "original_replica_count",
+        "last_active_time",
+        "external_metric_names",
+        "resource_metric_names",
+        "composite_scaler_name",
+        "conditions",
+        "health",
+        "paused_replica_count",
+        "hpa_name",
+        "triggers_types",
+        "authentications_types",
+    )
+    class HealthEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: KubernetesScaledObjectHealthStatus
+        def __init__(
+            self,
+            key: _Optional[str] = ...,
+            value: _Optional[_Union[KubernetesScaledObjectHealthStatus, _Mapping]] = ...,
+        ) -> None: ...
+
     SCALE_TARGET_KIND_FIELD_NUMBER: _ClassVar[int]
+    SCALE_TARGET_GVKR_FIELD_NUMBER: _ClassVar[int]
     ORIGINAL_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     LAST_ACTIVE_TIME_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_METRIC_NAMES_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_METRIC_NAMES_FIELD_NUMBER: _ClassVar[int]
+    COMPOSITE_SCALER_NAME_FIELD_NUMBER: _ClassVar[int]
+    CONDITIONS_FIELD_NUMBER: _ClassVar[int]
+    HEALTH_FIELD_NUMBER: _ClassVar[int]
     PAUSED_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     HPA_NAME_FIELD_NUMBER: _ClassVar[int]
+    TRIGGERS_TYPES_FIELD_NUMBER: _ClassVar[int]
+    AUTHENTICATIONS_TYPES_FIELD_NUMBER: _ClassVar[int]
     scale_target_kind: str
+    scale_target_gvkr: KubernetesGroupVersionKindResource
     original_replica_count: int
     last_active_time: _timestamp_pb2.Timestamp
+    external_metric_names: _containers.RepeatedScalarFieldContainer[str]
+    resource_metric_names: _containers.RepeatedScalarFieldContainer[str]
+    composite_scaler_name: str
+    conditions: _containers.RepeatedCompositeFieldContainer[KubernetesScaledObjectCondition]
+    health: _containers.MessageMap[str, KubernetesScaledObjectHealthStatus]
     paused_replica_count: int
     hpa_name: str
+    triggers_types: str
+    authentications_types: str
     def __init__(
         self,
         scale_target_kind: _Optional[str] = ...,
+        scale_target_gvkr: _Optional[_Union[KubernetesGroupVersionKindResource, _Mapping]] = ...,
         original_replica_count: _Optional[int] = ...,
         last_active_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        external_metric_names: _Optional[_Iterable[str]] = ...,
+        resource_metric_names: _Optional[_Iterable[str]] = ...,
+        composite_scaler_name: _Optional[str] = ...,
+        conditions: _Optional[_Iterable[_Union[KubernetesScaledObjectCondition, _Mapping]]] = ...,
+        health: _Optional[_Mapping[str, KubernetesScaledObjectHealthStatus]] = ...,
         paused_replica_count: _Optional[int] = ...,
         hpa_name: _Optional[str] = ...,
+        triggers_types: _Optional[str] = ...,
+        authentications_types: _Optional[str] = ...,
     ) -> None: ...
 
 class KubernetesScaledObjectSpec(_message.Message):
@@ -52,6 +240,10 @@ class KubernetesScaledObjectSpec(_message.Message):
         "cooldown_period",
         "idle_replica_count",
         "max_replica_count",
+        "advanced",
+        "triggers",
+        "fallback",
+        "min_replica_count",
     )
     SCALE_TARGET_REF_FIELD_NUMBER: _ClassVar[int]
     POLLING_INTERVAL_FIELD_NUMBER: _ClassVar[int]
@@ -59,12 +251,20 @@ class KubernetesScaledObjectSpec(_message.Message):
     COOLDOWN_PERIOD_FIELD_NUMBER: _ClassVar[int]
     IDLE_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     MAX_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
+    ADVANCED_FIELD_NUMBER: _ClassVar[int]
+    TRIGGERS_FIELD_NUMBER: _ClassVar[int]
+    FALLBACK_FIELD_NUMBER: _ClassVar[int]
+    MIN_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     scale_target_ref: KubernetesScaledObjectTargetRef
     polling_interval: int
     initial_cooldown_period: int
     cooldown_period: int
     idle_replica_count: int
     max_replica_count: int
+    advanced: KubernetesScaledObjectAdvancedConfig
+    triggers: _containers.RepeatedCompositeFieldContainer[KubernetesScaledObjectTrigger]
+    fallback: KubernetesScaledObjectFallback
+    min_replica_count: int
     def __init__(
         self,
         scale_target_ref: _Optional[_Union[KubernetesScaledObjectTargetRef, _Mapping]] = ...,
@@ -73,6 +273,10 @@ class KubernetesScaledObjectSpec(_message.Message):
         cooldown_period: _Optional[int] = ...,
         idle_replica_count: _Optional[int] = ...,
         max_replica_count: _Optional[int] = ...,
+        advanced: _Optional[_Union[KubernetesScaledObjectAdvancedConfig, _Mapping]] = ...,
+        triggers: _Optional[_Iterable[_Union[KubernetesScaledObjectTrigger, _Mapping]]] = ...,
+        fallback: _Optional[_Union[KubernetesScaledObjectFallback, _Mapping]] = ...,
+        min_replica_count: _Optional[int] = ...,
     ) -> None: ...
 
 class KubernetesScaledObjectData(_message.Message):
@@ -86,3 +290,49 @@ class KubernetesScaledObjectData(_message.Message):
         spec: _Optional[_Union[KubernetesScaledObjectSpec, _Mapping]] = ...,
         status: _Optional[_Union[KubernetesScaledObjectStatus, _Mapping]] = ...,
     ) -> None: ...
+
+class KubernetesScaledObject(_message.Message):
+    __slots__ = ("name", "namespace", "labels", "annotations", "creation_timestamp", "spec", "status", "cluster_name")
+    class LabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    class AnnotationsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
+    LABELS_FIELD_NUMBER: _ClassVar[int]
+    ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
+    CREATION_TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    SPEC_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    CLUSTER_NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    namespace: str
+    labels: _containers.ScalarMap[str, str]
+    annotations: _containers.ScalarMap[str, str]
+    creation_timestamp: int
+    spec: KubernetesScaledObjectSpec
+    status: KubernetesScaledObjectStatus
+    cluster_name: str
+    def __init__(
+        self,
+        name: _Optional[str] = ...,
+        namespace: _Optional[str] = ...,
+        labels: _Optional[_Mapping[str, str]] = ...,
+        annotations: _Optional[_Mapping[str, str]] = ...,
+        creation_timestamp: _Optional[int] = ...,
+        spec: _Optional[_Union[KubernetesScaledObjectSpec, _Mapping]] = ...,
+        status: _Optional[_Union[KubernetesScaledObjectStatus, _Mapping]] = ...,
+        cluster_name: _Optional[str] = ...,
+    ) -> None: ...

chalk/_gen/chalk/server/v1/benchmark_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/benchmark.proto
+# Protobuf Python Version: 4.25.3
+"""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 chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
+from chalk._gen.chalk.common.v1 import online_query_pb2 as chalk_dot_common_dot_v1_dot_online__query__pb2
+from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x1f\x63halk/server/v1/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1egoogle/protobuf/duration.proto"n\n\x1cSimpleOnlineQueryBulkRequest\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12\'\n\x0foutput_features\x18\x02 \x03(\tR\x0eoutputFeatures"\xfc\x02\n\x16\x43reateBenchmarkRequest\x12\x1d\n\nwarmup_qps\x18\x01 \x01(\x03R\twarmupQps\x12\x42\n\x0fwarmup_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0ewarmupDuration\x12\x10\n\x03qps\x18\x03 \x01(\x03R\x03qps\x12\x35\n\x08\x64uration\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12H\n\nquery_bulk\x18\x05 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\tqueryBulk\x12[\n\x11simple_query_bulk\x18\x06 \x01(\x0b\x32-.chalk.server.v1.SimpleOnlineQueryBulkRequestH\x00R\x0fsimpleQueryBulkB\x0f\n\rquery_request"S\n\x17\x43reateBenchmarkResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32 .chalk.server.v1.BenchmarkStatusR\x06status*\xe8\x01\n\x0f\x42\x65nchmarkStatus\x12 \n\x1c\x42\x45NCHMARK_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_QUEUED\x10\x01\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_WORKING\x10\x02\x12\x1e\n\x1a\x42\x45NCHMARK_STATUS_COMPLETED\x10\x03\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_FAILED\x10\x04\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_SKIPPED\x10\x05\x12\x1d\n\x19\x42\x45NCHMARK_STATUS_CANCELED\x10\x06\x32}\n\x10\x42\x65nchmarkService\x12i\n\x0f\x43reateBenchmark\x12\'.chalk.server.v1.CreateBenchmarkRequest\x1a(.chalk.server.v1.CreateBenchmarkResponse"\x03\x80}\x03\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0e\x42\x65nchmarkProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.benchmark_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\016BenchmarkProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSTATUS"]._serialized_start = 734
+    _globals["_BENCHMARKSTATUS"]._serialized_end = 966
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 153
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 263
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 266
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 646
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 648
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 731
+    _globals["_BENCHMARKSERVICE"]._serialized_start = 968
+    _globals["_BENCHMARKSERVICE"]._serialized_end = 1093
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/benchmark_pb2.pyi

@@ -0,0 +1,74 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from chalk._gen.chalk.common.v1 import online_query_pb2 as _online_query_pb2
+from google.protobuf import duration_pb2 as _duration_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class BenchmarkStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    BENCHMARK_STATUS_UNSPECIFIED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_QUEUED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_WORKING: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_COMPLETED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_FAILED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_SKIPPED: _ClassVar[BenchmarkStatus]
+    BENCHMARK_STATUS_CANCELED: _ClassVar[BenchmarkStatus]
+
+BENCHMARK_STATUS_UNSPECIFIED: BenchmarkStatus
+BENCHMARK_STATUS_QUEUED: BenchmarkStatus
+BENCHMARK_STATUS_WORKING: BenchmarkStatus
+BENCHMARK_STATUS_COMPLETED: BenchmarkStatus
+BENCHMARK_STATUS_FAILED: BenchmarkStatus
+BENCHMARK_STATUS_SKIPPED: BenchmarkStatus
+BENCHMARK_STATUS_CANCELED: BenchmarkStatus
+
+class SimpleOnlineQueryBulkRequest(_message.Message):
+    __slots__ = ("input_features", "output_features")
+    INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    input_features: _containers.RepeatedScalarFieldContainer[str]
+    output_features: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(
+        self, input_features: _Optional[_Iterable[str]] = ..., output_features: _Optional[_Iterable[str]] = ...
+    ) -> None: ...
+
+class CreateBenchmarkRequest(_message.Message):
+    __slots__ = ("warmup_qps", "warmup_duration", "qps", "duration", "query_bulk", "simple_query_bulk")
+    WARMUP_QPS_FIELD_NUMBER: _ClassVar[int]
+    WARMUP_DURATION_FIELD_NUMBER: _ClassVar[int]
+    QPS_FIELD_NUMBER: _ClassVar[int]
+    DURATION_FIELD_NUMBER: _ClassVar[int]
+    QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
+    SIMPLE_QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
+    warmup_qps: int
+    warmup_duration: _duration_pb2.Duration
+    qps: int
+    duration: _duration_pb2.Duration
+    query_bulk: _online_query_pb2.OnlineQueryBulkRequest
+    simple_query_bulk: SimpleOnlineQueryBulkRequest
+    def __init__(
+        self,
+        warmup_qps: _Optional[int] = ...,
+        warmup_duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
+        qps: _Optional[int] = ...,
+        duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
+        query_bulk: _Optional[_Union[_online_query_pb2.OnlineQueryBulkRequest, _Mapping]] = ...,
+        simple_query_bulk: _Optional[_Union[SimpleOnlineQueryBulkRequest, _Mapping]] = ...,
+    ) -> None: ...
+
+class CreateBenchmarkResponse(_message.Message):
+    __slots__ = ("status",)
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    status: BenchmarkStatus
+    def __init__(self, status: _Optional[_Union[BenchmarkStatus, str]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py

@@ -0,0 +1,78 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc
+
+from chalk._gen.chalk.server.v1 import benchmark_pb2 as chalk_dot_server_dot_v1_dot_benchmark__pb2
+
+
+class BenchmarkServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateBenchmark = channel.unary_unary(
+            "/chalk.server.v1.BenchmarkService/CreateBenchmark",
+            request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkResponse.FromString,
+        )
+
+
+class BenchmarkServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateBenchmark(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_BenchmarkServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "CreateBenchmark": grpc.unary_unary_rpc_method_handler(
+            servicer.CreateBenchmark,
+            request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.BenchmarkService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class BenchmarkService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateBenchmark(
+        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,
+            "/chalk.server.v1.BenchmarkService/CreateBenchmark",
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi

@@ -0,0 +1,36 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.benchmark_pb2 import (
+    CreateBenchmarkRequest,
+    CreateBenchmarkResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class BenchmarkServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    CreateBenchmark: UnaryUnaryMultiCallable[
+        CreateBenchmarkRequest,
+        CreateBenchmarkResponse,
+    ]
+
+class BenchmarkServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def CreateBenchmark(
+        self,
+        request: CreateBenchmarkRequest,
+        context: ServicerContext,
+    ) -> CreateBenchmarkResponse: ...
+
+def add_BenchmarkServiceServicer_to_server(servicer: BenchmarkServiceServicer, server: Server) -> None: ...