Merge pull request #46 from PalabraAI/feature/benchmark-refactoring

New auto-tempo defaults and bugfix

Author: yokotoka

src/palabra_ai/benchmark/__main__.py

@@ -4,7 +4,6 @@
 import bisect
 import re
 import wave
-from audioop import byteswap
 from base64 import b64decode
 from collections import defaultdict
 from dataclasses import dataclass

src/palabra_ai/constant.py

@@ -71,10 +71,10 @@
 SPEECH_TEMPO_ADJUSTMENT_FACTOR_DEFAULT = 0.75
 
 # Queue Config Constants
-DESIRED_QUEUE_LEVEL_MS_DEFAULT = 10000
+DESIRED_QUEUE_LEVEL_MS_DEFAULT = 5000
 MAX_QUEUE_LEVEL_MS_DEFAULT = 24000
-QUEUE_MIN_TEMPO = 1.0
-QUEUE_MAX_TEMPO = 1.2
+QUEUE_MIN_TEMPO = 1.15
+QUEUE_MAX_TEMPO = 1.45
 
 # TranscriptionMessage Constants
 MIN_ALIGNMENT_SCORE_DEFAULT = 0.2

src/palabra_ai/task/io/base.py

@@ -59,8 +59,8 @@ class Io(Task):
     @property
     def io_data(self) -> IoData:
         return IoData(
-            start_perf_ts=self.global_start_perf_ts,
-            start_utc_ts=self.global_start_utc_ts,
+            start_perf_ts=self.global_start_perf_ts or 0.0,
+            start_utc_ts=self.global_start_utc_ts or 0.0,
             in_sr=self.cfg.mode.input_sample_rate,
             out_sr=self.cfg.mode.output_sample_rate,
             mode=self.cfg.mode.name,

tests/test_config.py

@@ -222,7 +222,7 @@ def test_translation():
 def test_queue_configs():
     """Test QueueConfigs with alias"""
     qc = QueueConfigs()
-    assert qc.global_.desired_queue_level_ms == 10000
+    assert qc.global_.desired_queue_level_ms == 5000
     assert qc.global_.max_queue_level_ms == 24000
     assert qc.global_.auto_tempo is True
 

tests/test_constant.py

@@ -79,7 +79,7 @@ def test_tts_constants():
 
 def test_queue_config_constants():
     """Test queue config constants"""
-    assert constant.DESIRED_QUEUE_LEVEL_MS_DEFAULT == 10000
+    assert constant.DESIRED_QUEUE_LEVEL_MS_DEFAULT == 5000
     assert constant.MAX_QUEUE_LEVEL_MS_DEFAULT == 24000
 
 

tests/test_task_adapter_buffer.py

@@ -107,6 +107,20 @@ async def test_read_at_eof(self):
             mock_debug.assert_called_once()
             assert "EOF reached" in str(mock_debug.call_args[0][0])
 
+    @pytest.mark.asyncio
+    async def test_empty_buffer_immediate_eof(self):
+        """Test that empty buffer immediately returns EOF without crashing"""
+        empty_buffer = io.BytesIO(b"")
+        reader = BufferReader(buffer=empty_buffer)
+        reader.ready = TaskEvent()
+        +reader.ready
+
+        # Reading from empty buffer should return None and set EOF
+        data = await reader.read(100)
+        assert data is None
+        assert reader.eof.is_set()
+        assert reader._position == 0
+
 
 class TestBufferWriter:
     """Test BufferWriter class"""