rasa-pro 3.9.18__py3-none-any.whl

rasa/core/run.py

@@ -0,0 +1,331 @@
+import asyncio
+import logging
+import os
+import platform
+import uuid
+import warnings
+from asyncio import AbstractEventLoop
+from functools import partial
+from typing import (
+    Any,
+    Callable,
+    List,
+    Optional,
+    Text,
+    Tuple,
+    Union,
+    Dict,
+)
+
+from sanic import Sanic
+from sanic.worker.loader import AppLoader
+
+import rasa.core.utils
+import rasa.shared.utils.common
+import rasa.shared.utils.io
+import rasa.utils
+from rasa.utils import licensing
+import rasa.utils.common
+import rasa.utils.io
+from rasa import server, telemetry
+from rasa.constants import ENV_SANIC_BACKLOG
+from rasa.core import agent, channels, constants
+from rasa.core.agent import Agent
+from rasa.core.channels import console
+from rasa.core.channels.channel import InputChannel
+from rasa.core.utils import AvailableEndpoints
+from rasa.plugin import plugin_manager
+from rasa.shared.exceptions import RasaException
+from rasa.shared.utils.yaml import read_config_file
+
+logger = logging.getLogger()  # get the root logger
+
+
+def create_http_input_channels(
+    channel: Optional[Text], credentials_file: Optional[Text]
+) -> List["InputChannel"]:
+    """Instantiate the chosen input channel."""
+    if credentials_file:
+        all_credentials = read_config_file(credentials_file)
+    else:
+        all_credentials = {}
+
+    if channel:
+        if len(all_credentials) > 1:
+            logger.info(
+                "Connecting to channel '{}' which was specified by the "
+                "'--connector' argument. Any other channels will be ignored. "
+                "To connect to all given channels, omit the '--connector' "
+                "argument.".format(channel)
+            )
+        return [_create_single_channel(channel, all_credentials.get(channel))]
+    else:
+        return [_create_single_channel(c, k) for c, k in all_credentials.items()]
+
+
+def _create_single_channel(channel: Text, credentials: Dict[Text, Any]) -> Any:
+    from rasa.core.channels import BUILTIN_CHANNELS
+
+    if channel in BUILTIN_CHANNELS:
+        return BUILTIN_CHANNELS[channel].from_credentials(credentials)
+    else:
+        # try to load channel based on class name
+        try:
+            input_channel_class = rasa.shared.utils.common.class_from_module_path(
+                channel
+            )
+            return input_channel_class.from_credentials(credentials)
+        except (AttributeError, ImportError):
+            raise RasaException(
+                f"Failed to find input channel class for '{channel}'. Unknown "
+                f"input channel. Check your credentials configuration to "
+                f"make sure the mentioned channel is not misspelled. "
+                f"If you are creating your own channel, make sure it "
+                f"is a proper name of a class in a module."
+            )
+
+
+def _create_app_without_api(cors: Optional[Union[Text, List[Text]]] = None) -> Sanic:
+    app = Sanic("rasa_core_no_api", configure_logging=False)
+
+    # Reset Sanic warnings filter that allows the triggering of Sanic warnings
+    warnings.filterwarnings("ignore", category=DeprecationWarning, module=r"sanic.*")
+
+    server.add_root_route(app)
+    server.configure_cors(app, cors)
+    return app
+
+
+def _is_apple_silicon_system() -> bool:
+    # check if the system is MacOS
+    if platform.system().lower() != "darwin":
+        return False
+    # check for arm architecture, indicating apple silicon
+    return platform.machine().startswith("arm") or os.uname().machine.startswith("arm")
+
+
+def configure_app(
+    input_channels: Optional[List["InputChannel"]] = None,
+    cors: Optional[Union[Text, List[Text], None]] = None,
+    auth_token: Optional[Text] = None,
+    enable_api: bool = True,
+    response_timeout: int = constants.DEFAULT_RESPONSE_TIMEOUT,
+    jwt_secret: Optional[Text] = None,
+    jwt_private_key: Optional[Text] = None,
+    jwt_method: Optional[Text] = None,
+    route: Optional[Text] = "/webhooks/",
+    port: int = constants.DEFAULT_SERVER_PORT,
+    endpoints: Optional[AvailableEndpoints] = None,
+    log_file: Optional[Text] = None,
+    conversation_id: Optional[Text] = uuid.uuid4().hex,
+    use_syslog: bool = False,
+    syslog_address: Optional[Text] = None,
+    syslog_port: Optional[int] = None,
+    syslog_protocol: Optional[Text] = None,
+    request_timeout: Optional[int] = None,
+    server_listeners: Optional[List[Tuple[Callable, Text]]] = None,
+    use_uvloop: Optional[bool] = True,
+    keep_alive_timeout: int = constants.DEFAULT_KEEP_ALIVE_TIMEOUT,
+) -> Sanic:
+    """Run the agent."""
+    rasa.core.utils.configure_file_logging(
+        logger, log_file, use_syslog, syslog_address, syslog_port, syslog_protocol
+    )
+
+    if enable_api:
+        loader = AppLoader(
+            factory=partial(
+                server.create_app,
+                cors_origins=cors,
+                auth_token=auth_token,
+                response_timeout=response_timeout,
+                jwt_secret=jwt_secret,
+                jwt_private_key=jwt_private_key,
+                jwt_method=jwt_method,
+                endpoints=endpoints,
+            )
+        )
+    else:
+        loader = AppLoader(factory=partial(_create_app_without_api, cors))
+
+    app = loader.load()
+    app.config.KEEP_ALIVE_TIMEOUT = keep_alive_timeout
+
+    if _is_apple_silicon_system() or not use_uvloop:
+        app.config.USE_UVLOOP = False
+        # some library still sets the loop to uvloop, even if disabled for sanic
+        # using uvloop leads to breakingio errors, see
+        # https://rasahq.atlassian.net/browse/ENG-667
+        asyncio.set_event_loop_policy(None)
+
+    if input_channels:
+        channels.channel.register(input_channels, app, route=route)
+    else:
+        input_channels = []
+
+    if logger.isEnabledFor(logging.DEBUG):
+        rasa.core.utils.list_routes(app)
+
+    @app.main_process_start
+    async def configure_async_logging(running_app: Sanic) -> None:
+        if logger.isEnabledFor(logging.DEBUG):
+            rasa.utils.io.enable_async_loop_debugging(asyncio.get_event_loop())
+
+    if "cmdline" in {c.name() for c in input_channels}:
+
+        @app.after_server_start
+        async def run_cmdline_io(running_app: Sanic) -> None:
+            """Small wrapper to shut down the server once cmd io is done."""
+            await console.record_messages(
+                server_url=constants.DEFAULT_SERVER_FORMAT.format("http", port),
+                sender_id=conversation_id,
+                request_timeout=request_timeout,
+            )
+
+            logger.info("Killing Sanic server now.")
+            running_app.stop()  # kill the sanic server
+
+    @app.after_server_stop
+    async def after_server_stop(running_app: Sanic) -> None:
+        plugin_manager().hook.after_server_stop()
+
+    if server_listeners:
+        for listener, event in server_listeners:
+            app.register_listener(listener, event)
+
+    return app
+
+
+def serve_application(
+    model_path: Optional[Text] = None,
+    channel: Optional[Text] = None,
+    interface: Optional[Text] = constants.DEFAULT_SERVER_INTERFACE,
+    port: int = constants.DEFAULT_SERVER_PORT,
+    credentials: Optional[Text] = None,
+    cors: Optional[Union[Text, List[Text]]] = None,
+    auth_token: Optional[Text] = None,
+    enable_api: bool = True,
+    response_timeout: int = constants.DEFAULT_RESPONSE_TIMEOUT,
+    jwt_secret: Optional[Text] = None,
+    jwt_private_key: Optional[Text] = None,
+    jwt_method: Optional[Text] = None,
+    endpoints: Optional[AvailableEndpoints] = None,
+    remote_storage: Optional[Text] = None,
+    log_file: Optional[Text] = None,
+    ssl_certificate: Optional[Text] = None,
+    ssl_keyfile: Optional[Text] = None,
+    ssl_ca_file: Optional[Text] = None,
+    ssl_password: Optional[Text] = None,
+    conversation_id: Optional[Text] = uuid.uuid4().hex,
+    use_syslog: Optional[bool] = False,
+    syslog_address: Optional[Text] = None,
+    syslog_port: Optional[int] = None,
+    syslog_protocol: Optional[Text] = None,
+    request_timeout: Optional[int] = None,
+    server_listeners: Optional[List[Tuple[Callable, Text]]] = None,
+) -> None:
+    """Run the API entrypoint."""
+    if not channel and not credentials:
+        channel = "cmdline"
+
+    input_channels = create_http_input_channels(channel, credentials)
+
+    app = configure_app(
+        input_channels,
+        cors,
+        auth_token,
+        enable_api,
+        response_timeout,
+        jwt_secret,
+        jwt_private_key,
+        jwt_method,
+        port=port,
+        endpoints=endpoints,
+        log_file=log_file,
+        conversation_id=conversation_id,
+        use_syslog=use_syslog,
+        syslog_address=syslog_address,
+        syslog_port=syslog_port,
+        syslog_protocol=syslog_protocol,
+        request_timeout=request_timeout,
+        server_listeners=server_listeners,
+    )
+
+    ssl_context = server.create_ssl_context(
+        ssl_certificate, ssl_keyfile, ssl_ca_file, ssl_password
+    )
+    protocol = "https" if ssl_context else "http"
+
+    logger.info(f"Starting Rasa server on {protocol}://{interface}:{port}")
+
+    app.register_listener(
+        partial(load_agent_on_start, model_path, endpoints, remote_storage),
+        "before_server_start",
+    )
+
+    app.register_listener(
+        licensing.validate_limited_server_license, "after_server_start"
+    )
+
+    app.register_listener(close_resources, "after_server_stop")
+
+    number_of_workers = rasa.core.utils.number_of_sanic_workers(
+        endpoints.lock_store if endpoints else None
+    )
+
+    telemetry.track_server_start(
+        input_channels, endpoints, model_path, number_of_workers, enable_api
+    )
+
+    rasa.utils.common.update_sanic_log_level(
+        log_file, use_syslog, syslog_address, syslog_port, syslog_protocol
+    )
+
+    app.run(
+        host=interface,
+        port=port,
+        ssl=ssl_context,
+        backlog=int(os.environ.get(ENV_SANIC_BACKLOG, "100")),
+        workers=number_of_workers,
+        legacy=True,
+    )
+
+
+# noinspection PyUnusedLocal
+async def load_agent_on_start(
+    model_path: Text,
+    endpoints: AvailableEndpoints,
+    remote_storage: Optional[Text],
+    app: Sanic,
+    loop: AbstractEventLoop,
+) -> Agent:
+    """Load an agent.
+
+    Used to be scheduled on server start
+    (hence the `app` and `loop` arguments).
+    """
+    app.ctx.agent = await agent.load_agent(
+        model_path=model_path,
+        remote_storage=remote_storage,
+        endpoints=endpoints,
+        loop=loop,
+    )
+    logger.info("Rasa server is up and running.")
+    return app.ctx.agent
+
+
+async def close_resources(app: Sanic, _: AbstractEventLoop) -> None:
+    """Gracefully closes resources when shutting down server.
+
+    Args:
+        app: The Sanic application.
+        _: The current Sanic worker event loop.
+    """
+    current_agent = getattr(app.ctx, "agent", None)
+    if not current_agent:
+        logger.debug("No agent found when shutting down server.")
+        return
+
+    event_broker = current_agent.tracker_store.event_broker
+    if event_broker:
+        await event_broker.close()

rasa/core/secrets_manager/constants.py

@@ -0,0 +1,32 @@
+SECRET_MANAGER_ENV_NAME = "SECRET_MANAGER"
+SUPPORTED_SECRET_MANAGERS = ["vault"]
+
+SECRET_MANAGER_KEY = "secrets_manager"
+SECRETS_PATH = "secrets_path"
+TRANSIT_MOUNT_POINT_LABEL = "transit_mount_point"
+NAMESPACE_KEY_LABEL = "namespace"
+
+SECRET_MANAGER_PREFIX = "secrets_manager"
+
+SOURCE_KEY_LABEL = "source"
+SECRET_KEY_LABEL = "secret_key"
+TRANSIT_KEY_FOR_ENCRYPTION_LABEL = "transit_key"
+TRACKER_STORE_ENDPOINT_TYPE = "tracker_store"
+
+USERNAME_LABEL = "username"
+PASSWORD_LABEL = "password"
+
+VAULT_URL_ENV_NAME = "VAULT_URL"
+VAULT_TOKEN_ENV_NAME = "VAULT_TOKEN"
+VAULT_RASA_SECRETS_PATH_ENV_NAME = "VAULT_RASA_SECRETS_PATH"
+VAULT_TRANSIT_MOUNT_POINT_ENV_NAME = "VAULT_TRANSIT_MOUNT_POINT"
+VAULT_NAMESPACE_ENV_NAME = "VAULT_NAMESPACE"
+VAULT_DEFAULT_RASA_SECRETS_PATH = "rasa-secrets"
+VAULT_SECRET_MANAGER_NAME = "vault"
+
+
+VAULT_ENDPOINT_URL_LABEL = "url"
+VAULT_ENDPOINT_TOKEN_LABEL = "token"
+VAULT_ENDPOINT_SECRETS_PATH_LABEL = "secrets_path"
+VAULT_ENDPOINT_TRANSIT_MOUNT_POINT_LABEL = "transit_mount_point"
+VAULT_ENDPOINT_NAMESPACE_LABEL = "namespace"