PYTHON-5846 Fix typing: use Hello type in _HeartbeatTelemetry.succeeded

Author: blink1073

pymongo/_telemetry.py

@@ -37,6 +37,7 @@
 
 if TYPE_CHECKING:
     from bson.objectid import ObjectId
+    from pymongo.hello import Hello
     from pymongo.monitoring import _EventListeners
     from pymongo.typings import _Address, _DocumentOut
 
@@ -403,23 +404,22 @@ def emit_started_log(self, conn_id: int, server_conn_id: Optional[int]) -> None:
     def succeeded(
         self,
         round_trip_time: float,
-        reply: _DocumentOut,
-        awaited: bool,
+        response: Hello[Any],
         conn_id: int,
         server_conn_id: Optional[int],
     ) -> None:
         """Emit the SUCCEEDED log entry and APM event."""
         if self._should_publish:
             assert self._listeners is not None
             self._listeners.publish_server_heartbeat_succeeded(
-                self._address, round_trip_time, reply, awaited
+                self._address, round_trip_time, response, response.awaitable
             )
         self._emit_log(
             _SDAMStatusMessage.HEARTBEAT_SUCCESS,
             driverConnectionId=conn_id,
             serverConnectionId=server_conn_id,
             durationMS=round_trip_time * 1000,
-            reply=reply,
+            reply=response.document,
         )
 
     def failed(self, error: Exception, conn_id: Optional[int]) -> None:

pymongo/asynchronous/monitor.py

@@ -304,13 +304,7 @@ async def _check_once(self) -> ServerDescription:
 
             avg_rtt, min_rtt = await self._rtt_monitor.get()
             sd = ServerDescription(address, response, avg_rtt, min_round_trip_time=min_rtt)
-            hb.succeeded(
-                round_trip_time,
-                response.document,
-                response.awaitable,
-                conn.id,
-                conn.server_connection_id,
-            )
+            hb.succeeded(round_trip_time, response, conn.id, conn.server_connection_id)
             return sd
 
     async def _check_with_socket(self, conn: AsyncConnection) -> tuple[Hello, float]:  # type: ignore[type-arg]

pymongo/synchronous/monitor.py

@@ -302,13 +302,7 @@ def _check_once(self) -> ServerDescription:
 
             avg_rtt, min_rtt = self._rtt_monitor.get()
             sd = ServerDescription(address, response, avg_rtt, min_round_trip_time=min_rtt)
-            hb.succeeded(
-                round_trip_time,
-                response.document,
-                response.awaitable,
-                conn.id,
-                conn.server_connection_id,
-            )
+            hb.succeeded(round_trip_time, response, conn.id, conn.server_connection_id)
             return sd
 
     def _check_with_socket(self, conn: Connection) -> tuple[Hello, float]:  # type: ignore[type-arg]