Merge pull request #8 from ApeWorX/fix/boto3-types

fix: mypy_boto3_sqs dependency and conditional type checking import

Author: Mike Shultz

pyproject.toml

@@ -38,6 +38,7 @@ dev = [
     "flake8~=7.0.0",
     "isort~=5.13.2",
     "mypy~=1.8.0",
+    "mypy-boto3-sqs~=1.34.101",
     "pytest~=7.4.4",
     "requests~=2.31.0",
     "setuptools-scm~=8.0.4",

taskiq_sqs/broker.py

@@ -1,18 +1,24 @@
+from __future__ import (
+    annotations,  # Needed for conditional type import support
+)
+
 import asyncio
 import logging
 from datetime import datetime, timezone
-from typing import AsyncGenerator, Callable, Optional, Union
+from typing import TYPE_CHECKING, AsyncGenerator, Callable, Optional, Union
 
 import boto3
 from asyncer import asyncify
 from botocore.exceptions import ClientError
-from mypy_boto3_sqs.service_resource import Queue, SQSServiceResource
 from taskiq import AsyncBroker
 from taskiq.abc.result_backend import AsyncResultBackend
 from taskiq.acks import AckableMessage
 from taskiq.exceptions import BrokerError
 from taskiq.message import BrokerMessage
 
+if TYPE_CHECKING:
+    from mypy_boto3_sqs.service_resource import Queue, SQSServiceResource
+
 logger = logging.getLogger(__name__)