rasa-pro 3.12.18.dev1__py3-none-any.whl → 3.13.0a1.dev1__py3-none-any.whl

rasa/__init__.py

@@ -5,11 +5,5 @@ from rasa import version
 # define the version before the other imports since these need it
 __version__ = version.__version__
 
-from litellm.integrations.langfuse.langfuse import LangFuseLogger
-
-from rasa.monkey_patches import litellm_langfuse_logger_init_fixed
-
-# Monkey-patch the init method as early as possible before the class is used
-LangFuseLogger.__init__ = litellm_langfuse_logger_init_fixed  # type: ignore
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())

rasa/__main__.py

@@ -9,6 +9,7 @@ from rasa_sdk import __version__ as rasa_sdk_version
 
 import rasa.telemetry
 import rasa.utils.io
+import rasa.utils.licensing
 import rasa.utils.tensorflow.environment as tf_env
 from rasa import version
 from rasa.cli import (
@@ -105,11 +106,12 @@ def main(raw_arguments: Optional[List[str]] = None) -> None:
         raw_arguments: Arguments to parse. If not provided,
             arguments will be taken from the command line.
     """
+    rasa.utils.licensing.validate_license_from_env()
+
     warn_if_rasa_plus_package_installed()
     parse_last_positional_argument_as_model_path()
     arg_parser = create_argument_parser()
     cmdline_arguments = arg_parser.parse_args(raw_arguments)
-
     log_level = getattr(cmdline_arguments, "loglevel", None)
     logging_config_file = getattr(cmdline_arguments, "logging_config_file", None)
     configure_logging_and_warnings(
@@ -141,9 +143,6 @@ def main(raw_arguments: Optional[List[str]] = None) -> None:
             if not is_studio_command:
                 plugin_manager().hook.init_telemetry(endpoints_file=endpoints_file)
                 plugin_manager().hook.init_managers(endpoints_file=endpoints_file)
-                plugin_manager().hook.init_anonymization_pipeline(
-                    endpoints_file=endpoints_file
-                )
 
             cmdline_arguments.func(cmdline_arguments)
         elif hasattr(cmdline_arguments, "version"):

rasa/api.py

@@ -38,7 +38,7 @@ def run(
     """
     import rasa.core.run
     import rasa.shared.utils.common
-    from rasa.core.utils import AvailableEndpoints
+    from rasa.core.available_endpoints import AvailableEndpoints
     from rasa.shared.constants import DOCS_BASE_URL
     from rasa.shared.utils.cli import print_warning
 

rasa/builder/create_openai_vector_store.py

@@ -0,0 +1,69 @@
+import json
+from dataclasses import dataclass
+from pathlib import Path
+
+import openai
+
+DOCS_DIR = "rasa_docs_md"  # Folder with scraped Markdown files
+assistant_name = "Rasa Docs Assistant"
+
+
+@dataclass
+class FileWithAttributes:
+    file_id: str
+    file_name: str
+    attributes: dict
+
+
+def get_files_with_metadata(files) -> list[FileWithAttributes]:
+    with open("markdown_to_url.json", "r") as f:
+        markdown_to_url = json.load(f)
+
+    files_with_metadata = []
+    for file in files:
+        files_with_metadata.append(
+            FileWithAttributes(
+                file_id=file["file_id"],
+                file_name=file["file_name"],
+                attributes={"url": markdown_to_url[file["file_name"]]},
+            )
+        )
+
+    return files_with_metadata
+
+
+# Step 1: Upload files
+print("📤 Uploading files to OpenAI...")
+
+files = []
+
+# try to read the file ids from the json file
+if Path("file_ids.json").exists():
+    with open("file_ids.json", "r") as f:
+        files = json.load(f)
+else:
+    for md_file in Path(DOCS_DIR).glob("*.md"):
+        with open(md_file, "rb") as f:
+            uploaded = openai.files.create(file=f, purpose="assistants")
+            files.append({"file_id": uploaded.id, "file_name": md_file.name})
+            print(f"✅ Uploaded {md_file.name} → {uploaded.id}")
+
+    # persist the file ids in a json file
+    with open("file_ids.json", "w") as f:
+        json.dump(files, f, indent=2)
+
+# Step 2: Create Vector Store
+print("\n🤖 Creating vector store...")
+
+vector_store = openai.vector_stores.create(name="Rasa Docs")
+
+for file in get_files_with_metadata(files):
+    openai.vector_stores.files.create(
+        vector_store_id=vector_store.id,
+        file_id=file.file_id,
+        attributes=file.attributes,
+    )
+    print(f"✅ Added {file.file_name} to vector store.")
+
+print("\n🎉 Vector store created successfully!")
+print(f"Vector store ID: {vector_store.id}")

rasa/builder/llm-helper-schema.json

@@ -0,0 +1,69 @@
+{
+  "$schema": "http://json-schema.org/draft-07/schema#",
+  "type": "object",
+  "properties": {
+    "content_blocks": {
+      "type": "array",
+      "items": {
+        "oneOf": [
+          {
+            "type": "object",
+            "required": ["type", "text"],
+            "properties": {
+              "type": {
+                "type": "string",
+                "enum": ["text"]
+              },
+              "text": {
+                "type": "string"
+              }
+            }
+          },
+          {
+            "type": "object",
+            "required": ["type", "text"],
+            "properties": {
+              "type": {
+                "type": "string",
+                "enum": ["link"]
+              },
+              "text": {
+                "type": "string",
+                "format": "uri"
+              }
+            }
+          },
+          {
+            "type": "object",
+            "required": ["type", "file", "content"],
+            "properties": {
+              "type": {
+                "type": "string",
+                "enum": ["file"]
+              },
+              "file": {
+                "type": "string"
+              },
+              "content": {
+                "type": "string"
+              }
+            }
+          },
+          {
+            "type": "object",
+            "required": ["type", "text"],
+            "properties": {
+              "type": {
+                "type": "string",
+                "enum": ["code"]
+              },
+              "text": {
+                "type": "string"
+              }
+            }
+          }
+        ]
+      }
+    }
+  }
+}