chore(celery): refine utility helper typing

Assisted-by: GPT-5.4

Author: diogosilva30

instrumentation/opentelemetry-instrumentation-celery/src/opentelemetry/instrumentation/celery/utils.py

@@ -16,7 +16,7 @@
 
 import logging
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, Any, Optional, Tuple, cast
+from typing import TYPE_CHECKING, Any, Optional, Protocol, cast
 
 from celery import registry  # pylint: disable=no-name-in-module
 from celery.app.task import Task
@@ -30,8 +30,14 @@
 if TYPE_CHECKING:
     from contextlib import AbstractContextManager
 
-ContextTuple = Tuple[Span, "AbstractContextManager[Span]", Optional[object]]
-ContextDict = dict[tuple[str, bool], ContextTuple]
+ContextKey = tuple[str, bool]
+ContextTuple = tuple[Span, "AbstractContextManager[Span]", object | None]
+ContextDict = dict[ContextKey, ContextTuple]
+
+
+class ContextCarrier(Protocol):
+    def get(self, key: str, default: Any = None) -> Any: ...
+
 
 logger = logging.getLogger(__name__)
 
@@ -65,7 +71,7 @@
 # pylint:disable=too-many-branches
 def set_attributes_from_context(
     span: Span,
-    context: Mapping[str, Any],
+    context: ContextCarrier,
 ) -> None:
     """Helper to extract meta values from a Celery Context"""
     if not span.is_recording():