Add stt metrics to output

NeonGeckoCom · Oct 3, 2023 · d692b20 · d692b20
1 parent c3c27b5
commit d692b20
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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),