feat: update to API v0.7.0

Signed-off-by: Matthias Wende <[email protected]>

Author: matthias-wende-frequenz

pyproject.toml

@@ -45,7 +45,7 @@ dependencies = [
   "frequenz-channels >= 1.6.1, < 2",
   "frequenz-client-base >= 0.9.0, < 0.10.0",
   "frequenz-client-common >= 0.1.0, < 0.4.0",
-  "frequenz-api-electricity-trading >= 0.6.1, < 0.7.0",
+  "frequenz-api-electricity-trading >= 0.7.0, < 0.8.0",
   "protobuf >= 5.29.2, < 7",
 ]
 dynamic = ["version"]

src/frequenz/client/electricity_trading/_client.py

@@ -11,7 +11,7 @@
 import logging
 from datetime import datetime, timedelta, timezone
 from decimal import Decimal, InvalidOperation
-from typing import Any, AsyncIterator, Awaitable, Callable, cast
+from typing import Any, AsyncIterator, Awaitable, Callable, List, cast
 from zoneinfo import ZoneInfo
 
 import grpc
@@ -232,7 +232,7 @@ def __init__(
             PublicOrderBookFilter,
             GrpcStreamBroadcaster[
                 electricity_trading_pb2.ReceivePublicOrderBookStreamResponse,
-                PublicOrder,
+                List[PublicOrder],
             ],
         ] = {}
 
@@ -1012,7 +1012,7 @@ def receive_public_order_book(
         start_time: datetime | None = None,
         end_time: datetime | None = None,
     ) -> GrpcStreamBroadcaster[
-        electricity_trading_pb2.ReceivePublicOrderBookStreamResponse, PublicOrder
+        electricity_trading_pb2.ReceivePublicOrderBookStreamResponse, List[PublicOrder]
     ]:
         """
         Stream public orders with optional filters and time range.
@@ -1055,9 +1055,10 @@ def receive_public_order_book(
                             ),
                             metadata=self._metadata,
                         ),
-                        lambda response: PublicOrder.from_pb(
-                            response.public_order_book_record
-                        ),
+                        lambda response: [
+                            PublicOrder.from_pb(public_order)
+                            for public_order in response.public_order_book_records
+                        ],
                     )
                 )
             except grpc.RpcError as e:

src/frequenz/client/electricity_trading/_types.py

@@ -30,7 +30,7 @@
 
 
 def from_pb(
-    func: Callable[Concatenate[type[T], P], T]
+    func: Callable[Concatenate[type[T], P], T],
 ) -> Callable[Concatenate[type[T], P], T]:
     """Standardize from_pb methods like error handling with this decorator.
 
@@ -1968,7 +1968,7 @@ class PublicOrder:  # pylint: disable=too-many-instance-attributes
     quantity: Power
     """The quantity of the contract being traded."""
 
-    execution_option: OrderExecutionOption
+    execution_option: OrderExecutionOption | None
     """Order execution options such as All or None, Fill or Kill, etc."""
 
     create_time: datetime
@@ -2034,8 +2034,12 @@ def to_pb(self) -> electricity_trading_pb2.PublicOrderBookRecord:
             side=electricity_trading_pb2.MarketSide.ValueType(self.side.value),
             price=self.price.to_pb(),
             quantity=self.quantity.to_pb(),
-            execution_option=electricity_trading_pb2.OrderExecutionOption.ValueType(
-                self.execution_option.value
+            execution_option=(
+                electricity_trading_pb2.OrderExecutionOption.ValueType(
+                    self.execution_option.value
+                )
+                if self.execution_option
+                else None
             ),
             create_time=create_time,
             update_time=update_time,