Update base client version

Signed-off-by: camille-bouvy-frequenz <[email protected]>

Author: camille-bouvy-frequenz

pyproject.toml

@@ -30,7 +30,7 @@ dependencies = [
   "frequenz-api-common >= 0.6.3, < 0.7.0",
   "grpcio >= 1.66.2, < 2",
   "frequenz-channels >= 1.0.0, < 2",
-  "frequenz-client-base >= 0.6.1, < 0.7.0",
+  "frequenz-client-base >= 0.7.0, < 0.8.0",
   "frequenz-client-common >= 0.1.0, < 0.3.0",
   "frequenz-api-electricity-trading >= 0.2.4, < 1",
   "protobuf >= 5.28.0, < 6",

src/frequenz/client/electricity_trading/_client.py

@@ -13,10 +13,12 @@
 # pylint: disable=no-member
 from frequenz.api.electricity_trading.v1 import electricity_trading_pb2
 from frequenz.api.electricity_trading.v1.electricity_trading_pb2_grpc import (
+    ElectricityTradingServiceAsyncStub,
     ElectricityTradingServiceStub,
 )
 from frequenz.channels import Receiver
 from frequenz.client.base.client import BaseApiClient
+from frequenz.client.base.exception import ClientNotConnected
 from frequenz.client.base.streaming import GrpcStreamBroadcaster
 from frequenz.client.common.pagination import Params
 from google.protobuf import field_mask_pb2, struct_pb2
@@ -81,7 +83,7 @@ def validate_decimal_places(value: Decimal, decimal_places: int, name: str) -> N
         ) from exc
 
 
-class Client(BaseApiClient[ElectricityTradingServiceStub]):
+class Client(BaseApiClient):
     """Electricity trading client."""
 
     _instances: dict[tuple[str, str | None], "Client"] = {}
@@ -123,7 +125,11 @@ def __init__(
         if not hasattr(
             self, "_initialized"
         ):  # Prevent re-initialization of existing instances
-            super().__init__(server_url, ElectricityTradingServiceStub, connect=connect)
+            super().__init__(server_url, connect=connect)
+            self._stub = cast(
+                ElectricityTradingServiceAsyncStub,
+                ElectricityTradingServiceStub(self.channel),
+            )
             self._initialized = True
 
         self._gridpool_orders_streams: dict[
@@ -149,6 +155,21 @@ def __init__(
 
         self._metadata = (("key", auth_key),) if auth_key else ()
 
+    @property
+    def stub(self) -> ElectricityTradingServiceAsyncStub:
+        """
+        Get the gRPC stub for the Electricity Trading service.
+
+        Returns:
+            The gRPC stub.
+
+        Raises:
+            ClientNotConnected: If the client is not connected to the server.
+        """
+        if self._channel is None:
+            raise ClientNotConnected(server_url=self.server_url, operation="stub")
+        return self._stub
+
     async def stream_gridpool_orders(
         # pylint: disable=too-many-arguments, too-many-positional-arguments
         self,
@@ -192,7 +213,7 @@ async def stream_gridpool_orders(
             try:
                 self._gridpool_orders_streams[stream_key] = GrpcStreamBroadcaster(
                     f"electricity-trading-{stream_key}",
-                    lambda: self.stub.ReceiveGridpoolOrdersStream(  # type: ignore
+                    lambda: self.stub.ReceiveGridpoolOrdersStream(
                         electricity_trading_pb2.ReceiveGridpoolOrdersStreamRequest(
                             gridpool_id=gridpool_id,
                             filter=gridpool_order_filter.to_pb(),
@@ -251,7 +272,7 @@ async def stream_gridpool_trades(
             try:
                 self._gridpool_trades_streams[stream_key] = GrpcStreamBroadcaster(
                     f"electricity-trading-{stream_key}",
-                    lambda: self.stub.ReceiveGridpoolTradesStream(  # type: ignore
+                    lambda: self.stub.ReceiveGridpoolTradesStream(
                         electricity_trading_pb2.ReceiveGridpoolTradesStreamRequest(
                             gridpool_id=gridpool_id,
                             filter=gridpool_trade_filter.to_pb(),
@@ -303,7 +324,7 @@ async def stream_public_trades(
                 self._public_trades_streams[public_trade_filter] = (
                     GrpcStreamBroadcaster(
                         f"electricity-trading-{public_trade_filter}",
-                        lambda: self.stub.ReceivePublicTradesStream(  # type: ignore
+                        lambda: self.stub.ReceivePublicTradesStream(
                             electricity_trading_pb2.ReceivePublicTradesStreamRequest(
                                 filter=public_trade_filter.to_pb(),
                             ),