ara-cli 0.1.11.3__tar.gz → 0.1.13.5__tar.gz

{ara_cli-0.1.11.3 → ara_cli-0.1.13.5}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: ara_cli
-Version: 0.1.11.3
+Version: 0.1.13.5
 Summary: Powerful, open source command-line tool for managing, structuring and automating software development artifacts in line with Business-Driven Development (BDD) and AI-assisted processes
 Description-Content-Type: text/markdown
 Requires-Dist: langfuse
-Requires-Dist: litellm
+Requires-Dist: litellm==1.49.7
 Requires-Dist: llama-index
 Requires-Dist: llama-index-llms-openai
 Requires-Dist: llama-index-retrievers-bm25
@@ -21,6 +21,8 @@ Requires-Dist: python-docx
 Requires-Dist: pymupdf4llm
 Requires-Dist: typer
 Requires-Dist: psutil
+Requires-Dist: requests
+Requires-Dist: rich
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: requires-dist

ara_cli-0.1.13.5/ara_cli/__init__.py

@@ -0,0 +1,65 @@
+import warnings
+from .error_handler import ErrorHandler
+
+whitelisted_commands = [
+    "RERUN",
+    "SEND",
+    "EXTRACT",
+    "LOAD_IMAGE",
+    "CHOOSE_MODEL",
+    "CHOOSE_EXTRACTION_MODEL",
+    "CURRENT_MODEL",
+    "CURRENT_EXTRACTION_MODEL",
+    "LIST_MODELS",
+]
+
+error_handler = ErrorHandler()
+
+
+# ANSI escape codes for coloring
+YELLOW = "\033[93m"
+RESET = "\033[0m"
+
+
+def format_warning(message, category, *args, **kwargs):
+    return f"{YELLOW}[WARNING] {category.__name__}: {message}{RESET}\n"
+
+
+warnings.formatwarning = format_warning
+
+### CHAT VALUES ###
+
+CATEGORY_CHAT_CONTROL = "Chat control commands"
+CATEGORY_LLM_CONTROL = "Language model controls"
+CATEGORY_SCRIPT_CONTROL = "Script control commands"
+CATEGORY_AGENT_CONTROL = "Agent control commands"
+
+ROLE_PROMPT = "ara prompt"
+ROLE_RESPONSE = "ara response"
+
+INTRO = """/***************************************/
+                 araarar
+               aa       ara
+             aa    aa   aara
+             a        araarar
+             a        ar  ar
+           aa          ara
+          a               a
+          a               aa
+           a              a
+   ar      aa           aa
+    (c) ara chat by talsen team
+              aa      aa
+               aa    a
+                a aa
+                 aa
+/***************************************/
+Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat mode):"""
+
+BINARY_TYPE_MAPPING = {
+    ".png": "image/png",
+    ".jpg": "image/jpeg",
+    ".jpeg": "image/jpeg",
+}
+
+DOCUMENT_TYPE_EXTENSIONS = [".docx", ".doc", ".odt", ".pdf"]

{ara_cli-0.1.11.3 → ara_cli-0.1.13.5}/ara_cli/__main__.py

@@ -12,19 +12,27 @@ from ara_cli.ara_subcommands.list_tags import register as register_list_tags_cli
 from ara_cli.ara_subcommands.prompt import register as register_prompt_cli
 from ara_cli.ara_subcommands.chat import register as register_chat_cli
 from ara_cli.ara_subcommands.template import register as register_template_cli
-from ara_cli.ara_subcommands.fetch_templates import register as register_fetch_templates_cli
+from ara_cli.ara_subcommands.fetch_templates import (
+    register as register_fetch_templates_cli,
+)
 from ara_cli.ara_subcommands.fetch_scripts import register as register_fetch_scripts_cli
+from ara_cli.ara_subcommands.fetch_agents import register as register_fetch_agents_cli
+from ara_cli.ara_subcommands.fetch import register as register_fetch_cli
 from ara_cli.ara_subcommands.read import register as register_read_cli
 from ara_cli.ara_subcommands.reconnect import register as register_reconnect_cli
 from ara_cli.ara_subcommands.read_status import register as register_read_status_cli
 from ara_cli.ara_subcommands.read_user import register as register_read_user_cli
 from ara_cli.ara_subcommands.set_status import register as register_set_status_cli
 from ara_cli.ara_subcommands.set_user import register as register_set_user_cli
-from ara_cli.ara_subcommands.classifier_directory import register as register_classifier_directory_cli
+from ara_cli.ara_subcommands.classifier_directory import (
+    register as register_classifier_directory_cli,
+)
 from ara_cli.ara_subcommands.scan import register as register_scan_cli
 from ara_cli.ara_subcommands.autofix import register as register_autofix_cli
 from ara_cli.ara_subcommands.extract import register as register_extract_cli
 from ara_cli.ara_subcommands.load import register as register_load_cli
+from ara_cli.ara_subcommands.config import register as register_config_cli
+from ara_cli.ara_subcommands.convert import register as register_convert_cli
 
 from ara_cli.directory_navigator import DirectoryNavigator
 
@@ -37,7 +45,7 @@ def version_callback(value: bool):
 
 def is_debug_mode_enabled():
     """Check if debug mode is enabled via environment variable."""
-    return getenv('ARA_DEBUG', '').lower() in ('1', 'true', 'yes')
+    return getenv("ARA_DEBUG", "").lower() in ("1", "true", "yes")
 
 
 def configure_debug_mode(debug: bool, env_debug_mode: bool):
@@ -54,8 +62,11 @@ def check_ara_directory_exists():
 def prompt_create_ara_directory():
     """Prompt user to create ara directory and create it if confirmed."""
     # Print the prompt message
-    print("No 'ara' directory found. Create one in the current directory? (Y/n)",
-          end=" ", flush=True)
+    print(
+        "No 'ara' directory found. Create one in the current directory? (Y/n)",
+        end=" ",
+        flush=True,
+    )
 
     # Read user input
     try:
@@ -64,21 +75,21 @@ def prompt_create_ara_directory():
         typer.echo("\nOperation cancelled.")
         raise typer.Exit(1)
 
-    if response.lower() in ('y', 'yes', ''):
+    if response.lower() in ("y", "yes", ""):
         current_dir = os.getcwd()
-        ara_path = os.path.join(current_dir, 'ara')
+        ara_path = os.path.join(current_dir, "ara")
 
         # Create ara directory structure
         subdirectories = [
-            'businessgoals',
-            'capabilities',
-            'epics',
-            'examples',
-            'features',
-            'keyfeatures',
-            'tasks',
-            'userstories',
-            'vision'
+            "businessgoals",
+            "capabilities",
+            "epics",
+            "examples",
+            "features",
+            "keyfeatures",
+            "tasks",
+            "userstories",
+            "vision",
         ]
 
         try:
@@ -90,12 +101,13 @@ def prompt_create_ara_directory():
                 os.makedirs(os.path.join(ara_path, subdir), exist_ok=True)
 
             # Create .araconfig directory
-            araconfig_path = os.path.join(ara_path, '.araconfig')
+            araconfig_path = os.path.join(ara_path, ".araconfig")
             os.makedirs(araconfig_path, exist_ok=True)
 
             # Create default ara_config.json using ConfigManager
             from ara_cli.ara_config import ConfigManager, ARAconfig
-            config_file_path = os.path.join(araconfig_path, 'ara_config.json')
+
+            config_file_path = os.path.join(araconfig_path, "ara_config.json")
 
             # Reset ConfigManager to ensure clean state
             ConfigManager.reset()
@@ -103,6 +115,7 @@ def prompt_create_ara_directory():
             # Create default config and save it
             default_config = ARAconfig()
             from ara_cli.ara_config import save_data
+
             save_data(config_file_path, default_config)
 
             typer.echo(f"Created ara directory structure at {ara_path}")
@@ -156,15 +169,16 @@ ara artefact handling examples:
   > rename artefact and artefact data directory for e.g. a feature:                  ara rename feature {initial_feature_name} {new_feature_name}
   > create additional templates for a specific aspect (valid aspects are: customer,
     persona, concept, technology) related to an existing artefact like a feature:    ara create feature {feature_name} aspect {aspect_name}
-  > list artefact data with .md file extension                                       ara list --data {classifier} {artefact_name} --include-extension .md
-  > list artefact data with .md and .json file extensions                            ara list --data {classifier} {artefact_name} --include-extension .md .json
+  > list artefact data with .md file extension                                       ara list {classifier} {artefact_name} --data --include-extension .md
+  > list artefact data with .md and .json file extensions                            ara list {classifier} {artefact_name} --data --include-extension .md .json
   > list everything but userstories                                                  ara list --exclude-extension .userstory
   > list all existing features:                                                      ara list --include-extension .feature
+  > list all artefacts of a specific classifier:                                     ara list {classifier}
   > list all child artefacts contributing value to a parent artefact:                ara list --include-content "Contributes to {name_of_parent_artefact} {ara classifier_of_parent_artefact}"
   > list tasks which contain 'example content'                                       ara list --include-extension .task --include-content "example content"
-  > list children artefacts of a userstory                                           ara list --children userstory {name_of_userstory}
-  > list parent artefacts of a userstory                                             ara list --branch userstory {name_of_userstory}
-  > list parent businessgoal artefact of a userstory                                 ara list --branch userstory {name_of_userstory} --include-extension .businessgoal
+  > list children artefacts of a userstory                                           ara list userstory {name_of_userstory} --children
+  > list parent artefacts of a userstory                                             ara list userstory {name_of_userstory} --branch
+  > list parent businessgoal artefact of a userstory                                 ara list userstory {name_of_userstory} --branch --include-extension .businessgoal
   > print any artefact template for e.g. a feature file template in the terminal:    ara template feature
 
 ara prompt templates examples:
@@ -179,25 +193,32 @@ ara chat examples:
   > load selected templates in config_prompt_templates.md for the task {task_name}:  ara prompt load task {task_name}
   > create and send configured prompt of the task {task_name} to the configured LLM: ara prompt send task {task_name}
   > extract the selected LLM response in task.exploration.md and save to disk:       ara prompt extract task {task_name}
+
+ara config examples:
+  > show current configuration status:                                               ara config show
+  > show only LLM configurations:                                                    ara config show --llm
+  > reset LLM configurations to defaults:                                            ara config reset --llm-config -y
+  > preview reset without making changes:                                            ara config reset --all --dry-run
         """,
         no_args_is_help=True,
         add_completion=True,
-        rich_markup_mode="rich"
+        rich_markup_mode="rich",
     )
 
     @app.callback(invoke_without_command=True)
     def main(
         ctx: typer.Context,
         version: Optional[bool] = typer.Option(
-            None, "--version", "-v",
+            None,
+            "--version",
+            "-v",
             callback=version_callback,
             is_eager=True,
-            help="Show version and exit"
+            help="Show version and exit",
         ),
         debug: bool = typer.Option(
-            False, "--debug",
-            help="Enable debug mode for detailed error output"
-        )
+            False, "--debug", help="Enable debug mode for detailed error output"
+        ),
     ):
         """The ara cli terminal tool is a management tool for classified ara artefacts."""
         debug_mode = is_debug_mode_enabled()
@@ -211,9 +232,20 @@ ara chat examples:
         # Check for ara directory before executing any command
         # Skip check for commands that don't require ara directory
         commands_requiring_ara = {
-            'create', 'delete', 'rename', 'list', 'list-tags', 'prompt',
-            'read', 'reconnect', 'read-status', 'read-user', 'set-status',
-            'set-user', 'scan', 'autofix'
+            "create",
+            "delete",
+            "rename",
+            "list",
+            "list-tags",
+            "prompt",
+            "read",
+            "reconnect",
+            "read-status",
+            "read-user",
+            "set-status",
+            "set-user",
+            "scan",
+            "autofix",
         }
 
         if ctx.invoked_subcommand in commands_requiring_ara:
@@ -230,6 +262,8 @@ ara chat examples:
     register_template_cli(app)
     register_fetch_templates_cli(app)
     register_fetch_scripts_cli(app)
+    register_fetch_agents_cli(app)
+    register_fetch_cli(app)
     register_read_cli(app)
     register_reconnect_cli(app)
     register_read_status_cli(app)
@@ -241,6 +275,8 @@ ara chat examples:
     register_autofix_cli(app)
     register_extract_cli(app)
     register_load_cli(app)
+    register_config_cli(app)
+    register_convert_cli(app)
 
     return app
 

{ara_cli-0.1.11.3 → ara_cli-0.1.13.5}/ara_cli/ara_command_action.py

@@ -23,24 +23,33 @@ def create_action(args):
     from ara_cli.artefact_reader import ArtefactReader
     from ara_cli.artefact_fuzzy_search import find_closest_rule
 
-    check_validity(Classifier.is_valid_classifier(args.classifier), "Invalid classifier provided. Please provide a valid classifier.")
-    check_validity(is_valid_filename(args.parameter), "Invalid filename provided. Please provide a valid filename.")
+    check_validity(Classifier.is_valid_classifier(args.classifier),
+                   "Invalid classifier provided. Please provide a valid classifier.")
+    check_validity(is_valid_filename(args.parameter),
+                   "Invalid filename provided. Please provide a valid filename.")
 
     def handle_parent_arguments(args):
-        parent_classifier = args.parent_classifier if hasattr(args, "parent_classifier") else None
-        parent_name = args.parent_name if hasattr(args, "parent_name") else None
+        parent_classifier = args.parent_classifier if hasattr(
+            args, "parent_classifier") else None
+        parent_name = args.parent_name if hasattr(
+            args, "parent_name") else None
         rule = args.rule if hasattr(args, 'rule') else None
         invalid_classifier_message = "Invalid parent classifier provided. Please provide a valid classifier"
         invalid_name_message = "Invalid filename provided for parent. Please provide a valid filename."
         if parent_classifier and parent_name and rule:
-            check_validity(Classifier.is_valid_classifier(parent_classifier), invalid_classifier_message)
-            check_validity(is_valid_filename(parent_name), invalid_name_message)
-            parent_artefact = ArtefactReader.read_artefact(artefact_name=parent_name, classifier=parent_classifier)
+            check_validity(Classifier.is_valid_classifier(
+                parent_classifier), invalid_classifier_message)
+            check_validity(is_valid_filename(
+                parent_name), invalid_name_message)
+            parent_artefact = ArtefactReader.read_artefact(
+                artefact_name=parent_name, classifier=parent_classifier)
             rule = find_closest_rule(parent_artefact, rule)
             return parent_classifier, parent_name, rule
         if parent_classifier and parent_name:
-            check_validity(Classifier.is_valid_classifier(parent_classifier), invalid_classifier_message)
-            check_validity(is_valid_filename(parent_name), invalid_name_message)
+            check_validity(Classifier.is_valid_classifier(
+                parent_classifier), invalid_classifier_message)
+            check_validity(is_valid_filename(
+                parent_name), invalid_name_message)
             return parent_classifier, parent_name, rule
         return None, None, None
 
@@ -61,7 +70,8 @@ def create_action(args):
         return
 
     artefact_creator = ArtefactCreator()
-    artefact_creator.run(args.parameter, args.classifier, parent_classifier, parent_name, rule)
+    artefact_creator.run(args.parameter, args.classifier,
+                         parent_classifier, parent_name, rule)
 
 
 @handle_errors(context="delete action", error_handler=error_handler)
@@ -78,22 +88,33 @@ def rename_action(args):
     from ara_cli.classifier import Classifier
     from ara_cli.filename_validator import is_valid_filename
 
-    check_validity(is_valid_filename(args.parameter), "Invalid filename provided. Please provide a valid filename.")
-    check_validity(Classifier.is_valid_classifier(args.classifier), "Invalid classifier provided. Please provide a valid classifier.")
-    check_validity(is_valid_filename(args.aspect), "Invalid new filename provided. Please provide a valid filename.")
+    check_validity(is_valid_filename(args.parameter),
+                   "Invalid filename provided. Please provide a valid filename.")
+    check_validity(Classifier.is_valid_classifier(args.classifier),
+                   "Invalid classifier provided. Please provide a valid classifier.")
+    check_validity(is_valid_filename(
+        args.aspect), "Invalid new filename provided. Please provide a valid filename.")
 
     artefact_renamer = ArtefactRenamer()
     artefact_renamer.rename(args.parameter, args.aspect, args.classifier)
 
 
+def _execute_list_method(method, classifier, artefact_name, list_filter, flag_name):
+    """Helper function to validate and execute list methods."""
+    if not classifier or not artefact_name:
+        raise AraError(
+            f"Both classifier and artefact_name are required for --{flag_name}"
+        )
+    method(classifier=classifier, artefact_name=artefact_name, list_filter=list_filter)
+
+
 @handle_errors(context="rename action", error_handler=error_handler)
 def list_action(args):
     from ara_cli.artefact_lister import ArtefactLister
     from ara_cli.list_filter import ListFilter
 
-    branch_classifier, branch_artefact_name = args.branch_args
-    children_classifier, children_artefact_name = args.children_args
-    data_classifier, data_artefact_name = args.data_args
+    classifier = args.classifier
+    artefact_name = args.artefact_name
 
     artefact_lister = ArtefactLister()
 
@@ -106,29 +127,27 @@ def list_action(args):
         exclude_tags=args.exclude_tags
     )
 
-    if branch_classifier and branch_artefact_name:
-        artefact_lister.list_branch(
-            classifier=branch_classifier,
-            artefact_name=branch_artefact_name,
-            list_filter=list_filter
-        )
-        return
-
-    if children_classifier and children_artefact_name:
-        artefact_lister.list_children(
-            classifier=children_classifier,
-            artefact_name=children_artefact_name,
-            list_filter=list_filter
-        )
-        return
+    # Map flags to their corresponding methods
+    flag_method_map = {
+        "branch": (args.branch, artefact_lister.list_branch),
+        "children": (args.children, artefact_lister.list_children),
+        "data": (args.data, artefact_lister.list_data),
+    }
 
-    if data_classifier and data_artefact_name:
-        artefact_lister.list_data(
-            classifier=data_classifier,
-            artefact_name=data_artefact_name,
-            list_filter=list_filter
+    for flag_name, (flag_value, method) in flag_method_map.items():
+        if flag_value:
+            _execute_list_method(
+                method, classifier, artefact_name, list_filter, flag_name
+            )
+            return
+
+    # If both classifier and artefact_name are present, but no specific action flag (branch, children, data)
+    # was provided, raise an error as per requirements.
+    if classifier and artefact_name:
+        raise AraError(
+            f"To list specific info for '{classifier} {artefact_name}', "
+            "you must provide one of: --children, --branch, or --data."
         )
-        return
 
     artefact_lister.list_files(list_filter=list_filter)
 
@@ -171,8 +190,10 @@ def prompt_action(args):
     from ara_cli.classifier import Classifier
     from ara_cli.filename_validator import is_valid_filename
 
-    check_validity(Classifier.is_valid_classifier(args.classifier), "Invalid classifier provided. Please provide a valid classifier.")
-    check_validity(is_valid_filename(args.parameter), "Invalid filename provided. Please provide a valid filename.")
+    check_validity(Classifier.is_valid_classifier(args.classifier),
+                   "Invalid classifier provided. Please provide a valid classifier.")
+    check_validity(is_valid_filename(args.parameter),
+                   "Invalid filename provided. Please provide a valid filename.")
 
     classifier = args.classifier
     param = args.parameter
@@ -207,7 +228,8 @@ def prompt_action(args):
         from ara_cli.update_config_prompt import update_artefact_config_prompt_files
         extract_and_save_prompt_results(classifier, param, write=write)
         print(f"automatic update after extract")
-        update_artefact_config_prompt_files(classifier, param, automatic_update=True)
+        update_artefact_config_prompt_files(
+            classifier, param, automatic_update=True)
 
     def handle_chat():
         from ara_cli.prompt_chat import initialize_prompt_chat_mode
@@ -216,11 +238,13 @@ def prompt_action(args):
         output_mode = args.output_mode
         append_strings = args.append
         restricted = args.restricted
-        initialize_prompt_chat_mode(classifier, param, chat_name, reset=reset, output_mode=output_mode, append_strings=append_strings, restricted=restricted)
+        initialize_prompt_chat_mode(classifier, param, chat_name, reset=reset,
+                                    output_mode=output_mode, append_strings=append_strings, restricted=restricted)
 
     def handle_update():
         from ara_cli.update_config_prompt import update_artefact_config_prompt_files
-        update_artefact_config_prompt_files(classifier, param, automatic_update=True)
+        update_artefact_config_prompt_files(
+            classifier, param, automatic_update=True)
 
     command_dispatcher = {
         'init': handle_init,
@@ -255,7 +279,8 @@ def chat_action(args):
     chat_file_path = join(cwd, chat_name)
 
     with suppress_stdout(output_mode):
-        chat = Chat(chat_file_path, reset=reset) if not restricted else Chat(chat_file_path, reset=reset, enable_commands=whitelisted_commands)
+        chat = Chat(chat_file_path, reset=reset) if not restricted else Chat(
+            chat_file_path, reset=reset, enable_commands=whitelisted_commands)
 
     if append_strings:
         chat.append_strings(append_strings)
@@ -305,7 +330,8 @@ def load_action(args):
     default_pattern = default_patterns.get(template_type)
 
     if not template_name and not default_pattern:
-        raise AraError(f"A template name is required for template type '{template_type}'.")
+        raise AraError(
+            f"A template name is required for template type '{template_type}'.")
 
     loader = TemplateLoader()  # No chat instance for CLI context
     success = loader.load_template(
@@ -324,7 +350,8 @@ def template_action(args):
     from ara_cli.classifier import Classifier
     from ara_cli.template_manager import TemplatePathManager
 
-    check_validity(Classifier.is_valid_classifier(args.classifier), "Invalid classifier provided. Please provide a valid classifier.")
+    check_validity(Classifier.is_valid_classifier(args.classifier),
+                   "Invalid classifier provided. Please provide a valid classifier.")
 
     template_manager = TemplatePathManager()
     content = template_manager.get_template_content(args.classifier)
@@ -345,9 +372,11 @@ def fetch_templates_action(args):
 
     subdirs = ["commands", "rules", "intentions", "blueprints"]
 
-    os.makedirs(join(prompt_templates_dir, "global-prompt-modules"), exist_ok=True)
+    os.makedirs(join(prompt_templates_dir,
+                "global-prompt-modules"), exist_ok=True)
     for subdir in subdirs:
-        target_dir = join(prompt_templates_dir, "global-prompt-modules", subdir)
+        target_dir = join(prompt_templates_dir,
+                          "global-prompt-modules", subdir)
         source_dir = join(global_prompt_templates_path, subdir)
         os.makedirs(target_dir, exist_ok=True)
         for item in os.listdir(source_dir):
@@ -356,7 +385,8 @@ def fetch_templates_action(args):
             shutil.copy2(source, target)
 
     custom_prompt_templates_subdir = config.custom_prompt_templates_subdir
-    local_prompt_modules_dir = join(prompt_templates_dir, custom_prompt_templates_subdir)
+    local_prompt_modules_dir = join(
+        prompt_templates_dir, custom_prompt_templates_subdir)
     os.makedirs(local_prompt_modules_dir, exist_ok=True)
     for subdir in subdirs:
         os.makedirs(join(local_prompt_modules_dir, subdir), exist_ok=True)
@@ -459,9 +489,11 @@ def read_status_action(args):
     artefact_info = file_classifier.classify_files()
     artefact_info_dicts = artefact_info.get(classifier, [])
 
-    all_artefact_names = [artefact_info["title"] for artefact_info in artefact_info_dicts]
+    all_artefact_names = [artefact_info["title"]
+                          for artefact_info in artefact_info_dicts]
     if artefact_name not in all_artefact_names:
-        suggest_close_name_matches(artefact_name, all_artefact_names, report_as_error=True)
+        suggest_close_name_matches(
+            artefact_name, all_artefact_names, report_as_error=True)
         return
 
     artefact_info = next(filter(
@@ -492,9 +524,11 @@ def read_user_action(args):
     artefact_info = file_classifier.classify_files()
     artefact_info_dicts = artefact_info.get(classifier, [])
 
-    all_artefact_names = [artefact_info["title"] for artefact_info in artefact_info_dicts]
+    all_artefact_names = [artefact_info["title"]
+                          for artefact_info in artefact_info_dicts]
     if artefact_name not in all_artefact_names:
-        suggest_close_name_matches(artefact_name, all_artefact_names, report_as_error=True)
+        suggest_close_name_matches(
+            artefact_name, all_artefact_names, report_as_error=True)
         return
 
     artefact_info = next(filter(
@@ -529,12 +563,14 @@ def set_status_action(args):
     if new_status.startswith('@'):
         new_status = new_status.lstrip('@')
 
-    check_validity(new_status in status_tags, "Invalid status provided. Please provide a valid status.")
+    check_validity(new_status in status_tags,
+                   "Invalid status provided. Please provide a valid status.")
 
     file_classifier = FileClassifier(os)
     classified_artefacts_info = file_classifier.classify_files()
     classified_artefact_dict = classified_artefacts_info.get(classifier, [])
-    all_artefact_names = [artefact_info["title"] for artefact_info in classified_artefact_dict]
+    all_artefact_names = [artefact_info["title"]
+                          for artefact_info in classified_artefact_dict]
 
     if artefact_name not in all_artefact_names:
         suggest_close_name_matches(artefact_name, all_artefact_names)
@@ -554,7 +590,8 @@ def set_status_action(args):
     with open(f"{artefact_info['file_path']}", 'w', encoding='utf-8') as file:
         file.write(serialized_content)
 
-    print(f"Status of task '{artefact_name}' has been updated to '{new_status}'.")
+    print(
+        f"Status of task '{artefact_name}' has been updated to '{new_status}'.")
 
 
 @handle_errors(context="set-user action", error_handler=error_handler)
@@ -572,7 +609,8 @@ def set_user_action(args):
     file_classifier = FileClassifier(os)
     classified_artefacts_info = file_classifier.classify_files()
     classified_artefact_dict = classified_artefacts_info.get(classifier, [])
-    all_artefact_names = [artefact_info["title"] for artefact_info in classified_artefact_dict]
+    all_artefact_names = [artefact_info["title"]
+                          for artefact_info in classified_artefact_dict]
 
     if artefact_name not in all_artefact_names:
         suggest_close_name_matches(artefact_name, all_artefact_names)
@@ -673,4 +711,4 @@ def extract_action(args):
         write=write,
         output=lambda msg: print(msg, file=sys.stdout)
     )
-    command.execute()
+    command.execute()