Remove agent state concept (#58)

livekit · May 20, 2024 · ba9aedf · ba9aedf
1 parent 2d7d3f3
commit ba9aedf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 52 deletions.
diff --git a/src/components/playground/Playground.tsx b/src/components/playground/Playground.tsx
@@ -15,7 +15,6 @@ import {
 import { AgentMultibandAudioVisualizer } from "@/components/visualization/AgentMultibandAudioVisualizer";
 import { useConfig } from "@/hooks/useConfig";
 import { useMultibandTrackVolume } from "@/hooks/useTrackVolume";
-import { AgentState } from "@/lib/types";
 import {
   VideoTrack,
   useChat,
@@ -56,7 +55,6 @@ export default function Playground({
 }: PlaygroundProps) {
   const { config, setUserSettings } = useConfig();
   const { name } = useRoomInfo();
-  const [agentState, setAgentState] = useState<AgentState>("offline");
   const [messages, setMessages] = useState<ChatMessageType[]>([]);
   const [transcripts, setTranscripts] = useState<ChatMessageType[]>([]);
   const { localParticipant } = useLocalParticipant();
@@ -65,6 +63,7 @@ export default function Playground({
     updateOnlyOn: [RoomEvent.ParticipantMetadataChanged],
   });
   const agentParticipant = participants.find((p) => p.isAgent);
+  const isAgentConnected = agentParticipant !== undefined;
 
   const { send: sendChat, chatMessages } = useChat();
   const roomState = useConnectionState();
@@ -109,24 +108,6 @@ export default function Playground({
     20
   );
 
-  useEffect(() => {
-    if (!agentParticipant) {
-      setAgentState("offline");
-      return;
-    }
-    let agentMd: any = {};
-    if (agentParticipant.metadata) {
-      agentMd = JSON.parse(agentParticipant.metadata);
-    }
-    if (agentMd.agent_state) {
-      setAgentState(agentMd.agent_state);
-    } else {
-      setAgentState("starting");
-    }
-  }, [agentParticipant, agentParticipant?.metadata]);
-
-  const isAgentConnected = agentState !== "offline";
-
   const onDataReceived = useCallback(
     (msg: any) => {
       if (msg.topic === "transcription") {
@@ -233,10 +214,11 @@ export default function Playground({
       </div>
     );
 
+    // TODO: keep it in the speaking state until we come up with a better protocol for agent states
     const visualizerContent = (
       <div className="flex items-center justify-center w-full">
         <AgentMultibandAudioVisualizer
-          state={agentState}
+          state="speaking"
           barWidth={30}
           minBarHeight={30}
           maxBarHeight={150}
@@ -260,7 +242,6 @@ export default function Playground({
     return visualizerContent;
   }, [
     agentAudioTrack,
-    agentState,
     config.settings.theme_color,
     subscribedVolumes,
     roomState,
@@ -334,24 +315,6 @@ export default function Playground({
                   : "gray-500"
               }
             />
-            <NameValueRow
-              name="Agent status"
-              value={
-                agentState !== "offline" && agentState !== "speaking" ? (
-                  <div className="flex gap-2 items-center">
-                    <LoadingSVG diameter={12} strokeWidth={2} />
-                    {agentState}
-                  </div>
-                ) : (
-                  agentState
-                )
-              }
-              valueColor={
-                agentState === "speaking"
-                  ? `${config.settings.theme_color}-500`
-                  : "gray-500"
-              }
-            />
           </div>
         </ConfigurationPanelItem>
         {localVideoTrack && (
@@ -405,7 +368,6 @@ export default function Playground({
     name,
     roomState,
     isAgentConnected,
-    agentState,
     localVideoTrack,
     localMicTrack,
     localMultibandVolume,

diff --git a/src/components/visualization/AgentMultibandAudioVisualizer.tsx b/src/components/visualization/AgentMultibandAudioVisualizer.tsx
@@ -1,8 +1,8 @@
-import { AgentState } from "@/lib/types";
 import { useEffect, useState } from "react";
 
+type VisualizerState = "listening" | "idle" | "speaking" | "thinking";
 type AgentMultibandAudioVisualizerProps = {
-  state: AgentState;
+  state: VisualizerState;
   barWidth: number;
   minBarHeight: number;
   maxBarHeight: number;

diff --git a/src/lib/types.ts b/src/lib/types.ts
@@ -13,12 +13,4 @@ export interface SessionProps {
 export interface TokenResult {
   identity: string;
   accessToken: string;
-}
-
-export type AgentState =
-  | "idle"
-  | "listening"
-  | "speaking"
-  | "thinking"
-  | "offline"
-  | "starting";
+}