fix: FastStream 0.5.0 compat

Author: Lancetnik

taskiq_faststream/__about__.py

@@ -1,2 +1,2 @@
 """FastStream - taskiq integration to schedule FastStream tasks."""
-__version__ = "0.1.7"
+__version__ = "0.1.8"

taskiq_faststream/broker.py

@@ -1,9 +1,9 @@
 import typing
 import warnings
+from typing import Any
 
 import anyio
 from faststream.app import FastStream
-from faststream.broker.core.asynchronous import BrokerAsyncUsecase
 from faststream.types import SendableMessage
 from taskiq import AsyncBroker, BrokerMessage
 from taskiq.acks import AckableMessage
@@ -31,7 +31,7 @@ class BrokerWrapper(AsyncBroker):
         task : Register FastStream scheduled task.
     """
 
-    def __init__(self, broker: BrokerAsyncUsecase[typing.Any, typing.Any]) -> None:
+    def __init__(self, broker: Any) -> None:
         super().__init__()
         self.serializer = PatchedSerializer()
         self.broker = broker
@@ -131,7 +131,7 @@ async def kick(self, message: BrokerMessage) -> None:
 
 
 async def _broker_publish(
-    broker: BrokerAsyncUsecase[typing.Any, typing.Any],
+    broker: Any,
     message: BrokerMessage,
 ) -> None:
     labels = message.labels

tests/testcase.py

@@ -4,7 +4,6 @@
 from unittest.mock import MagicMock
 
 import pytest
-from faststream.broker.core.asynchronous import BrokerAsyncUsecase
 from faststream.utils.functions import timeout_scope
 from taskiq import AsyncBroker, TaskiqScheduler
 from taskiq.cli.scheduler.args import SchedulerArgs
@@ -20,14 +19,14 @@ class SchedulerTestcase:
     subj_name: str
 
     @staticmethod
-    def build_taskiq_broker(broker: BrokerAsyncUsecase[Any, Any]) -> AsyncBroker:
+    def build_taskiq_broker(broker: Any) -> AsyncBroker:
         """Build Taskiq compatible object."""
         return BrokerWrapper(broker)
 
     async def test_task(
         self,
         subject: str,
-        broker: BrokerAsyncUsecase[Any, Any],
+        broker: Any,
         mock: MagicMock,
         event: asyncio.Event,
     ) -> None: