Enforce from __future__ import annotations

Author: Kludex

exporter/opentelemetry-exporter-prometheus-remote-write/src/opentelemetry/exporter/prometheus_remote_write/__init__.py

@@ -11,12 +11,13 @@
 # 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.
+from __future__ import annotations
 
 import logging
 import re
 from collections import defaultdict
 from itertools import chain
-from typing import Dict, Mapping, Sequence
+from typing import Mapping, Sequence
 
 import requests
 import snappy
@@ -71,14 +72,14 @@ class PrometheusRemoteWriteMetricsExporter(MetricExporter):
     def __init__(
         self,
         endpoint: str,
-        basic_auth: Dict = None,
-        headers: Dict = None,
+        basic_auth: dict = None,
+        headers: dict = None,
         timeout: int = 30,
-        tls_config: Dict = None,
-        proxies: Dict = None,
+        tls_config: dict = None,
+        proxies: dict = None,
         resources_as_labels: bool = True,
-        preferred_temporality: Dict[type, AggregationTemporality] = None,
-        preferred_aggregation: Dict = None,
+        preferred_temporality: dict[type, AggregationTemporality] = None,
+        preferred_aggregation: dict = None,
     ):
         self.endpoint = endpoint
         self.basic_auth = basic_auth
@@ -115,7 +116,7 @@ def basic_auth(self):
         return self._basic_auth
 
     @basic_auth.setter
-    def basic_auth(self, basic_auth: Dict):
+    def basic_auth(self, basic_auth: dict):
         if basic_auth:
             if "username" not in basic_auth:
                 raise ValueError("username required in basic_auth")
@@ -147,7 +148,7 @@ def tls_config(self):
         return self._tls_config
 
     @tls_config.setter
-    def tls_config(self, tls_config: Dict):
+    def tls_config(self, tls_config: dict):
         if tls_config:
             new_config = {}
             if "ca_file" in tls_config:
@@ -170,15 +171,15 @@ def proxies(self):
         return self._proxies
 
     @proxies.setter
-    def proxies(self, proxies: Dict):
+    def proxies(self, proxies: dict):
         self._proxies = proxies
 
     @property
     def headers(self):
         return self._headers
 
     @headers.setter
-    def headers(self, headers: Dict):
+    def headers(self, headers: dict):
         self._headers = headers
 
     def export(
@@ -355,7 +356,7 @@ def _build_message(timeseries: Sequence[TimeSeries]) -> bytes:
         serialized_message = write_request.SerializeToString()
         return snappy.compress(serialized_message)
 
-    def _build_headers(self) -> Dict:
+    def _build_headers(self) -> dict:
         headers = {
             "Content-Encoding": "snappy",
             "Content-Type": "application/x-protobuf",
@@ -367,7 +368,7 @@ def _build_headers(self) -> Dict:
         return headers
 
     def _send_message(
-        self, message: bytes, headers: Dict
+        self, message: bytes, headers: dict
     ) -> MetricExportResult:
         auth = None
         if self.basic_auth:

exporter/opentelemetry-exporter-richconsole/src/opentelemetry/exporter/richconsole/__init__.py

@@ -50,11 +50,12 @@
 .. _Rich: https://rich.readthedocs.io/
 .. _OpenTelemetry: https://github.com/open-telemetry/opentelemetry-python/
 """
+
 # pylint: disable=import-error
+from __future__ import annotations
 
 import datetime
 import typing
-from typing import Dict, Optional
 
 from rich.console import Console
 from rich.syntax import Syntax
@@ -150,7 +151,7 @@ class RichConsoleSpanExporter(SpanExporter):
 
     def __init__(
         self,
-        service_name: Optional[str] = None,
+        service_name: str | None = None,
     ):
         self.service_name = service_name
         self.console = Console()
@@ -165,7 +166,7 @@ def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         return SpanExportResult.SUCCESS
 
     @staticmethod
-    def spans_to_tree(spans: typing.Sequence[ReadableSpan]) -> Dict[str, Tree]:
+    def spans_to_tree(spans: typing.Sequence[ReadableSpan]) -> dict[str, Tree]:
         trees = {}
         parents = {}
         spans = list(spans)

instrumentation-genai/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/patch.py

@@ -11,9 +11,7 @@
 # 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.
-
-
-from typing import Optional
+from __future__ import annotations
 
 from openai import Stream
 
@@ -201,12 +199,12 @@ def append_tool_call(self, tool_call):
 
 class StreamWrapper:
     span: Span
-    response_id: Optional[str] = None
-    response_model: Optional[str] = None
-    service_tier: Optional[str] = None
+    response_id: str | None = None
+    response_model: str | None = None
+    service_tier: str | None = None
     finish_reasons: list = []
-    prompt_tokens: Optional[int] = 0
-    completion_tokens: Optional[int] = 0
+    prompt_tokens: int | None = 0
+    completion_tokens: int | None = 0
 
     def __init__(
         self,

instrumentation-genai/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/utils.py

@@ -11,9 +11,10 @@
 # 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.
+from __future__ import annotations
 
 from os import environ
-from typing import Mapping, Optional, Union
+from typing import Mapping
 from urllib.parse import urlparse
 
 from httpx import URL
@@ -179,7 +180,7 @@ def is_streaming(kwargs):
     return non_numerical_value_is_set(kwargs.get("stream"))
 
 
-def non_numerical_value_is_set(value: Optional[Union[bool, str]]):
+def non_numerical_value_is_set(value: bool | str | None):
     return bool(value) and value != NOT_GIVEN
 
 

instrumentation-genai/opentelemetry-instrumentation-openai-v2/tests/test_async_chat_completions.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # pylint: disable=too-many-locals
-
-from typing import Optional
+from __future__ import annotations
 
 import pytest
 from openai import APIConnectionError, AsyncOpenAI, NotFoundError
@@ -802,8 +801,8 @@ def assert_all_attributes(
     request_model: str,
     response_id: str = None,
     response_model: str = None,
-    input_tokens: Optional[int] = None,
-    output_tokens: Optional[int] = None,
+    input_tokens: int | None = None,
+    output_tokens: int | None = None,
     operation_name: str = "chat",
     server_address: str = "api.openai.com",
 ):

instrumentation-genai/opentelemetry-instrumentation-openai-v2/tests/test_chat_completions.py

@@ -12,8 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # pylint: disable=too-many-locals
-
-from typing import Optional
+from __future__ import annotations
 
 import pytest
 from openai import APIConnectionError, NotFoundError, OpenAI
@@ -779,8 +778,8 @@ def assert_all_attributes(
     request_model: str,
     response_id: str = None,
     response_model: str = None,
-    input_tokens: Optional[int] = None,
-    output_tokens: Optional[int] = None,
+    input_tokens: int | None = None,
+    output_tokens: int | None = None,
     operation_name: str = "chat",
     server_address: str = "api.openai.com",
 ):

instrumentation/opentelemetry-instrumentation-aio-pika/src/opentelemetry/instrumentation/aio_pika/callback_decorator.py

@@ -11,7 +11,9 @@
 # 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.
-from typing import Any, Callable, Optional
+from __future__ import annotations
+
+from typing import Any, Callable
 
 from aio_pika import Queue
 from aio_pika.abc import AbstractIncomingMessage
@@ -28,7 +30,7 @@ def __init__(self, tracer: Tracer, queue: Queue):
         self._tracer = tracer
         self._queue = queue
 
-    def _get_span(self, message: AbstractIncomingMessage) -> Optional[Span]:
+    def _get_span(self, message: AbstractIncomingMessage) -> Span | None:
         builder = SpanBuilder(self._tracer)
         builder.set_as_consumer()
         builder.set_operation(MessagingOperationValues.RECEIVE)

instrumentation/opentelemetry-instrumentation-aio-pika/src/opentelemetry/instrumentation/aio_pika/publish_decorator.py

@@ -11,7 +11,9 @@
 # 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.
-from typing import Callable, Optional
+from __future__ import annotations
+
+from typing import Callable
 
 import aiormq
 from aio_pika import Exchange
@@ -29,7 +31,7 @@ def __init__(self, tracer: Tracer, exchange: Exchange):
 
     def _get_publish_span(
         self, message: AbstractMessage, routing_key: str
-    ) -> Optional[Span]:
+    ) -> Span | None:
         builder = SpanBuilder(self._tracer)
         builder.set_as_producer()
         builder.set_destination(f"{self._exchange.name},{routing_key}")
@@ -40,7 +42,7 @@ def _get_publish_span(
     def decorate(self, publish: Callable) -> Callable:
         async def decorated_publish(
             message: AbstractMessage, routing_key: str, **kwargs
-        ) -> Optional[aiormq.abc.ConfirmationFrameType]:
+        ) -> aiormq.abc.ConfirmationFrameType | None:
             span = self._get_publish_span(message, routing_key)
             if not span:
                 return await publish(message, routing_key, **kwargs)

instrumentation/opentelemetry-instrumentation-aio-pika/src/opentelemetry/instrumentation/aio_pika/span_builder.py

@@ -11,7 +11,7 @@
 # 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.
-from typing import Optional
+from __future__ import annotations
 
 from aio_pika.abc import AbstractChannel, AbstractMessage
 
@@ -77,7 +77,7 @@ def set_message(self, message: AbstractMessage):
                 properties.correlation_id
             )
 
-    def build(self) -> Optional[Span]:
+    def build(self) -> Span | None:
         if not is_instrumentation_enabled():
             return None
         if self._operation:

instrumentation/opentelemetry-instrumentation-aio-pika/tests/test_publish_decorator.py

@@ -11,8 +11,9 @@
 # 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.
+from __future__ import annotations
+
 import asyncio
-from typing import Type
 from unittest import TestCase, mock, skipIf
 from unittest.mock import MagicMock
 
@@ -70,7 +71,7 @@ def test_get_publish_span(self):
             attributes=self.EXPECTED_ATTRIBUTES,
         )
 
-    def _test_publish(self, exchange_type: Type[Exchange]):
+    def _test_publish(self, exchange_type: type[Exchange]):
         exchange = exchange_type(CONNECTION_7, CHANNEL_7, EXCHANGE_NAME)
         with mock.patch.object(
             PublishDecorator, "_get_publish_span"
@@ -149,7 +150,7 @@ def test_get_publish_span(self):
             attributes=self.EXPECTED_ATTRIBUTES,
         )
 
-    def _test_publish(self, exchange_type: Type[Exchange]):
+    def _test_publish(self, exchange_type: type[Exchange]):
         exchange = exchange_type(CONNECTION_8, CHANNEL_8, EXCHANGE_NAME)
         with mock.patch.object(
             PublishDecorator, "_get_publish_span"