|
8 | 8 | import anyio |
9 | 9 | import stompman |
10 | 10 | from fast_depends.dependencies import Dependant |
11 | | -from faststream import ContextRepo, PublishType |
| 11 | +from faststream import BaseMiddleware, ContextRepo, PublishType |
12 | 12 | from faststream._internal.basic_types import LoggerProto, SendableMessage |
13 | 13 | from faststream._internal.broker import BrokerUsecase |
14 | 14 | from faststream._internal.broker.registrator import Registrator |
@@ -47,7 +47,8 @@ def _handle_listen_task_done(listen_task: asyncio.Task[None]) -> None: |
47 | 47 | except asyncio.CancelledError: |
48 | 48 | return |
49 | 49 | if isinstance(task_exception, ExceptionGroup) and isinstance( |
50 | | - task_exception.exceptions[0], stompman.FailedAllConnectAttemptsError |
| 50 | + task_exception.exceptions[0], |
| 51 | + stompman.FailedAllConnectAttemptsError, |
51 | 52 | ): |
52 | 53 | raise SystemExit(1) |
53 | 54 |
|
@@ -94,7 +95,7 @@ def __init__( |
94 | 95 | decoder: CustomCallable | None = None, |
95 | 96 | parser: CustomCallable | None = None, |
96 | 97 | dependencies: Iterable[Dependant] = (), |
97 | | - middlewares: Sequence[BrokerMiddleware[stompman.MessageFrame, StompPublishCommand]] = (), |
| 98 | + middlewares: Sequence[type[BaseMiddleware] | BrokerMiddleware[stompman.MessageFrame, StompPublishCommand]] = (), |
98 | 99 | graceful_timeout: float | None = 15.0, |
99 | 100 | routers: Sequence[Registrator[stompman.MessageFrame]] = (), |
100 | 101 | # Logging args |
|
0 commit comments