rasa-pro 3.11.0__py3-none-any.whl → 3.11.0a2__py3-none-any.whl

rasa/core/actions/action_repeat_bot_messages.py

@@ -1,89 +0,0 @@
-from typing import Optional, Dict, Any, List
-
-from rasa.core.actions.action import Action
-from rasa.core.channels import OutputChannel
-from rasa.core.nlg import NaturalLanguageGenerator
-from rasa.dialogue_understanding.patterns.collect_information import (
-    CollectInformationPatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.repeat import (
-    RepeatBotMessagesPatternFlowStackFrame,
-)
-from rasa.dialogue_understanding.patterns.user_silence import (
-    UserSilencePatternFlowStackFrame,
-)
-from rasa.shared.core.constants import ACTION_REPEAT_BOT_MESSAGES
-from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import Event, BotUttered, UserUttered
-from rasa.shared.core.trackers import DialogueStateTracker
-
-
-class ActionRepeatBotMessages(Action):
-    """Action to repeat bot messages"""
-
-    def name(self) -> str:
-        """Return the name of the action."""
-        return ACTION_REPEAT_BOT_MESSAGES
-
-    def _get_last_bot_events(self, tracker: DialogueStateTracker) -> List[Event]:
-        """Get the last consecutive bot events before the most recent user message.
-
-        This function scans the dialogue history in reverse to find the last sequence of
-        bot responses that occurred without any user interruption. It filters out all
-        non-utterance events and stops when it encounters a user message after finding
-        bot messages.
-
-        Args:
-            tracker: DialogueStateTracker containing the conversation events.
-
-        Returns:
-            List[Event]: A list of consecutive BotUttered events that occurred
-                most recently, in chronological order. Returns an empty list
-                if no bot messages are found or if the last message was from
-                the user.
-
-        Example:
-            For events: [User1, Bot1, Bot2, User2, Bot4, Bot5, User3]
-            Returns: [Bot4, Bot5] (the last two bot events)
-            The elif condition doesn't break when it sees User3 event.
-            But it does at User2 event.
-        """
-        # Skip action if we are in a collect information step whose
-        # default behavior is to repeat anyways
-        top_frame = tracker.stack.top(
-            lambda frame: isinstance(frame, RepeatBotMessagesPatternFlowStackFrame)
-            or isinstance(frame, UserSilencePatternFlowStackFrame)
-        )
-        if isinstance(top_frame, CollectInformationPatternFlowStackFrame):
-            return []
-        # filter user and bot events
-        filtered = [
-            e for e in tracker.events if isinstance(e, (BotUttered, UserUttered))
-        ]
-        bot_events: List[Event] = []
-
-        # find the last BotUttered events
-        for e in reversed(filtered):
-            if isinstance(e, BotUttered):
-                # insert instead of append because the list is reversed
-                bot_events.insert(0, e)
-
-            # stop if a UserUttered event is found
-            # only if we have collected some bot events already
-            # this condition skips the first N UserUttered events
-            elif bot_events:
-                break
-
-        return bot_events
-
-    async def run(
-        self,
-        output_channel: OutputChannel,
-        nlg: NaturalLanguageGenerator,
-        tracker: DialogueStateTracker,
-        domain: Domain,
-        metadata: Optional[Dict[str, Any]] = None,
-    ) -> List[Event]:
-        """Send the last bot messages to the channel again"""
-        bot_events = self._get_last_bot_events(tracker)
-        return bot_events

rasa/core/channels/inspector/dist/assets/flowDiagram-v2-855bc5b3-736177bf.js

@@ -1 +0,0 @@
-import{p as e,f as o}from"./flowDb-1972c806-fca3bfe4.js";import{f as t,g as a}from"./styles-080da4f6-23ffa4fc.js";import{t as s}from"./index-e7cef9de.js";import"./layout-b9885fb6.js";import"./index-2c4b9a3b-f55afcdf.js";import"./edges-f2ad444c-6166330c.js";import"./createText-62fc7601-b0ec81d6.js";import"./line-7c59abb6.js";import"./array-9f3ba611.js";import"./path-53f90ab3.js";const k={parser:e,db:o,renderer:t,styles:a,init:r=>{r.flowchart||(r.flowchart={}),r.flowchart.arrowMarkerAbsolute=r.arrowMarkerAbsolute,s({flowchart:{arrowMarkerAbsolute:r.arrowMarkerAbsolute}}),t.setConf(r.flowchart),o.clear(),o.setGen("gen-2")}};export{k as diagram};

rasa/core/channels/voice_stream/asr/azure.py

@@ -1,129 +0,0 @@
-import os
-from dataclasses import dataclass
-from typing import Any, Dict, Optional, AsyncIterator
-import asyncio
-
-from rasa.core.channels.voice_stream.asr.asr_engine import ASREngine, ASREngineConfig
-from rasa.core.channels.voice_stream.asr.asr_event import (
-    ASREvent,
-    NewTranscript,
-    UserIsSpeaking,
-)
-from rasa.core.channels.voice_stream.audio_bytes import HERTZ, RasaAudioBytes
-from rasa.shared.constants import AZURE_SPEECH_API_KEY_ENV_VAR
-from rasa.shared.exceptions import ConnectionException
-
-
-@dataclass
-class AzureASRConfig(ASREngineConfig):
-    language: Optional[str] = None
-    speech_region: Optional[str] = None
-
-
-class AzureASR(ASREngine[AzureASRConfig]):
-    required_env_vars = (AZURE_SPEECH_API_KEY_ENV_VAR,)
-    required_packages = ("azure.cognitiveservices.speech",)
-
-    def __init__(self, config: Optional[AzureASRConfig] = None):
-        super().__init__(config)
-
-        import azure.cognitiveservices.speech as speechsdk
-
-        self.speech_recognizer: Optional[speechsdk.SpeechRecognizer] = None
-        self.stream: Optional[speechsdk.audio.PushAudioInputStream] = None
-        self.is_recognizing = False
-        self.queue: asyncio.Queue[speechsdk.SpeechRecognitionEventArgs] = (
-            asyncio.Queue()
-        )
-
-    @staticmethod
-    def validate_environment() -> None:
-        """Make sure all needed requirements for this component are met."""
-
-    def signal_user_is_speaking(self, event: Any) -> None:
-        """Replace the azure event with a generic is speaking event."""
-        self.fill_queue(UserIsSpeaking())
-
-    def fill_queue(self, event: Any) -> None:
-        """Either puts the event or a dedicated ASR Event into the queue."""
-        self.queue.put_nowait(event)
-
-    async def connect(self) -> None:
-        import azure.cognitiveservices.speech as speechsdk
-
-        speech_config = speechsdk.SpeechConfig(
-            subscription=os.environ[AZURE_SPEECH_API_KEY_ENV_VAR],
-            region=self.config.speech_region,
-        )
-        audio_format = speechsdk.audio.AudioStreamFormat(
-            samples_per_second=HERTZ,
-            bits_per_sample=8,
-            channels=1,
-            wave_stream_format=speechsdk.AudioStreamWaveFormat.MULAW,
-        )
-        self.stream = speechsdk.audio.PushAudioInputStream(stream_format=audio_format)
-        audio_config = speechsdk.audio.AudioConfig(stream=self.stream)
-        self.speech_recognizer = speechsdk.SpeechRecognizer(
-            speech_config=speech_config,
-            language=self.config.language,
-            audio_config=audio_config,
-        )
-        self.speech_recognizer.recognized.connect(self.fill_queue)
-        self.speech_recognizer.recognizing.connect(self.signal_user_is_speaking)
-        self.speech_recognizer.start_continuous_recognition_async()
-        self.is_recognizing = True
-
-    async def close_connection(self) -> None:
-        if self.speech_recognizer is None:
-            raise ConnectionException("Websocket not connected.")
-        self.speech_recognizer.stop_continuous_recognition_async()
-
-    async def signal_audio_done(self) -> None:
-        """Signal to the ASR Api that you are done sending data."""
-        self.is_recognizing = False
-
-    def rasa_audio_bytes_to_engine_bytes(self, chunk: RasaAudioBytes) -> bytes:
-        """Convert RasaAudioBytes to bytes usable by this engine."""
-        return chunk
-
-    async def send_audio_chunks(self, chunk: RasaAudioBytes) -> None:
-        """Send audio chunks to the ASR system via the websocket."""
-        if self.speech_recognizer is None or self.stream is None:
-            raise ConnectionException("ASR not connected.")
-        engine_bytes = self.rasa_audio_bytes_to_engine_bytes(chunk)
-        self.stream.write(engine_bytes)
-
-    async def stream_asr_events(self) -> AsyncIterator[ASREvent]:
-        """Stream the events returned by the ASR system as it is fed audio bytes."""
-        if self.speech_recognizer is None:
-            raise ConnectionException("Websocket not connected.")
-        while self.is_recognizing or not self.queue.empty():
-            try:
-                message = await asyncio.wait_for(self.queue.get(), timeout=2)
-                asr_event = self.engine_event_to_asr_event(message)
-                if asr_event:
-                    yield asr_event
-            except asyncio.TimeoutError:
-                pass
-
-    def engine_event_to_asr_event(self, e: Any) -> Optional[ASREvent]:
-        """Translate an engine event to a common ASREvent."""
-        import azure.cognitiveservices.speech as speechsdk
-
-        if isinstance(e, speechsdk.SpeechRecognitionEventArgs) and isinstance(
-            e.result, speechsdk.SpeechRecognitionResult
-        ):
-            return NewTranscript(e.result.text)
-        if isinstance(e, ASREvent):
-            # transformation happened before
-            return e
-
-        return None
-
-    @staticmethod
-    def get_default_config() -> AzureASRConfig:
-        return AzureASRConfig("en-US", "germanywestcentral")
-
-    @classmethod
-    def from_config_dict(cls, config: Dict) -> "AzureASR":
-        return AzureASR(AzureASRConfig.from_dict(config))

rasa/core/channels/voice_stream/call_state.py

@@ -1,23 +0,0 @@
-import asyncio
-from contextvars import ContextVar
-from werkzeug.local import LocalProxy
-from dataclasses import dataclass
-from typing import Optional
-
-
-# Per voice session data
-# This is similar to how flask makes the "request" object available as a global variable
-# It's a "global" variable that is local to an async task (i.e. websocket session)
-@dataclass
-class CallState:
-    is_user_speaking: bool = False
-    is_bot_speaking: bool = False
-    silence_timeout_watcher: Optional[asyncio.Task] = None
-    silence_timeout: Optional[float] = None
-    latest_bot_audio_id: Optional[str] = None
-    should_hangup: bool = False
-    connection_failed: bool = False
-
-
-_call_state: ContextVar[CallState] = ContextVar("call_state")
-call_state = LocalProxy(_call_state)

rasa/dialogue_understanding/commands/repeat_bot_messages_command.py

@@ -1,60 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
-from rasa.dialogue_understanding.patterns.repeat import (
-    RepeatBotMessagesPatternFlowStackFrame,
-)
-from rasa.shared.core.events import Event
-from rasa.shared.core.flows import FlowsList
-from rasa.shared.core.trackers import DialogueStateTracker
-
-
-@dataclass
-class RepeatBotMessagesCommand(Command):
-    """A command to indicate that the bot should repeat its last messages."""
-
-    @classmethod
-    def command(cls) -> str:
-        """Returns the command type."""
-        return "repeat"
-
-    @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> RepeatBotMessagesCommand:
-        """Converts the dictionary to a command.
-
-        Returns:
-            The converted dictionary.
-        """
-        return RepeatBotMessagesCommand()
-
-    def run_command_on_tracker(
-        self,
-        tracker: DialogueStateTracker,
-        all_flows: FlowsList,
-        original_tracker: DialogueStateTracker,
-    ) -> List[Event]:
-        """Runs the command on the tracker.
-        Get all the bot utterances until last user utterance and repeat them.
-
-        Args:
-            tracker: The tracker to run the command on.
-            all_flows: All flows in the assistant.
-            original_tracker: The tracker before any command was executed.
-
-        Returns:
-            The events to apply to the tracker.
-        """
-        stack = tracker.stack
-        stack.push(RepeatBotMessagesPatternFlowStackFrame())
-        return tracker.create_stack_updated_events(stack)
-
-    def __hash__(self) -> int:
-        return hash(self.command())
-
-    def __eq__(self, other: object) -> bool:
-        if not isinstance(other, RepeatBotMessagesCommand):
-            return False
-
-        return True

rasa/dialogue_understanding/commands/user_silence_command.py

@@ -1,59 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any, Dict, List
-from rasa.dialogue_understanding.commands import Command
-from rasa.dialogue_understanding.patterns.user_silence import (
-    UserSilencePatternFlowStackFrame,
-)
-from rasa.shared.core.events import Event
-from rasa.shared.core.flows import FlowsList
-from rasa.shared.core.trackers import DialogueStateTracker
-
-
-@dataclass
-class UserSilenceCommand(Command):
-    """A command to indicate user silence."""
-
-    @classmethod
-    def command(cls) -> str:
-        """Returns the command type."""
-        return "user silence"
-
-    @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> UserSilenceCommand:
-        """Converts the dictionary to a command.
-
-        Returns:
-            The converted dictionary.
-        """
-        return UserSilenceCommand()
-
-    def run_command_on_tracker(
-        self,
-        tracker: DialogueStateTracker,
-        all_flows: FlowsList,
-        original_tracker: DialogueStateTracker,
-    ) -> List[Event]:
-        """Runs the command on the tracker.
-
-        Args:
-            tracker: The tracker to run the command on.
-            all_flows: All flows in the assistant.
-            original_tracker: The tracker before any command was executed.
-
-        Returns:
-            The events to apply to the tracker.
-        """
-        stack = tracker.stack
-        stack.push(UserSilencePatternFlowStackFrame())
-        return tracker.create_stack_updated_events(stack)
-
-    def __hash__(self) -> int:
-        return hash(self.command())
-
-    def __eq__(self, other: object) -> bool:
-        if not isinstance(other, UserSilenceCommand):
-            return False
-
-        return True

rasa/dialogue_understanding/patterns/repeat.py

@@ -1,37 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any, Dict
-
-from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
-from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
-
-FLOW_PATTERN_REPEAT = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "repeat_bot_messages"
-
-
-@dataclass
-class RepeatBotMessagesPatternFlowStackFrame(PatternFlowStackFrame):
-    """A flow stack frame that can get added when bot messages should be repeated"""
-
-    flow_id: str = FLOW_PATTERN_REPEAT
-    """The ID of the flow."""
-
-    @classmethod
-    def type(cls) -> str:
-        """Returns the type of the frame."""
-        return FLOW_PATTERN_REPEAT
-
-    @staticmethod
-    def from_dict(data: Dict[str, Any]) -> RepeatBotMessagesPatternFlowStackFrame:
-        """Creates a `DialogueStackFrame` from a dictionary.
-
-        Args:
-            data: The dictionary to create the `DialogueStackFrame` from.
-
-        Returns:
-            The created `DialogueStackFrame`.
-        """
-        return RepeatBotMessagesPatternFlowStackFrame(
-            frame_id=data["frame_id"],
-            step_id=data["step_id"],
-        )

rasa/dialogue_understanding/patterns/user_silence.py

@@ -1,37 +0,0 @@
-from __future__ import annotations
-from dataclasses import dataclass
-from typing import Any, Dict
-
-from rasa.dialogue_understanding.stack.frames import PatternFlowStackFrame
-from rasa.shared.constants import RASA_DEFAULT_FLOW_PATTERN_PREFIX
-
-
-FLOW_PATTERN_USER_SILENCE = RASA_DEFAULT_FLOW_PATTERN_PREFIX + "user_silence"
-
-
-@dataclass
-class UserSilencePatternFlowStackFrame(PatternFlowStackFrame):
-    """A flow stack frame that can get added when the silence timeout trips."""
-
-    flow_id: str = FLOW_PATTERN_USER_SILENCE
-    """The ID of the flow."""
-
-    @classmethod
-    def type(cls) -> str:
-        """Returns the type of the frame."""
-        return FLOW_PATTERN_USER_SILENCE
-
-    @staticmethod
-    def from_dict(data: Dict[str, Any]) -> UserSilencePatternFlowStackFrame:
-        """Creates a `DialogueStackFrame` from a dictionary.
-
-        Args:
-            data: The dictionary to create the `DialogueStackFrame` from.
-
-        Returns:
-            The created `DialogueStackFrame`.
-        """
-        return UserSilencePatternFlowStackFrame(
-            frame_id=data["frame_id"],
-            step_id=data["step_id"],
-        )

rasa/model_manager/config.py

@@ -1,40 +0,0 @@
-import sys
-import os
-
-DEFAULT_SERVER_BASE_WORKING_DIRECTORY = "working-data"
-
-SERVER_BASE_WORKING_DIRECTORY = os.environ.get(
-    "RASA_MODEL_SERVER_BASE_DIRECTORY", DEFAULT_SERVER_BASE_WORKING_DIRECTORY
-)
-
-SERVER_PORT = os.environ.get("RASA_MODEL_SERVER_PORT", 8000)
-
-SERVER_BASE_URL = os.environ.get("RASA_MODEL_SERVER_BASE_URL", None)
-
-# defaults to storing on the local hard drive
-SERVER_MODEL_REMOTE_STORAGE = os.environ.get("RASA_REMOTE_STORAGE", None)
-
-# The path to the python executable that is running this script
-# we will use the same python to run training / bots
-RASA_PYTHON_PATH = sys.executable
-
-# the max limit for parallel training requests
-DEFAULT_MAX_PARALLEL_TRAININGS = 10
-
-MAX_PARALLEL_TRAININGS = os.getenv(
-    "MAX_PARALLEL_TRAININGS", DEFAULT_MAX_PARALLEL_TRAININGS
-)
-# the max limit for parallel running bots
-DEFAULT_MAX_PARALLEL_BOT_RUNS = 10
-
-MAX_PARALLEL_BOT_RUNS = os.getenv(
-    "MAX_PARALLEL_BOT_RUNS", DEFAULT_MAX_PARALLEL_BOT_RUNS
-)
-
-DEFAULT_SERVER_PATH_PREFIX = "talk"
-
-DEFAULT_MIN_REQUIRED_DISCSPACE_MB = 1
-
-MIN_REQUIRED_DISCSPACE_MB = int(
-    os.getenv("MIN_REQUIRED_DISCSPACE_MB", DEFAULT_MIN_REQUIRED_DISCSPACE_MB)
-)