fixed warnings

Author: daniel-sanche

google/cloud/bigtable/data/_async/metrics_interceptor.py

@@ -28,6 +28,7 @@
 if CrossSync.is_async:
     from grpc.aio import UnaryUnaryClientInterceptor
     from grpc.aio import UnaryStreamClientInterceptor
+    from grpc.aio import AioRpcError
 else:
     from grpc import UnaryUnaryClientInterceptor
     from grpc import UnaryStreamClientInterceptor
@@ -86,12 +87,18 @@ def _end_attempt(operation, exc, metadata):
 
 
 @CrossSync.convert
-async def _get_metadata(source):
+async def _get_metadata(source) -> dict[str, str|bytes] | None:
     """Helper to extract metadata from a call or RpcError"""
     try:
-        return (await source.trailing_metadata() or {}) + (
-            await source.initial_metadata() or {}
-        )
+        if isinstance(source, AioRpcError) and CrossSync.is_async:
+            return {
+                k:v for k,v 
+                in list(source.trailing_metadata()) + list(source.initial_metadata())
+            }
+        else:
+            return (await source.trailing_metadata() or {}) + (
+                await source.initial_metadata() or {}
+            )
     except Exception:
         # ignore errors while fetching metadata
         return None

google/cloud/bigtable/data/_metrics/data_model.py

@@ -25,6 +25,7 @@
 from dataclasses import dataclass
 from dataclasses import field
 from grpc import StatusCode
+from grpc.aio import AioRpcError
 
 import google.cloud.bigtable.data.exceptions as bt_exceptions
 from google.cloud.bigtable_v2.types.response_params import ResponseParams
@@ -389,6 +390,8 @@ def _exc_to_status(exc: BaseException) -> StatusCode:
             and exc.__cause__.grpc_status_code is not None
         ):
             return exc.__cause__.grpc_status_code
+        if isinstance(exc, AioRpcError):
+            return exc.code()
         return StatusCode.UNKNOWN
 
     @staticmethod