rasa-pro 3.12.21__py3-none-any.whl → 3.13.0__py3-none-any.whl

rasa/dialogue_understanding_test/du_test_schema.yml

@@ -5,12 +5,12 @@ mapping:
     sequence:
       - type: map
         mapping:
-          regex;(^[a-zA-Z_]+[a-zA-Z0-9_\-]*$):
+          regex;(^[a-zA-Z_]+[a-zA-Z0-9_]*$):
             type: "seq"
             sequence:
               - type: map
                 mapping:
-                  regex;(^[a-zA-Z_]+[a-zA-Z0-9_\-]*$):
+                  regex;(^[a-zA-Z_]+[a-zA-Z0-9_]*$):
                     type: any
 
   metadata:
@@ -129,7 +129,7 @@ mapping:
             type: "seq"
             sequence:
               - type: "str"
-                pattern: ^[a-zA-Z_]+[a-zA-Z0-9_\-]*$
+                pattern: ^[a-zA-Z_]+[a-zA-Z0-9_]*$
           metadata:
             type: "str"
             pattern: ^[a-zA-Z_]+[a-zA-Z0-9_]*$

rasa/dialogue_understanding_test/io.py

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Union
 import rich
 
 import rasa.shared.data
-from rasa.dialogue_understanding_test.command_metric_calculation import CommandMetrics
+from rasa.dialogue_understanding_test.command_metrics import CommandMetrics
 from rasa.dialogue_understanding_test.constants import SCHEMA_FILE_PATH
 from rasa.dialogue_understanding_test.du_test_case import (
     KEY_CHOICES,
@@ -13,6 +13,9 @@ from rasa.dialogue_understanding_test.du_test_case import (
     KEY_PROMPT_TOKENS,
 )
 from rasa.dialogue_understanding_test.du_test_result import (
+    KEY_COMMANDS_F1_MACRO,
+    KEY_COMMANDS_F1_MICRO,
+    KEY_COMMANDS_F1_WEIGHTED,
     DialogueUnderstandingTestSuiteResult,
     FailedTestStep,
 )
@@ -274,6 +277,7 @@ def print_test_results(
         # print failed test steps
         print_failed_cases(test_suite_result, output_prompt=output_prompt)
 
+    print_f1_summary(test_suite_result)
     print_command_summary(test_suite_result.command_metrics)
     print_latency_and_token_metrics(test_suite_result)
     print_final_line(test_suite_result)
@@ -350,11 +354,26 @@ def print_llm_output(step: FailedTestStep) -> None:
     for component, component_prompts in step.prompts.items():
         for prompt_data in component_prompts:
             if KEY_CHOICES in prompt_data:
-                rich.print("\n[red3]-- CHOICES --[/red3]")
+                rich.print(f"\n[red3]-- LLM ouptut for {component} --[/red3]")
                 rich.print(prompt_data.get(KEY_CHOICES))
                 rich.print("[red3]-------------[/red3]")
 
 
+def print_f1_summary(result: DialogueUnderstandingTestSuiteResult) -> None:
+    """Print the f1 summary."""
+    print()
+    rasa.shared.utils.cli.print_info(rasa.shared.utils.cli.pad("COMMANDS F1"))
+    rasa.shared.utils.cli.print_info(
+        f"macro           : {result.f1_score[KEY_COMMANDS_F1_MACRO]:.8f}"
+    )
+    rasa.shared.utils.cli.print_info(
+        f"micro           : {result.f1_score[KEY_COMMANDS_F1_MICRO]:.8f}"
+    )
+    rasa.shared.utils.cli.print_info(
+        f"weighted average: {result.f1_score[KEY_COMMANDS_F1_WEIGHTED]:.8f}"
+    )
+
+
 def print_command_summary(metrics: Dict[str, CommandMetrics]) -> None:
     """Print the command summary.
 
@@ -390,16 +409,24 @@ def print_latency_and_token_metrics(
     """Print the latency and token metrics."""
     print()
     rasa.shared.utils.cli.print_info(rasa.shared.utils.cli.pad("LATENCY METRICS"))
-    for key, value in result.latency_metrics.items():
-        rasa.shared.utils.cli.print_info(f"{key}: {value:.8f}")
+    for component, latency_metric in result.latency_metrics.items():
+        rasa.shared.utils.cli.print_info(f"--- {component} ---")
+        for key, value in latency_metric.items():
+            rasa.shared.utils.cli.print_info(f"{key}: {value:.8f}")
+
     rasa.shared.utils.cli.print_info(rasa.shared.utils.cli.pad("PROMPT TOKEN METRICS"))
-    for key, value in result.prompt_token_metrics.items():
-        rasa.shared.utils.cli.print_info(f"{key}: {value:.2f}")
+    for component, prompt_token_metric in result.prompt_token_metrics.items():
+        rasa.shared.utils.cli.print_info(f"--- {component} ---")
+        for key, value in prompt_token_metric.items():
+            rasa.shared.utils.cli.print_info(f"{key}: {value:.2f}")
+
     rasa.shared.utils.cli.print_info(
         rasa.shared.utils.cli.pad("COMPLETION TOKEN METRICS")
     )
-    for key, value in result.completion_token_metrics.items():
-        rasa.shared.utils.cli.print_info(f"{key}: {value:.2f}")
+    for component, completion_token_metric in result.completion_token_metrics.items():
+        rasa.shared.utils.cli.print_info(f"--- {component} ---")
+        for key, value in completion_token_metric.items():
+            rasa.shared.utils.cli.print_info(f"{key}: {value:.2f}")
 
 
 def print_final_line(test_suite_result: DialogueUnderstandingTestSuiteResult) -> None:

rasa/e2e_test/constants.py

@@ -40,7 +40,7 @@ DEFAULT_GROUNDEDNESS_PROMPT_TEMPLATE_FILE_NAME = "groundedness_prompt_template.j
 DEFAULT_ANSWER_RELEVANCE_PROMPT_TEMPLATE_FILE_NAME = (
     "answer_relevance_prompt_template.jinja2"
 )
-DEFAULT_E2E_TESTING_MODEL = "gpt-4o-mini"
+DEFAULT_E2E_TESTING_MODEL = "gpt-4.1-mini-2025-04-14"
 KEY_SCORE = "score"
 KEY_JUSTIFICATION = "justification"
 KEY_EXTRA_PARAMETERS = "extra_parameters"

rasa/e2e_test/e2e_test_runner.py

@@ -13,11 +13,11 @@ import structlog
 from tqdm import tqdm
 
 import rasa.shared.utils.io
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.channels import CollectingOutputChannel, UserMessage
 from rasa.core.constants import ACTIVE_FLOW_METADATA_KEY, STEP_ID_METADATA_KEY
 from rasa.core.exceptions import AgentNotReady
 from rasa.core.persistor import StorageType
-from rasa.core.utils import AvailableEndpoints
 from rasa.dialogue_understanding_test.du_test_case import DialogueUnderstandingTestCase
 from rasa.e2e_test.constants import TEST_CASE_NAME, TEST_FILE_NAME
 from rasa.e2e_test.e2e_config import create_llm_judge_config

rasa/e2e_test/e2e_test_schema.yml

@@ -5,12 +5,12 @@ mapping:
     sequence:
       - type: map
         mapping:
-          regex;(^[a-zA-Z_]+[a-zA-Z0-9_\-]*$):
+          regex;(^[a-zA-Z_]+[a-zA-Z0-9_]*$):
             type: "seq"
             sequence:
               - type: map
                 mapping:
-                  regex;(^[a-zA-Z_]+[a-zA-Z0-9_\-]*$):
+                  regex;(^[a-zA-Z_]+[a-zA-Z0-9_]*$):
                     type: any
 
   metadata:
@@ -129,7 +129,7 @@ mapping:
             type: "seq"
             sequence:
               - type: "str"
-                pattern: ^[a-zA-Z_]+[a-zA-Z0-9_\-]*$
+                pattern: ^[a-zA-Z_]+[a-zA-Z0-9_]*$
           metadata:
             type: "str"
             pattern: ^[a-zA-Z_]+[a-zA-Z0-9_]*$

rasa/engine/constants.py

@@ -1,7 +1,7 @@
 from typing import List, Optional
 
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.channels import UserMessage
-from rasa.core.utils import AvailableEndpoints
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.importers.importer import TrainingDataImporter
 

rasa/engine/graph.py

@@ -500,9 +500,9 @@ class GraphNode:
                 structlogger.warning(
                     "graph.node.input_not_resolved",
                     node_name=self._node_name,
-                    input_name=i,
+                    input_name=i,  # no PII
                     event_info=(
-                        "Node input was not resolved, there is no putput. "
+                        "Node input was not resolved, there is no output. "
                         "Another component should have provided this as an output."
                     ),
                 )

rasa/engine/recipes/default_recipe.py

@@ -770,7 +770,7 @@ class DefaultV1Recipe(Recipe):
     @staticmethod
     def _intentless_policy_used(nodes: Dict[Text, SchemaNode]) -> bool:
         """Checks if the intentless policy is used in the nodes."""
-        from rasa.core import IntentlessPolicy
+        from rasa.core.policies.intentless_policy import IntentlessPolicy
 
         for schema_node in nodes.values():
             if schema_node.matches_type(IntentlessPolicy):

rasa/engine/validation.py

@@ -23,9 +23,10 @@ import structlog
 import typing_utils
 
 import rasa.utils.common
-from rasa.core import ContextualResponseRephraser, IntentlessPolicy
+from rasa.core.available_endpoints import AvailableEndpoints
+from rasa.core.nlg.contextual_response_rephraser import ContextualResponseRephraser
+from rasa.core.policies.intentless_policy import IntentlessPolicy
 from rasa.core.policies.policy import PolicyPrediction
-from rasa.core.utils import AvailableEndpoints
 from rasa.dialogue_understanding.coexistence.constants import (
     CALM_ENTRY,
     NLU_ENTRY,

rasa/hooks.py

@@ -9,10 +9,9 @@ import pluggy
 #  across the codebase.
 
 if TYPE_CHECKING:
-    from rasa.anonymization.anonymization_pipeline import AnonymizationPipeline
     from rasa.cli import SubParsersAction
     from rasa.core.brokers.broker import EventBroker
-    from rasa.core.tracker_store import TrackerStore
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
     from rasa.shared.core.domain import Domain
     from rasa.utils.endpoints import EndpointConfig
 
@@ -80,7 +79,7 @@ def create_tracker_store(
     domain: "Domain",
     event_broker: Optional["EventBroker"],
 ) -> "TrackerStore":
-    from rasa.core.auth_retry_tracker_store import AuthRetryTrackerStore
+    from rasa.core.tracker_stores.auth_retry_tracker_store import AuthRetryTrackerStore
     from rasa.utils.endpoints import EndpointConfig
 
     if isinstance(endpoint_config, EndpointConfig):
@@ -88,30 +87,3 @@ def create_tracker_store(
             endpoint_config=endpoint_config, domain=domain, event_broker=event_broker
         )
     return endpoint_config
-
-
-@hookimpl  # type: ignore[misc]
-def init_anonymization_pipeline(endpoints_file: Optional[Text]) -> None:
-    """Hook implementation for initializing the anonymization pipeline."""
-    from rasa.anonymization.anonymization_pipeline import load_anonymization_pipeline
-
-    load_anonymization_pipeline(endpoints_file)
-
-
-@hookimpl  # type: ignore[misc]
-def get_anonymization_pipeline() -> Optional["AnonymizationPipeline"]:
-    """Hook implementation for getting the anonymization pipeline."""
-    from rasa.anonymization.anonymization_pipeline import AnonymizationPipelineProvider
-
-    return AnonymizationPipelineProvider().get_anonymization_pipeline()
-
-
-@hookimpl  # type: ignore[misc]
-def after_server_stop() -> None:
-    """Hook implementation for stopping the anonymization pipeline."""
-    from rasa.anonymization.anonymization_pipeline import AnonymizationPipelineProvider
-
-    anon_pipeline = AnonymizationPipelineProvider().get_anonymization_pipeline()
-
-    if anon_pipeline is not None:
-        anon_pipeline.stop()

rasa/llm_fine_tuning/paraphrasing/conversation_rephraser.py

@@ -22,11 +22,7 @@ from rasa.shared.constants import (
 from rasa.shared.exceptions import ProviderClientAPIException
 from rasa.shared.providers.mappings import OPENAI_PROVIDER
 from rasa.shared.utils.constants import LOG_COMPONENT_SOURCE_METHOD_INIT
-from rasa.shared.utils.llm import (
-    USER,
-    get_prompt_template,
-    llm_factory,
-)
+from rasa.shared.utils.llm import USER, get_prompt_template, llm_factory
 
 SEPARATOR = "\n\n"
 BACKUP_SEPARATOR = "\nUSER:"
@@ -39,7 +35,7 @@ DEFAULT_REPHRASING_PROMPT_TEMPLATE = importlib.resources.read_text(
 
 DEFAULT_LLM_CONFIG = {
     PROVIDER_CONFIG_KEY: OPENAI_PROVIDER,
-    MODEL_CONFIG_KEY: "gpt-4o-mini",
+    MODEL_CONFIG_KEY: "gpt-4.1-mini-2025-04-14",
     TIMEOUT_CONFIG_KEY: 7,
     TEMPERATURE_CONFIG_KEY: 0.0,
     MAX_COMPLETION_TOKENS_CONFIG_KEY: 4096,

rasa/model_manager/model_api.py

@@ -7,12 +7,16 @@ from typing import Any, Callable, Dict, Optional, Union
 import dotenv
 import psutil
 import structlog
+from ruamel.yaml import YAMLError
 from sanic import Blueprint, Sanic, response
 from sanic.exceptions import NotFound
 from sanic.request import Request
 from sanic.response import json
 from socketio import AsyncServer
 
+import rasa
+from rasa.cli.project_templates.defaults import get_rasa_defaults
+from rasa.cli.scaffold import ProjectTemplateName, scaffold_path
 from rasa.constants import MODEL_ARCHIVE_EXTENSION
 from rasa.exceptions import ModelNotFound
 from rasa.model_manager import config
@@ -45,6 +49,10 @@ from rasa.model_manager.warm_rasa_process import (
     initialize_warm_rasa_process,
     shutdown_warm_rasa_processes,
 )
+from rasa.server import ErrorResponse
+from rasa.shared.exceptions import InvalidConfigException
+from rasa.shared.utils.yaml import dump_obj_as_yaml_to_string
+from rasa.studio.upload import build_calm_import_parts
 
 dotenv.load_dotenv()
 
@@ -476,6 +484,86 @@ def internal_blueprint() -> Blueprint:
         except ModelNotFound:
             return response.raw(b"", status=404)
 
+    @bp.post("/defaults")
+    async def get_defaults(request: Request) -> response.HTTPResponse:
+        """Returns the system defaults like prompts, patterns, etc."""
+        body = request.json or {}
+        config_yaml = body.get("config")
+        if config_yaml is None:
+            exc = ErrorResponse(
+                HTTPStatus.BAD_REQUEST,
+                "BadRequest",
+                "Missing `config` key in request body.",
+            )
+            return response.json(exc.error_info, status=exc.status)
+
+        endpoints_yaml = body.get("endpoints")
+        if endpoints_yaml is None:
+            exc = ErrorResponse(
+                HTTPStatus.BAD_REQUEST,
+                "BadRequest",
+                "Missing `endpoints` key in request body.",
+            )
+            return response.json(exc.error_info, status=exc.status)
+
+        try:
+            defaults = get_rasa_defaults(config_yaml, endpoints_yaml)
+        except (YAMLError, InvalidConfigException) as e:
+            exc = ErrorResponse(
+                HTTPStatus.INTERNAL_SERVER_ERROR,
+                "InitDataError",
+                f"Failed to load defaults. Error: {e!s}",
+            )
+            return response.json(exc.error_info, status=exc.status)
+        return response.json(defaults.model_dump(exclude_none=True))
+
+    @bp.get("/project_template")
+    async def get_project_template(request: Request) -> response.HTTPResponse:
+        """Return initial project template data."""
+        template = request.args.get("template", ProjectTemplateName.DEFAULT.value)
+
+        try:
+            template_enum = ProjectTemplateName(template)
+        except ValueError:
+            valid_templates = ", ".join([t.value for t in ProjectTemplateName])
+            exc = ErrorResponse(
+                HTTPStatus.BAD_REQUEST,
+                "BadRequest",
+                f"Unknown template '{template}'. Valid templates: "
+                f"{valid_templates}",
+            )
+            return response.json(exc.error_info, status=exc.status)
+
+        template_dir = scaffold_path(template_enum)
+        if not os.path.isdir(template_dir):
+            exc = ErrorResponse(
+                HTTPStatus.INTERNAL_SERVER_ERROR,
+                "InitDataError",
+                f"Template directory '{template_dir}' not found.",
+            )
+            return response.json(exc.error_info, status=exc.status)
+
+        assistant_name, parts = build_calm_import_parts(
+            data_path=f"{template_dir}/data",
+            domain_path=f"{template_dir}/domain",
+            config_path=f"{template_dir}/config.yml",
+            endpoints_path=f"{template_dir}/endpoints.yml",
+            assistant_name=template_enum.value,
+        )
+
+        defaults = get_rasa_defaults(
+            config_yaml=dump_obj_as_yaml_to_string(parts.config),
+            endpoints_yaml=dump_obj_as_yaml_to_string(parts.endpoints),
+        )
+        return response.json(
+            {
+                **parts.model_dump(exclude_none=True),
+                "assistantName": assistant_name,
+                "defaults": defaults.model_dump(exclude_none=True),
+                "version": rasa.__version__,
+            }
+        )
+
     return bp
 
 
@@ -483,7 +571,7 @@ def external_blueprint() -> Blueprint:
     """Create a blueprint for the model manager API."""
     from rasa.core.channels.socketio import SocketBlueprint
 
-    sio = AsyncServer(async_mode="sanic", cors_allowed_origins="*")
+    sio = AsyncServer(async_mode="sanic", cors_allowed_origins=[])
     bp = SocketBlueprint(sio, "", "model_api_external")
 
     create_bridge_server(sio, running_bots)

rasa/model_manager/runner_service.py

@@ -2,7 +2,8 @@ import os
 import shutil
 import subprocess
 from enum import Enum
-from typing import Dict, Optional
+from pathlib import Path
+from typing import Dict, Optional, Union
 
 import aiohttp
 import structlog
@@ -18,6 +19,7 @@ from rasa.model_manager.utils import (
     write_encoded_data_to_file,
 )
 from rasa.model_manager.warm_rasa_process import start_rasa_process
+from rasa.studio.prompts import handle_prompts
 
 structlogger = structlog.get_logger()
 
@@ -121,11 +123,25 @@ def get_open_port() -> int:
 
 
 def write_encoded_config_data_to_files(
-    encoded_configs: Dict[str, bytes], base_path: str
+    encoded_configs: Dict[str, Union[bytes, Dict[str, str]]], base_path: str
 ) -> None:
     """Write the encoded config data to files."""
-    for key, value in encoded_configs.items():
-        write_encoded_data_to_file(value, subpath(base_path, f"{key}.yml"))
+    endpoints_encoded = encoded_configs.get("endpoints")
+    if endpoints_encoded:
+        write_encoded_data_to_file(
+            endpoints_encoded, subpath(base_path, "endpoints.yml")
+        )
+    config_encoded = encoded_configs.get("config")
+    if config_encoded:
+        write_encoded_data_to_file(config_encoded, subpath(base_path, "config.yml"))
+    credentials_encoded = encoded_configs.get("credentials")
+    if credentials_encoded:
+        write_encoded_data_to_file(
+            credentials_encoded, subpath(base_path, "credentials.yml")
+        )
+
+    if prompts := encoded_configs.get("prompts"):
+        handle_prompts(prompts, Path(base_path))
 
 
 def prepare_bot_directory(

rasa/model_manager/socket_bridge.py

@@ -131,13 +131,6 @@ async def create_bridge_client(
         structlogger.debug("model_runner.bot_message", deployment_id=deployment_id)
         await sio.emit("bot_message", data, room=sid)
 
-    @client.event  # type: ignore[misc]
-    async def error(data: Dict[str, Any]) -> None:
-        structlogger.debug(
-            "model_runner.bot_error", deployment_id=deployment_id, data=data
-        )
-        await sio.emit("error", data, room=sid)
-
     @client.event  # type: ignore[misc]
     async def tracker(data: Dict[str, Any]) -> None:
         await sio.emit("tracker", json.loads(data), room=sid)

rasa/model_manager/trainer_service.py

@@ -2,12 +2,13 @@ import os
 import shutil
 import subprocess
 from enum import Enum
+from pathlib import Path
 from typing import Any, Dict, Optional
 
 import structlog
 from pydantic import BaseModel, ConfigDict
 
-from rasa.constants import MODEL_ARCHIVE_EXTENSION
+from rasa.constants import MODEL_ARCHIVE_EXTENSION, RASA_DIR_NAME
 from rasa.model_manager import config
 from rasa.model_manager.utils import (
     ensure_base_directory_exists,
@@ -20,6 +21,7 @@ from rasa.model_manager.warm_rasa_process import (
     start_rasa_process,
 )
 from rasa.model_training import generate_random_model_name
+from rasa.studio.prompts import handle_prompts
 
 structlogger = structlog.get_logger()
 
@@ -171,7 +173,7 @@ def seed_training_directory_with_rasa_cache(
             training_base_path=training_base_path,
         )
         # copy the cache to the training directory
-        shutil.copytree(src=cache_path, dst=subpath(training_base_path, ".rasa"))
+        shutil.copytree(src=cache_path, dst=subpath(training_base_path, RASA_DIR_NAME))
 
 
 def persist_rasa_cache(assistant_id: str, training_base_path: str) -> None:
@@ -184,12 +186,12 @@ def persist_rasa_cache(assistant_id: str, training_base_path: str) -> None:
     cache_path = cache_for_assistant_path(assistant_id)
 
     # if the training failed and didn't create a cache, skip this step
-    if not os.path.exists(subpath(training_base_path, ".rasa")):
+    if not os.path.exists(subpath(training_base_path, RASA_DIR_NAME)):
         return
 
     # clean up the cache directory first
     shutil.rmtree(cache_path, ignore_errors=True)
-    shutil.copytree(src=subpath(training_base_path, ".rasa"), dst=cache_path)
+    shutil.copytree(src=subpath(training_base_path, RASA_DIR_NAME), dst=cache_path)
 
 
 def write_training_data_to_files(
@@ -208,6 +210,7 @@ def write_training_data_to_files(
         "stories": "base64 encoded stories.yml",
         "rules": "base64 encoded rules.yml",
         "nlu": "base64 encoded nlu.yml"
+        "prompts": "dictionary with the prompts",
     }
     ```
     """
@@ -230,6 +233,9 @@ def write_training_data_to_files(
             subpath(training_base_path + "/" + parent_path, file_name),
         )
 
+    if prompts := encoded_training_data.get("prompts"):
+        handle_prompts(prompts, Path(training_base_path))
+
 
 def prepare_training_directory(
     training_base_path: str, assistant_id: str, encoded_training_data: Dict[str, Any]

rasa/plugin.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import argparse
 import functools
 import sys
-from typing import TYPE_CHECKING, Any, List, Optional, Text, Union
+from typing import TYPE_CHECKING, List, Optional, Text, Union
 
 import pluggy
 
@@ -11,7 +11,7 @@ from rasa.cli import SubParsersAction
 
 if TYPE_CHECKING:
     from rasa.core.brokers.broker import EventBroker
-    from rasa.core.tracker_store import TrackerStore
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
     from rasa.shared.core.domain import Domain
     from rasa.shared.core.trackers import DialogueStateTracker
     from rasa.utils.endpoints import EndpointConfig
@@ -32,11 +32,8 @@ def plugin_manager() -> pluggy.PluginManager:
 
 def init_hooks(manager: pluggy.PluginManager) -> None:
     """Initialise hooks into rasa."""
-    import rasa.utils.licensing
     from rasa import hooks
 
-    rasa.utils.licensing.validate_license_from_env()
-
     manager.register(hooks)
 
 
@@ -72,16 +69,6 @@ def create_tracker_store(  # type: ignore[empty-body]
     """Hook specification for wrapping with AuthRetryTrackerStore."""
 
 
-@hookspec(firstresult=True)  # type: ignore[misc]
-def init_anonymization_pipeline(endpoints_file: Optional[Text]) -> None:
-    """Hook specification for initialising the anonymization pipeline."""
-
-
-@hookspec(firstresult=True)  # type: ignore[misc]
-def get_anonymization_pipeline() -> Optional[Any]:
-    """Hook specification for getting the anonymization pipeline."""
-
-
 @hookspec  # type: ignore[misc]
 def after_server_stop() -> None:
     """Hook specification for stopping the server.

rasa/privacy/constants.py

@@ -0,0 +1,83 @@
+PRIVACY_CONFIG_SCHEMA = "privacy/privacy_config_schema.json"
+REDACTION_CHAR_KEY = "redaction_char"
+KEEP_LEFT_KEY = "keep_left"
+KEEP_RIGHT_KEY = "keep_right"
+DELETION_KEY = "deletion"
+ANONYMIZATION_KEY = "anonymization"
+TRACKER_STORE_SETTINGS = "tracker_store_settings"
+SLOT_KEY = "slot"
+TEXT_KEY = "text"
+ENTITIES_KEY = "entities"
+VALUE_KEY = "value"
+ENTITY_LABEL_KEY = "label"
+
+USER_CHAT_INACTIVITY_IN_MINUTES_ENV_VAR_NAME = "USER_CHAT_INACTIVITY_IN_MINUTES"
+GLINER_MODEL_PATH_ENV_VAR_NAME = "GLINER_MODEL_PATH"
+HUGGINGFACE_CACHE_DIR_ENV_VAR_NAME = "HUGGINGFACE_HUB_CACHE_DIR"
+
+DEFAULT_PII_MODEL = "urchade/gliner_multi_pii-v1"
+GLINER_LABELS = [
+    "person",
+    "organization",
+    "company",
+    "phone number",
+    "address",
+    "full address",
+    "postcode",
+    "zip code",
+    "passport number",
+    "email",
+    "credit card number",
+    "social security number",
+    "health insurance id number",
+    "date of birth",
+    "mobile phone number",
+    "bank account number",
+    "medication",
+    "cpf",
+    "driver's license number",
+    "tax identification number",
+    "medical condition",
+    "identity card number",
+    "national id number",
+    "ip address",
+    "email address",
+    "iban",
+    "credit card expiration date",
+    "username",
+    "health insurance number",
+    "registration number",
+    "student id number",
+    "insurance number",
+    "membership number",
+    "booking number",
+    "landline phone number",
+    "blood type",
+    "cvv",
+    "reservation number",
+    "digital signature",
+    "social media handle",
+    "license plate number",
+    "cnpj",
+    "postal code",
+    "passport_number",
+    "serial number",
+    "vehicle registration number",
+    "fax number",
+    "visa number",
+    "insurance company",
+    "identity document number",
+    "transaction number",
+    "national health insurance number",
+    "cvc",
+    "birth certificate number",
+    "train ticket number",
+    "passport expiration date",
+    "social_security_number",
+    "personally identifiable information",
+    "banking routing number",
+    "sort code",
+    "routing number",
+    "tax number",
+    "swift code",
+]