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

rasa/llm_fine_tuning/annotation_module.py

@@ -1,17 +1,17 @@
 import asyncio
 from contextlib import contextmanager
-from typing import List, Generator, Optional, Tuple, Union
+from typing import Generator, List, Optional, Tuple, Union
 
 import structlog
 
 from rasa.dialogue_understanding.commands import Command
-from rasa.e2e_test.e2e_test_case import TestSuite, TestCase, ActualStepOutput, TestStep
-from rasa.e2e_test.e2e_test_runner import E2ETestRunner, TEST_TURNS_TYPE
+from rasa.e2e_test.e2e_test_case import ActualStepOutput, TestCase, TestStep, TestSuite
+from rasa.e2e_test.e2e_test_runner import TEST_TURNS_TYPE, E2ETestRunner
 from rasa.llm_fine_tuning.conversations import Conversation, ConversationStep
 from rasa.llm_fine_tuning.storage import StorageContext
 from rasa.shared.core.constants import USER
 from rasa.shared.core.trackers import DialogueStateTracker
-from rasa.shared.nlu.constants import LLM_PROMPT, LLM_COMMANDS
+from rasa.shared.nlu.constants import LLM_COMMANDS, LLM_PROMPT
 from rasa.shared.utils.llm import tracker_as_readable_transcript
 
 ANNOTATION_MODULE_STORAGE_LOCATION = "1_command_annotations"

rasa/llm_fine_tuning/conversations.py

@@ -1,16 +1,16 @@
 from dataclasses import dataclass, field
-from typing import List, Union, Iterator, Any, Dict, Optional
+from typing import Any, Dict, Iterator, List, Optional, Union
 
 from rasa.dialogue_understanding.commands import (
-    Command,
-    StartFlowCommand,
-    SetSlotCommand,
     CancelFlowCommand,
     ChitChatAnswerCommand,
-    SkipQuestionCommand,
+    ClarifyCommand,
+    Command,
     HumanHandoffCommand,
     KnowledgeAnswerCommand,
-    ClarifyCommand,
+    SetSlotCommand,
+    SkipQuestionCommand,
+    StartFlowCommand,
 )
 from rasa.e2e_test.e2e_test_case import TestCase, TestStep
 from rasa.shared.core.constants import USER

rasa/llm_fine_tuning/llm_data_preparation_module.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import List, Dict, Any, Optional
+from typing import Any, Dict, List, Optional
 
 import structlog
 from tqdm import tqdm

rasa/llm_fine_tuning/paraphrasing/conversation_rephraser.py

@@ -1,6 +1,6 @@
 import importlib.resources
 import re
-from typing import Dict, Any, List, Tuple, Optional
+from typing import Any, Dict, List, Optional, Tuple
 
 import structlog
 from jinja2 import Template
@@ -10,9 +10,9 @@ from rasa.llm_fine_tuning.paraphrasing.rephrased_user_message import (
     RephrasedUserMessage,
 )
 from rasa.shared.constants import (
-    MODEL_NAME_CONFIG_KEY,
-    MODEL_CONFIG_KEY,
     LLM_CONFIG_KEY,
+    MODEL_CONFIG_KEY,
+    MODEL_NAME_CONFIG_KEY,
     PROMPT_TEMPLATE_CONFIG_KEY,
     PROVIDER_CONFIG_KEY,
     TIMEOUT_CONFIG_KEY,
@@ -20,9 +20,9 @@ from rasa.shared.constants import (
 from rasa.shared.exceptions import ProviderClientAPIException
 from rasa.shared.providers.mappings import OPENAI_PROVIDER
 from rasa.shared.utils.llm import (
+    USER,
     get_prompt_template,
     llm_factory,
-    USER,
 )
 
 SEPARATOR = "\n\n"

rasa/llm_fine_tuning/paraphrasing/rephrase_validator.py

@@ -1,4 +1,4 @@
-from typing import Dict, Any, List
+from typing import Any, Dict, List
 
 import structlog
 

rasa/llm_fine_tuning/paraphrasing_module.py

@@ -1,4 +1,4 @@
-from typing import List, Dict, Any, Tuple
+from typing import Any, Dict, List, Tuple
 
 import structlog
 from tqdm import tqdm

rasa/llm_fine_tuning/storage.py

@@ -1,9 +1,9 @@
 import json
 import os
-from abc import abstractmethod, ABC
+from abc import ABC, abstractmethod
 from enum import Enum
 from pathlib import Path
-from typing import List, Optional, TYPE_CHECKING
+from typing import TYPE_CHECKING, List, Optional
 
 import rasa.shared.utils.io
 import rasa.utils.io
@@ -11,9 +11,9 @@ from rasa.llm_fine_tuning.conversations import Conversation
 from rasa.shared.utils.yaml import write_yaml
 
 if TYPE_CHECKING:
+    from rasa.e2e_test.e2e_test_case import TestSuite
     from rasa.llm_fine_tuning.llm_data_preparation_module import LLMDataExample
     from rasa.llm_fine_tuning.train_test_split_module import DataExampleFormat
-    from rasa.e2e_test.e2e_test_case import TestSuite
 
 
 class StorageType(Enum):

rasa/markers/marker.py

@@ -1,15 +1,14 @@
 import logging
 from typing import List, Optional, Text
 
-from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import ActionExecuted, Event, SlotSet, UserUttered
-
 from rasa.markers.marker_base import (
     ConditionMarker,
     Marker,
     MarkerRegistry,
     OperatorMarker,
 )
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.events import ActionExecuted, Event, SlotSet, UserUttered
 
 logger = logging.getLogger(__name__)
 

rasa/markers/marker_base.py

@@ -6,9 +6,9 @@ from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from pathlib import Path
 from typing import (
-    ClassVar,
     TYPE_CHECKING,
     Any,
+    ClassVar,
     Dict,
     Iterator,
     List,
@@ -30,7 +30,6 @@ from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import ActionExecuted, Event, UserUttered
 from rasa.shared.exceptions import InvalidConfigException, RasaException
 from rasa.utils.io import WriteRow
-
 from rasa.utils.yaml import (
     collect_configs_from_yaml_files,
     collect_yaml_files_from_path,

rasa/markers/upload.py

@@ -3,10 +3,10 @@ from pathlib import Path
 from typing import Any, Dict, List, Text
 
 import requests
-from rasa.shared.core.domain import Domain
-from rasa.shared.utils.cli import print_error_and_exit
 
 from rasa.markers.validate import validate_marker_file
+from rasa.shared.core.domain import Domain
+from rasa.shared.utils.cli import print_error_and_exit
 from rasa.utils.yaml import (
     YAML_CONFIGS,
     collect_configs_from_yaml_files,

rasa/markers/validate.py

@@ -1,10 +1,9 @@
 from pathlib import Path
 
-from rasa.shared.core.domain import Domain
-from rasa.shared.utils.cli import print_error_and_exit
-
 from rasa.markers.marker import OrMarker
 from rasa.markers.marker_base import Marker
+from rasa.shared.core.domain import Domain
+from rasa.shared.utils.cli import print_error_and_exit
 
 
 def validate_marker_file(domain: Domain, markers_path: Path) -> None:

rasa/model.py

@@ -3,13 +3,11 @@ import hashlib
 import logging
 import os
 from pathlib import Path
-from subprocess import check_output, DEVNULL, CalledProcessError
-from typing import Text, Optional, Union
-
-from rasa.shared.constants import DEFAULT_MODELS_PATH
+from subprocess import DEVNULL, CalledProcessError, check_output
+from typing import Optional, Text, Union
 
 from rasa.exceptions import ModelNotFound
-
+from rasa.shared.constants import DEFAULT_MODELS_PATH
 
 logger = logging.getLogger(__name__)
 

rasa/model_manager/config.py

@@ -1,5 +1,5 @@
-import sys
 import os
+import sys
 
 DEFAULT_SERVER_BASE_WORKING_DIRECTORY = "working-data"
 

rasa/model_manager/model_api.py

@@ -1,21 +1,22 @@
 import asyncio
-from functools import wraps
 import os
+from functools import wraps
 from http import HTTPStatus
 from typing import Any, Callable, Dict, Optional, Union
+
 import dotenv
 import psutil
+import structlog
 from sanic import Blueprint, Sanic, response
-from sanic.response import json
 from sanic.exceptions import NotFound
 from sanic.request import Request
-import structlog
+from sanic.response import json
 from socketio import AsyncServer
 
+from rasa.constants import MODEL_ARCHIVE_EXTENSION
 from rasa.exceptions import ModelNotFound
 from rasa.model_manager import config
 from rasa.model_manager.config import SERVER_BASE_URL
-from rasa.constants import MODEL_ARCHIVE_EXTENSION
 from rasa.model_manager.runner_service import (
     BotSession,
     BotSessionStatus,

rasa/model_manager/runner_service.py

@@ -1,22 +1,22 @@
 import os
 import shutil
+import subprocess
+from enum import Enum
 from typing import Dict, Optional
+
 import aiohttp
 import structlog
-import subprocess
 from pydantic import BaseModel, ConfigDict
-from enum import Enum
 
+from rasa.constants import MODEL_ARCHIVE_EXTENSION
 from rasa.exceptions import ModelNotFound
+from rasa.model_manager import config
 from rasa.model_manager.utils import (
+    logs_path,
     models_base_path,
     subpath,
     write_encoded_data_to_file,
 )
-from rasa.constants import MODEL_ARCHIVE_EXTENSION
-
-from rasa.model_manager import config
-from rasa.model_manager.utils import logs_path
 from rasa.model_manager.warm_rasa_process import start_rasa_process
 
 structlogger = structlog.get_logger()

rasa/model_manager/socket_bridge.py

@@ -1,7 +1,8 @@
+import json
 from typing import Any, Dict, Optional
 
-from socketio import AsyncServer
 import structlog
+from socketio import AsyncServer
 from socketio.asyncio_client import AsyncClient
 
 from rasa.model_manager.runner_service import BotSession
@@ -92,7 +93,7 @@ def create_bridge_server(sio: AsyncServer, running_bots: Dict[str, BotSession])
 
     @sio.on("*")
     async def handle_message(event: str, sid: str, data: Dict[str, Any]) -> None:
-        """ "Bridge messages between user and bot.
+        """Bridge messages between user and bot.
 
         Both incoming user messages to the bot_url and
         bot responses sent back to the client need to
@@ -111,7 +112,7 @@ async def create_bridge_client(
 ) -> AsyncClient:
     """Create a new socket bridge client.
 
-    Forwards messages comming from the bot to the user.
+    Forwards messages coming from the bot to the user.
     """
     client = AsyncClient()
 
@@ -129,6 +130,10 @@ 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 tracker(data: Dict[str, Any]) -> None:
+        await sio.emit("tracker", json.loads(data), room=sid)
+
     @client.event  # type: ignore[misc]
     async def disconnect() -> None:
         structlogger.debug(

rasa/model_manager/studio_jwt_auth.py

@@ -6,6 +6,7 @@ from typing import Any, Dict, Optional
 import jwt
 import requests
 import structlog
+
 from rasa.shared.exceptions import RasaException
 
 structlogger = structlog.get_logger()

rasa/model_manager/trainer_service.py

@@ -1,23 +1,25 @@
 import os
-from typing import Any, Dict, Optional
 import shutil
-import structlog
 import subprocess
+from enum import Enum
+from typing import Any, Dict, Optional
+
+import structlog
+from pydantic import BaseModel, ConfigDict
+
 from rasa.constants import MODEL_ARCHIVE_EXTENSION
+from rasa.model_manager import config
 from rasa.model_manager.utils import (
+    ensure_base_directory_exists,
+    logs_path,
     models_base_path,
     subpath,
     write_encoded_data_to_file,
 )
-from pydantic import BaseModel, ConfigDict
-from enum import Enum
-
-from rasa.model_manager import config
 from rasa.model_manager.warm_rasa_process import (
     start_rasa_process,
 )
 from rasa.model_training import generate_random_model_name
-from rasa.model_manager.utils import ensure_base_directory_exists, logs_path
 
 structlogger = structlog.get_logger()
 

rasa/model_manager/utils.py

@@ -1,5 +1,5 @@
-import os
 import base64
+import os
 from typing import Optional
 
 import structlog

rasa/model_manager/warm_rasa_process.py

@@ -1,12 +1,16 @@
+import os
 import shlex
 import subprocess
-from rasa.__main__ import main
-import os
+import uuid
+from dataclasses import dataclass
 from typing import List
+
 import structlog
-from dataclasses import dataclass
-import uuid
 
+from rasa.__main__ import main
+from rasa.dialogue_understanding.constants import (
+    RASA_RECORD_COMMANDS_AND_PROMPTS_ENV_VAR_NAME,
+)
 from rasa.model_manager import config
 from rasa.model_manager.utils import ensure_base_directory_exists, logs_path
 
@@ -43,6 +47,7 @@ def _create_warm_rasa_process() -> WarmRasaProcess:
 
     envs = os.environ.copy()
     envs["RASA_TELEMETRY_ENABLED"] = "false"
+    envs[RASA_RECORD_COMMANDS_AND_PROMPTS_ENV_VAR_NAME] = "true"
 
     log_id = uuid.uuid4().hex
     log_path = logs_path(log_id)
@@ -146,14 +151,15 @@ def warmup() -> None:
     directly start.
     """
     try:
-        import presidio_analyzer  # noqa: F401
-        import litellm  # noqa: F401
         import langchain  # noqa: F401
-        import tensorflow  # noqa: F401
+        import litellm  # noqa: F401
         import matplotlib  # noqa: F401
-        import pandas  # noqa: F401
         import numpy  # noqa: F401
+        import pandas  # noqa: F401
+        import presidio_analyzer  # noqa: F401
         import spacy  # noqa: F401
+        import tensorflow  # noqa: F401
+
         import rasa.validator  # noqa: F401
     except ImportError:
         pass

rasa/model_service.py

@@ -1,18 +1,15 @@
 import logging
+from urllib.parse import urlparse
 
-from sanic import Sanic
 import structlog
+from sanic import Sanic
 
 from rasa.cli.scaffold import print_error_and_exit
 from rasa.core.persistor import RemoteStorageType, get_persistor
 from rasa.core.utils import list_routes
-from rasa.model_manager import model_api
-from rasa.model_manager import config
+from rasa.model_manager import config, model_api
 from rasa.model_manager.config import SERVER_BASE_URL, SERVER_PORT
 from rasa.utils.common import configure_logging_and_warnings
-import rasa.utils.licensing
-from urllib.parse import urlparse
-
 from rasa.utils.log_utils import configure_structlog
 from rasa.utils.sanic_error_handler import register_custom_sanic_error_handler
 
@@ -64,6 +61,8 @@ def main() -> None:
     The API server can receive requests to train models, run bots, and manage
     the lifecycle of models and bots.
     """
+    import rasa.utils.licensing
+
     log_level = logging.DEBUG
     configure_logging_and_warnings(
         log_level=log_level,

rasa/model_testing.py

@@ -1,32 +1,31 @@
 import copy
 import logging
 import os
+from pathlib import Path
 from typing import (
-    Text,
-    Dict,
-    Optional,
-    List,
     Any,
+    Dict,
     Iterable,
+    List,
+    Optional,
+    Text,
     Tuple,
     Union,
 )
-from pathlib import Path
 
-from rasa.core.agent import Agent
-from rasa.engine.storage.local_model_storage import LocalModelStorage
+import rasa.model
+import rasa.shared.nlu.training_data.loading
 import rasa.shared.utils.cli
 import rasa.shared.utils.common
 import rasa.shared.utils.io
 import rasa.utils.common
-from rasa.constants import RESULTS_FILE, NUMBER_OF_TRAINING_STORIES_FILE
+from rasa.constants import NUMBER_OF_TRAINING_STORIES_FILE, RESULTS_FILE
+from rasa.core.agent import Agent
+from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.exceptions import ModelNotFound
 from rasa.shared.constants import DEFAULT_RESULTS_PATH
-import rasa.shared.nlu.training_data.loading
 from rasa.shared.data import TrainingType
 from rasa.shared.nlu.training_data.training_data import TrainingData
-import rasa.model
-
 
 logger = logging.getLogger(__name__)
 
@@ -245,10 +244,9 @@ async def compare_nlu_models(
     exclusion_percentages: List[int],
 ) -> None:
     """Trains multiple models, compares them and saves the results."""
-    from rasa.nlu.test import drop_intents_below_freq
+    from rasa.nlu.test import compare_nlu, drop_intents_below_freq
     from rasa.nlu.utils import write_json_to_file
     from rasa.utils.io import create_path
-    from rasa.nlu.test import compare_nlu
 
     test_data = drop_intents_below_freq(test_data, cutoff=5)
 
@@ -313,10 +311,10 @@ async def perform_nlu_cross_validation(
             cross-validation, like number of `disable_plotting`.
     """
     from rasa.nlu.test import (
-        drop_intents_below_freq,
         cross_validate,
-        log_results,
+        drop_intents_below_freq,
         log_entity_results,
+        log_results,
     )
 
     additional_arguments = additional_arguments or {}