Add timing context handling to get_stt and add test for timing hand…

…ling in `audio_input`
NeonGeckoCom · Nov 13, 2023 · edfb492 · edfb492
1 parent 229c752
commit edfb492
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/neon_speech/service.py b/neon_speech/service.py
@@ -347,6 +347,12 @@ def handle_get_stt(self, message: Message):
         try:
             _, parser_data, transcriptions = \
                 self._get_stt_from_file(wav_file_path, lang)
+            received_time = time()
+            sent_time = message.context.get("timing", {}).get("client_sent",
+                                                              received_time)
+            if received_time != sent_time:
+                message.context['timing']['mq_from_client'] = \
+                    received_time - sent_time
             self.bus.emit(message.reply(ident,
                                         data={"parser_data": parser_data,
                                               "transcripts": transcriptions}))
@@ -380,7 +386,8 @@ def build_context(msg: Message):
         sent_time = message.context.get("timing", {}).get("client_sent",
                                                           received_time)
         if received_time != sent_time:
-            message.context['timing']['mq_from_client'] = received_time - sent_time
+            message.context['timing']['mq_from_client'] = \
+                received_time - sent_time
         ident = message.context.get("ident") or "neon.audio_input.response"
         LOG.info(f"Handling audio input: {ident}")
         if message.data.get("audio_data"):

diff --git a/tests/api_method_tests.py b/tests/api_method_tests.py
@@ -176,7 +176,8 @@ def test_audio_input_valid(self):
         context = {"client": "tester",
                    "ident": "11111",
                    "user": "TestRunner",
-                   "extra_data": "something"}
+                   "extra_data": "something",
+                   "timing": {"client_sent": time()}}
         audio_data = encode_file_to_base64_string(os.path.join(AUDIO_FILE_PATH,
                                                                "stop.wav"))
         stt_resp = self.bus.wait_for_response(Message(
@@ -199,6 +200,10 @@ def test_audio_input_valid(self):
         self.assertIn("stop", message.data["utterances"],
                       message.data.get("utterances"))
         self.assertIsInstance(message.context["timing"], dict)
+        self.assertIsInstance(stt_resp.context['timing']['mq_from_client'],
+                              float, stt_resp.context)
+        self.assertIsInstance(stt_resp.context['timing']['transcribed'], float,
+                              stt_resp.context)
         self.assertEqual(message.context["destination"], ["skills"])
 
     def test_wake_words_state(self):