types-kafka-python 2.3.0.20260523__py3-none-any.whl

kafka-stubs/future.pyi

@@ -0,0 +1,19 @@
+from _typeshed import Incomplete
+
+log: Incomplete
+
+class Future:
+    error_on_callbacks: bool
+    is_done: bool
+    value: Incomplete
+    exception: Incomplete
+    def __init__(self) -> None: ...
+    def succeeded(self): ...
+    def failed(self): ...
+    def retriable(self): ...
+    def success(self, value): ...
+    def failure(self, e): ...
+    def add_callback(self, f, *args, **kwargs): ...
+    def add_errback(self, f, *args, **kwargs): ...
+    def add_both(self, f, *args, **kwargs): ...
+    def chain(self, future): ...

kafka-stubs/metrics/__init__.pyi

@@ -0,0 +1,10 @@
+from kafka.metrics.compound_stat import NamedMeasurable as NamedMeasurable
+from kafka.metrics.dict_reporter import DictReporter as DictReporter
+from kafka.metrics.kafka_metric import KafkaMetric as KafkaMetric
+from kafka.metrics.measurable import AnonMeasurable as AnonMeasurable
+from kafka.metrics.metric_config import MetricConfig as MetricConfig
+from kafka.metrics.metric_name import MetricName as MetricName
+from kafka.metrics.metrics import Metrics as Metrics
+from kafka.metrics.quota import Quota as Quota
+
+__all__ = ["AnonMeasurable", "DictReporter", "KafkaMetric", "MetricConfig", "MetricName", "Metrics", "NamedMeasurable", "Quota"]

kafka-stubs/metrics/compound_stat.pyi

@@ -0,0 +1,13 @@
+import abc
+
+from kafka.metrics.stat import AbstractStat
+
+class AbstractCompoundStat(AbstractStat, metaclass=abc.ABCMeta):
+    def stats(self) -> None: ...
+
+class NamedMeasurable:
+    def __init__(self, metric_name, measurable_stat) -> None: ...
+    @property
+    def name(self): ...
+    @property
+    def stat(self): ...

kafka-stubs/metrics/dict_reporter.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+
+from kafka.metrics.metrics_reporter import AbstractMetricsReporter
+
+logger: Incomplete
+
+class DictReporter(AbstractMetricsReporter):
+    def __init__(self, prefix: str = "") -> None: ...
+    def snapshot(self): ...
+    def init(self, metrics) -> None: ...
+    def metric_change(self, metric) -> None: ...
+    def metric_removal(self, metric): ...
+    def get_category(self, metric): ...
+    def configure(self, configs) -> None: ...
+    def close(self) -> None: ...

kafka-stubs/metrics/kafka_metric.pyi

@@ -0,0 +1,13 @@
+class KafkaMetric:
+    def __init__(self, metric_name, measurable, config) -> None: ...
+    @property
+    def metric_name(self): ...
+    @property
+    def measurable(self): ...
+
+    @property
+    def config(self): ...
+    @config.setter
+    def config(self, config) -> None: ...
+
+    def value(self, time_ms=None): ...

kafka-stubs/metrics/measurable.pyi

@@ -0,0 +1,9 @@
+import abc
+
+class AbstractMeasurable(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def measure(self, config, now): ...
+
+class AnonMeasurable(AbstractMeasurable):
+    def __init__(self, measure_fn) -> None: ...
+    def measure(self, config, now): ...

kafka-stubs/metrics/measurable_stat.pyi

@@ -0,0 +1,6 @@
+import abc
+
+from kafka.metrics.measurable import AbstractMeasurable
+from kafka.metrics.stat import AbstractStat
+
+class AbstractMeasurableStat(AbstractStat, AbstractMeasurable, metaclass=abc.ABCMeta): ...

kafka-stubs/metrics/metric_config.pyi

@@ -0,0 +1,13 @@
+from _typeshed import Incomplete
+
+class MetricConfig:
+    quota: Incomplete
+    event_window: Incomplete
+    time_window_ms: Incomplete
+    tags: Incomplete
+    def __init__(self, quota=None, samples: int = 2, event_window=..., time_window_ms=30000, tags=None) -> None: ...
+
+    @property
+    def samples(self): ...
+    @samples.setter
+    def samples(self, value) -> None: ...

kafka-stubs/metrics/metric_name.pyi

@@ -0,0 +1,13 @@
+class MetricName:
+    def __init__(self, name, group, description=None, tags=None) -> None: ...
+    @property
+    def name(self): ...
+    @property
+    def group(self): ...
+    @property
+    def description(self): ...
+    @property
+    def tags(self): ...
+    def __hash__(self): ...
+    def __eq__(self, other): ...
+    def __ne__(self, other): ...

kafka-stubs/metrics/metrics.pyi

@@ -0,0 +1,24 @@
+from _typeshed import Incomplete
+
+logger: Incomplete
+
+class Metrics:
+    def __init__(self, default_config=None, reporters=None, enable_expiration: bool = False) -> None: ...
+    @property
+    def config(self): ...
+    @property
+    def metrics(self): ...
+    def metric_name(self, name, group, description: str = "", tags=None): ...
+    def get_sensor(self, name): ...
+    def sensor(self, name, config=None, inactive_sensor_expiration_time_seconds=..., parents=None): ...
+    def remove_sensor(self, name) -> None: ...
+    def add_metric(self, metric_name, measurable, config=None) -> None: ...
+    def remove_metric(self, metric_name): ...
+    def add_reporter(self, reporter) -> None: ...
+    def register_metric(self, metric) -> None: ...
+
+    class ExpireSensorTask:
+        @staticmethod
+        def run(metrics) -> None: ...
+
+    def close(self) -> None: ...

kafka-stubs/metrics/metrics_reporter.pyi

@@ -0,0 +1,13 @@
+import abc
+
+class AbstractMetricsReporter(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def init(self, metrics): ...
+    @abc.abstractmethod
+    def metric_change(self, metric): ...
+    @abc.abstractmethod
+    def metric_removal(self, metric): ...
+    @abc.abstractmethod
+    def configure(self, configs): ...
+    @abc.abstractmethod
+    def close(self): ...

kafka-stubs/metrics/quota.pyi

@@ -0,0 +1,13 @@
+class Quota:
+    def __init__(self, bound, is_upper) -> None: ...
+    @staticmethod
+    def upper_bound(upper_bound): ...
+    @staticmethod
+    def lower_bound(lower_bound): ...
+    def is_upper_bound(self): ...
+    @property
+    def bound(self): ...
+    def is_acceptable(self, value): ...
+    def __hash__(self): ...
+    def __eq__(self, other): ...
+    def __ne__(self, other): ...

kafka-stubs/metrics/stat.pyi

@@ -0,0 +1,5 @@
+import abc
+
+class AbstractStat(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def record(self, config, value, time_ms): ...

kafka-stubs/metrics/stats/__init__.pyi

@@ -0,0 +1,12 @@
+from kafka.metrics.stats.avg import Avg as Avg
+from kafka.metrics.stats.count import Count as Count
+from kafka.metrics.stats.histogram import Histogram as Histogram
+from kafka.metrics.stats.max_stat import Max as Max
+from kafka.metrics.stats.min_stat import Min as Min
+from kafka.metrics.stats.percentile import Percentile as Percentile
+from kafka.metrics.stats.percentiles import Percentiles as Percentiles
+from kafka.metrics.stats.rate import Rate as Rate
+from kafka.metrics.stats.sensor import Sensor as Sensor
+from kafka.metrics.stats.total import Total as Total
+
+__all__ = ["Avg", "Count", "Histogram", "Max", "Min", "Percentile", "Percentiles", "Rate", "Sensor", "Total"]

kafka-stubs/metrics/stats/avg.pyi

@@ -0,0 +1,6 @@
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class Avg(AbstractSampledStat):
+    def __init__(self) -> None: ...
+    def update(self, sample, config, value, now) -> None: ...
+    def combine(self, samples, config, now): ...

kafka-stubs/metrics/stats/count.pyi

@@ -0,0 +1,6 @@
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class Count(AbstractSampledStat):
+    def __init__(self) -> None: ...
+    def update(self, sample, config, value, now) -> None: ...
+    def combine(self, samples, config, now): ...

kafka-stubs/metrics/stats/histogram.pyi

@@ -0,0 +1,21 @@
+class Histogram:
+    def __init__(self, bin_scheme) -> None: ...
+    def record(self, value) -> None: ...
+    def value(self, quantile): ...
+    @property
+    def counts(self): ...
+    def clear(self) -> None: ...
+
+    class ConstantBinScheme:
+        def __init__(self, bins, min_val, max_val) -> None: ...
+        @property
+        def bins(self): ...
+        def from_bin(self, b): ...
+        def to_bin(self, x): ...
+
+    class LinearBinScheme:
+        def __init__(self, num_bins, max_val) -> None: ...
+        @property
+        def bins(self): ...
+        def from_bin(self, b): ...
+        def to_bin(self, x): ...

kafka-stubs/metrics/stats/max_stat.pyi

@@ -0,0 +1,6 @@
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class Max(AbstractSampledStat):
+    def __init__(self) -> None: ...
+    def update(self, sample, config, value, now) -> None: ...
+    def combine(self, samples, config, now): ...

kafka-stubs/metrics/stats/min_stat.pyi

@@ -0,0 +1,6 @@
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class Min(AbstractSampledStat):
+    def __init__(self) -> None: ...
+    def update(self, sample, config, value, now) -> None: ...
+    def combine(self, samples, config, now): ...

kafka-stubs/metrics/stats/percentile.pyi

@@ -0,0 +1,6 @@
+class Percentile:
+    def __init__(self, metric_name, percentile) -> None: ...
+    @property
+    def name(self): ...
+    @property
+    def percentile(self): ...

kafka-stubs/metrics/stats/percentiles.pyi

@@ -0,0 +1,21 @@
+from _typeshed import Incomplete
+
+from kafka.metrics.compound_stat import AbstractCompoundStat
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class BucketSizing:
+    CONSTANT: int
+    LINEAR: int
+
+class Percentiles(AbstractSampledStat, AbstractCompoundStat):
+    bin_scheme: Incomplete
+    def __init__(self, size_in_bytes, bucketing, max_val, min_val: float = 0.0, percentiles=None) -> None: ...
+    def stats(self): ...
+    def value(self, config, now, quantile): ...
+    def combine(self, samples, config, now): ...
+    def new_sample(self, time_ms): ...
+    def update(self, sample, config, value, time_ms) -> None: ...
+
+    class HistogramSample(AbstractSampledStat.Sample):
+        histogram: Incomplete
+        def __init__(self, scheme, now) -> None: ...

kafka-stubs/metrics/stats/rate.pyi

@@ -0,0 +1,28 @@
+from _typeshed import Incomplete
+
+from kafka.metrics.measurable_stat import AbstractMeasurableStat
+from kafka.metrics.stats.sampled_stat import AbstractSampledStat
+
+class TimeUnit:
+    NANOSECONDS: Incomplete
+    MICROSECONDS: Incomplete
+    MILLISECONDS: Incomplete
+    SECONDS: Incomplete
+    MINUTES: Incomplete
+    HOURS: Incomplete
+    DAYS: Incomplete
+    @staticmethod
+    def get_name(time_unit): ...
+
+class Rate(AbstractMeasurableStat):
+    def __init__(self, time_unit=3, sampled_stat=None) -> None: ...
+    def unit_name(self): ...
+    def record(self, config, value, time_ms) -> None: ...
+    def measure(self, config, now): ...
+    def window_size(self, config, now): ...
+    def convert(self, time_ms): ...
+
+class SampledTotal(AbstractSampledStat):
+    def __init__(self, initial_value=None) -> None: ...
+    def update(self, sample, config, value, time_ms) -> None: ...
+    def combine(self, samples, config, now): ...

kafka-stubs/metrics/stats/sampled_stat.pyi

@@ -0,0 +1,26 @@
+import abc
+from _typeshed import Incomplete
+
+from kafka.metrics.measurable_stat import AbstractMeasurableStat
+
+class AbstractSampledStat(AbstractMeasurableStat, metaclass=abc.ABCMeta):
+    def __init__(self, initial_value) -> None: ...
+    @abc.abstractmethod
+    def update(self, sample, config, value, time_ms): ...
+    @abc.abstractmethod
+    def combine(self, samples, config, now): ...
+    def record(self, config, value, time_ms) -> None: ...
+    def new_sample(self, time_ms): ...
+    def measure(self, config, now): ...
+    def current(self, time_ms): ...
+    def oldest(self, now): ...
+    def purge_obsolete_samples(self, config, now) -> None: ...
+
+    class Sample:
+        initial_value: Incomplete
+        event_count: int
+        last_window_ms: Incomplete
+        value: Incomplete
+        def __init__(self, initial_value, now) -> None: ...
+        def reset(self, now) -> None: ...
+        def is_complete(self, time_ms, config): ...

kafka-stubs/metrics/stats/sensor.pyi

@@ -0,0 +1,10 @@
+class Sensor:
+    def __init__(self, registry, name, parents, config, inactive_sensor_expiration_time_seconds) -> None: ...
+    @property
+    def name(self): ...
+    @property
+    def metrics(self): ...
+    def record(self, value: float = 1.0, time_ms=None) -> None: ...
+    def add_compound(self, compound_stat, config=None) -> None: ...
+    def add(self, metric_name, stat, config=None) -> None: ...
+    def has_expired(self): ...

kafka-stubs/metrics/stats/total.pyi

@@ -0,0 +1,6 @@
+from kafka.metrics.measurable_stat import AbstractMeasurableStat
+
+class Total(AbstractMeasurableStat):
+    def __init__(self, value: float = 0.0) -> None: ...
+    def record(self, config, value, now) -> None: ...
+    def measure(self, config, now): ...

kafka-stubs/partitioner/__init__.pyi

@@ -0,0 +1,3 @@
+from kafka.partitioner.default import DefaultPartitioner as DefaultPartitioner, murmur2 as murmur2
+
+__all__ = ["DefaultPartitioner", "murmur2"]

kafka-stubs/partitioner/default.pyi

@@ -0,0 +1,5 @@
+class DefaultPartitioner:
+    @classmethod
+    def __call__(cls, key, all_partitions, available): ...
+
+def murmur2(data): ...

kafka-stubs/producer/__init__.pyi

@@ -0,0 +1,3 @@
+from kafka.producer.kafka import KafkaProducer as KafkaProducer
+
+__all__ = ["KafkaProducer"]

kafka-stubs/producer/future.pyi

@@ -0,0 +1,37 @@
+from _typeshed import Incomplete
+from typing import NamedTuple
+
+from kafka.future import Future
+from kafka.structs import TopicPartition
+
+class FutureProduceResult(Future):
+    topic_partition: TopicPartition
+    def __init__(self, topic_partition: TopicPartition) -> None: ...
+    def success(self, value): ...
+    def failure(self, error): ...
+    def wait(self, timeout=None): ...
+
+class FutureRecordMetadata(Future):
+    args: Incomplete
+    def __init__(
+        self,
+        produce_future,
+        batch_index,
+        timestamp_ms,
+        checksum,
+        serialized_key_size,
+        serialized_value_size,
+        serialized_header_size,
+    ) -> None: ...
+    def get(self, timeout=None): ...
+
+class RecordMetadata(NamedTuple):
+    topic: str
+    partition: int
+    topic_partition: TopicPartition
+    offset: int
+    timestamp: int
+    checksum: int | None
+    serialized_key_size: int
+    serialized_value_size: int
+    serialized_header_size: int

kafka-stubs/producer/kafka.pyi

@@ -0,0 +1,110 @@
+import selectors
+import ssl
+from _typeshed import Incomplete
+from collections.abc import Callable, Mapping, Sequence
+from typing import Literal, TypeAlias, TypedDict, type_check_only
+from typing_extensions import Unpack
+
+from kafka.producer.future import FutureRecordMetadata
+from kafka.serializer.abstract import Serializer
+from kafka.structs import OffsetAndMetadata, TopicPartition
+
+_ApiVersion: TypeAlias = tuple[int, ...]
+_BootstrapServers: TypeAlias = str | Sequence[str]
+_KafkaClientFactory: TypeAlias = Callable[..., object]
+_Partitioner: TypeAlias = Callable[[bytes | None, Sequence[int], Sequence[int]], int]
+_ProducerSerializer: TypeAlias = Serializer | Callable[[object], bytes]
+_SaslMechanism: TypeAlias = Literal["PLAIN", "GSSAPI", "OAUTHBEARER", "SCRAM-SHA-256", "SCRAM-SHA-512"]
+_SecurityProtocol: TypeAlias = Literal["PLAINTEXT", "SSL", "SASL_PLAINTEXT", "SASL_SSL"]
+_SocketOption: TypeAlias = tuple[int, int, int]
+
+@type_check_only
+class _KafkaProducerConfig(TypedDict, total=False):
+    bootstrap_servers: _BootstrapServers
+    client_id: str | None
+    key_serializer: _ProducerSerializer | None
+    value_serializer: _ProducerSerializer | None
+    enable_idempotence: bool
+    transactional_id: str | None
+    transaction_timeout_ms: int
+    delivery_timeout_ms: float
+    acks: int | Literal["all"]
+    bootstrap_topics_filter: set[str]
+    compression_type: Literal["gzip", "snappy", "lz4", "zstd"] | None
+    retries: int | float
+    batch_size: int
+    linger_ms: int
+    partitioner: _Partitioner
+    connections_max_idle_ms: int
+    max_block_ms: int
+    max_request_size: int
+    allow_auto_create_topics: bool
+    metadata_max_age_ms: int
+    retry_backoff_ms: int
+    request_timeout_ms: int
+    receive_buffer_bytes: int | None
+    send_buffer_bytes: int | None
+    socket_options: Sequence[_SocketOption]
+    sock_chunk_bytes: int
+    sock_chunk_buffer_count: int
+    reconnect_backoff_ms: int
+    reconnect_backoff_max_ms: int
+    max_in_flight_requests_per_connection: int
+    security_protocol: _SecurityProtocol
+    ssl_context: ssl.SSLContext | None
+    ssl_check_hostname: bool
+    ssl_cafile: str | None
+    ssl_certfile: str | None
+    ssl_keyfile: str | None
+    ssl_crlfile: str | None
+    ssl_password: str | None
+    ssl_ciphers: str | None
+    api_version: _ApiVersion | None
+    api_version_auto_timeout_ms: int
+    metric_reporters: Sequence[type[object]]
+    metrics_enabled: bool
+    metrics_num_samples: int
+    metrics_sample_window_ms: int
+    selector: type[selectors.BaseSelector]
+    sasl_mechanism: _SaslMechanism | None
+    sasl_plain_username: str | None
+    sasl_plain_password: str | None
+    sasl_kerberos_name: object | None
+    sasl_kerberos_service_name: str
+    sasl_kerberos_domain_name: str | None
+    sasl_oauth_token_provider: object | None
+    socks5_proxy: str | None
+    kafka_client: _KafkaClientFactory
+
+log: Incomplete
+PRODUCER_CLIENT_ID_SEQUENCE: Incomplete
+
+class KafkaProducer:
+    DEFAULT_CONFIG: Incomplete
+    DEPRECATED_CONFIGS: Incomplete
+    config: Incomplete
+    def __init__(self, **configs: Unpack[_KafkaProducerConfig]) -> None: ...
+    def bootstrap_connected(self): ...
+    def __del__(self) -> None: ...
+    def close(self, timeout: float | None = None, null_logger: bool = False) -> None: ...
+    def partitions_for(self, topic: str) -> set[int]: ...
+    @classmethod
+    def max_usable_produce_magic(cls, api_version): ...
+    def init_transactions(self) -> None: ...
+    def begin_transaction(self) -> None: ...
+    def send_offsets_to_transaction(
+        self, offsets: Mapping[TopicPartition, OffsetAndMetadata], consumer_group_id: str
+    ) -> None: ...
+    def commit_transaction(self) -> None: ...
+    def abort_transaction(self) -> None: ...
+    def send(
+        self,
+        topic: str,
+        value: object = None,
+        key: object = None,
+        headers: Sequence[tuple[str, bytes]] | None = None,
+        partition: int | None = None,
+        timestamp_ms: int | None = None,
+    ) -> FutureRecordMetadata: ...
+    def flush(self, timeout: float | None = None) -> None: ...
+    def metrics(self, raw: bool = False) -> dict[str, dict[str, object]] | dict[object, object] | None: ...

kafka-stubs/producer/producer_batch.pyi

@@ -0,0 +1,42 @@
+from _typeshed import Incomplete
+from enum import IntEnum
+
+log: Incomplete
+
+class FinalState(IntEnum):
+    ABORTED = 0
+    FAILED = 1
+    SUCCEEDED = 2
+
+class ProducerBatch:
+    max_record_size: int
+    created: Incomplete
+    drained: Incomplete
+    attempts: int
+    last_attempt: Incomplete
+    last_append: Incomplete
+    records: Incomplete
+    topic_partition: Incomplete
+    produce_future: Incomplete
+    def __init__(self, tp, records, now=None) -> None: ...
+    @property
+    def final_state(self): ...
+    @property
+    def record_count(self): ...
+    @property
+    def producer_id(self): ...
+    @property
+    def producer_epoch(self): ...
+    @property
+    def has_sequence(self): ...
+    def try_append(self, timestamp_ms, key, value, headers, now=None): ...
+    def abort(self, exception): ...
+    def complete(self, base_offset, log_append_time): ...
+    def complete_exceptionally(self, top_level_exception, record_exceptions_fn): ...
+    def done(self, base_offset=None, timestamp_ms=None, top_level_exception=None, record_exceptions_fn=None): ...
+    def has_reached_delivery_timeout(self, delivery_timeout_ms, now=None): ...
+    def in_retry(self): ...
+    def retry(self, now=None) -> None: ...
+    @property
+    def is_done(self): ...
+    def __lt__(self, other): ...

kafka-stubs/producer/record_accumulator.pyi

@@ -0,0 +1,45 @@
+from _typeshed import Incomplete
+
+log: Incomplete
+
+class AtomicInteger:
+    def __init__(self, val: int = 0) -> None: ...
+    def increment(self): ...
+    def decrement(self): ...
+    def get(self): ...
+
+class RecordAccumulator:
+    DEFAULT_CONFIG: Incomplete
+    config: Incomplete
+    muted: Incomplete
+    def __init__(self, **configs) -> None: ...
+    @property
+    def delivery_timeout_ms(self): ...
+    @property
+    def next_expiry_time_ms(self): ...
+    def append(self, tp, timestamp_ms, key, value, headers, now=None): ...
+    def reset_next_batch_expiry_time(self) -> None: ...
+    def maybe_update_next_batch_expiry_time(self, batch) -> None: ...
+    def expired_batches(self, now=None): ...
+    def reenqueue(self, batch, now=None) -> None: ...
+    def ready(self, cluster, now=None): ...
+    def has_undrained(self): ...
+    def drain_batches_for_one_node(self, cluster, node_id, max_size, now=None): ...
+    def drain(self, cluster, nodes, max_size, now=None): ...
+    def deallocate(self, batch) -> None: ...
+    def flush_in_progress(self): ...
+    def begin_flush(self) -> None: ...
+    def await_flush_completion(self, timeout=None) -> None: ...
+    @property
+    def has_incomplete(self): ...
+    def abort_incomplete_batches(self) -> None: ...
+    def abort_undrained_batches(self, error) -> None: ...
+    def close(self) -> None: ...
+
+class IncompleteProducerBatches:
+    def __init__(self) -> None: ...
+    def add(self, batch) -> None: ...
+    def remove(self, batch) -> None: ...
+    def all(self): ...
+    def __bool__(self) -> bool: ...
+    __nonzero__ = __bool__

kafka-stubs/producer/sender.pyi

@@ -0,0 +1,47 @@
+import threading
+from _typeshed import Incomplete
+from typing import NamedTuple
+
+log: Incomplete
+
+class PartitionResponse(NamedTuple):
+    error: Incomplete = ...
+    base_offset: Incomplete = ...
+    last_offset: Incomplete = ...
+    log_append_time: Incomplete = ...
+    log_start_offset: Incomplete = ...
+    record_errors: Incomplete = ...
+    error_message: Incomplete = ...
+    current_leader: Incomplete = ...
+
+class Sender(threading.Thread):
+    DEFAULT_CONFIG: Incomplete
+    config: Incomplete
+    name: Incomplete
+    def __init__(self, client, metadata, accumulator, **configs) -> None: ...
+    def run(self) -> None: ...
+    def run_once(self) -> None: ...
+    def initiate_close(self) -> None: ...
+    def force_close(self) -> None: ...
+    def add_topic(self, topic) -> None: ...
+    def wakeup(self) -> None: ...
+    def bootstrap_connected(self): ...
+
+class SenderMetrics:
+    metrics: Incomplete
+    batch_size_sensor: Incomplete
+    compression_rate_sensor: Incomplete
+    queue_time_sensor: Incomplete
+    records_per_request_sensor: Incomplete
+    byte_rate_sensor: Incomplete
+    retry_sensor: Incomplete
+    error_sensor: Incomplete
+    max_record_size_sensor: Incomplete
+    def __init__(self, metrics, client, metadata) -> None: ...
+    def add_metric(
+        self, metric_name, measurable, group_name: str = "producer-metrics", description=None, tags=None, sensor_name=None
+    ) -> None: ...
+    def maybe_register_topic_metrics(self, topic): ...
+    def update_produce_request_metrics(self, batches_map) -> None: ...
+    def record_retries(self, topic, count) -> None: ...
+    def record_errors(self, topic, count) -> None: ...