rasa-pro 3.14.0.dev3__py3-none-any.whl → 3.14.0.dev4__py3-none-any.whl

rasa/tracing/instrumentation/instrumentation.py

@@ -443,7 +443,7 @@ def instrument(
             tracer_provider.get_tracer(lock_store_class.__module__),
             attribute_extractors.extract_attrs_for_lock_store,
         )
-        lock_store_class.lock = contextlib.asynccontextmanager(traced_lock_method)  # type: ignore[assignment]
+        lock_store_class.lock = contextlib.asynccontextmanager(traced_lock_method)  # type: ignore[method-assign]
 
         logger.debug(f"Instrumented '{lock_store_class.__name__}.lock'.")
 
@@ -790,7 +790,7 @@ def _instrument_nlu_command_adapter_predict_commands(
 
         return wrapper
 
-    nlu_command_adapter_class.predict_commands = (  # type: ignore[assignment]
+    nlu_command_adapter_class.predict_commands = (  # type: ignore[method-assign]
         tracing_nlu_command_adapter_predict_commands_wrapper(
             nlu_command_adapter_class.predict_commands
         )
@@ -839,7 +839,7 @@ def _instrument_multi_step_llm_command_generator_parse_commands(
 
         return wrapper
 
-    multi_step_llm_command_generator_class.parse_commands = (  # type: ignore[assignment]
+    multi_step_llm_command_generator_class.parse_commands = (  # type: ignore[method-assign]
         tracing_multi_step_llm_command_generator_parse_commands_wrapper(
             multi_step_llm_command_generator_class.parse_commands
         )
@@ -873,7 +873,7 @@ def _instrument_information_retrieval_search(
 
         return wrapper
 
-    vector_store_class.search = tracing_information_retrieval_search_wrapper(  # type: ignore[assignment]
+    vector_store_class.search = tracing_information_retrieval_search_wrapper(  # type: ignore[method-assign]
         vector_store_class.search
     )
 
@@ -1013,7 +1013,7 @@ def _instrument_get_tracker(
 
         return wrapper
 
-    processor_class.get_tracker = tracing_get_tracker_wrapper(  # type: ignore[assignment]
+    processor_class.get_tracker = tracing_get_tracker_wrapper(  # type: ignore[method-assign]
         processor_class.get_tracker
     )
 
@@ -1225,7 +1225,7 @@ def _instrument_run_action(
 
         return wrapper
 
-    processor_class._run_action = tracing_run_action_wrapper(  # type: ignore[assignment]
+    processor_class._run_action = tracing_run_action_wrapper(  # type: ignore[method-assign]
         processor_class._run_action
     )
 
@@ -1284,7 +1284,7 @@ def _instrument_endpoint_config(
 
         return wrapper
 
-    endpoint_config_class.request = tracing_endpoint_config_wrapper(  # type: ignore[assignment]
+    endpoint_config_class.request = tracing_endpoint_config_wrapper(  # type: ignore[method-assign]
         endpoint_config_class.request
     )
 
@@ -1316,7 +1316,7 @@ def _instrument_grpc_custom_action_executor(
 
         return wrapper
 
-    grpc_custom_action_executor_class.run = tracing_grpc_custom_action_executor_wrapper(  # type: ignore[assignment]
+    grpc_custom_action_executor_class.run = tracing_grpc_custom_action_executor_wrapper(  # type: ignore[method-assign]
         grpc_custom_action_executor_class.run
     )
 

rasa/tracing/instrumentation/intentless_policy_instrumentation.py

@@ -37,7 +37,7 @@ def _instrument_select_response_examples(
 
         return wrapper
 
-    policy_class.select_response_examples = tracing_select_response_examples_wrapper(  # type: ignore[assignment]
+    policy_class.select_response_examples = tracing_select_response_examples_wrapper(  # type: ignore[method-assign]
         policy_class.select_response_examples
     )
 
@@ -72,7 +72,7 @@ def _instrument_select_few_shot_conversations(
 
         return wrapper
 
-    policy_class.select_few_shot_conversations = (  # type: ignore[assignment]
+    policy_class.select_few_shot_conversations = (  # type: ignore[method-assign]
         tracing_select_few_shot_conversations_wrapper(
             policy_class.select_few_shot_conversations
         )
@@ -104,7 +104,7 @@ def _instrument_extract_ai_responses(
 
         return wrapper
 
-    policy_class.extract_ai_responses = tracing_extract_ai_responses_wrapper(  # type: ignore[assignment]
+    policy_class.extract_ai_responses = tracing_extract_ai_responses_wrapper(  # type: ignore[method-assign]
         policy_class.extract_ai_responses
     )
 
@@ -137,7 +137,7 @@ def _instrument_generate_answer(
 
         return wrapper
 
-    policy_class.generate_answer = tracing_generate_answer_wrapper(  # type: ignore[assignment]
+    policy_class.generate_answer = tracing_generate_answer_wrapper(  # type: ignore[method-assign]
         policy_class.generate_answer
     )
 

rasa/utils/log_utils.py

@@ -90,7 +90,7 @@ def configure_structlog(
         ]
 
     structlog.configure(
-        processors=processors,  # type: ignore
+        processors=processors,
         context_class=dict,
         # `logger_factory` is used to create wrapped loggers that are used for
         # OUTPUT. This one returns a `logging.Logger`. The final value (a JSON

rasa/utils/ml_utils.py

@@ -119,7 +119,7 @@ def response_for_template(
     rsps = responses.get(template_name, [])
     usuable_responses = [r for r in rsps if r.get(KEY_RESPONSES_TEXT)]
     if usuable_responses:
-        selected_response = np.random.choice(usuable_responses)  # type: ignore
+        selected_response = np.random.choice(usuable_responses)
         return selected_response.get(KEY_RESPONSES_TEXT)
     else:
         structlogger.warning(

rasa/utils/tensorflow/rasa_layers.py

@@ -291,7 +291,7 @@ class ConcatenateSparseDenseFeatures(RasaCustomLayer):
             [
                 config[DENSE_DIMENSION][attribute]
                 if signature.is_sparse
-                else signature.units
+                else (signature.units or 0)
                 for signature in feature_type_signature
             ]
         )

rasa/utils/train_utils.py

@@ -11,7 +11,7 @@ from rasa.nlu.constants import NUMBER_OF_SUB_TOKENS
 from rasa.shared.constants import NEXT_MAJOR_VERSION_FOR_DEPRECATIONS
 from rasa.shared.exceptions import InvalidConfigException
 from rasa.shared.nlu.constants import SPLIT_ENTITIES_BY_COMMA
-from rasa.utils.tensorflow.callback import RasaModelCheckpoint, RasaTrainingLogger
+from rasa.utils.tensorflow.callback import RasaModelCheckpoint
 from rasa.utils.tensorflow.constants import (
     AUTO,
     CHECKPOINT_MODEL,
@@ -371,20 +371,20 @@ def create_common_callbacks(
     Returns:
         A list of callbacks.
     """
-    import tensorflow as tf
-
-    callbacks = [RasaTrainingLogger(epochs, silent=False)]
-
-    if tensorboard_log_dir:
-        callbacks.append(
-            tf.keras.callbacks.TensorBoard(
-                log_dir=tensorboard_log_dir,
-                update_freq=tensorboard_log_level,
-                write_graph=True,
-                write_images=True,
-                histogram_freq=10,
-            )
-        )
+    # import tensorflow as tf
+
+    callbacks = []
+
+    # if tensorboard_log_dir:
+    #     callbacks.append(
+    #         tf.keras.callbacks.TensorBoard(
+    #             log_dir=tensorboard_log_dir,
+    #             update_freq=tensorboard_log_level,
+    #             write_graph=True,
+    #             write_images=True,
+    #             histogram_freq=10,
+    #         )
+    #     )
 
     if checkpoint_dir:
         callbacks.append(RasaModelCheckpoint(checkpoint_dir))

rasa/validator.py

@@ -9,7 +9,7 @@ from jinja2 import Template
 from pypred import Predicate
 from pypred.ast import CompareOperator, Literal, NegateOperator
 
-import rasa.core.training.story_conflict
+# import rasa.core.training.story_conflict
 import rasa.shared.nlu.constants
 import rasa.shared.utils.cli
 import rasa.shared.utils.io
@@ -60,7 +60,8 @@ from rasa.shared.core.flows.utils import (
     get_invalid_slot_persistence_config_error_message,
     warn_deprecated_collect_step_config,
 )
-from rasa.shared.core.generator import TrainingDataGenerator
+
+# from rasa.shared.core.generator import TrainingDataGenerator
 from rasa.shared.core.slot_mappings import CoexistenceSystemType
 from rasa.shared.core.slots import BooleanSlot, CategoricalSlot, ListSlot, Slot
 from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
@@ -70,7 +71,7 @@ from rasa.shared.core.training_data.structures import StoryGraph
 from rasa.shared.data import create_regex_pattern_reader
 from rasa.shared.exceptions import RasaException
 from rasa.shared.importers.importer import TrainingDataImporter
-from rasa.shared.nlu.constants import COMMANDS
+from rasa.shared.nlu.constants import COMMANDS, INTENT
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.telemetry import track_validation_error_log
@@ -130,7 +131,7 @@ class Validator:
         """Compares list of intents in domain with intents in NLU training data."""
         everything_is_alright = True
 
-        nlu_data_intents = {e.data["intent"] for e in self.intents.intent_examples}
+        nlu_data_intents = {e.data[INTENT] for e in self.intents.intent_examples}
 
         for intent in self._non_default_intents():
             if intent not in nlu_data_intents:
@@ -172,7 +173,7 @@ class Validator:
         duplication_hash = defaultdict(set)
         for example in self.intents.intent_examples:
             text = example.get(rasa.shared.nlu.constants.TEXT)
-            duplication_hash[text].add(example.get("intent"))
+            duplication_hash[text].add(example.get(INTENT))
 
         for text, intents in duplication_hash.items():
             if len(duplication_hash[text]) > 1:
@@ -446,17 +447,18 @@ class Validator:
             event_info="Story structure validation...",
         )
 
-        trackers = TrainingDataGenerator(
-            self.story_graph,
-            domain=self.domain,
-            remove_duplicates=False,
-            augmentation_factor=0,
-        ).generate_story_trackers()
+        # trackers = TrainingDataGenerator(
+        #     self.story_graph,
+        #     domain=self.domain,
+        #     remove_duplicates=False,
+        #     augmentation_factor=0,
+        # ).generate_story_trackers()
 
         # Create a list of `StoryConflict` objects
-        conflicts = rasa.core.training.story_conflict.find_story_conflicts(
-            trackers, self.domain, max_history
-        )
+        # conflicts = rasa.core.training.story_conflict.find_story_conflicts(
+        #     trackers, self.domain, max_history
+        # )
+        conflicts = None
 
         if not conflicts:
             structlogger.info(

rasa/version.py

@@ -1,3 +1,3 @@
 # this file will automatically be changed,
 # do not add anything but the version number here!
-__version__ = "3.14.0.dev3"
+__version__ = "3.14.0.dev4"

{rasa_pro-3.14.0.dev3.dist-info → rasa_pro-3.14.0.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.14.0.dev3
+Version: 3.14.0.dev4
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
 Author: Rasa Technologies GmbH
@@ -24,9 +24,10 @@ Provides-Extra: transformers
 Requires-Dist: CacheControl (>=0.14.2,<0.15.0)
 Requires-Dist: PyJWT[crypto] (>=2.8.0,<3.0.0)
 Requires-Dist: SQLAlchemy (>=2.0.42,<2.1.0)
+Requires-Dist: a2a-sdk (>=0.3.2,<0.4.0)
 Requires-Dist: absl-py (>=2.0,<2.1)
 Requires-Dist: aio-pika (>=8.2.3,<9.4.4)
-Requires-Dist: aiogram (>=3.15,<3.16)
+Requires-Dist: aiogram (>=3.20.0,<3.21.0)
 Requires-Dist: aiohttp (>=3.10,<3.11)
 Requires-Dist: apscheduler (>=3.10,<3.11)
 Requires-Dist: attrs (>=23.1,<23.2)
@@ -60,7 +61,7 @@ Requires-Dist: jinja2 (>=3.1.6,<3.2.0)
 Requires-Dist: jsonpatch (>=1.33,<2.0)
 Requires-Dist: jsonpickle (>=3.3.0,<3.4)
 Requires-Dist: jsonschema (>=4.22)
-Requires-Dist: keras (==2.14.0)
+Requires-Dist: keras (>=3.5.0,<3.6.0)
 Requires-Dist: langchain (>=0.2.17,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.19,<0.3.0)
 Requires-Dist: langcodes (>=3.5.0,<4.0.0)
@@ -73,21 +74,21 @@ Requires-Dist: numpy (>=1.26.4,<1.27.0)
 Requires-Dist: onnxruntime (==1.19.2) ; extra == "pii"
 Requires-Dist: openai (>=1.68.2,<1.69.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: opentelemetry-api (>=1.33.0,<1.34.0)
+Requires-Dist: opentelemetry-exporter-jaeger (>=1.21.0,<1.22.0)
+Requires-Dist: opentelemetry-exporter-otlp (>=1.33.0,<1.34.0)
+Requires-Dist: opentelemetry-sdk (>=1.33.0,<1.34.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: prompt-toolkit (>=3.0.28,<3.0.29)
-Requires-Dist: protobuf (>=4.25.8,<4.26.0)
+Requires-Dist: protobuf (>=5.29.5,<5.30.0)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: psycopg2-binary (>=2.9.10,<2.10.0)
 Requires-Dist: pydot (>=1.4,<1.5)
 Requires-Dist: pykwalify (>=1.8,<1.9)
-Requires-Dist: pymilvus (>=2.4.1,<2.4.2)
+Requires-Dist: pymilvus (>=2.5.0,<2.6.0)
 Requires-Dist: pymongo (>=4.10.1,<4.11.0)
 Requires-Dist: pypred (>=0.4.0,<0.5.0)
 Requires-Dist: python-dateutil (>=2.8.2,<2.9.0)
@@ -100,7 +101,7 @@ Requires-Dist: pyyaml (>=6.0.2,<6.1.0)
 Requires-Dist: qdrant-client (>=1.9.1,<1.10.0)
 Requires-Dist: questionary (>=1.10.0,<2.1.0)
 Requires-Dist: randomname (>=0.2.1,<0.3.0)
-Requires-Dist: rasa-sdk (==3.14.0.dev2)
+Requires-Dist: rasa-sdk (==3.14.0.dev3)
 Requires-Dist: redis (>=4.6.0,<6.0)
 Requires-Dist: regex (>=2024.7.24,<2024.8.0)
 Requires-Dist: requests (>=2.32.3,<2.33.0)
@@ -125,15 +126,10 @@ Requires-Dist: structlog (>=25.4.0,<25.5.0)
 Requires-Dist: structlog-sentry (>=2.0.3,<2.1.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.9.0,<0.10.0)
@@ -146,6 +142,7 @@ 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: werkzeug (>=1.0.1)
 Requires-Dist: wheel (>=0.40.0)
 Project-URL: Documentation, https://rasa.com/docs
 Project-URL: Homepage, https://rasa.com