mypy

Author: sentrivana

sentry_sdk/_metrics.py

@@ -4,18 +4,18 @@
 """
 
 import time
-from typing import Any, Optional, TYPE_CHECKING
+from typing import Any, Optional, TYPE_CHECKING, Union
 
 import sentry_sdk
 from sentry_sdk.utils import safe_repr
 
 if TYPE_CHECKING:
-    from sentry_sdk._types import Metric
+    from sentry_sdk._types import Metric, MetricType
 
 
 def _capture_metric(
     name,  # type: str
-    metric_type,  # type: str
+    metric_type,  # type: MetricType
     value,  # type: float
     unit=None,  # type: Optional[str]
     attributes=None,  # type: Optional[dict[str, Any]]

sentry_sdk/_metrics_batcher.py

@@ -2,7 +2,7 @@
 import random
 import threading
 from datetime import datetime, timezone
-from typing import Optional, List, Callable, TYPE_CHECKING, Any
+from typing import Optional, List, Callable, TYPE_CHECKING, Any, Union
 
 from sentry_sdk.utils import format_timestamp, safe_repr
 from sentry_sdk.envelope import Envelope, Item, PayloadRef

sentry_sdk/_types.py

@@ -254,7 +254,7 @@ class SDKInfo(TypedDict):
             "type": MetricType,
             "value": float,
             "unit": Optional[str],
-            "attributes": dict[str, MetricAttributeValue],
+            "attributes": dict[str, str | bool | float | int],
         },
     )
 

sentry_sdk/client.py

@@ -68,6 +68,7 @@
     from sentry_sdk.spotlight import SpotlightClient
     from sentry_sdk.transport import Transport
     from sentry_sdk._log_batcher import LogBatcher
+    from sentry_sdk._metrics_batcher import MetricsBatcher
 
     I = TypeVar("I", bound=Integration)  # noqa: E741
 

sentry_sdk/utils.py

@@ -59,7 +59,7 @@
 
     from gevent.hub import Hub
 
-    from sentry_sdk._types import Event, ExcInfo, Log, Hint
+    from sentry_sdk._types import Event, ExcInfo, Log, Hint, Metric
 
     P = ParamSpec("P")
     R = TypeVar("R")