rasa-pro 3.17.1.dev2__tar.gz → 3.17.1.dev3__tar.gz

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.17.1.dev2
+Version: 3.17.1.dev3
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
 Author: Rasa Technologies GmbH

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.17.1.dev2"
+version = "3.17.1.dev3"
 description = "State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development."
 authors = [ "Rasa Technologies GmbH <hi@rasa.com>",]
 maintainers = [ "Tom Bocklisch <tom@rasa.com>",]

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/rasa/core/channels/voice_stream/call_recorder.py

@@ -47,11 +47,18 @@ def _stereo_wav_header(
     )
 
 
-def recording_filename(call_id: str) -> Path:
-    """Return a timestamped WAV path under ``recordings/``."""
+def recording_basename(call_id: str) -> str:
+    """Return a timestamped WAV filename (no directory)."""
     ts = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     safe_id = re.sub(r"[^\w.-]", "_", call_id or "unknown")
-    return RECORDINGS_DIR / f"{ts}_{safe_id}.wav"
+    return f"{ts}_{safe_id}.wav"
+
+
+def recording_filename(
+    call_id: str, base_dir: Optional[Path] = None
+) -> Path:
+    """Return a timestamped WAV path under ``recordings/`` (or ``base_dir``)."""
+    return (base_dir or RECORDINGS_DIR) / recording_basename(call_id)
 
 
 class CallRecorder:
@@ -70,8 +77,13 @@ class CallRecorder:
     of the previous bot audio (silence fills the gap).
     """
 
-    def __init__(self, path: Path, sample_rate: int, sample_width: int = 2) -> None:
-        self.path = path
+    def __init__(
+        self,
+        filename: str,
+        sample_rate: int,
+        sample_width: int = 2,
+    ) -> None:
+        self.filename = filename
         self._sample_rate = sample_rate
         self._sample_width = sample_width
         self._anchor: Optional[float] = time.monotonic()
@@ -153,8 +165,8 @@ class CallRecorder:
         staged_seconds = len(self._bot_staged) / self._bytes_per_second()
         return min(staged_seconds, time.monotonic() - self._bot_playback_start)
 
-    def stop(self) -> float:
-        """Finalize the recording and return call duration in seconds."""
+    def stop(self) -> tuple[float, bytes]:
+        """Finalize the recording and return ``(duration_seconds, wav_bytes)``."""
         duration = time.monotonic() - self._anchor if self._anchor is not None else 0.0
 
         wall_bytes = int(duration * self._sample_rate) * self._sample_width
@@ -164,16 +176,11 @@ class CallRecorder:
         self._pad_to_byte_position(self._bot_buffer, total_bytes)
 
         stereo_pcm = self._interleave_stereo()
-        self.path.parent.mkdir(parents=True, exist_ok=True)
-        with open(self.path, "wb") as f:
-            f.write(
-                _stereo_wav_header(
-                    len(stereo_pcm), self._sample_rate, self._sample_width
-                )
-            )
-            f.write(stereo_pcm)
-
-        return duration
+        wav_bytes = _stereo_wav_header(
+            len(stereo_pcm), self._sample_rate, self._sample_width
+        ) + stereo_pcm
+
+        return duration, wav_bytes
 
     def _interleave_stereo(self) -> bytes:
         """Interleave two mono PCM buffers into L0, R0, L1, R1, ..."""

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/rasa/core/channels/voice_stream/recording_storage.py

@@ -38,8 +38,8 @@ class RecordingStore(ABC):
     """Read/write access to finalized call recordings."""
 
     @abstractmethod
-    def persist(self, local_path: Path) -> None:
-        """Persist a finalized local WAV file."""
+    def persist_wav(self, filename: str, wav_data: bytes) -> None:
+        """Persist finalized WAV bytes."""
 
     @abstractmethod
     def list_recordings(self) -> List[Dict[str, Any]]:
@@ -53,8 +53,14 @@ class RecordingStore(ABC):
 class LocalRecordingStore(RecordingStore):
     """Store recordings on the local filesystem under ``RECORDINGS_DIR``."""
 
-    def persist(self, local_path: Path) -> None:
-        return
+    def persist_wav(self, filename: str, wav_data: bytes) -> None:
+        safe_name = validate_recording_name(filename)
+        if safe_name is None:
+            return
+
+        path = RECORDINGS_DIR / safe_name
+        path.parent.mkdir(parents=True, exist_ok=True)
+        path.write_bytes(wav_data)
 
     def list_recordings(self) -> List[Dict[str, Any]]:
         base = RECORDINGS_DIR
@@ -113,29 +119,28 @@ class S3RecordingStore(RecordingStore):
             return f"{self._prefix}/{filename}"
         return filename
 
-    def persist(self, local_path: Path) -> None:
-        if not local_path.is_file():
+    def persist_wav(self, filename: str, wav_data: bytes) -> None:
+        safe_name = validate_recording_name(filename)
+        if safe_name is None or not wav_data:
             return
 
-        key = self._object_key(local_path.name)
+        key = self._object_key(safe_name)
         try:
-            self._s3.Object(self._bucket, key).upload_file(
-                str(local_path),
-                ExtraArgs={"ContentType": "audio/wav"},
+            self._s3.Object(self._bucket, key).put(
+                Body=wav_data,
+                ContentType="audio/wav",
             )
             logger.info(
                 "recording_storage.uploaded",
                 bucket=self._bucket,
                 key=key,
-                local_path=str(local_path),
+                size_bytes=len(wav_data),
             )
-            local_path.unlink()
         except Exception as exc:
             logger.warning(
                 "recording_storage.upload_failed",
                 bucket=self._bucket,
                 key=key,
-                local_path=str(local_path),
                 error=str(exc),
             )
 

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/rasa/core/channels/voice_stream/voice_channel.py

@@ -48,7 +48,7 @@ from rasa.core.channels.voice_stream.audio_bytes import (
 )
 from rasa.core.channels.voice_stream.call_recorder import (
     CallRecorder,
-    recording_filename,
+    recording_basename,
 )
 from rasa.core.channels.voice_stream.recording_storage import (
     recording_store_from_config,
@@ -2546,11 +2546,11 @@ class VoiceInputChannel(InputChannel):
             return
 
         sample_rate = self.audio_format.sample_rate
-        path = recording_filename(call_parameters.call_id or "unknown")
-        self.call_recorder = CallRecorder(path, sample_rate)
+        filename = recording_basename(call_parameters.call_id or "unknown")
+        self.call_recorder = CallRecorder(filename, sample_rate)
         logger.info(
             "call_recording.started",
-            path=str(path),
+            filename=filename,
             call_id=call_parameters.call_id,
             sample_rate=self.audio_format.sample_rate,
         )
@@ -2560,16 +2560,15 @@ class VoiceInputChannel(InputChannel):
             return
 
         self._abort_bot_playback_recording()
-        path = self.call_recorder.path
-        duration = self.call_recorder.stop()
+        filename = self.call_recorder.filename
+        duration, wav_bytes = self.call_recorder.stop()
         self.call_recorder = None
         logger.info(
             "call_recording.stopped",
-            path=str(path),
+            filename=filename,
             duration_seconds=round(duration, 2),
         )
-        if path.is_file():
-            self._recording_store.persist(path)
+        self._recording_store.persist_wav(filename, wav_bytes)
 
     def list_recordings(self) -> List[Dict[str, Any]]:
         """Return recording metadata for the recordings viewer API."""

{rasa_pro-3.17.1.dev2 → rasa_pro-3.17.1.dev3}/rasa/version.py

@@ -1,3 +1,3 @@
 # this file will automatically be changed,
 # do not add anything but the version number here!
-__version__ = "3.17.1.dev2"
+__version__ = "3.17.1.dev3"