Fix typing

Author: WiZeYAR

src/asyncapi_python/amqp/connection.py

@@ -14,7 +14,6 @@
 
 
 from functools import cache
-from typing import TypeAlias
 from aio_pika.robust_connection import (
     AbstractRobustConnection,
     AbstractRobustChannel,

src/asyncapi_python/amqp/endpoint/receiver.py

@@ -45,8 +45,8 @@ async def start(self) -> None:
         print("start", self._op)
         if self._fn:
             async with self._params.pool.acquire() as ch:
-                if self._params.amqp_params.prefetch_count:
-                    await ch.set_qos(prefetch_count=self._params.amqp_params.prefetch_count)
+                if prefetch_count := self._params.amqp_params.get("prefetch_count"):
+                    await ch.set_qos(prefetch_count=prefetch_count)
                 q = self._queue = await self._declare(ch)
                 self._consumer_tag = await q.consume(self._consumer)
             return

src/asyncapi_python/amqp/params.py

@@ -1,4 +1,5 @@
 from typing import TypedDict
 
+
 class AmqpParams(TypedDict, total=False):
-    prefetch_count: str
+    prefetch_count: int