44from propcache import cached_property
55
66from homeassistant .helpers .entity import Entity
7- from tesla_fleet_api import EnergySpecific , VehicleSpecific
7+ from tesla_fleet_api .teslemetry .vehicles import TeslemetryVehicleFleet
8+ from tesla_fleet_api .tesla .energysite import EnergySite
89from tesla_fleet_api .const import Scope
910from teslemetry_stream import Signal
1011
@@ -28,7 +29,7 @@ class TeslemetryRootEntity(Entity):
2829
2930 _attr_has_entity_name = True
3031 scoped : bool
31- api : VehicleSpecific | EnergySpecific
32+ api : TeslemetryVehicleFleet | EnergySite
3233
3334 def raise_for_scope (self , scope : Scope ):
3435 """Raise an error if a scope is not available."""
@@ -196,7 +197,7 @@ def __init__(
196197 | TeslemetryEnergySiteLiveCoordinator
197198 | TeslemetryEnergySiteInfoCoordinator
198199 | TeslemetryEnergyHistoryCoordinator ,
199- api : VehicleSpecific | EnergySpecific ,
200+ api : TeslemetryVehicleFleet | EnergySite ,
200201 key : str ,
201202 ) -> None :
202203 """Initialize common aspects of a Teslemetry entity."""
@@ -253,7 +254,7 @@ def _async_update_attrs(self) -> None:
253254class TeslemetryVehicleEntity (TeslemetryCoordinatorEntity ):
254255 """Parent class for polled Teslemetry Vehicle entities."""
255256
256- api : VehicleSpecific
257+ api : TeslemetryVehicleFleet
257258
258259 def __init__ (
259260 self ,
@@ -287,7 +288,7 @@ async def handle_command(self, command) -> dict[str, Any]:
287288class TeslemetryEnergyLiveEntity (TeslemetryCoordinatorEntity ):
288289 """Parent class for Teslemetry Energy Site Live entities."""
289290
290- api : EnergySpecific
291+ api : EnergySite
291292
292293 def __init__ (
293294 self ,
@@ -306,7 +307,7 @@ def __init__(
306307class TeslemetryEnergyInfoEntity (TeslemetryCoordinatorEntity ):
307308 """Parent class for Teslemetry Energy Site Info Entities."""
308309
309- api : EnergySpecific
310+ api : EnergySite
310311
311312 def __init__ (
312313 self ,
@@ -324,7 +325,7 @@ def __init__(
324325class TeslemetryEnergyHistoryEntity (TeslemetryCoordinatorEntity ):
325326 """Parent class for Teslemetry Energy History Entities."""
326327
327- api : EnergySpecific
328+ api : EnergySite
328329
329330 def __init__ (
330331 self ,
@@ -346,7 +347,7 @@ class TeslemetryWallConnectorEntity(
346347):
347348 """Parent class for Teslemetry Wall Connector Entities."""
348349
349- api : EnergySpecific
350+ api : EnergySite
350351
351352 _attr_has_entity_name = True
352353
0 commit comments