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

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

@@ -1 +0,0 @@
-import{p as e,f as o}from"./flowDb-1972c806-201c5bbc.js";import{f as t,g as a}from"./styles-080da4f6-75ad421d.js";import{t as s}from"./index-37817b51.js";import"./layout-89e6403a.js";import"./index-2c4b9a3b-42ba3e3d.js";import"./edges-f2ad444c-f1878e0a.js";import"./createText-62fc7601-ef476ecd.js";import"./line-dc73d3fc.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/shared/providers/_configs/azure_entra_id_client_creds.py

@@ -1,40 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass, field
-from typing import Any, Dict, List
-
-from azure.identity import ClientSecretCredential
-
-from rasa.shared.providers._configs.oauth_config import OAuth
-
-
-@dataclass
-class AzureEntraIDClientCreds(OAuth):
-    client_id: str
-    client_secret: str
-    tenant_id: str
-    scopes: List[str] = field(default_factory=list)
-
-    @classmethod
-    def from_config(cls, config: Dict[str, Any]) -> AzureEntraIDClientCreds:
-        scopes = config.get("scopes")
-        if isinstance(scopes, str):
-            scopes = [scopes]
-
-        return cls(
-            client_id=config.get("client_id"),
-            client_secret=config.get("client_secret"),
-            tenant_id=config.get("tenant_id"),
-            scopes=scopes,
-        )
-
-    def get_bearer_token(self) -> str:
-        return (
-            ClientSecretCredential(
-                client_id=self.client_id,
-                client_secret=self.client_secret,
-                tenant_id=self.tenant_id,
-            )
-            .get_token(*self.scopes)
-            .token
-        )

rasa/shared/providers/_configs/azure_entra_id_config.py

@@ -1,533 +0,0 @@
-from __future__ import annotations
-
-import abc
-import logging
-from dataclasses import dataclass
-from enum import Enum
-from functools import lru_cache
-from typing import Any, Callable, Dict, List, Optional, Set, Type, TypeVar
-
-import structlog
-from azure.core.credentials import TokenProvider
-from azure.identity import (
-    CertificateCredential,
-    ClientSecretCredential,
-    DefaultAzureCredential,
-)
-
-from rasa.shared.providers._configs.oauth_config import OAUTH_TYPE_FIELD, OAuth
-
-AZURE_CLIENT_ID_FIELD = "client_id"
-AZURE_CLIENT_SECRET_FIELD = "client_secret"
-AZURE_TENANT_ID_FIELD = "tenant_id"
-AZURE_CERTIFICATE_PATH_FIELD = "certificate_path"
-AZURE_CERTIFICATE_PASSWORD_FIELD = "certificate_password"
-AZURE_SEND_CERTIFICATE_CHAIN_FIELD = "send_certificate_chain"
-AZURE_SCOPES_FIELD = "scopes"
-AZURE_AUTHORITY_FIELD = "authority_host"
-AZURE_DISABLE_INSTANCE_DISCOVERY_FIELD = "disable_instance_discovery"
-
-
-azure_logger = logging.getLogger("azure")
-azure_logger.setLevel(logging.DEBUG)
-
-structlogger = structlog.get_logger()
-
-
-class AzureEntraIDOAuthType(str, Enum):
-    """Azure Entra ID OAuth types."""
-
-    AZURE_ENTRA_ID_DEFAULT = "azure_entra_id_default"
-    AZURE_ENTRA_ID_CLIENT_SECRET = "azure_entra_id_client_secret"
-    AZURE_ENTRA_ID_CLIENT_CERTIFICATE = "azure_entra_id_client_certificate"
-
-    # Invalid type is used to indicate that the type
-    # configuration is invalid EntraID or not set.
-    INVALID = "invalid"
-
-    @staticmethod
-    def from_string(value: Optional[str]) -> AzureEntraIDOAuthType:
-        """Converts a string to an AzureOAuthType."""
-        if value is None or value not in AzureEntraIDOAuthType.valid_string_values():
-            return AzureEntraIDOAuthType.INVALID
-
-        return AzureEntraIDOAuthType(value)
-
-    @staticmethod
-    def valid_string_values() -> Set[str]:
-        """Returns the valid string values for the AzureOAuthType."""
-        return {e.value for e in AzureEntraIDOAuthType.valid_values()}
-
-    @staticmethod
-    def valid_values() -> Set[AzureEntraIDOAuthType]:
-        """Returns the valid values for the AzureOAuthType."""
-        return {
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_DEFAULT,
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_CLIENT_SECRET,
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_CLIENT_CERTIFICATE,
-        }
-
-
-# BearerTokenProvider is a callable that returns a bearer token.
-BearerTokenProvider = Callable[[], str]
-
-AzureEntraIDTokenProviderConfigType = TypeVar(
-    "AzureEntraIDTokenProviderConfigType", bound="AzureEntraIDTokenProviderConfig"
-)
-
-
-class AzureEntraIDTokenProviderConfig(abc.ABC):
-    """Interface for Azure Entra ID OAuth credential configuration."""
-
-    @abc.abstractmethod
-    def create_azure_token_provider(self) -> TokenProvider:
-        """Create an Azure Entra ID token provider."""
-        ...
-
-    @classmethod
-    def from_config(
-        cls: AzureEntraIDTokenProviderConfigType, config: Dict[str, Any]
-    ) -> AzureEntraIDTokenProviderConfigType:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            config: (dict) The config from which to initialize.
-
-        Returns:
-            AzureEntraIDCredential
-        """
-        ...
-
-
-@dataclass
-class AzureEntraIDClientCredentialsConfig(AzureEntraIDTokenProviderConfig):
-    """Azure Entra ID OAuth client credentials configuration.
-
-    Attributes:
-        client_id: The client ID.
-        client_secret: The client secret.
-        tenant_id: The tenant ID.
-        authority_host: The authority host.
-        disable_instance_discovery: Whether to disable instance discovery. This is used
-            to disable fetching metadata from the Azure Instance Metadata Service.
-    """
-
-    client_id: str
-    client_secret: str
-    tenant_id: str
-    authority_host: Optional[str] = None
-    disable_instance_discovery: bool = False
-
-    @staticmethod
-    def required_fields() -> Set[str]:
-        """Returns the required fields for the configuration."""
-        return {AZURE_CLIENT_ID_FIELD, AZURE_TENANT_ID_FIELD, AZURE_CLIENT_SECRET_FIELD}
-
-    @staticmethod
-    def config_has_required_fields(config: Dict[str, Any]) -> bool:
-        """Check if the configuration has all the required fields."""
-        return AzureEntraIDClientCredentialsConfig.required_fields().issubset(
-            set(config.keys())
-        )
-
-    @classmethod
-    def from_config(cls, config: Dict[str, Any]) -> AzureEntraIDClientCredentialsConfig:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            config: (dict) The config from which to initialize.
-
-        Returns:
-            AzureClientCredentialsConfig
-        """
-        if not cls.config_has_required_fields(config):
-            message = (
-                f"A configuration for Azure client credentials "
-                f"must contain the following keys: {cls.required_fields()}"
-            )
-            structlogger.error(
-                "azure_client_credentials_config.missing_required_keys",
-                message=message,
-                config=config,
-            )
-            raise ValueError(message)
-
-        return cls(
-            client_id=config.pop(AZURE_CLIENT_ID_FIELD),
-            client_secret=config.pop(AZURE_CLIENT_SECRET_FIELD),
-            tenant_id=config.pop(AZURE_TENANT_ID_FIELD),
-            authority_host=config.pop(AZURE_AUTHORITY_FIELD, None),
-            disable_instance_discovery=config.pop(
-                AZURE_DISABLE_INSTANCE_DISCOVERY_FIELD, False
-            ),
-        )
-
-    def create_azure_token_provider(self) -> TokenProvider:
-        """Create a ClientSecretCredential for Azure Entra ID."""
-        return create_azure_entra_id_client_credentials(
-            client_id=self.client_id,
-            client_secret=self.client_secret,
-            tenant_id=self.tenant_id,
-            authority_host=self.authority_host,
-            disable_instance_discovery=self.disable_instance_discovery,
-        )
-
-
-@lru_cache
-def create_azure_entra_id_client_credentials(
-    client_id: str,
-    client_secret: str,
-    tenant_id: str,
-    authority_host: Optional[str] = None,
-    disable_instance_discovery: bool = False,
-) -> ClientSecretCredential:
-    """Creates a ClientSecretCredential for Azure Entra ID.
-
-    We cache the result of this function to avoid creating multiple instances
-    of the same credential. This makes it possible to utilise the token caching
-    and token refreshing functionality of the azure-identity library.
-
-    Args:
-        client_id: The client ID.
-        client_secret: The client secret.
-        tenant_id: The tenant ID.
-        authority_host: The authority host.
-        disable_instance_discovery: Whether to disable instance discovery. This is used
-            to disable fetching metadata from the Azure Instance Metadata Service.
-
-    Returns:
-        ClientSecretCredential
-    """
-    return ClientSecretCredential(
-        client_id=client_id,
-        client_secret=client_secret,
-        tenant_id=tenant_id,
-        authority=authority_host,
-        disable_instance_discovery=disable_instance_discovery,
-    )
-
-
-@dataclass
-class AzureEntraIDClientCertificateConfig(AzureEntraIDTokenProviderConfig):
-    """Azure Entra ID OAuth client certificate configuration.
-
-    Attributes:
-        client_id: The client ID.
-        tenant_id: The tenant ID.
-        certificate_path: The path to the certificate file.
-        certificate_password: The certificate password.
-        send_certificate_chain: Whether to send the certificate chain.
-        authority_host: The authority host.
-        disable_instance_discovery: Whether to disable instance discovery. This is used
-            to disable fetching metadata from the Azure Instance Metadata Service.
-    """
-
-    client_id: str
-    tenant_id: str
-    certificate_path: str
-    certificate_password: Optional[str] = None
-    send_certificate_chain: bool = False
-    authority_host: Optional[str] = None
-    disable_instance_discovery: bool = False
-
-    @staticmethod
-    def required_fields() -> Set[str]:
-        """Returns the required fields for the configuration."""
-        return {
-            AZURE_CLIENT_ID_FIELD,
-            AZURE_TENANT_ID_FIELD,
-            AZURE_CERTIFICATE_PATH_FIELD,
-            AZURE_CERTIFICATE_PASSWORD_FIELD,
-        }
-
-    @staticmethod
-    def config_has_required_fields(config: Dict[str, Any]) -> bool:
-        """Check if the configuration has all the required fields."""
-        return AzureEntraIDClientCertificateConfig.required_fields().issubset(
-            set(config.keys())
-        )
-
-    @classmethod
-    def from_config(
-        cls, config: Dict[str, Any]
-    ) -> Optional[AzureEntraIDClientCertificateConfig]:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            config: (dict) The config from which to initialize.
-
-        Returns:
-            AzureClientCertificateConfig
-        """
-        if not cls.config_has_required_fields(config):
-            message = (
-                f"A configuration for Azure client certificate "
-                f"must contain "
-                f"the following keys: {cls.required_fields()}"
-            )
-            structlogger.error(
-                "azure_client_certificate_config.validation_error",
-                message=message,
-                config=config,
-            )
-            raise ValueError(message)
-
-        return cls(
-            client_id=config[AZURE_CLIENT_ID_FIELD],
-            tenant_id=config[AZURE_TENANT_ID_FIELD],
-            certificate_path=config[AZURE_CERTIFICATE_PATH_FIELD],
-            certificate_password=config.get(AZURE_CERTIFICATE_PASSWORD_FIELD, None),
-            authority_host=config.get(AZURE_AUTHORITY_FIELD, None),
-            send_certificate_chain=config.get(
-                AZURE_SEND_CERTIFICATE_CHAIN_FIELD, False
-            ),
-            disable_instance_discovery=config.get(
-                AZURE_DISABLE_INSTANCE_DISCOVERY_FIELD, False
-            ),
-        )
-
-    def create_azure_token_provider(self) -> TokenProvider:
-        """Creates a CertificateCredential for Azure Entra ID."""
-        return create_azure_entra_id_certificate_credentials(
-            client_id=self.client_id,
-            tenant_id=self.tenant_id,
-            certificate_path=self.certificate_path,
-            password=self.certificate_password,
-            send_certificate_chain=self.send_certificate_chain,
-            authority_host=self.authority_host,
-            disable_instance_discovery=self.disable_instance_discovery,
-        )
-
-
-@lru_cache
-def create_azure_entra_id_certificate_credentials(
-    tenant_id: str,
-    client_id: str,
-    certificate_path: Optional[str] = None,
-    password: Optional[str] = None,
-    send_certificate_chain: bool = False,
-    authority_host: Optional[str] = None,
-    disable_instance_discovery: bool = False,
-) -> CertificateCredential:
-    """Creates a CertificateCredential for Azure Entra ID.
-
-    We cache the result of this function to avoid creating multiple instances
-    of the same credential. This makes it possible to utilise the token caching
-    and token refreshing functionality of the azure-identity library.
-
-    Args:
-        tenant_id: The tenant ID.
-        client_id: The client ID.
-        certificate_path: The path to the certificate file.
-        password: The certificate password.
-        send_certificate_chain: Whether to send the certificate chain.
-        authority_host: The authority host.
-        disable_instance_discovery: Whether to disable instance discovery. This is used
-
-    Returns:
-        CertificateCredential
-    """
-
-    return CertificateCredential(
-        client_id=client_id,
-        tenant_id=tenant_id,
-        certificate_path=certificate_path,
-        password=password.encode("utf-8") if password else None,
-        send_certificate_chain=send_certificate_chain,
-        authority=authority_host,
-        disable_instance_discovery=disable_instance_discovery,
-    )
-
-
-@dataclass
-class AzureEntraIDDefaultCredentialsConfig(AzureEntraIDTokenProviderConfig):
-    """Azure Entra ID OAuth default credentials configuration.
-
-    Attributes:
-        authority_host: The authority host.
-    """
-
-    authority_host: Optional[str] = None
-
-    @classmethod
-    def from_config(
-        cls, config: Dict[str, Any]
-    ) -> AzureEntraIDDefaultCredentialsConfig:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            config: (dict) The config from which to initialize.
-
-        Returns:
-            AzureOAuthDefaultCredentialsConfig
-        """
-        return cls(authority_host=config.pop(AZURE_AUTHORITY_FIELD, None))
-
-    def create_azure_token_provider(self) -> TokenProvider:
-        """Creates a DefaultAzureCredential."""
-        return create_azure_entra_id_default_credentials(
-            authority_host=self.authority_host
-        )
-
-
-@lru_cache
-def create_azure_entra_id_default_credentials(
-    authority_host: Optional[str] = None,
-) -> DefaultAzureCredential:
-    """Creates a DefaultAzureCredential.
-
-    We cache the result of this function to avoid creating multiple instances
-    of the same credential. This makes it possible to utilise the token caching
-    functionality of the azure-identity library.
-
-    Args:
-        authority_host: The authority host.
-
-    Returns:
-        DefaultAzureCredential
-    """
-    return DefaultAzureCredential(authority=authority_host)
-
-
-@dataclass
-class AzureEntraIDOAuthConfig(OAuth):
-    """Azure Entra ID OAuth configuration.
-
-    It consists of the scopes and the Azure Entra ID OAuth credentials.
-    """
-
-    scopes: List[str]
-    azure_entra_id_token_provider_config: Optional[AzureEntraIDTokenProviderConfig] = (
-        None
-    )
-
-    @staticmethod
-    def _supported_azure_oauth() -> (
-        Dict[AzureEntraIDOAuthType, Type[AzureEntraIDTokenProviderConfig]]
-    ):
-        """Returns a mapping of supported Azure Entra ID OAuth types to their"""
-        return {
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_DEFAULT: AzureEntraIDDefaultCredentialsConfig,
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_CLIENT_SECRET: AzureEntraIDClientCredentialsConfig,
-            AzureEntraIDOAuthType.AZURE_ENTRA_ID_CLIENT_CERTIFICATE: AzureEntraIDClientCertificateConfig,
-        }
-
-    @staticmethod
-    def _get_azure_oauth_by_type(
-        oauth_type: AzureEntraIDOAuthType,
-    ) -> Type[AzureEntraIDTokenProviderConfig]:
-        """Returns the Azure Entra ID OAuth class based on the type.
-
-        Args:
-            oauth_type: (AzureOAuthType) The type of the Azure Entra ID OAuth.
-
-        Returns:
-            The Azure Entra ID OAuth class
-
-        Raises:
-            ValueError: If the passed oauth_type is not supported or invalid.
-        """
-        azure_oauth_types = AzureEntraIDOAuthConfig._supported_azure_oauth()
-        azure_oauth_class = azure_oauth_types.get(oauth_type)
-
-        if azure_oauth_class is None:
-            message = (
-                f"Unsupported Azure Entra ID oauth type: {oauth_type}. "
-                f"Supported types are: {AzureEntraIDOAuthType.valid_string_values()}"
-            )
-            structlogger.error(
-                "azure_oauth_config.unsupported_azure_oauth_type",
-                message=message,
-            )
-            raise ValueError(message)
-
-        return azure_oauth_class
-
-    @classmethod
-    def from_config(cls, oauth_config: Dict[str, Any]) -> AzureEntraIDOAuthConfig:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            oauth_config: (dict) The config from which to initialize.
-
-        Returns:
-            AzureOAuthConfig
-        """
-        scopes = AzureEntraIDOAuthConfig._read_scopes_from_config(oauth_config)
-        azure_credentials = (
-            AzureEntraIDOAuthConfig._create_azure_entra_id_client_from_config(
-                oauth_config
-            )
-        )
-        return cls(
-            azure_entra_id_token_provider_config=azure_credentials, scopes=scopes
-        )
-
-    @staticmethod
-    def _read_scopes_from_config(oauth_config: Dict[str, Any]) -> List[str]:
-        """Reads scopes from the configuration.
-
-        The original scopes are removed from the configuration.
-
-        Args:
-            oauth_config: (dict) The configuration from which to read the scopes.
-
-        Returns:
-            List[str]: The list of scopes.
-        """
-        scopes = oauth_config.pop(AZURE_SCOPES_FIELD, "")
-
-        if not scopes:
-            message = "Azure Entra ID scopes cannot be empty."
-            structlogger.error(
-                "azure_oauth_config.scopes_empty",
-                message=message,
-            )
-            raise ValueError(message)
-
-        if isinstance(scopes, str):
-            scopes = [scopes]
-
-        return scopes
-
-    @staticmethod
-    def _create_azure_entra_id_client_from_config(
-        oauth_config: Dict[str, Any],
-    ) -> AzureEntraIDTokenProviderConfig:
-        """Creates an Azure Entra ID client from the configuration.
-
-        Args:
-            oauth_config: (dict) The configuration from which to create the credential.
-
-        Returns:
-            AzureEntraIDTokenProviderConfig: The Azure OAuth credential.
-        """
-
-        oauth_type = AzureEntraIDOAuthType.from_string(
-            oauth_config.pop(OAUTH_TYPE_FIELD, None)
-        )
-
-        if oauth_type == AzureEntraIDOAuthType.INVALID:
-            message = (
-                "Azure Entra ID oauth configuration must contain "
-                f"'{OAUTH_TYPE_FIELD}' field and it must be set to one of the "
-                f"following values: {AzureEntraIDOAuthType.valid_string_values()}, "
-            )
-            structlogger.error(
-                "azure_oauth_config.missing_azure_oauth_type",
-                message=message,
-            )
-            raise ValueError(message)
-
-        azure_oauth_class = AzureEntraIDOAuthConfig._get_azure_oauth_by_type(oauth_type)
-        return azure_oauth_class.from_config(oauth_config)
-
-    def _create_azure_credential(
-        self,
-    ) -> TokenProvider:
-        """Create an Azure Entra ID client which can be used to get a bearer token."""
-        return self.azure_entra_id_token_provider_config.create_azure_token_provider()
-
-    def get_bearer_token(self) -> str:
-        """Returns a bearer token."""
-        return self._create_azure_credential().get_token(*self.scopes).token

rasa/shared/providers/_configs/oauth_config.py

@@ -1,33 +0,0 @@
-import abc
-from typing import Any, Dict, TypeVar
-
-OAUTH_TYPE_FIELD = "type"
-OAUTH_KEY = "oauth"
-
-OAuthType = TypeVar("OAuthType", bound="OAuth")
-
-
-class OAuth(abc.ABC):
-    """Interface for OAuth configuration."""
-
-    @classmethod
-    @abc.abstractmethod
-    def from_config(cls: OAuthType, config: Dict[str, Any]) -> OAuthType:
-        """Initializes a dataclass from the passed config.
-
-        Args:
-            config: (dict) The config from which to initialize.
-
-        Returns:
-            OAuth
-        """
-        ...
-
-    @abc.abstractmethod
-    def get_bearer_token(self) -> str:
-        """Returns a bearer token.
-
-        Bear token is used to authenticate requests to the Azure Oopen AI instance's API protected
-        by the Gateway.
-        """
-        ...

rasa/shared/providers/constants.py

@@ -1,6 +0,0 @@
-DEFAULT_AZURE_API_KEY_NAME = "AZURE_API_KEY"
-LITE_LLM_API_BASE_FIELD = "api_base"
-LITE_LLM_API_KEY_FIELD = "api_key"
-LITE_LLM_API_VERSION_FIELD = "api_version"
-LITE_LLM_MODEL_FIELD = "model"
-LITE_LLM_AZURE_AD_TOKEN = "azure_ad_token"