From d692b20f3b47cc7e3efb7b48e9351d1624fdc513 Mon Sep 17 00:00:00 2001 From: Daniel McKnight Date: Mon, 2 Oct 2023 19:28:49 -0700 Subject: [PATCH] Add stt metrics to output --- neon_minerva/integration/text.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/neon_minerva/integration/text.py b/neon_minerva/integration/text.py index b4c7faa..4b461d9 100644 --- a/neon_minerva/integration/text.py +++ b/neon_minerva/integration/text.py @@ -66,12 +66,16 @@ def run_test(self) -> dict: "get_tts": [], "intent_handler": [], "total": []} + if self.test_audio: + aggregated_results['get_stt'] = [] for result in self._results: aggregated_results['save_transcript'].append(result['save_transcript']) aggregated_results['text_parsers'].append(result['text_parsers']) aggregated_results['get_tts'].append(result['get_tts']) aggregated_results['intent_handler'].append(result['speech_start'] - result['handle_utterance']) aggregated_results['total'].append(result['finished'] - result['transcribed']) + if self.test_audio: + aggregated_results['get_stt'].append(result['get_stt']) formatted_results = dict() for key, values in aggregated_results.items(): formatted_results[key] = {"average": sum(values) / len(values),