rasa-pro 3.12.0.dev2__py3-none-any.whl → 3.12.0.dev4__py3-none-any.whl

rasa/cli/telemetry.py

@@ -2,11 +2,11 @@ import argparse
 import textwrap
 from typing import List
 
+import rasa.cli.utils
+import rasa.shared.utils.cli
 from rasa import telemetry
 from rasa.cli import SubParsersAction
-import rasa.cli.utils
 from rasa.shared.constants import DOCS_URL_TELEMETRY
-import rasa.shared.utils.cli
 from rasa.utils import licensing
 
 

rasa/cli/test.py

@@ -2,31 +2,31 @@ import argparse
 import asyncio
 import logging
 import os
-from typing import List, Optional, Text, Dict, Union, Any
+from typing import Any, Dict, List, Optional, Text, Union
 
-from rasa.cli import SubParsersAction
+import rasa.cli.utils
 import rasa.shared.data
-from rasa.shared.exceptions import YamlException
-import rasa.shared.utils.io
 import rasa.shared.utils.cli
+import rasa.shared.utils.io
+import rasa.utils.common
+from rasa.cli import SubParsersAction
 from rasa.cli.arguments import test as arguments
 from rasa.core.constants import (
     FAILED_STORIES_FILE,
-    SUCCESSFUL_STORIES_FILE,
     STORIES_WITH_WARNINGS_FILE,
+    SUCCESSFUL_STORIES_FILE,
 )
 from rasa.shared.constants import (
     CONFIG_SCHEMA_FILE,
-    DEFAULT_E2E_TESTS_PATH,
     DEFAULT_CONFIG_PATH,
-    DEFAULT_MODELS_PATH,
     DEFAULT_DATA_PATH,
+    DEFAULT_E2E_TESTS_PATH,
+    DEFAULT_MODELS_PATH,
     DEFAULT_RESULTS_PATH,
 )
-from rasa.shared.utils.yaml import validate_raw_yaml_using_schema_file
-import rasa.cli.utils
-import rasa.utils.common
+from rasa.shared.exceptions import YamlException
 from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.shared.utils.yaml import validate_raw_yaml_using_schema_file
 
 logger = logging.getLogger(__name__)
 
@@ -95,9 +95,9 @@ def _print_core_test_execution_info(args: argparse.Namespace) -> None:
 async def run_core_test_async(args: argparse.Namespace) -> None:
     """Run core tests."""
     from rasa.model_testing import (
-        test_core_models_in_directory,
         test_core,
         test_core_models,
+        test_core_models_in_directory,
     )
 
     stories = rasa.cli.utils.get_validated_path(

rasa/cli/utils.py

@@ -1,30 +1,32 @@
-import json
 import argparse
-import structlog
 import importlib
+import json
 import os
 import sys
 import time
 from pathlib import Path
 from types import FrameType
-from typing import Any, Dict, List, Optional, TYPE_CHECKING, Text, Union, overload
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Union, overload
+
 import randomname
+import structlog
 
 import rasa.shared.utils.cli
 import rasa.shared.utils.io
-from rasa.shared.importers.importer import TrainingDataImporter
+from rasa import telemetry
 from rasa.shared.constants import (
     ASSISTANT_ID_DEFAULT_VALUE,
     ASSISTANT_ID_KEY,
     DEFAULT_CONFIG_PATH,
 )
-from rasa import telemetry
+from rasa.shared.importers.importer import TrainingDataImporter
 from rasa.shared.utils.yaml import read_config_file
 from rasa.utils.io import write_yaml
 
 if TYPE_CHECKING:
     from questionary import Question
     from typing_extensions import Literal
+
     from rasa.validator import Validator
 
 structlogger = structlog.get_logger()

rasa/core/__init__.py

@@ -1,7 +1,6 @@
 import logging
 
 import rasa
-
 from rasa.core.nlg.contextual_response_rephraser import ContextualResponseRephraser
 from rasa.core.policies.enterprise_search_policy import EnterpriseSearchPolicy
 from rasa.core.policies.intentless_policy import IntentlessPolicy

rasa/core/actions/action.py

@@ -41,10 +41,17 @@ from rasa.nlu.constants import (
     RESPONSE_SELECTOR_UTTER_ACTION_KEY,
 )
 from rasa.shared.constants import (
+    ATTACHMENT,
+    BUTTONS,
+    CUSTOM,
     DEFAULT_NLU_FALLBACK_INTENT_NAME,
     DOCS_BASE_URL,
+    ELEMENTS,
     FLOW_PREFIX,
+    IMAGE,
+    QUICK_REPLIES,
     ROUTE_TO_CALM_SLOT,
+    TEXT,
     UTTER_PREFIX,
 )
 from rasa.shared.core.constants import (
@@ -55,6 +62,8 @@ from rasa.shared.core.constants import (
     ACTION_DEFAULT_FALLBACK_NAME,
     ACTION_EXTRACT_SLOTS,
     ACTION_LISTEN_NAME,
+    ACTION_METADATA_EXECUTION_ERROR_MESSAGE,
+    ACTION_METADATA_EXECUTION_SUCCESS,
     ACTION_RESET_ROUTING,
     ACTION_RESTART_NAME,
     ACTION_REVERT_FALLBACK_EVENTS_NAME,
@@ -109,12 +118,12 @@ logger = logging.getLogger(__name__)
 def default_actions(action_endpoint: Optional[EndpointConfig] = None) -> List["Action"]:
     """List default actions."""
     from rasa.core.actions.action_clean_stack import ActionCleanStack
+    from rasa.core.actions.action_hangup import ActionHangup
+    from rasa.core.actions.action_repeat_bot_messages import ActionRepeatBotMessages
     from rasa.core.actions.action_run_slot_rejections import ActionRunSlotRejections
     from rasa.core.actions.action_trigger_chitchat import ActionTriggerChitchat
     from rasa.core.actions.action_trigger_search import ActionTriggerSearch
     from rasa.core.actions.two_stage_fallback import TwoStageFallbackAction
-    from rasa.core.actions.action_hangup import ActionHangup
-    from rasa.core.actions.action_repeat_bot_messages import ActionRepeatBotMessages
     from rasa.dialogue_understanding.patterns.cancel import ActionCancelFlow
     from rasa.dialogue_understanding.patterns.clarify import ActionClarifyFlows
     from rasa.dialogue_understanding.patterns.correction import ActionCorrectFlowSlot
@@ -255,18 +264,18 @@ def action_for_name_or_text(
 def create_bot_utterance(message: Dict[Text, Any]) -> BotUttered:
     """Create BotUttered event from message."""
     bot_message = BotUttered(
-        text=message.pop("text", None),
+        text=message.pop(TEXT, None),
         data={
-            "elements": message.pop("elements", None),
-            "quick_replies": message.pop("quick_replies", None),
-            "buttons": message.pop("buttons", None),
+            ELEMENTS: message.pop(ELEMENTS, None),
+            QUICK_REPLIES: message.pop(QUICK_REPLIES, None),
+            BUTTONS: message.pop(BUTTONS, None),
             # for legacy / compatibility reasons we need to set the image
             # to be the attachment if there is no other attachment (the
             # `.get` is intentional - no `pop` as we still need the image`
             # property to set it in the following line)
-            "attachment": message.pop("attachment", None) or message.get("image", None),
-            "image": message.pop("image", None),
-            "custom": message.pop("custom", None),
+            ATTACHMENT: message.pop(ATTACHMENT, None) or message.get(IMAGE, None),
+            IMAGE: message.pop(IMAGE, None),
+            CUSTOM: message.pop(CUSTOM, None),
         },
         metadata=message,
     )
@@ -310,22 +319,30 @@ class Action:
         return f"{self.__class__.__name__}('{self.name()}')"
 
     def event_for_successful_execution(
-        self, prediction: PolicyPrediction
+        self,
+        prediction: PolicyPrediction,
+        was_successful: bool,
+        error_message: Optional[str],
     ) -> ActionExecuted:
         """Event which should be logged for the successful execution of this action.
 
         Args:
             prediction: Prediction which led to the execution of this event.
+            was_successful: Whether the action was executed successfully.
+            error_message: Error message if the action was not executed successfully.
 
         Returns:
             Event which should be logged onto the tracker.
         """
+        metadata = prediction.action_metadata or {}
+        metadata[ACTION_METADATA_EXECUTION_SUCCESS] = was_successful
+        metadata[ACTION_METADATA_EXECUTION_ERROR_MESSAGE] = error_message
         return ActionExecuted(
             self.name(),
             prediction.policy_name,
             prediction.max_confidence,
             hide_rule_turn=prediction.hide_rule_turn,
-            metadata=prediction.action_metadata,
+            metadata=metadata,
         )
 
 
@@ -412,22 +429,31 @@ class ActionEndToEndResponse(Action):
         return [create_bot_utterance(message)]
 
     def event_for_successful_execution(
-        self, prediction: PolicyPrediction
+        self,
+        prediction: PolicyPrediction,
+        was_successful: bool,
+        error_message: Optional[str],
     ) -> ActionExecuted:
         """Event which should be logged for the successful execution of this action.
 
         Args:
             prediction: Prediction which led to the execution of this event.
+            was_successful: Whether the action was executed successfully.
+            error_message: Error message if the action was not executed successfully.
 
         Returns:
             Event which should be logged onto the tracker.
         """
+        metadata = prediction.action_metadata or {}
+        metadata[ACTION_METADATA_EXECUTION_SUCCESS] = was_successful
+        metadata[ACTION_METADATA_EXECUTION_ERROR_MESSAGE] = error_message
+
         return ActionExecuted(
             policy=prediction.policy_name,
             confidence=prediction.max_confidence,
             action_text=self.action_text,
             hide_rule_turn=prediction.hide_rule_turn,
-            metadata=prediction.action_metadata,
+            metadata=metadata,
         )
 
 
@@ -727,10 +753,7 @@ class ActionDeactivateLoop(Action):
 
 
 class RemoteActionJSONValidator:
-    """
-    A validator class for ensuring that the JSON response from a custom action executor
-    adheres to the expected schema.
-    """
+    """Check action responses for JSON schema validation."""
 
     @staticmethod
     def action_response_format_spec() -> Dict[Text, Any]:
@@ -753,8 +776,7 @@ class RemoteActionJSONValidator:
 
     @classmethod
     def validate(cls, result: Dict[Text, Any]) -> bool:
-        """
-        Validate the given JSON result against the expected Action response schema.
+        """Validate the given JSON result against the expected Action response schema.
 
         This method uses a cached JSON schema validator to check if the provided result
         conforms to the predefined schema.
@@ -788,8 +810,7 @@ class RemoteActionJSONValidator:
     @classmethod
     @lru_cache(maxsize=1)
     def get_action_response_validator(cls) -> Draft202012Validator:
-        """
-        Retrieve a cached JSON schema validator for the Action response schema.
+        """Retrieve a cached JSON schema validator for the Action response schema.
 
         Returns:
             Draft202012Validator: An instance of the JSON schema validator.

rasa/core/actions/action_hangup.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional
 
 from rasa.core.actions.action import Action
 from rasa.core.channels import OutputChannel

rasa/core/actions/action_repeat_bot_messages.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional
 
 from rasa.core.actions.action import Action
 from rasa.core.channels import OutputChannel
@@ -14,7 +14,7 @@ from rasa.dialogue_understanding.patterns.user_silence import (
 )
 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.events import BotUttered, Event, UserUttered
 from rasa.shared.core.trackers import DialogueStateTracker
 
 

rasa/core/actions/action_run_slot_rejections.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional, TYPE_CHECKING, Text, Union
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Union
 
 import structlog
 from jinja2 import Template
@@ -20,8 +20,8 @@ from rasa.shared.core.slots import (
 )
 
 if TYPE_CHECKING:
-    from rasa.core.nlg import NaturalLanguageGenerator
     from rasa.core.channels.channel import OutputChannel
+    from rasa.core.nlg import NaturalLanguageGenerator
     from rasa.shared.core.domain import Domain
     from rasa.shared.core.trackers import DialogueStateTracker
 

rasa/core/actions/action_trigger_chitchat.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional
 
 from rasa.core.actions.action import Action
 from rasa.core.channels import OutputChannel

rasa/core/actions/action_trigger_flow.py

@@ -1,15 +1,16 @@
-from typing import Any, Dict, Optional, Text, List
+from typing import Any, Dict, List, Optional, Text
 
 import structlog
+
+from rasa.core.actions import action
+from rasa.core.channels import OutputChannel
+from rasa.core.nlg import NaturalLanguageGenerator
 from rasa.dialogue_understanding.stack.frames.flow_stack_frame import (
     FlowStackFrameType,
     UserFlowStackFrame,
 )
 from rasa.dialogue_understanding.stack.utils import top_user_flow_frame
-from rasa.core.actions import action
-from rasa.core.channels import OutputChannel
 from rasa.shared.constants import FLOW_PREFIX
-
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import (
     ActiveLoop,
@@ -17,7 +18,6 @@ from rasa.shared.core.events import (
     FlowInterrupted,
     SlotSet,
 )
-from rasa.core.nlg import NaturalLanguageGenerator
 from rasa.shared.core.trackers import DialogueStateTracker
 
 structlogger = structlog.get_logger(__name__)

rasa/core/actions/action_trigger_search.py

@@ -1,4 +1,4 @@
-from typing import Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional
 
 from rasa.core.actions.action import Action
 from rasa.core.channels import OutputChannel

rasa/core/actions/forms.py

@@ -1,33 +1,35 @@
 import copy
-from typing import Text, List, Optional, Union, Any, Dict, Set
 import itertools
+import json
 import logging
+from typing import Any, Dict, List, Optional, Set, Text, Union
+
 import structlog
-import json
 
 from rasa.core.actions import action
-from rasa.core.actions.loops import LoopAction
-from rasa.core.channels import OutputChannel
-from rasa.shared.core.domain import Domain, KEY_SLOTS
-from rasa.shared.core.constants import SlotMappingType, SLOT_MAPPINGS, MAPPING_TYPE
-
 from rasa.core.actions.action import RemoteAction
 from rasa.core.actions.action_exceptions import ActionExecutionRejection
+from rasa.core.actions.loops import LoopAction
+from rasa.core.channels import OutputChannel
+from rasa.core.nlg import NaturalLanguageGenerator
+from rasa.shared.constants import UTTER_PREFIX
 from rasa.shared.core.constants import (
     ACTION_EXTRACT_SLOTS,
     ACTION_LISTEN_NAME,
+    MAPPING_TYPE,
     REQUESTED_SLOT,
+    SLOT_MAPPINGS,
+    SlotMappingType,
 )
-from rasa.shared.constants import UTTER_PREFIX
+from rasa.shared.core.domain import KEY_SLOTS, Domain
 from rasa.shared.core.events import (
-    Event,
-    SlotSet,
     ActionExecuted,
-    ActiveLoop,
     ActionExecutionRejected,
+    ActiveLoop,
+    Event,
     Restarted,
+    SlotSet,
 )
-from rasa.core.nlg import NaturalLanguageGenerator
 from rasa.shared.core.slot_mappings import SlotMapping
 from rasa.shared.core.slots import ListSlot
 from rasa.shared.core.trackers import DialogueStateTracker

rasa/core/actions/http_custom_action_executor.py

@@ -1,6 +1,6 @@
 import json
 import logging
-from typing import Any, Dict, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, Any, Dict, Optional
 
 import aiohttp
 
@@ -94,6 +94,13 @@ class HTTPCustomActionExecutor(CustomActionExecutor):
                 )
                 logger.error(exception.message)
                 raise exception
+            elif e.status == 404:
+                message = (
+                    f"Custom action implementation with name '{self.action_name}' "
+                    f"not found."
+                )
+                logger.error(message)
+                raise RasaException(message) from e
             else:
                 raise RasaException(
                     f"Failed to execute custom action '{self.action_name}'"

rasa/core/actions/loops.py

@@ -1,13 +1,13 @@
 from abc import ABC
-from typing import Any, Dict, List, TYPE_CHECKING, Optional, Text
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text
 
 from rasa.core.actions.action import Action
-from rasa.shared.core.events import Event, ActiveLoop
+from rasa.shared.core.events import ActiveLoop, Event
 
 if TYPE_CHECKING:
     from rasa.core.channels import OutputChannel
-    from rasa.shared.core.domain import Domain
     from rasa.core.nlg import NaturalLanguageGenerator
+    from rasa.shared.core.domain import Domain
     from rasa.shared.core.trackers import DialogueStateTracker
 
 

rasa/core/actions/two_stage_fallback.py

@@ -1,29 +1,29 @@
 import copy
 import time
-from typing import List, Text, Optional
+from typing import List, Optional, Text
 
 from rasa.core.actions import action
 from rasa.core.actions.loops import LoopAction
 from rasa.core.channels import OutputChannel
-from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import (
-    Event,
-    UserUtteranceReverted,
-    ActionExecuted,
-    UserUttered,
-    ActiveLoop,
-)
 from rasa.core.nlg import NaturalLanguageGenerator
-from rasa.shared.core.trackers import DialogueStateTracker, EventVerbosity
 from rasa.shared.constants import DEFAULT_NLU_FALLBACK_INTENT_NAME
 from rasa.shared.core.constants import (
-    USER_INTENT_OUT_OF_SCOPE,
-    ACTION_LISTEN_NAME,
-    ACTION_DEFAULT_FALLBACK_NAME,
     ACTION_DEFAULT_ASK_AFFIRMATION_NAME,
     ACTION_DEFAULT_ASK_REPHRASE_NAME,
+    ACTION_DEFAULT_FALLBACK_NAME,
+    ACTION_LISTEN_NAME,
     ACTION_TWO_STAGE_FALLBACK_NAME,
+    USER_INTENT_OUT_OF_SCOPE,
+)
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.events import (
+    ActionExecuted,
+    ActiveLoop,
+    Event,
+    UserUtteranceReverted,
+    UserUttered,
 )
+from rasa.shared.core.trackers import DialogueStateTracker, EventVerbosity
 from rasa.shared.nlu.constants import INTENT, PREDICTED_CONFIDENCE_KEY
 from rasa.utils.endpoints import EndpointConfig
 

rasa/core/auth_retry_tracker_store.py

@@ -2,13 +2,12 @@ import logging
 from typing import Iterable, Optional, Text
 
 from rasa.core.brokers.broker import EventBroker
+from rasa.core.secrets_manager.secret_manager import EndpointResolver
 from rasa.core.tracker_store import TrackerStore, create_tracker_store
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.utils.endpoints import EndpointConfig
 
-from rasa.core.secrets_manager.secret_manager import EndpointResolver
-
 logger = logging.getLogger(__name__)
 
 DEFAULT_RETRIES = 3

rasa/core/brokers/broker.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
+
 import logging
 from asyncio import AbstractEventLoop
-from typing import Any, Dict, Text, Optional, Union, TypeVar, Type
+from typing import Any, Dict, Optional, Text, Type, TypeVar, Union
 
 import aiormq
 

rasa/core/brokers/file.py

@@ -2,7 +2,7 @@ import json
 import logging
 import typing
 from asyncio import AbstractEventLoop
-from typing import Optional, Text, Dict
+from typing import Dict, Optional, Text
 
 from rasa.core.brokers.broker import EventBroker
 

rasa/core/brokers/kafka.py

@@ -1,23 +1,23 @@
 import asyncio
-import os
 import json
 import logging
+import os
+import threading
+import time
+from asyncio import AbstractEventLoop
 from functools import cached_property
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Union
 
 import structlog
-import threading
-from asyncio import AbstractEventLoop
-from typing import Any, Text, List, Optional, Union, Dict, TYPE_CHECKING
-import time
 
+import rasa.shared.utils.common
 from rasa.core.brokers.broker import EventBroker
 from rasa.core.exceptions import KafkaProducerInitializationError
 from rasa.shared.utils.io import DEFAULT_ENCODING
 from rasa.utils.endpoints import EndpointConfig
-import rasa.shared.utils.common
 
 if TYPE_CHECKING:
-    from confluent_kafka import KafkaError, Producer, Message
+    from confluent_kafka import KafkaError, Message, Producer
 
 logger = logging.getLogger(__name__)
 structlogger = structlog.get_logger()
@@ -293,7 +293,7 @@ def kafka_error_callback(err: "KafkaError") -> None:
     Any exception raised from this callback will be re-raised from the
     triggering flush() call.
     """
-    from confluent_kafka import KafkaException, KafkaError
+    from confluent_kafka import KafkaError, KafkaException
 
     # handle authentication / connection related issues, likely pointing
     # to a configuration error

rasa/core/brokers/pika.py

@@ -1,25 +1,24 @@
 import asyncio
 import json
 import logging
-from functools import cached_property
-
-import structlog
 import os
 import ssl
 from asyncio import AbstractEventLoop
 from collections import deque
-from typing import Deque, Dict, Optional, Text, Union, Any, List, Set, Tuple, cast
+from functools import cached_property
+from typing import Any, Deque, Dict, List, Optional, Set, Text, Tuple, Union, cast
 from urllib.parse import urlparse
 
 import aio_pika
+import structlog
 
-from rasa.shared.exceptions import RasaException
-from rasa.shared.constants import DOCS_URL_PIKA_EVENT_BROKER
-from rasa.core.brokers.broker import EventBroker
+import rasa.shared.utils.common
 import rasa.shared.utils.io
-from rasa.utils.endpoints import EndpointConfig
+from rasa.core.brokers.broker import EventBroker
+from rasa.shared.constants import DOCS_URL_PIKA_EVENT_BROKER
+from rasa.shared.exceptions import RasaException
 from rasa.shared.utils.io import DEFAULT_ENCODING
-import rasa.shared.utils.common
+from rasa.utils.endpoints import EndpointConfig
 
 logger = logging.getLogger(__name__)
 structlogger = structlog.get_logger()

rasa/core/brokers/sql.py

@@ -2,11 +2,11 @@ import contextlib
 import json
 import logging
 from asyncio import AbstractEventLoop
-from typing import Any, Dict, Optional, Text, Generator
+from typing import Any, Dict, Generator, Optional, Text
 
-from sqlalchemy.orm import Session
 from sqlalchemy import Column, Integer, String
 from sqlalchemy import Text as SqlAlchemyText  # to avoid name clash with typing.Text
+from sqlalchemy.orm import Session
 
 from rasa.core.brokers.broker import EventBroker
 from rasa.utils.endpoints import EndpointConfig
@@ -44,9 +44,10 @@ class SQLEventBroker(EventBroker):
         password: Optional[Text] = None,
     ) -> None:
         """Initializes `SQLBrokerEvent`."""
-        from rasa.core.tracker_store import SQLTrackerStore
         import sqlalchemy.orm
 
+        from rasa.core.tracker_store import SQLTrackerStore
+
         engine_url = SQLTrackerStore.get_db_url(
             dialect, host, port, db, username, password
         )

rasa/core/channels/__init__.py

@@ -1,3 +1,4 @@
+# ruff: noqa: I001
 from typing import Text, Dict, List, Type
 
 from rasa.core.channels.channel import (  # noqa: F401
@@ -31,6 +32,7 @@ from rasa.core.channels.vier_cvg import CVGInput
 from rasa.core.channels.voice_stream.twilio_media_streams import (
     TwilioMediaStreamsInputChannel,
 )
+from rasa.core.channels.studio_chat import StudioChatInput
 
 input_channel_classes: List[Type[InputChannel]] = [
     CmdlineInput,
@@ -53,6 +55,7 @@ input_channel_classes: List[Type[InputChannel]] = [
     JambonzVoiceReadyInput,
     TwilioMediaStreamsInputChannel,
     BrowserAudioInputChannel,
+    StudioChatInput,
 ]
 
 # Mapping from an input channel name to its class to allow name based lookup.

rasa/core/channels/botframework.py

@@ -3,7 +3,7 @@ import json
 import logging
 import re
 from http import HTTPStatus
-from typing import Text, Dict, Any, List, Iterable, Callable, Awaitable, Optional
+from typing import Any, Awaitable, Callable, Dict, Iterable, List, Optional, Text
 
 import jwt
 import requests
@@ -14,7 +14,7 @@ from sanic import Blueprint, response
 from sanic.request import Request
 from sanic.response import HTTPResponse
 
-from rasa.core.channels.channel import UserMessage, OutputChannel, InputChannel
+from rasa.core.channels.channel import InputChannel, OutputChannel, UserMessage
 
 MICROSOFT_OPEN_ID_URI = (
     "https://login.botframework.com/v1/.well-known/openidconfiguration"