rasa-pro 3.13.1a14__py3-none-any.whl → 3.13.1a16__py3-none-any.whl

rasa/builder/project_generator.py

@@ -12,6 +12,7 @@ import structlog
 from rasa.builder import config
 from rasa.builder.exceptions import ProjectGenerationError, ValidationError
 from rasa.builder.llm_service import get_skill_generation_messages, llm_service
+from rasa.builder.models import BotFiles
 from rasa.builder.validation_service import validate_project
 from rasa.cli.scaffold import ProjectTemplateName, create_initial_project
 from rasa.shared.core.flows import yaml_flows_io
@@ -25,7 +26,7 @@ structlogger = structlog.get_logger()
 class ProjectGenerator:
     """Service for generating Rasa projects from skill descriptions."""
 
-    def __init__(self, project_folder: str):
+    def __init__(self, project_folder: str) -> None:
         """Initialize the project generator with a folder for file persistence.
 
         Args:
@@ -34,7 +35,7 @@ class ProjectGenerator:
         self.project_folder = Path(project_folder)
         self.project_folder.mkdir(parents=True, exist_ok=True)
 
-    def init_from_template(self, template: ProjectTemplateName):
+    def init_from_template(self, template: ProjectTemplateName) -> None:
         """Create the initial project files."""
         self.cleanup()
         create_initial_project(self.project_folder.as_posix(), template)
@@ -150,7 +151,7 @@ class ProjectGenerator:
 
         return await _generate_with_retry(initial_messages, max_retries)
 
-    async def _validate_generated_project(self):
+    async def _validate_generated_project(self) -> None:
         """Validate the generated project using the validation service."""
         importer = self._create_importer()
         validation_error = await validate_project(importer)
@@ -178,9 +179,9 @@ class ProjectGenerator:
         except Exception as e:
             raise ValidationError(f"Failed to create importer: {e}")
 
-    def get_bot_files(self) -> Dict[str, Optional[str]]:
+    def get_bot_files(self) -> BotFiles:
         """Get the current bot files by reading from disk."""
-        bot_files = {}
+        bot_files: BotFiles = {}
 
         for file in self.project_folder.glob("**/*"):
             # Skip directories
@@ -255,7 +256,7 @@ class ProjectGenerator:
             shutil.rmtree(flows_folder)
         flows_folder.mkdir(parents=True, exist_ok=True)
 
-    def update_bot_files(self, files: Dict[str, str]) -> None:
+    def update_bot_files(self, files: Dict[str, Optional[str]]) -> None:
         """Update bot files with new content by writing to disk."""
         for filename, content in files.items():
             file_path = Path(subpath(self.project_folder, filename))

rasa/builder/scrape_rasa_docs.py

@@ -17,18 +17,18 @@ to_visit = [DOCS_ROOT]
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 
-def is_valid_doc_url(url):
+def is_valid_doc_url(url: str) -> bool:
     return url.startswith(DOCS_ROOT) and not any(
         [url.endswith(".pdf"), "#" in url, "mailto:" in url]
     )
 
 
-def slugify_url(url):
+def slugify_url(url: str) -> str:
     path = urlparse(url).path.strip("/").replace("/", "_")
     return path if path else "index"
 
 
-def clean_text(html):
+def clean_text(html: str) -> str:
     soup = BeautifulSoup(html, "html.parser")
 
     # Remove navs, footers, and code tabs (customize if needed)
@@ -47,7 +47,7 @@ def clean_text(html):
     return text
 
 
-def save_as_markdown(text) -> str:
+def save_as_markdown(text: str, url: str) -> str:
     slug = slugify_url(url)
     file_name = f"{slug}.md"
     md_path = Path(OUTPUT_DIR) / file_name