fix: polish middleware types in favor to Pycharm

Author: Lancetnik

docs/docs.py

@@ -5,7 +5,6 @@
 from http.server import HTTPServer, SimpleHTTPRequestHandler
 from pathlib import Path
 
-import mkdocs.commands.build
 import mkdocs.commands.serve
 import typer
 from create_api_docs import create_api_docs, remove_api_dir, render_navigation

faststream/confluent/prometheus/middleware.py

@@ -14,8 +14,8 @@
 
 class KafkaPrometheusMiddleware(
     PrometheusMiddleware[
-        KafkaPublishCommand,
         Message | Sequence[Message],
+        KafkaPublishCommand,
     ],
 ):
     def __init__(

faststream/kafka/prometheus/middleware.py

@@ -14,8 +14,8 @@
 
 class KafkaPrometheusMiddleware(
     PrometheusMiddleware[
-        KafkaPublishCommand,
         ConsumerRecord | Sequence[ConsumerRecord],
+        KafkaPublishCommand,
     ],
 ):
     def __init__(

faststream/nats/prometheus/middleware.py

@@ -13,7 +13,7 @@
 
 
 class NatsPrometheusMiddleware(
-    PrometheusMiddleware[NatsPublishCommand, Msg | Sequence[Msg]],
+    PrometheusMiddleware[Msg | Sequence[Msg], NatsPublishCommand],
 ):
     def __init__(
         self,

faststream/opentelemetry/middleware.py

@@ -2,7 +2,7 @@
 from collections import defaultdict
 from collections.abc import Callable
 from copy import copy
-from typing import TYPE_CHECKING, Any, Generic, Optional, cast
+from typing import TYPE_CHECKING, Any, Optional, cast
 
 from opentelemetry import baggage, context, metrics, trace
 from opentelemetry.baggage.propagation import W3CBaggagePropagator
@@ -12,7 +12,7 @@
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
 
 from faststream._internal.middlewares import BaseMiddleware
-from faststream._internal.types import PublishCommandType
+from faststream._internal.types import BrokerMiddleware, PublishCommandType
 from faststream.opentelemetry.baggage import Baggage
 from faststream.opentelemetry.consts import (
     ERROR_TYPE,
@@ -42,7 +42,7 @@
 _TRACE_PROPAGATOR = TraceContextTextMapPropagator()
 
 
-class TelemetryMiddleware(Generic[PublishCommandType]):
+class TelemetryMiddleware(BrokerMiddleware[Any, PublishCommandType]):
     __slots__ = (
         "_meter",
         "_metrics",

faststream/prometheus/middleware.py

@@ -4,7 +4,7 @@
 
 from faststream._internal.constants import EMPTY
 from faststream._internal.middlewares import BaseMiddleware
-from faststream._internal.types import AnyMsg, PublishCommandType
+from faststream._internal.types import AnyMsg, BrokerMiddleware, PublishCommandType
 from faststream.exceptions import IgnoredException
 from faststream.message import SourceType
 from faststream.prometheus.consts import (
@@ -25,7 +25,7 @@
     from faststream.message.message import StreamMessage
 
 
-class PrometheusMiddleware(Generic[PublishCommandType, AnyMsg]):
+class PrometheusMiddleware(BrokerMiddleware[AnyMsg, PublishCommandType]):
     __slots__ = (
         "_dynamic_labels",
         "_metrics_container",

faststream/rabbit/prometheus/middleware.py

@@ -13,7 +13,7 @@
 
 
 class RabbitPrometheusMiddleware(
-    PrometheusMiddleware[RabbitPublishCommand, IncomingMessage],
+    PrometheusMiddleware[IncomingMessage, RabbitPublishCommand],
 ):
     def __init__(
         self,

faststream/redis/prometheus/middleware.py

@@ -11,7 +11,7 @@
 
 
 class RedisPrometheusMiddleware(
-    PrometheusMiddleware[RedisPublishCommand, dict[str, Any]]
+    PrometheusMiddleware[dict[str, Any], RedisPublishCommand]
 ):
     def __init__(
         self,