feat: Add aio module and update mypy configuration to include aio files (#942)

Signed-off-by: Albert Callarisa <albert@diagrid.io>
(cherry picked from commit 8589d2e)

Author: acroca

dapr/aio/__init__.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+
+"""
+Copyright 2026 The Dapr Authors
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+    http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""

dapr/aio/clients/__init__.py

@@ -103,7 +103,7 @@ async def invoke_method(
         self,
         app_id: str,
         method_name: str,
-        data: Union[bytes, str, GrpcMessage],
+        data: Union[bytes, str, GrpcMessage] = b'',
         content_type: Optional[str] = None,
         metadata: Optional[MetadataTuple] = None,
         http_verb: Optional[str] = None,

dapr/aio/clients/grpc/client.py

@@ -19,7 +19,7 @@
 import time
 import uuid
 from datetime import datetime
-from typing import Any, Awaitable, Callable, Dict, List, Optional, Sequence, Text, Union
+from typing import Any, Awaitable, Callable, Dict, List, Optional, Sequence, Text, Tuple, Union
 from urllib.parse import urlencode
 from warnings import warn
 
@@ -154,7 +154,7 @@ def __init__(
 
         useragent = f'dapr-sdk-python/{__version__}'
         if not max_grpc_message_length:
-            options = [
+            options: List[Tuple[str, Any]] = [
                 ('grpc.primary_user_agent', useragent),
             ]
         else:
@@ -205,7 +205,7 @@ def __init__(
 
     @staticmethod
     def get_credentials():
-        return grpc.ssl_channel_credentials()
+        return grpc.ssl_channel_credentials()  # type: ignore[attr-defined]
 
     async def close(self):
         """Closes Dapr runtime gRPC channel."""
@@ -604,7 +604,7 @@ async def subscribe_with_handler(
         self,
         pubsub_name: str,
         topic: str,
-        handler_fn: Callable[..., TopicEventResponse],
+        handler_fn: Callable[..., Awaitable[TopicEventResponse]],
         metadata: Optional[dict] = None,
         dead_letter_topic: Optional[str] = None,
     ) -> Callable[[], Awaitable[None]]:
@@ -1020,7 +1020,7 @@ async def execute_state_transaction(
                 operationType=o.operation_type.value,
                 request=common_v1.StateItem(
                     key=o.key,
-                    value=to_bytes(o.data),
+                    value=to_bytes(o.data) if o.data is not None else to_bytes(''),
                     etag=common_v1.Etag(value=o.etag) if o.etag is not None else None,
                 ),
             )
@@ -1360,7 +1360,7 @@ async def try_lock(
         response = await call
         return TryLockResponse(
             success=response.success,
-            client=self,
+            client=self,  # type: ignore[arg-type]
             store_name=store_name,
             resource_id=resource_id,
             lock_owner=lock_owner,
@@ -1697,7 +1697,7 @@ async def raise_workflow_event(
             else:
                 encoded_data = bytes([])
         # Actual workflow raise event invocation
-        req = api_v1.raise_workflow_event(
+        req = api_v1.RaiseEventWorkflowRequest(
             instance_id=instance_id,
             workflow_component=workflow_component,
             event_name=event_name,

dapr/aio/clients/grpc/interceptors.py

@@ -92,9 +92,9 @@ async def _intercept_call(self, client_call_details: ClientCallDetails) -> Clien
             :class: `ClientCallDetails` modified call details
         """
 
-        metadata = []
+        metadata: List[Tuple[str, str]] = []
         if client_call_details.metadata is not None:
-            metadata = list(client_call_details.metadata)
+            metadata = list(client_call_details.metadata)  # type: ignore[arg-type]
         metadata.extend(self._metadata)
 
         new_call_details = _ClientCallDetailsAsync(

dapr/aio/clients/grpc/subscription.py

@@ -1,6 +1,6 @@
 import asyncio
 
-from grpc import StatusCode
+from grpc import StatusCode  # type: ignore[attr-defined]
 from grpc.aio import AioRpcError
 
 from dapr.aio.clients.health import DaprHealth
@@ -21,7 +21,7 @@ def __init__(self, stub, pubsub_name, topic, metadata=None, dead_letter_topic=No
         self._metadata = metadata or {}
         self._dead_letter_topic = dead_letter_topic or ''
         self._stream = None
-        self._send_queue = asyncio.Queue()
+        self._send_queue: asyncio.Queue[api_v1.SubscribeTopicEventsRequestAlpha1] = asyncio.Queue()
         self._stream_active = asyncio.Event()
 
     async def start(self):

mypy.ini

@@ -9,6 +9,7 @@ non_interactive = True
 
 files =
     dapr/actor/**/*.py,
+    dapr/aio/**/*.py,
     dapr/clients/**/*.py,
     dapr/conf/**/*.py,
     dapr/serializers/**/*.py,