adds mute agent state manager (vocodedev#305)

Author: ajar98

vocode/streaming/agent/base_agent.py

@@ -151,6 +151,7 @@ def __init__(
  self.transcript: Optional[Transcript] = None
 
  self.functions = self.get_functions() if self.agent_config.actions else None
+ self.is_muted = False
 
  def get_functions(self):
  raise NotImplementedError
@@ -249,6 +250,9 @@ async def handle_respond(
  return False
 
  async def process(self, item: InterruptibleEvent[AgentInput]):
+ if self.is_muted:
+ self.logger.debug("Agent is muted, skipping processing")
+ return
  assert self.transcript is not None
  try:
  agent_input = item.payload

vocode/streaming/utils/state_manager.py

@@ -21,13 +21,19 @@ def set_transcriber_endpointing_config(self, endpointing_config: EndpointingConf
  self._conversation.transcriber.get_transcriber_config().endpointing_config = (
  endpointing_config
  )
- 
+
  def disable_synthesis(self):
  self._conversation.synthesis_enabled = False
- 
+
  def enable_synthesis(self):
  self._conversation.synthesis_enabled = True
- 
+
+ def mute_agent(self):
+ self._conversation.agent.is_muted = True
+
+ def unmute_agent(self):
+ self._conversation.agent.is_muted = False
+
  async def terminate_conversation(self):
  await self._conversation.terminate()