Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-09-27 04:22:46 +02:00
Code Issues Projects Releases Wiki Activity

Add wake word capability to voice assistant (#519)

Browse Source
This commit is contained in:
Jesse Hills 2023-08-10 16:03:15 +12:00 committed by GitHub
parent 53074cfd3c
commit 2c7b2ed1a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 79 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aioesphomeapi/api.proto
View File

@ -1445,7 +1445,7 @@ message VoiceAssistantRequest {
bool start = 1;
string conversation_id = 2;
bool use_vad = 3;
int32 flags = 3;
}
message VoiceAssistantResponse {
@ -1467,6 +1467,8 @@ enum VoiceAssistantEvent {
VOICE_ASSISTANT_INTENT_END = 6;
VOICE_ASSISTANT_TTS_START = 7;
VOICE_ASSISTANT_TTS_END = 8;
VOICE_ASSISTANT_WAKE_WORD_START = 9;
VOICE_ASSISTANT_WAKE_WORD_END = 10;
}
message VoiceAssistantEventData {

130
aioesphomeapi/api_pb2.py
View File

File diff suppressed because one or more lines are too long

4
aioesphomeapi/client.py
View File

@ -1342,7 +1342,7 @@ class APIClient:
async def subscribe_voice_assistant(
self,
handle_start: Callable[[str, bool], Coroutine[Any, Any, int | None]],
handle_start: Callable[[str, int], Coroutine[Any, Any, int | None]],
handle_stop: Callable[[], Coroutine[Any, Any, None]],
) -> Callable[[], None]:
"""Subscribes to voice assistant messages from the device.
@ -1371,7 +1371,7 @@ class APIClient:
command = VoiceAssistantCommand.from_pb(msg)
if command.start:
start_task = asyncio.create_task(
handle_start(command.conversation_id, command.use_vad)
handle_start(command.conversation_id, command.flags)
)
start_task.add_done_callback(_started)
# We hold a reference to the start_task in unsub function

9
aioesphomeapi/model.py
View File

@ -1063,11 +1063,16 @@ class BluetoothDeviceRequestType(APIIntEnum):
CLEAR_CACHE = 6
class VoiceAssistantCommandFlag(enum.IntFlag):
USE_VAD = 1 << 0
USE_WAKE_WORD = 1 << 1
@_frozen_dataclass_decorator
class VoiceAssistantCommand(APIModelBase):
start: bool = False
conversation_id: str = ""
use_vad: bool = False
flags: int = False
class LogLevel(APIIntEnum):
@ -1091,3 +1096,5 @@ class VoiceAssistantEventType(APIIntEnum):
VOICE_ASSISTANT_INTENT_END = 6
VOICE_ASSISTANT_TTS_START = 7
VOICE_ASSISTANT_TTS_END = 8
VOICE_ASSISTANT_WAKE_WORD_START = 9
VOICE_ASSISTANT_WAKE_WORD_END = 10