Desmond-Dong commited on
Commit
58fd2a5
·
1 Parent(s): a7510f3

Fix import error: use ListEntitiesMediaPlayerResponse (singular)

Browse files
Files changed (1) hide show
  1. reachy_mini_ha_voice/entity.py +5 -5
reachy_mini_ha_voice/entity.py CHANGED
@@ -5,9 +5,9 @@ from typing import Dict, List, Optional
5
 
6
  # pylint: disable=no-name-in-module
7
  from aioesphomeapi.api_pb2 import ( # type: ignore[attr-defined]
8
- ListEntitiesMediaPlayersResponse,
 
9
  MediaPlayerCommandRequest,
10
- TextSensorStateResponse,
11
  )
12
  from aioesphomeapi.model import MediaPlayerState
13
 
@@ -34,16 +34,16 @@ class MediaPlayerEntity(Entity):
34
 
35
  def handle_message(self, msg):
36
  """Handle a message."""
37
- if isinstance(msg, ListEntitiesMediaPlayersResponse):
38
  yield self.get_list_entities_response()
39
  elif isinstance(msg, MediaPlayerCommandRequest):
40
  self.handle_command(msg)
41
 
42
  def get_list_entities_response(self):
43
  """Get list entities response."""
44
- from aioesphomeapi.api_pb2 import ListEntitiesMediaPlayersResponse
45
 
46
- return ListEntitiesMediaPlayersResponse(
47
  object_id=self.object_id,
48
  key=self.key,
49
  name=self.name,
 
5
 
6
  # pylint: disable=no-name-in-module
7
  from aioesphomeapi.api_pb2 import ( # type: ignore[attr-defined]
8
+ ListEntitiesDoneResponse,
9
+ ListEntitiesRequest,
10
  MediaPlayerCommandRequest,
 
11
  )
12
  from aioesphomeapi.model import MediaPlayerState
13
 
 
34
 
35
  def handle_message(self, msg):
36
  """Handle a message."""
37
+ if isinstance(msg, ListEntitiesRequest):
38
  yield self.get_list_entities_response()
39
  elif isinstance(msg, MediaPlayerCommandRequest):
40
  self.handle_command(msg)
41
 
42
  def get_list_entities_response(self):
43
  """Get list entities response."""
44
+ from aioesphomeapi.api_pb2 import ListEntitiesMediaPlayerResponse
45
 
46
+ return ListEntitiesMediaPlayerResponse(
47
  object_id=self.object_id,
48
  key=self.key,
49
  name=self.name,