fixup! Add missing retry strategy reset in streaming

Author: llucax

src/frequenz/client/base/streaming.py

@@ -19,6 +19,10 @@
 _logger = logging.getLogger(__name__)
 
 
+RequestT = TypeVar("RequestT")
+"""The request type of the stream."""
+
+
 InputT = TypeVar("InputT")
 """The input type of the stream."""
 
@@ -105,7 +109,7 @@ def async_range() -> AsyncIterable[int]:
     def __init__(  # pylint: disable=too-many-arguments,too-many-positional-arguments
         self,
         stream_name: str,
-        stream_method: Callable[[], AsyncIterable[InputT]],
+        stream_method: Callable[[], grpc.aio.UnaryStreamCall[RequestT, InputT]],
         transform: Callable[[InputT], OutputT],
         retry_strategy: retry.Strategy | None = None,
         retry_on_exhausted_stream: bool = False,
@@ -182,6 +186,7 @@ async def _run(self) -> None:
             _logger.info("%s: starting to stream", self._stream_name)
             try:
                 call = self._stream_method()
+                await call.initial_metadata()
                 self._retry_strategy.reset()
                 await sender.send(StreamStarted())
                 async for msg in call: