Tweaks to voice assistant hook (#946)

livekit · Aug 20, 2024 · 45a95dd · 45a95dd
1 parent d35dffd
commit 45a95dd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/packages/react/etc/components-react.api.md b/packages/react/etc/components-react.api.md
@@ -1239,7 +1239,7 @@ export interface VoiceAssistant {
 export const VoiceAssistantContext: React_2.Context<VoiceAssistant | undefined>;
 
 // @alpha (undocumented)
-export type VoiceAssistantState = 'disconnected' | 'connecting' | 'listening' | 'thinking' | 'speaking';
+export type VoiceAssistantState = 'disconnected' | 'connecting' | 'initializing' | 'listening' | 'thinking' | 'speaking';
 
 // @public (undocumented)
 export type WidgetState = {

diff --git a/packages/react/src/hooks/useVoiceAssistant.ts b/packages/react/src/hooks/useVoiceAssistant.ts
@@ -14,6 +14,7 @@ import { useParticipantAttributes } from './useParticipantAttributes';
 export type VoiceAssistantState =
   | 'disconnected'
   | 'connecting'
+  | 'initializing'
   | 'listening'
   | 'thinking'
   | 'speaking';
@@ -46,10 +47,14 @@ export function useVoiceAssistant(): VoiceAssistant {
   const state: VoiceAssistantState = React.useMemo(() => {
     if (connectionState === ConnectionState.Disconnected) {
       return 'disconnected';
-    } else if (connectionState === ConnectionState.Connecting || !agent || !attributes?.state) {
+    } else if (
+      connectionState === ConnectionState.Connecting ||
+      !agent ||
+      !attributes?.['agent.state']
+    ) {
       return 'connecting';
     } else {
-      return attributes.state as VoiceAssistantState;
+      return attributes['agent.state'] as VoiceAssistantState;
     }
   }, [attributes, agent, connectionState]);