Merge pull request #4 from liberate-org/fix/torch-hub-fix

Fix/torch hub fix
liberate-org · Feb 29, 2024 · b2cfd91 · b2cfd91
2 parents 322c0fc + d144f52
commit b2cfd91
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/vocode/streaming/input_device/silero_vad.py b/vocode/streaming/input_device/silero_vad.py
@@ -22,15 +22,17 @@ def _load_model(self, use_onnx: bool = False) -> torch.nn.Module:
                 repo_or_dir='silero-vad',
                 model='silero_vad',
                 source='local',
-                onnx=use_onnx
+                onnx=use_onnx,
+                trust_repo=True
             )
         except FileNotFoundError:
             self.logger.warning("Could not find local VAD model, downloading from GitHub!")
             model, _ = torch.hub.load(
                 repo_or_dir='snakers4/silero-vad',
                 model='silero_vad',
                 source='github',
-                onnx=use_onnx
+                onnx=use_onnx,
+                trust_repo=True
             )
         return model
 

diff --git a/vocode/streaming/streaming_conversation.py b/vocode/streaming/streaming_conversation.py
@@ -320,6 +320,7 @@ async def process(self, item: InterruptibleAgentResponseEvent[AgentResponse]):
                     is_interruptible=item.is_interruptible,
                     agent_response_tracker=item.agent_response_tracker,
                 )
+                self.conversation.mark_last_agent_response()
             except asyncio.CancelledError:
                 pass
 
@@ -385,7 +386,6 @@ async def process(
                             await self.conversation.terminate()
                     except asyncio.TimeoutError:
                         pass
-                self.conversation.mark_last_agent_response()
             except asyncio.CancelledError:
                 pass
 
@@ -712,6 +712,7 @@ async def send_speech_to_output(
                 "Sent chunk {} with size {}".format(chunk_idx, len(chunk_result.chunk))
             )
             self.mark_last_action_timestamp()
+            self.mark_last_agent_response()
             chunk_idx += 1
             seconds_spoken += seconds_per_chunk
             if transcript_message:
@@ -790,7 +791,7 @@ async def check_if_human_should_be_prompted(self):
             if self.last_agent_response and self.last_final_transcript_from_human:
                 last_human_touchpoint = time.time() - self.last_final_transcript_from_human
                 last_agent_touchpoint = time.time() - self.last_agent_response
-                if last_human_touchpoint >= reengage_timeout and last_agent_touchpoint >= reengage_timeout:
+                if (last_human_touchpoint >= reengage_timeout) and (last_agent_touchpoint >= reengage_timeout):
                     reengage_statement = random.choice(reengage_options)
                     self.logger.debug(f"Prompting user with {reengage_statement}: no interaction has happened in {reengage_timeout} seconds")
                     self.chunk_size = (
@@ -812,7 +813,7 @@ async def check_if_human_should_be_prompted(self):
                         agent_response_tracker=asyncio.Event(),
                     )
                     self.mark_last_agent_response()
-                await asyncio.sleep(1)
+                await asyncio.sleep(2.5)
             else:
                 await asyncio.sleep(1)
         self.logger.debug("stopped check if human should be prompted")