ara-cli 0.1.10.5__py3-none-any.whl → 0.1.14.0__py3-none-any.whl

ara_cli/__init__.py

@@ -1,20 +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"]
-
+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'
+YELLOW = "\033[93m"
+RESET = "\033[0m"
 
 
 def format_warning(message, category, *args, **kwargs):
-    return f'{YELLOW}[WARNING] {category.__name__}: {message}{RESET}\n'
+    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", ".xlsx", ".xls", ".pptx"]

ara_cli/__main__.py

@@ -1,9 +1,7 @@
 import typer
-import sys
 import os
 from typing import Optional
 from os import getenv
-from ara_cli.error_handler import AraError
 from ara_cli.version import __version__
 from ara_cli import error_handler
 from ara_cli.ara_subcommands.create import register as register_create_cli
@@ -14,18 +12,29 @@ 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
 
 
 def version_callback(value: bool):
@@ -36,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):
@@ -45,95 +54,74 @@ def configure_debug_mode(debug: bool, env_debug_mode: bool):
         error_handler.debug_mode = True
 
 
-def find_ara_directory_root():
-    """Find the root ara directory by traversing up the directory tree."""
-    current_dir = os.getcwd()
-    
-    # Check if we're already inside an ara directory structure
-    path_parts = current_dir.split(os.sep)
-    
-    # Look for 'ara' in the path parts
-    if 'ara' in path_parts:
-        ara_index = path_parts.index('ara')
-        # Reconstruct path up to and including 'ara'
-        ara_root_parts = path_parts[:ara_index + 1]
-        potential_ara_root = os.sep.join(ara_root_parts)
-        if os.path.exists(potential_ara_root) and os.path.isdir(potential_ara_root):
-            return potential_ara_root
-    
-    # If not inside ara directory, check current directory and parents
-    check_dir = current_dir
-    while check_dir != os.path.dirname(check_dir):  # Stop at filesystem root
-        ara_path = os.path.join(check_dir, 'ara')
-        if os.path.exists(ara_path) and os.path.isdir(ara_path):
-            return ara_path
-        check_dir = os.path.dirname(check_dir)
-    
-    return None
-
-
 def check_ara_directory_exists():
     """Check if ara directory exists or if we're inside ara directory tree."""
-    return find_ara_directory_root() is not None
+    return DirectoryNavigator.find_ara_directory_root() is not None
 
 
 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:
         response = input().strip()
     except (EOFError, KeyboardInterrupt):
         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:
             # Create main ara directory
             os.makedirs(ara_path, exist_ok=True)
-            
+
             # Create subdirectories for artefact types
             for subdir in subdirectories:
                 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()
-            
+
             # 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}")
             typer.echo(f"Created default configuration at {config_file_path}")
             return True
-            
+
         except OSError as e:
             typer.echo(f"Error creating ara directory: {e}", err=True)
             raise typer.Exit(1)
@@ -181,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:
@@ -204,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()
@@ -232,15 +228,26 @@ ara chat examples:
         if ctx.invoked_subcommand is None:
             ctx.get_help()
             ctx.exit()
-        
+
         # 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', 'classifier-directory', '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:
             requires_ara_directory()
 
@@ -254,6 +261,9 @@ ara chat examples:
     register_chat_cli(app)
     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)
@@ -265,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
 
@@ -281,4 +293,4 @@ def cli():
 
 
 if __name__ == "__main__":
-    cli()
+    cli()