From c13907b9394cc3f8eec5279d1339b4a0e5fae69e Mon Sep 17 00:00:00 2001 From: reqww Date: Wed, 12 Nov 2025 19:41:44 +0300 Subject: [PATCH] Add faststream compatibility --- packages/faststream-stomp/faststream_stomp/broker.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/packages/faststream-stomp/faststream_stomp/broker.py b/packages/faststream-stomp/faststream_stomp/broker.py index 907d97b..4c5e5a9 100644 --- a/packages/faststream-stomp/faststream_stomp/broker.py +++ b/packages/faststream-stomp/faststream_stomp/broker.py @@ -8,7 +8,7 @@ import anyio import stompman from fast_depends.dependencies import Dependant -from faststream import ContextRepo, PublishType +from faststream import BaseMiddleware, ContextRepo, PublishType from faststream._internal.basic_types import LoggerProto, SendableMessage from faststream._internal.broker import BrokerUsecase from faststream._internal.broker.registrator import Registrator @@ -47,7 +47,8 @@ def _handle_listen_task_done(listen_task: asyncio.Task[None]) -> None: except asyncio.CancelledError: return if isinstance(task_exception, ExceptionGroup) and isinstance( - task_exception.exceptions[0], stompman.FailedAllConnectAttemptsError + task_exception.exceptions[0], + stompman.FailedAllConnectAttemptsError, ): raise SystemExit(1) @@ -94,7 +95,7 @@ def __init__( decoder: CustomCallable | None = None, parser: CustomCallable | None = None, dependencies: Iterable[Dependant] = (), - middlewares: Sequence[BrokerMiddleware[stompman.MessageFrame, StompPublishCommand]] = (), + middlewares: Sequence[type[BaseMiddleware] | BrokerMiddleware[stompman.MessageFrame, StompPublishCommand]] = (), graceful_timeout: float | None = 15.0, routers: Sequence[Registrator[stompman.MessageFrame]] = (), # Logging args