From 1fa849097c883bc487b364d610750175c5abe2ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9mence=20Lesn=C3=A9?= <clemence@lesne.pro>
Date: Thu, 21 Nov 2024 11:33:21 +0100
Subject: [PATCH] fix: Test run after new VAD algoritm implementation

---
 tests/llm.py | 49 ++++++++++++++++++++++++++++++-------------------
 1 file changed, 30 insertions(+), 19 deletions(-)

diff --git a/tests/llm.py b/tests/llm.py
index eeeab28d..337bcf0c 100644
--- a/tests/llm.py
+++ b/tests/llm.py
@@ -3,6 +3,7 @@
 import re
 from datetime import datetime
 
+import aiojobs
 import pytest
 from deepeval import assert_test
 from deepeval.metrics import (
@@ -23,10 +24,11 @@
     on_end_call,
     on_ivr_recognized,
     on_play_completed,
-    on_speech_recognized,
 )
+from app.helpers.call_llm import _out_answer
 from app.helpers.logging import logger
 from app.models.call import CallStateModel
+from app.models.message import MessageModel, PersonaEnum as MessagePersonaEnum
 from app.models.reminder import ReminderModel
 from app.models.training import TrainingModel
 from tests.conftest import CallAutomationClientMock, with_conversations
@@ -285,24 +287,33 @@ async def _training_callback(_call: CallStateModel) -> None:
     )
 
     # Simulate conversation with speech recognition
-    for speech in speeches:
-        # Respond
-        await on_speech_recognized(
-            call=call,
-            client=automation_client,
-            post_callback=_post_callback,
-            text=speech,
-            training_callback=_training_callback,
-        )
-        # Receip
-        await on_play_completed(
-            call=call,
-            client=automation_client,
-            contexts=call_client.last_contexts,
-            post_callback=_post_callback,
-        )
-        # Reset contexts
-        call_client.last_contexts.clear()
+    async with aiojobs.Scheduler() as scheduler:
+        for speech in speeches:
+            # Add message to history
+            call.messages.append(
+                MessageModel(
+                    content=speech,
+                    persona=MessagePersonaEnum.HUMAN,
+                )
+            )
+
+            # Respond
+            await _out_answer(
+                call=call,
+                client=automation_client,
+                post_callback=_post_callback,
+                scheduler=scheduler,
+                training_callback=_training_callback,
+            )
+            # Receip
+            await on_play_completed(
+                call=call,
+                client=automation_client,
+                contexts=call_client.last_contexts,
+                post_callback=_post_callback,
+            )
+            # Reset contexts
+            call_client.last_contexts.clear()
 
     # Disconnect call
     await on_call_disconnected(