rasa-pro 3.13.1a17__py3-none-any.whl → 3.13.2__py3-none-any.whl

rasa/builder/skill_to_bot_prompt.jinja2

@@ -1,164 +0,0 @@
-# Rasa CALM Flow and Domain YAML Generator
-
-You are an expert in creating Rasa CALM flows and domain JSON configurations. Your task is to generate these JSON files based on a user's description of a conversational skill. The user may not be familiar with Rasa, so it's crucial to interpret their requirements accurately and create a well-structured, functional Rasa configuration.
-
-## Input
-You will receive a description of a conversational skill. This description will be inserted where you see [USER_SKILL_DESCRIPTION] in this prompt.
-
-## Output
-Generate a JSON configurations that includes:
-1. A Rasa CALM flow JSON
-2. A corresponding Rasa domain JSON
-
-Ensure that both parts of the JSONs are complete, well-structured, and compatible with each other.
-
-## Example
-Here's an example of a skill description and the corresponding JSON outputs and custom action code
-
-Skill Description: "Create a skill for transferring money. It should ask for the recipient and amount, check the user has enough balance, and then confirm with the user before finalizing the transfer."
-
-CALM BOT JSON:
-```json
-{
-  "flows": {
-    "transfer_money": {
-      "description": "This flow lets users send money to friends and family.",
-      "steps": [
-        {
-          "collect": "recipient"
-        },
-        {
-          "collect": "amount",
-          "description": "the number of US dollars to send"
-        },
-        {
-          "action": "action_check_sufficient_funds",
-          "next": [
-            {
-              "if": "not slots.has_sufficient_funds",
-              "then": [
-                {
-                  "action": "utter_insufficient_funds",
-                  "next": "END"
-                }
-              ]
-            },
-            {
-              "else": "final_confirmation"
-            }
-          ]
-        },
-        {
-          "collect": "final_confirmation",
-          "id": "final_confirmation",
-          "next": [
-            {
-              "if": "not slots.final_confirmation",
-              "then": [
-                {
-                  "action": "utter_transfer_cancelled",
-                  "next": "END"
-                }
-              ]
-            },
-            {
-              "else": "transfer_successful"
-            }
-          ]
-        },
-        {
-          "action": "utter_transfer_complete",
-          "id": "transfer_successful"
-        }
-      ]
-    }
-  },
-  "domain": {
-    "actions": [
-      "action_check_sufficient_funds"
-    ],
-    "slots": {
-      "recipient": {
-        "type": "Text",
-        "mappings": [
-          {
-            "type": "from_llm"
-          }
-        ]
-      },
-      "has_sufficient_funds": {
-        "type": "bool",
-        "mappings": [
-          {
-            "type": "custom"
-          }
-        ]
-      }
-    },
-    "responses": {
-      "utter_ask_recipient": [
-        {
-          "text": "Who would you like to send money to?"
-        }
-      ],
-      "utter_ask_final_confirmation": [
-        {
-          "text": "Please confirm: you want to transfer {amount} to {recipient}?"
-        }
-      ],
-      "utter_transfer_cancelled": [
-        {
-          "text": "Your transfer has been cancelled."
-        }
-      ],
-      "utter_insufficient_funds": [
-        {
-          "text": "You do not have enough funds to make this transaction."
-        }
-      ]
-    }
-  }
-}
-```
-
-## Guidelines for CALM Flow JSON:
-- Start a `flow` with the name of the flow as the key in `flows`
-- Include a `description` for the skill
-- Use `steps` to outline the conversation flow
-- Implement `collect` steps for gathering information
-- `collect` steps should reference an existing slot
-- For a collect step referencing a slot `A`, there should be a corresponding `utter_ask_A`
-  utterance that is used in the collect step to ask for the information to be stored in `A`
-- Use `action` steps for custom actions
-- Implement conditional logic with `if`, `then`, and `else` where appropriate
-- Use `next` to define the flow between steps. If the flow should end after a step, add next: END.
-- The content after `then` or `else` can be: the id of another step defined in the flow, a list of steps, or an END
-- End the flow with an appropriate action or message
-
-## Guidelines for Domain JSON:
-- Include all necessary `actions`
-- Define all required `slots` with appropriate types. Type should be one of 'float', 'bool', 'text', or 'categorical'
-- Slots filled by a 'collect' step should have mapping 'from_llm' and slots set by custom actions should have a 'custom' mapping
-- Provide `responses` for all bot turns
-
-## Guidelines for Custom Actions
-- Ensure all actions mentioned in the flow are properly defined in the domains list of actions.
-
-## Important Notes:
-- Ensure that the flow logic is coherent and follows a natural conversation pattern
-- All custom actions in the flow should be listed in the domain's `actions` section
-- All slots collected in the flow should be defined in the domain's `slots` section
-- Provide appropriate `utter_` responses for each user interaction
-- Use realistic and appropriate names for actions, slots, and responses
-- Consider error handling and alternative conversation paths
-- Aim for a balance between simplicity and functionality
-
-Now, please generate Rasa CALM flow and domain JSON based on the initial bot data.
-Modify the initial bot data where it makes sense based on the users description
-of the skill (e.g. the initial bot greeting utter_greet).
-
-INITIAL BOT DATA:
-{{project_data|tojson}}
-
-USER_SKILL_DESCRIPTION:
-{{skill_description}}

rasa/builder/training_service.py

@@ -1,132 +0,0 @@
-"""Functions for training and loading Rasa models."""
-
-import os
-import tempfile
-
-import structlog
-
-from rasa.builder import config
-from rasa.builder.exceptions import AgentLoadError, TrainingError
-from rasa.core import agent
-from rasa.core.utils import AvailableEndpoints, read_endpoints_from_path
-from rasa.model_training import TrainingResult, train
-from rasa.shared.importers.importer import TrainingDataImporter
-from rasa.shared.utils.yaml import dump_obj_as_yaml_to_string
-
-structlogger = structlog.get_logger()
-
-
-async def train_and_load_agent(importer: TrainingDataImporter) -> agent.Agent:
-    """Train a model and load an agent.
-
-    Args:
-        importer: Training data importer with domain, flows, and config
-
-    Returns:
-        Loaded and ready agent
-
-    Raises:
-        TrainingError: If training fails
-        AgentLoadError: If agent loading fails
-    """
-    try:
-        # Setup endpoints for training validation
-        await _setup_endpoints()
-
-        # Train the model
-        training_result = await _train_model(importer)
-
-        # Load the agent
-        agent_instance = await _load_agent(training_result.model)
-
-        # Verify agent is ready
-        if not agent_instance.is_ready():
-            raise AgentLoadError("Agent failed to load properly - model is not ready")
-
-        structlogger.info("training.agent_ready", model_path=training_result.model)
-
-        return agent_instance
-
-    except (TrainingError, AgentLoadError):
-        raise
-    except Exception as e:
-        raise TrainingError(f"Unexpected error during training: {e}")
-    except SystemExit as e:
-        raise TrainingError(f"SystemExit during training: {e}")
-
-
-async def _setup_endpoints() -> None:
-    """Setup endpoints configuration for training."""
-    try:
-        with tempfile.NamedTemporaryFile(
-            mode="w", suffix=".yml", delete=False
-        ) as temp_file:
-            endpoints_yaml = dump_obj_as_yaml_to_string(config.get_default_endpoints())
-            temp_file.write(endpoints_yaml)
-            temp_file.flush()
-
-            # Reset and load endpoints
-            AvailableEndpoints.reset_instance()
-            read_endpoints_from_path(temp_file.name)
-
-            structlogger.debug("training.endpoints_setup", temp_file=temp_file.name)
-
-    except Exception as e:
-        raise TrainingError(f"Failed to setup endpoints: {e}")
-
-
-async def _train_model(importer: TrainingDataImporter) -> TrainingResult:
-    """Train the Rasa model."""
-    try:
-        structlogger.info("training.started")
-
-        training_result = await train(
-            domain="",
-            config="",
-            training_files=None,
-            file_importer=importer,
-        )
-
-        if not training_result or not training_result.model:
-            raise TrainingError("Training completed but no model was produced")
-
-        structlogger.info("training.completed", model_path=training_result.model)
-
-        return training_result
-
-    except Exception as e:
-        raise TrainingError(f"Model training failed: {e}")
-
-
-async def _load_agent(model_path: str) -> agent.Agent:
-    """Load the trained agent."""
-    try:
-        structlogger.info("training.loading_agent", model_path=model_path)
-
-        available_endpoints = AvailableEndpoints.get_instance()
-        if available_endpoints is None:
-            raise AgentLoadError("No endpoints available for agent loading")
-
-        structlogger.debug(
-            "training.loading_agent.cwd",
-            cwd=os.getcwd(),
-            model_path=model_path,
-        )
-
-        agent_instance = await agent.load_agent(
-            model_path=model_path,
-            remote_storage=None,
-            endpoints=available_endpoints,
-        )
-
-        if agent_instance is None:
-            raise AgentLoadError("Agent loading returned None")
-
-        structlogger.info("training.agent_loaded", model_path=model_path)
-
-        return agent_instance
-
-    except AgentLoadError:
-        raise
-    except Exception as e:
-        raise AgentLoadError(f"Failed to load agent: {e}")

rasa/builder/validation_service.py

@@ -1,93 +0,0 @@
-"""Functions for validating Rasa projects."""
-
-import sys
-from contextlib import contextmanager
-from typing import Optional
-
-import structlog
-from structlog.testing import capture_logs
-
-from rasa.builder import config
-from rasa.builder.exceptions import ValidationError
-from rasa.cli.utils import validate_files
-from rasa.shared.importers.importer import TrainingDataImporter
-
-structlogger = structlog.get_logger()
-
-
-@contextmanager
-def _mock_sys_exit():
-    """Context manager to prevent sys.exit from being called during validation."""
-    was_sys_exit_called = {"value": False}
-
-    def sys_exit_mock(code: int = 0):
-        was_sys_exit_called["value"] = True
-
-    original_exit = sys.exit
-    sys.exit = sys_exit_mock
-
-    try:
-        yield was_sys_exit_called
-    finally:
-        sys.exit = original_exit
-
-
-async def validate_project(importer: TrainingDataImporter) -> Optional[str]:
-    """Validate a Rasa project.
-
-    Args:
-        importer: Training data importer with domain, flows, and config
-
-    Returns:
-        None if validation passes, error message if validation fails.
-
-    Raises:
-        ValidationError: If validation fails
-    """
-    try:
-        with _mock_sys_exit() as exit_tracker:
-            with capture_logs() as cap_logs:
-                validate_files(
-                    fail_on_warnings=config.VALIDATION_FAIL_ON_WARNINGS,
-                    max_history=config.VALIDATION_MAX_HISTORY,
-                    importer=importer,
-                )
-
-            if exit_tracker["value"]:
-                error_logs = [
-                    log for log in cap_logs if log.get("log_level") != "debug"
-                ]
-                structlogger.error(
-                    "validation.failed.sys_exit",
-                    error_logs=error_logs,
-                )
-                raise ValidationError(
-                    "Validation failed with sys.exit", validation_logs=error_logs
-                )
-
-            structlogger.info("validation.success")
-            return None
-
-    except ValidationError:
-        raise
-
-    except Exception as e:
-        error_msg = f"Validation failed with exception: {e}"
-
-        error_logs = [log for log in cap_logs if log.get("log_level") != "debug"]
-
-        structlogger.error(
-            "validation.failed.exception", error=str(e), validation_logs=error_logs
-        )
-        raise ValidationError(error_msg, validation_logs=error_logs)
-
-    except SystemExit as e:
-        error_logs = [log for log in cap_logs if log.get("log_level") != "debug"]
-
-        structlogger.error(
-            "validation.failed.sys_exit",
-            error_logs=error_logs,
-        )
-        raise ValidationError(
-            f"SystemExit during validation: {e}", validation_logs=error_logs
-        )

rasa/cli/project_templates/finance/actions/action_add_payee.py

@@ -1,47 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.events import SlotSet
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionAddPayee(Action):
-    def name(self) -> Text:
-        return "action_add_payee"
-
-    def run(
-        self,
-        dispatcher: CollectingDispatcher,
-        tracker: Tracker,
-        domain: Dict[Text, Any],
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-        payee_name = tracker.get_slot("payee_name")
-        account_number = tracker.get_slot("account_number")
-        sort_code = tracker.get_slot("sort_code")
-        payee_type = tracker.get_slot("payee_type")
-        reference = tracker.get_slot("reference") or ""
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Add the payee
-        success = db.add_payee(
-            user["id"], payee_name, sort_code, account_number, payee_type, reference
-        )
-
-        if success:
-            dispatcher.utter_message(
-                text=f"Payee {payee_name} has been added successfully."
-            )
-            return [SlotSet("payee_added", True)]
-        else:
-            dispatcher.utter_message(text="Failed to add payee. Please try again.")
-            return [SlotSet("payee_added", False)]

rasa/cli/project_templates/finance/actions/action_ask_account.py

@@ -1,50 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.events import SlotSet
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionAskAccount(Action):
-    def name(self) -> Text:
-        return "action_ask_account"
-
-    def run(
-        self,
-        dispatcher: CollectingDispatcher,
-        tracker: Tracker,
-        domain: Dict[Text, Any],
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Get all accounts for the user
-        accounts = db.get_accounts_by_user(user["id"])
-
-        if not accounts:
-            dispatcher.utter_message(text="No accounts found for this user.")
-            return []
-
-        buttons = [
-            {
-                "content_type": "text",
-                "title": f"{account['number']} ({account['type'].title()})",
-                "payload": str(account["number"]),
-            }
-            for account in accounts
-        ]
-        message = "Which account would you like the balance for?"
-        dispatcher.utter_message(text=message, buttons=buttons)
-
-        selected_account = tracker.get_slot("account")
-
-        return [SlotSet("account", selected_account)]

rasa/cli/project_templates/finance/actions/action_ask_account_from.py

@@ -1,50 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionAskAccountFrom(Action):
-    def name(self) -> Text:
-        return "action_ask_account_from"
-
-    def run(
-        self,
-        dispatcher: CollectingDispatcher,
-        tracker: Tracker,
-        domain: Dict[Text, Any],
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Get all accounts for the user
-        accounts = db.get_accounts_by_user(user["id"])
-
-        if not accounts:
-            dispatcher.utter_message(text="No accounts found for this user.")
-            return []
-
-        buttons = [
-            {
-                "content_type": "text",
-                "title": (
-                    f"{account['type'].title()} (Balance: "
-                    f"${float(account['balance']):.2f})"
-                ),
-                "payload": str(account["number"]),
-            }
-            for account in accounts
-        ]
-        message = "Which account would you like to transfer money from?"
-        dispatcher.utter_message(text=message, buttons=buttons)
-
-        return []

rasa/cli/project_templates/finance/actions/action_ask_card.py

@@ -1,47 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionAskCard(Action):
-    def name(self) -> Text:
-        return "action_ask_card"
-
-    def run(
-        self,
-        dispatcher: CollectingDispatcher,
-        tracker: Tracker,
-        domain: Dict[Text, Any],
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Get all cards for the user
-        cards = db.get_cards_by_user(user["id"])
-
-        if not cards:
-            dispatcher.utter_message(text="No cards found for this user.")
-            return []
-
-        buttons = [
-            {
-                "content_type": "text",
-                "title": f"{i + 1}: x{card['number'][-4:]} ({card['type'].title()})",
-                "payload": f"/SetSlots(card_selection={card['number']!s})",
-            }
-            for i, card in enumerate(cards)
-        ]
-        message = "Select the card you require assistance with:"
-        dispatcher.utter_message(text=message, buttons=buttons)
-
-        return []

rasa/cli/project_templates/finance/actions/action_check_balance.py

@@ -1,40 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionCheckBalance(Action):
-    def name(self) -> Text:
-        return "action_check_balance"
-
-    def run(
-        self,
-        dispatcher: CollectingDispatcher,
-        tracker: Tracker,
-        domain: Dict[Text, Any],
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-        account_number = tracker.get_slot("account")
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Get account information
-        account = db.get_account_by_user_and_number(user["id"], account_number)
-        if not account:
-            dispatcher.utter_message(text="Account not found.")
-            return []
-
-        current_balance = float(account["balance"])
-
-        message = f"The balance is: ${current_balance}"
-        dispatcher.utter_message(text=message)
-        return []

rasa/cli/project_templates/finance/actions/action_check_card_existence.py

@@ -1,35 +0,0 @@
-from typing import Any, Dict, List, Text
-
-from rasa_sdk import Action, Tracker
-from rasa_sdk.events import SlotSet
-from rasa_sdk.executor import CollectingDispatcher
-
-from actions.database import Database
-
-
-class ActionCheckCardExistence(Action):
-    def name(self) -> str:
-        return "action_check_card_existence"
-
-    def run(
-        self, dispatcher: CollectingDispatcher, tracker: Tracker, domain: Dict[str, Any]
-    ) -> List[Dict[Text, Any]]:
-        username = tracker.get_slot("username")
-        card_number = tracker.get_slot("card_number")
-
-        db = Database()
-
-        # Get user information
-        user = db.get_user_by_name(username)
-        if not user:
-            dispatcher.utter_message(text="User not found.")
-            return []
-
-        # Get all cards for the user
-        cards = db.get_cards_by_user(user["id"])
-        card_numbers = [card["number"] for card in cards]
-
-        if card_number in card_numbers:
-            return [SlotSet("card_found", True)]
-        else:
-            return [SlotSet("card_found", False)]