diff --git a/pyproject.toml b/pyproject.toml index 180253f..51fd711 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -41,7 +41,7 @@ classifiers = [ dependencies = [ "taskiq>=0.12.1,<0.13.0", - "faststream>=0.3.14,<0.7", + "faststream>=0.3.14,<0.8", ] [project.optional-dependencies] diff --git a/taskiq_faststream/broker.py b/taskiq_faststream/broker.py index 6ed7d97..ca3f43a 100644 --- a/taskiq_faststream/broker.py +++ b/taskiq_faststream/broker.py @@ -56,7 +56,9 @@ async def startup(self) -> None: async def shutdown(self) -> None: """Shutdown wrapped FastStream broker.""" - await self.broker.close() + # `stop` replaced `close` in FastStream 0.5.44; `close` removed in 0.7.0. + stop = getattr(self.broker, "stop", None) or self.broker.close + await stop() await super().shutdown() async def kick(self, message: PatchedMessage) -> None: # type: ignore[override]