first

speech_to_speech/api/openai_realtime/handlers/base.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+import logging
+from queue import Queue
+from threading import Event as ThreadingEvent
+from typing import TYPE_CHECKING
+
+from openai.types.realtime import RealtimeErrorEvent
+
+from speech_to_speech.utils.utils import _generate_id
+
+if TYPE_CHECKING:
+    from speech_to_speech.api.openai_realtime.service import ConnState, RealtimeService
+
+logger = logging.getLogger(__name__)
+
+
+class RealtimeBaseHandler:
+    """Shared base for domain handlers.
+
+    Provides conn_id-keyed accessors for per-connection state, config,
+    and queues.  Each connection owns its own ``RuntimeConfig`` via
+    ``ConnState``.
+    """
+
+    def __init__(self, service: RealtimeService) -> None:
+        self._service = service
+
+    # ── conn_id-keyed accessors ──
+
+    def _queue(self, conn_id: str) -> Queue | None:
+        return self._service.text_prompt_queue
+
+    def _should_listen(self, conn_id: str) -> ThreadingEvent | None:
+        return self._service.should_listen
+
+    # ── shared helpers ────────────────────────────────
+
+    def _state(self, conn_id: str) -> ConnState:
+        return self._service._state(conn_id)
+
+    def _input_item_id(self, conn_id: str) -> str:
+        st = self._state(conn_id)
+        return st.speculative_input_item_id or self._service.response._current_item_id(conn_id)
+
+    def _next_input_content_index(self, conn_id: str) -> int:
+        st = self._state(conn_id)
+        idx = st.input_content_index
+        st.input_content_index += 1
+        return idx
+
+    @staticmethod
+    def _next_event_id() -> str:
+        return _generate_id("event")
+
+    def make_error(self, message: str, _type: str) -> RealtimeErrorEvent:
+        return self._service.make_error(message, _type)