rasa-pro 3.10.11__tar.gz → 3.10.13__tar.gz

rasa_pro-3.10.13/PKG-INFO

@@ -0,0 +1,196 @@
+Metadata-Version: 2.1
+Name: rasa-pro
+Version: 3.10.13
+Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
+Home-page: https://rasa.com
+Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
+Author: Rasa Technologies GmbH
+Author-email: hi@rasa.com
+Maintainer: Tom Bocklisch
+Maintainer-email: tom@rasa.com
+Requires-Python: >=3.9,<3.11
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Topic :: Software Development :: Libraries
+Provides-Extra: full
+Provides-Extra: gh-release-notes
+Provides-Extra: jieba
+Provides-Extra: metal
+Provides-Extra: mlflow
+Provides-Extra: spacy
+Provides-Extra: transformers
+Requires-Dist: CacheControl (>=0.12.14,<0.13.0)
+Requires-Dist: PyJWT[crypto] (>=2.8.0,<3.0.0)
+Requires-Dist: SQLAlchemy (>=2.0.22,<2.1.0)
+Requires-Dist: absl-py (>=2.0,<2.1)
+Requires-Dist: aio-pika (>=8.2.3,<8.2.4)
+Requires-Dist: aiogram (>=2.15,<2.26)
+Requires-Dist: aiohttp (>=3.9.4,<3.10)
+Requires-Dist: apscheduler (>=3.10,<3.11)
+Requires-Dist: attrs (>=23.1,<23.2)
+Requires-Dist: azure-storage-blob (>=12.16.0,<12.17.0)
+Requires-Dist: boto3 (>=1.35.5,<1.36.0)
+Requires-Dist: certifi (>=2024.07.04)
+Requires-Dist: colorama (>=0.4.6,<0.5.0) ; sys_platform == "win32"
+Requires-Dist: colorclass (>=2.2,<2.3)
+Requires-Dist: coloredlogs (>=15,<16)
+Requires-Dist: colorhash (>=2.0,<2.1.0)
+Requires-Dist: confluent-kafka (>=2.3.0,<3.0.0)
+Requires-Dist: cryptography (>=42.0.5)
+Requires-Dist: cvg-python-sdk (>=0.5.1,<0.6.0)
+Requires-Dist: dask (==2022.10.2) ; python_version >= "3.9" and python_version < "3.11"
+Requires-Dist: diskcache (>=5.6.3,<5.7.0)
+Requires-Dist: dnspython (==2.6.1)
+Requires-Dist: faiss-cpu (>=1.7.4,<2.0.0)
+Requires-Dist: faker (>=26.0.0,<27.0.0)
+Requires-Dist: fbmessenger (>=6.0.0,<6.1.0)
+Requires-Dist: github3.py (>=3.2.0,<3.3.0) ; extra == "gh-release-notes"
+Requires-Dist: gitpython (>=3.1.41,<3.2.0) ; extra == "full"
+Requires-Dist: google-auth (>=2.23.4,<3)
+Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0)
+Requires-Dist: hvac (>=1.2.1,<2.0.0)
+Requires-Dist: importlib-metadata (>=8.5.0,<8.6.0)
+Requires-Dist: importlib-resources (==6.1.3)
+Requires-Dist: jieba (>=0.42.1,<0.43) ; extra == "jieba" or extra == "full"
+Requires-Dist: jinja2 (>=3.1.4,<4.0.0)
+Requires-Dist: jsonpatch (>=1.33,<2.0)
+Requires-Dist: jsonpickle (>=3.0,<3.1)
+Requires-Dist: jsonschema (>=4.22)
+Requires-Dist: keras (==2.14.0)
+Requires-Dist: langchain (>=0.2.0,<0.3.0)
+Requires-Dist: langchain-community (>=0.2.0,<0.3.0)
+Requires-Dist: litellm (>=1.52.6,<1.53.0)
+Requires-Dist: matplotlib (>=3.7,<3.8)
+Requires-Dist: mattermostwrapper (>=2.2,<2.3)
+Requires-Dist: mlflow (>=2.15.1,<3.0.0) ; extra == "mlflow"
+Requires-Dist: networkx (>=3.1,<3.2)
+Requires-Dist: numpy (>=1.23.5,<1.25.0) ; python_version >= "3.9" and python_version < "3.11"
+Requires-Dist: openai (>=1.54.0,<1.55.0)
+Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
+Requires-Dist: opentelemetry-api (>=1.16.0,<1.17.0)
+Requires-Dist: opentelemetry-exporter-jaeger (>=1.16.0,<1.17.0)
+Requires-Dist: opentelemetry-exporter-otlp (>=1.16.0,<1.17.0)
+Requires-Dist: opentelemetry-sdk (>=1.16.0,<1.17.0)
+Requires-Dist: packaging (>=23.2,<23.3)
+Requires-Dist: pep440-version-utils (>=1.1.0,<1.2.0)
+Requires-Dist: pluggy (>=1.2.0,<2.0.0)
+Requires-Dist: portalocker (>=2.7.0,<3.0.0)
+Requires-Dist: presidio-analyzer (>=2.2.33,<2.2.34)
+Requires-Dist: presidio-anonymizer (>=2.2.354,<3.0.0)
+Requires-Dist: prompt-toolkit (>=3.0.28,<3.0.29)
+Requires-Dist: protobuf (>=4.23.3,<4.25.4)
+Requires-Dist: psutil (>=5.9.5,<6.0.0)
+Requires-Dist: psycopg2-binary (>=2.9.9,<2.10.0)
+Requires-Dist: pycountry (>=22.3.5,<23.0.0)
+Requires-Dist: pydantic (>=2.0,<3.0)
+Requires-Dist: pydot (>=1.4,<1.5)
+Requires-Dist: pykwalify (>=1.8,<1.9)
+Requires-Dist: pymilvus (<2.4.2)
+Requires-Dist: pymongo[srv,tls] (>=4.6.3,<4.7)
+Requires-Dist: pypred (>=0.4.0,<0.5.0)
+Requires-Dist: python-dateutil (>=2.8.2,<2.9.0)
+Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: python-engineio (>=4.5.1,<6,!=5.0.0)
+Requires-Dist: python-keycloak (>=3.12.0,<4.0.0)
+Requires-Dist: python-socketio (>=5.8,<6)
+Requires-Dist: pytz (>=2022.7.1,<2023.0)
+Requires-Dist: pyyaml (>=6.0)
+Requires-Dist: qdrant-client (>=1.9.0,<2.0.0)
+Requires-Dist: questionary (>=1.10.0,<2.1.0)
+Requires-Dist: randomname (>=0.2.1,<0.3.0)
+Requires-Dist: rasa-sdk (>=3.10.0,<3.11.0)
+Requires-Dist: redis (>=4.6.0,<6.0)
+Requires-Dist: regex (>=2022.10.31,<2022.11)
+Requires-Dist: requests (>=2.31.0,<2.32.0)
+Requires-Dist: rich (>=13.4.2,<14.0.0)
+Requires-Dist: rocketchat_API (>=1.30.0,<1.31.0)
+Requires-Dist: ruamel.yaml (>=0.17.21,<0.17.22)
+Requires-Dist: safetensors (>=0.4.5,<0.5.0)
+Requires-Dist: sanic (>=22.12,<22.13)
+Requires-Dist: sanic-cors (>=2.2.0,<2.3.0)
+Requires-Dist: sanic-jwt (>=1.8.0,<2.0.0)
+Requires-Dist: sanic-routing (>=22.8.0,<23.0.0)
+Requires-Dist: scikit-learn (>=1.1.3,<1.2) ; python_version >= "3.9" and python_version < "3.11"
+Requires-Dist: scipy (>=1.10.1,<1.11.0) ; python_version >= "3.9" and python_version < "3.11"
+Requires-Dist: sentencepiece[sentencepiece] (>=0.1.99,<0.2.0) ; extra == "transformers" or extra == "full"
+Requires-Dist: sentry-sdk (>=1.14.0,<1.15.0)
+Requires-Dist: setuptools (>=70.0.0,<70.1.0)
+Requires-Dist: sklearn-crfsuite (>=0.3.6,<0.4.0)
+Requires-Dist: skops (>=0.10.0,<0.11.0)
+Requires-Dist: slack-sdk (>=3.27.1,<4.0.0)
+Requires-Dist: spacy (>=3.5.4,<4.0.0) ; extra == "spacy" or extra == "full"
+Requires-Dist: structlog (>=23.1.0,<23.2.0)
+Requires-Dist: structlog-sentry (>=2.0.3,<3.0.0)
+Requires-Dist: tarsafe (>=0.0.5,<0.0.6)
+Requires-Dist: tenacity (>=8.4.1,<8.5.0)
+Requires-Dist: tensorflow (==2.14.1) ; sys_platform != "darwin" or platform_machine != "arm64"
+Requires-Dist: tensorflow-cpu-aws (==2.14.1) ; sys_platform == "linux" and (platform_machine == "arm64" or platform_machine == "aarch64")
+Requires-Dist: tensorflow-intel (==2.14.1) ; sys_platform == "win32"
+Requires-Dist: tensorflow-io-gcs-filesystem (==0.31) ; sys_platform == "win32"
+Requires-Dist: tensorflow-io-gcs-filesystem (==0.34) ; sys_platform == "darwin" and platform_machine != "arm64"
+Requires-Dist: tensorflow-io-gcs-filesystem (==0.34) ; sys_platform == "linux"
+Requires-Dist: tensorflow-macos (==2.14.1) ; sys_platform == "darwin" and platform_machine == "arm64"
+Requires-Dist: tensorflow-metal (==1.1.0) ; (sys_platform == "darwin" and platform_machine == "arm64") and (extra == "metal")
+Requires-Dist: tensorflow-text (==2.14.0) ; sys_platform != "win32" and (platform_machine != "arm64" and platform_machine != "aarch64")
+Requires-Dist: tensorflow_hub (>=0.13.0,<0.14.0)
+Requires-Dist: terminaltables (>=3.1.10,<3.2.0)
+Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
+Requires-Dist: tqdm (>=4.66.2,<5.0.0)
+Requires-Dist: transformers (>=4.36.2,<4.37.0) ; extra == "transformers" or extra == "full"
+Requires-Dist: twilio (>=8.4,<8.5)
+Requires-Dist: types-protobuf (==4.25.0.20240417)
+Requires-Dist: typing-extensions (>=4.7.1,<5.0.0)
+Requires-Dist: typing-utils (>=0.1.0,<0.2.0)
+Requires-Dist: ujson (>=5.8,<6.0)
+Requires-Dist: webexteamssdk (>=1.6.1,<1.7.0)
+Requires-Dist: websockets (>=10.4,<11.0)
+Requires-Dist: wheel (>=0.40.0)
+Project-URL: Documentation, https://rasa.com/docs
+Project-URL: Repository, https://github.com/rasahq/rasa
+Description-Content-Type: text/markdown
+
+<h1 align="center">Rasa Pro</h1>
+
+<div align="center">
+
+[![Build Status](https://github.com/RasaHQ/rasa-private/workflows/Continuous%20Integration/badge.svg)](https://github.com/RasaHQ/rasa-private/actions)
+[![Quality Gate Status](https://sonarcloud.io/api/project_badges/measure?project=RasaHQ_rasa&metric=alert_status)](https://sonarcloud.io/summary/new_code?id=RasaHQ_rasa)
+[![Documentation Status](https://img.shields.io/badge/docs-stable-brightgreen.svg)](https://rasa.com/docs/rasa-pro/)
+
+</div>
+
+<hr />
+
+
+Rasa Pro is a framework for building scalable, dynamic conversational AI assistants that integrate large language models (LLMs) to enable more contextually aware and agentic interactions. Whether you’re new to conversational AI or an experienced developer, Rasa Pro offers enhanced flexibility, control, and performance for mission-critical applications.
+
+Building on the foundation of Rasa Open Source, Rasa Pro adds advanced features like CALM (Conversational AI with Language Models) and Dialogue Understanding (DU), which enable developers to shift from traditional intent-driven systems to LLM-based agents. This allows for more robust, responsive interactions that adhere strictly to business logic, while reducing risks like prompt injection and minimizing hallucinations.
+
+**Key Features:**
+
+- **Flows for Business Logic:** Easily define business logic through Flows, a simplified way to describe how your AI assistant should handle conversations. Flows help streamline the development process, focusing on key tasks and reducing the complexity involved in managing conversations.
+- **Automatic Conversation Repair:** Ensure seamless interactions by automatically handling interruptions or unexpected inputs. Developers have full control to customize these repairs based on specific use cases.
+- **Customizable and Open:** Fully customizable code that allows developers to modify Rasa Pro to meet specific requirements, ensuring flexibility and adaptability to various conversational AI needs.
+- **Robustness and Control:** Maintain strict adherence to business logic, preventing unwanted behaviors like prompt injection and hallucinations, leading to more reliable responses and secure interactions.
+- **Built-in Security:** Safeguard sensitive data, control access, and ensure secure deployment, essential for production environments that demand high levels of security and compliance.
+
+
+
+A [free developer license](https://rasa.com/docs/rasa-pro/developer-edition/) is available so you can explore and get to know Rasa Pro. For small production deployments, the Extended Developer License allows you to take your assistant live in a limited capacity. A paid license is required for larger-scale production use, but all code is visible and can be customized as needed.
+
+To get started right now, you can 
+
+`pip install rasa-pro` 
+
+Check out our 
+
+- [Rasa-pro Quickstart](https://rasa.com/docs/rasa-pro/installation/quickstart/),
+- [Conversational AI with Language Models (CALM) conceptual rundown](https://rasa.com/docs/rasa-pro/calm/),
+- [Rasa Pro / CALM tutorial](https://rasa.com/docs/rasa-pro/tutorial), and
+- [Rasa pro changelog](https://rasa.com/docs/rasa/rasa-pro-changelog/)
+
+for more. Also feel free to reach out to us on the [Rasa forum](https://forum.rasa.com/).
+

rasa_pro-3.10.13/README.md

@@ -0,0 +1,41 @@
+<h1 align="center">Rasa Pro</h1>
+
+<div align="center">
+
+[![Build Status](https://github.com/RasaHQ/rasa-private/workflows/Continuous%20Integration/badge.svg)](https://github.com/RasaHQ/rasa-private/actions)
+[![Quality Gate Status](https://sonarcloud.io/api/project_badges/measure?project=RasaHQ_rasa&metric=alert_status)](https://sonarcloud.io/summary/new_code?id=RasaHQ_rasa)
+[![Documentation Status](https://img.shields.io/badge/docs-stable-brightgreen.svg)](https://rasa.com/docs/rasa-pro/)
+
+</div>
+
+<hr />
+
+
+Rasa Pro is a framework for building scalable, dynamic conversational AI assistants that integrate large language models (LLMs) to enable more contextually aware and agentic interactions. Whether you’re new to conversational AI or an experienced developer, Rasa Pro offers enhanced flexibility, control, and performance for mission-critical applications.
+
+Building on the foundation of Rasa Open Source, Rasa Pro adds advanced features like CALM (Conversational AI with Language Models) and Dialogue Understanding (DU), which enable developers to shift from traditional intent-driven systems to LLM-based agents. This allows for more robust, responsive interactions that adhere strictly to business logic, while reducing risks like prompt injection and minimizing hallucinations.
+
+**Key Features:**
+
+- **Flows for Business Logic:** Easily define business logic through Flows, a simplified way to describe how your AI assistant should handle conversations. Flows help streamline the development process, focusing on key tasks and reducing the complexity involved in managing conversations.
+- **Automatic Conversation Repair:** Ensure seamless interactions by automatically handling interruptions or unexpected inputs. Developers have full control to customize these repairs based on specific use cases.
+- **Customizable and Open:** Fully customizable code that allows developers to modify Rasa Pro to meet specific requirements, ensuring flexibility and adaptability to various conversational AI needs.
+- **Robustness and Control:** Maintain strict adherence to business logic, preventing unwanted behaviors like prompt injection and hallucinations, leading to more reliable responses and secure interactions.
+- **Built-in Security:** Safeguard sensitive data, control access, and ensure secure deployment, essential for production environments that demand high levels of security and compliance.
+
+
+
+A [free developer license](https://rasa.com/docs/rasa-pro/developer-edition/) is available so you can explore and get to know Rasa Pro. For small production deployments, the Extended Developer License allows you to take your assistant live in a limited capacity. A paid license is required for larger-scale production use, but all code is visible and can be customized as needed.
+
+To get started right now, you can 
+
+`pip install rasa-pro` 
+
+Check out our 
+
+- [Rasa-pro Quickstart](https://rasa.com/docs/rasa-pro/installation/quickstart/),
+- [Conversational AI with Language Models (CALM) conceptual rundown](https://rasa.com/docs/rasa-pro/calm/),
+- [Rasa Pro / CALM tutorial](https://rasa.com/docs/rasa-pro/tutorial), and
+- [Rasa pro changelog](https://rasa.com/docs/rasa/rasa-pro-changelog/)
+
+for more. Also feel free to reach out to us on the [Rasa forum](https://forum.rasa.com/).

{rasa_pro-3.10.11 → rasa_pro-3.10.13}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.10.11"
+version = "3.10.13"
 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>",]
@@ -120,7 +120,6 @@ sanic-cors = "~2.2.0"
 sanic-jwt = "^1.8.0"
 sanic-routing = "^22.8.0"
 websockets = ">=10.4,<11.0"
-cloudpickle = ">=2.2.1,<3.1"
 aiohttp = ">=3.9.4,<3.10"
 questionary = ">=1.10.0,<2.1.0"
 prompt-toolkit = "^3.0.28,<3.0.29"
@@ -136,7 +135,6 @@ tensorflow_hub = "^0.13.0"
 setuptools = "~70.0.0"
 ujson = ">=5.8,<6.0"
 regex = ">=2022.10.31,<2022.11"
-joblib = ">=1.2.0,<1.3.0"
 sentry-sdk = ">=1.14.0,<1.15.0"
 aio-pika = ">=8.2.3,<8.2.4"
 aiogram = ">=2.15,<2.26"
@@ -188,6 +186,8 @@ types-protobuf = "4.25.0.20240417"
 litellm = "~1.52.6"
 openpyxl = "^3.1.5"
 diskcache = "~5.6.3"
+skops = "~0.10.0"
+safetensors = "~0.4.5"
 [[tool.poetry.dependencies.tensorflow-io-gcs-filesystem]]
 version = "==0.31" # newer versions doesn't provide wheels for Windows
 markers = "sys_platform == 'win32'"

{rasa_pro-3.10.11 → rasa_pro-3.10.13}/rasa/cli/studio/studio.py

@@ -1,5 +1,5 @@
 import argparse
-from typing import List, Optional
+from typing import List, Optional, Tuple
 from urllib.parse import ParseResult, urlparse
 
 import questionary
@@ -149,7 +149,7 @@ def _configure_studio_url() -> Optional[str]:
     return studio_url
 
 
-def _get_advanced_config(studio_url: str) -> tuple:
+def _get_advanced_config(studio_url: str) -> Tuple:
     """Get the advanced configuration values for Rasa Studio."""
     keycloak_url = questionary.text(
         "Please provide your Rasa Studio Keycloak URL",
@@ -167,7 +167,7 @@ def _get_advanced_config(studio_url: str) -> tuple:
     return keycloak_url, realm_name, client_id
 
 
-def _get_default_config(studio_url: str) -> tuple:
+def _get_default_config(studio_url: str) -> Tuple:
     """Get the default configuration values for Rasa Studio."""
     keycloak_url = studio_url + "auth/"
     realm_name = DEFAULT_REALM_NAME
@@ -178,6 +178,7 @@ def _get_default_config(studio_url: str) -> tuple:
         f"Keycloak URL: {keycloak_url}, "
         f"Realm Name: '{realm_name}', "
         f"Client ID: '{client_id}'. "
+        f"SSL verification is enabled."
         f"You can use '--advanced' to configure these settings."
     )
 
@@ -185,7 +186,11 @@ def _get_default_config(studio_url: str) -> tuple:
 
 
 def _create_studio_config(
-    studio_url: str, keycloak_url: str, realm_name: str, client_id: str
+    studio_url: str,
+    keycloak_url: str,
+    realm_name: str,
+    client_id: str,
+    disable_verify: bool = False,
 ) -> StudioConfig:
     """Create a StudioConfig object with the provided parameters."""
     return StudioConfig(
@@ -193,6 +198,7 @@ def _create_studio_config(
         studio_url=studio_url + "api/graphql/",
         client_id=client_id,
         realm_name=realm_name,
+        disable_verify=disable_verify,
     )
 
 
@@ -227,19 +233,23 @@ def _configure_studio_config(args: argparse.Namespace) -> StudioConfig:
 
     # create a configuration and auth object to try to reach the studio
     studio_config = _create_studio_config(
-        studio_url, keycloak_url, realm_name, client_id
+        studio_url,
+        keycloak_url,
+        realm_name,
+        client_id,
+        disable_verify=args.disable_verify,
     )
 
-    if args.disable_verify:
+    if studio_config.disable_verify:
         rasa.shared.utils.cli.print_info(
             "Disabling SSL verification for the Rasa Studio authentication server."
         )
-        studio_auth = StudioAuth(studio_config, verify=False)
     else:
         rasa.shared.utils.cli.print_info(
             "Enabling SSL verification for the Rasa Studio authentication server."
         )
-        studio_auth = StudioAuth(studio_config, verify=True)
+
+    studio_auth = StudioAuth(studio_config)
 
     if _check_studio_auth(studio_auth):
         return studio_config

{rasa_pro-3.10.11 → rasa_pro-3.10.13}/rasa/constants.py

@@ -18,7 +18,7 @@ CONFIG_TELEMETRY_ID = "rasa_user_id"
 CONFIG_TELEMETRY_ENABLED = "enabled"
 CONFIG_TELEMETRY_DATE = "date"
 
-MINIMUM_COMPATIBLE_VERSION = "3.10.0rc1"
+MINIMUM_COMPATIBLE_VERSION = "3.10.12"
 
 GLOBAL_USER_CONFIG_PATH = os.path.expanduser("~/.config/rasa/global.yml")
 

{rasa_pro-3.10.11 → rasa_pro-3.10.13}/rasa/core/featurizers/single_state_featurizer.py

@@ -1,7 +1,8 @@
 import logging
+from typing import List, Optional, Dict, Text, Set, Any
+
 import numpy as np
 import scipy.sparse
-from typing import List, Optional, Dict, Text, Set, Any
 
 from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
 from rasa.nlu.extractors.extractor import EntityTagSpec
@@ -360,6 +361,26 @@ class SingleStateFeaturizer:
             for action in domain.action_names_or_texts
         ]
 
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "action_texts": self.action_texts,
+            "entity_tag_specs": self.entity_tag_specs,
+            "feature_states": self._default_feature_states,
+        }
+
+    @classmethod
+    def create_from_dict(
+        cls, data: Dict[str, Any]
+    ) -> Optional["SingleStateFeaturizer"]:
+        if not data:
+            return None
+
+        featurizer = SingleStateFeaturizer()
+        featurizer.action_texts = data["action_texts"]
+        featurizer._default_feature_states = data["feature_states"]
+        featurizer.entity_tag_specs = data["entity_tag_specs"]
+        return featurizer
+
 
 class IntentTokenizerSingleStateFeaturizer(SingleStateFeaturizer):
     """A SingleStateFeaturizer for use with policies that predict intent labels."""

{rasa_pro-3.10.11 → rasa_pro-3.10.13}/rasa/core/featurizers/tracker_featurizers.py

@@ -1,11 +1,9 @@
 from __future__ import annotations
-from pathlib import Path
-from collections import defaultdict
-from abc import abstractmethod
-import jsonpickle
-import logging
 
-from tqdm import tqdm
+import logging
+from abc import abstractmethod
+from collections import defaultdict
+from pathlib import Path
 from typing import (
     Tuple,
     List,
@@ -18,25 +16,30 @@ from typing import (
     Set,
     DefaultDict,
     cast,
+    Type,
+    Callable,
+    ClassVar,
 )
+
 import numpy as np
+from tqdm import tqdm
 
-from rasa.core.featurizers.single_state_featurizer import SingleStateFeaturizer
-from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
-from rasa.core.exceptions import InvalidTrackerFeaturizerUsageError
 import rasa.shared.core.trackers
 import rasa.shared.utils.io
-from rasa.shared.nlu.constants import TEXT, INTENT, ENTITIES, ACTION_NAME
-from rasa.shared.nlu.training_data.features import Features
-from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.core.domain import State, Domain
-from rasa.shared.core.events import Event, ActionExecuted, UserUttered
+from rasa.core.exceptions import InvalidTrackerFeaturizerUsageError
+from rasa.core.featurizers.precomputation import MessageContainerForCoreFeaturization
+from rasa.core.featurizers.single_state_featurizer import SingleStateFeaturizer
 from rasa.shared.core.constants import (
     USER,
     ACTION_UNLIKELY_INTENT_NAME,
     PREVIOUS_ACTION,
 )
+from rasa.shared.core.domain import State, Domain
+from rasa.shared.core.events import Event, ActionExecuted, UserUttered
+from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.exceptions import RasaException
+from rasa.shared.nlu.constants import TEXT, INTENT, ENTITIES, ACTION_NAME
+from rasa.shared.nlu.training_data.features import Features
 from rasa.utils.tensorflow.constants import LABEL_PAD_ID
 from rasa.utils.tensorflow.model_data import ragged_array_to_ndarray
 
@@ -64,6 +67,10 @@ class InvalidStory(RasaException):
 class TrackerFeaturizer:
     """Base class for actual tracker featurizers."""
 
+    # Class registry to store all subclasses
+    _registry: ClassVar[Dict[str, Type["TrackerFeaturizer"]]] = {}
+    _featurizer_type: str = "TrackerFeaturizer"
+
     def __init__(
         self, state_featurizer: Optional[SingleStateFeaturizer] = None
     ) -> None:
@@ -74,6 +81,36 @@ class TrackerFeaturizer:
         """
         self.state_featurizer = state_featurizer
 
+    @classmethod
+    def register(cls, featurizer_type: str) -> Callable:
+        """Decorator to register featurizer subclasses."""
+
+        def wrapper(subclass: Type["TrackerFeaturizer"]) -> Type["TrackerFeaturizer"]:
+            cls._registry[featurizer_type] = subclass
+            # Store the type identifier in the class for serialization
+            subclass._featurizer_type = featurizer_type
+            return subclass
+
+        return wrapper
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "TrackerFeaturizer":
+        """Create featurizer instance from dictionary."""
+        featurizer_type = data.pop("type")
+
+        if featurizer_type not in cls._registry:
+            raise ValueError(f"Unknown featurizer type: {featurizer_type}")
+
+        # Get the correct subclass and instantiate it
+        subclass = cls._registry[featurizer_type]
+        return subclass.create_from_dict(data)
+
+    @classmethod
+    @abstractmethod
+    def create_from_dict(cls, data: Dict[str, Any]) -> "TrackerFeaturizer":
+        """Each subclass must implement its own creation from dict method."""
+        pass
+
     @staticmethod
     def _create_states(
         tracker: DialogueStateTracker,
@@ -465,9 +502,7 @@ class TrackerFeaturizer:
             self.state_featurizer.entity_tag_specs = []
 
         # noinspection PyTypeChecker
-        rasa.shared.utils.io.write_text_file(
-            str(jsonpickle.encode(self)), featurizer_file
-        )
+        rasa.shared.utils.io.dump_obj_as_json_to_file(featurizer_file, self.to_dict())
 
     @staticmethod
     def load(path: Union[Text, Path]) -> Optional[TrackerFeaturizer]:
@@ -481,7 +516,17 @@ class TrackerFeaturizer:
         """
         featurizer_file = Path(path) / FEATURIZER_FILE
         if featurizer_file.is_file():
-            return jsonpickle.decode(rasa.shared.utils.io.read_file(featurizer_file))
+            data = rasa.shared.utils.io.read_json_file(featurizer_file)
+
+            if "type" not in data:
+                logger.error(
+                    f"Couldn't load featurizer for policy. "
+                    f"File '{featurizer_file}' does not contain all "
+                    f"necessary information. 'type' is missing."
+                )
+                return None
+
+            return TrackerFeaturizer.from_dict(data)
 
         logger.error(
             f"Couldn't load featurizer for policy. "
@@ -508,7 +553,16 @@ class TrackerFeaturizer:
             )
         ]
 
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "type": self.__class__._featurizer_type,
+            "state_featurizer": (
+                self.state_featurizer.to_dict() if self.state_featurizer else None
+            ),
+        }
+
 
+@TrackerFeaturizer.register("FullDialogueTrackerFeaturizer")
 class FullDialogueTrackerFeaturizer(TrackerFeaturizer):
     """Creates full dialogue training data for time distributed architectures.
 
@@ -646,7 +700,20 @@ class FullDialogueTrackerFeaturizer(TrackerFeaturizer):
 
         return trackers_as_states
 
+    def to_dict(self) -> Dict[str, Any]:
+        return super().to_dict()
 
+    @classmethod
+    def create_from_dict(cls, data: Dict[str, Any]) -> "FullDialogueTrackerFeaturizer":
+        state_featurizer = SingleStateFeaturizer.create_from_dict(
+            data["state_featurizer"]
+        )
+        return cls(
+            state_featurizer,
+        )
+
+
+@TrackerFeaturizer.register("MaxHistoryTrackerFeaturizer")
 class MaxHistoryTrackerFeaturizer(TrackerFeaturizer):
     """Truncates the tracker history into `max_history` long sequences.
 
@@ -884,7 +951,25 @@ class MaxHistoryTrackerFeaturizer(TrackerFeaturizer):
 
         return trackers_as_states
 
+    def to_dict(self) -> Dict[str, Any]:
+        data = super().to_dict()
+        data.update(
+            {
+                "remove_duplicates": self.remove_duplicates,
+                "max_history": self.max_history,
+            }
+        )
+        return data
+
+    @classmethod
+    def create_from_dict(cls, data: Dict[str, Any]) -> "MaxHistoryTrackerFeaturizer":
+        state_featurizer = SingleStateFeaturizer.create_from_dict(
+            data["state_featurizer"]
+        )
+        return cls(state_featurizer, data["max_history"], data["remove_duplicates"])
 
+
+@TrackerFeaturizer.register("IntentMaxHistoryTrackerFeaturizer")
 class IntentMaxHistoryTrackerFeaturizer(MaxHistoryTrackerFeaturizer):
     """Truncates the tracker history into `max_history` long sequences.
 
@@ -1159,6 +1244,18 @@ class IntentMaxHistoryTrackerFeaturizer(MaxHistoryTrackerFeaturizer):
 
         return trackers_as_states
 
+    def to_dict(self) -> Dict[str, Any]:
+        return super().to_dict()
+
+    @classmethod
+    def create_from_dict(
+        cls, data: Dict[str, Any]
+    ) -> "IntentMaxHistoryTrackerFeaturizer":
+        state_featurizer = SingleStateFeaturizer.create_from_dict(
+            data["state_featurizer"]
+        )
+        return cls(state_featurizer, data["max_history"], data["remove_duplicates"])
+
 
 def _is_prev_action_unlikely_intent_in_state(state: State) -> bool:
     prev_action_name = state.get(PREVIOUS_ACTION, {}).get(ACTION_NAME)