ara-cli 0.1.9.96__tar.gz → 0.1.10.0__tar.gz

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: ara_cli
-Version: 0.1.9.96
+Version: 0.1.10.0
 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: llama-index
 Requires-Dist: llama-index-llms-openai

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/ara_cli/__init__.py

@@ -14,7 +14,7 @@ RESET = '\033[0m'
 
 
 def format_warning(message, category, *args, **kwargs):
-    return f'{YELLOW}{category.__name__}: {message}{RESET}\n'
+    return f'{YELLOW}[WARNING] {category.__name__}: {message}{RESET}\n'
 
 
 warnings.formatwarning = format_warning

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/ara_cli/__main__.py

@@ -21,7 +21,8 @@ from ara_cli.ara_command_action import (
     classifier_directory_action,
     scan_action,
     autofix_action,
-    extract_action
+    extract_action,
+    load_action
 )
 from . import error_handler
 import argcomplete
@@ -49,7 +50,8 @@ def define_action_mapping():
         "classifier-directory": classifier_directory_action,
         "scan": scan_action,
         "autofix": autofix_action,
-        "extract": extract_action
+        "extract": extract_action,
+        "load": load_action
     }
 
 

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/ara_cli/ara_command_action.py

@@ -261,6 +261,59 @@ def chat_action(args):
     chat.start()
 
 
+def _find_chat_file(chat_name: str) -> str | None:
+    """Resolves the chat file path based on common naming conventions."""
+    # Logic from setup_chat for finding existing files.
+    if os.path.exists(chat_name) and os.path.isfile(chat_name):
+        return chat_name
+
+    chat_name_md = f"{chat_name}.md"
+    if os.path.exists(chat_name_md) and os.path.isfile(chat_name_md):
+        return chat_name_md
+
+    chat_name_chat_md = f"{chat_name}_chat.md"
+    if os.path.exists(chat_name_chat_md) and os.path.isfile(chat_name_chat_md):
+        return chat_name_chat_md
+
+    return None
+
+
+@handle_errors(context="load action", error_handler=error_handler)
+def load_action(args):
+    from ara_cli.template_loader import TemplateLoader
+
+    chat_name = args.chat_name
+    template_type = args.template_type
+    template_name = args.template_name
+
+    chat_file_path = _find_chat_file(chat_name)
+
+    if not chat_file_path:
+        raise AraError(f"Chat file for '{chat_name}' not found.")
+
+    default_patterns = {
+        "rules": "*.rules.md",
+        "intention": "*.intention.md",
+        "commands": "*.commands.md"
+    }
+
+    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}'.")
+
+    loader = TemplateLoader()  # No chat instance for CLI context
+    success = loader.load_template(
+        template_name=template_name,
+        template_type=template_type,
+        chat_file_path=chat_file_path,
+        default_pattern=default_pattern
+    )
+
+    if not success:
+        sys.exit(1)
+
+
 @handle_errors(context="template action", error_handler=error_handler)
 def template_action(args):
     from ara_cli.classifier import Classifier
@@ -615,4 +668,4 @@ def extract_action(args):
         write=write,
         output=lambda msg: print(msg, file=sys.stdout)
     )
-    command.execute()
+    command.execute()

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/ara_cli/ara_command_parser.py

@@ -7,7 +7,8 @@ from ara_cli.commandline_completer import (
     StatusCompleter,
 )
 from ara_cli.template_manager import SpecificationBreakdownAspects
-
+import os
+import glob
 
 classifiers = Classifier.ordered_classifiers()
 aspects = SpecificationBreakdownAspects.VALID_ASPECTS
@@ -191,6 +192,44 @@ def list_tags_parser(subparsers):
     )
 
 
+class TemplateNameCompleter:
+    """Provides command-line completion for template names."""
+    def __call__(self, prefix, parsed_args, **kwargs):
+        import os
+        from ara_cli.template_loader import TemplateLoader
+
+        if not hasattr(parsed_args, 'template_type'):
+            return []
+
+        template_type = parsed_args.template_type
+        context_path = os.getcwd()
+
+        loader = TemplateLoader()
+        templates = loader.get_available_templates(template_type, context_path)
+
+        return [t for t in templates if t.startswith(prefix)]
+
+
+def load_parser(subparsers):
+    load_parser = subparsers.add_parser(
+        "load", help="Load a template into a chat file."
+    )
+    load_parser.add_argument(
+        "chat_name", help="Name of the chat file to load template into (without extension)."
+    )
+    load_parser.add_argument(
+        "template_type",
+        choices=['rules', 'intention', 'commands', 'blueprint'],
+        help="Type of template to load."
+    )
+    load_parser.add_argument(
+        "template_name",
+        nargs='?',
+        default="",
+        help="Name of the template to load. Supports wildcards and 'global/' prefix."
+    ).completer = TemplateNameCompleter()
+
+
 def add_chat_arguments(chat_parser):
     chat_parser.add_argument(
         "chat_name",
@@ -560,6 +599,7 @@ def action_parser():
     classifier_directory_parser(subparsers)
     scan_parser(subparsers)
     autofix_parser(subparsers)
+    load_parser(subparsers)
     extract_parser(subparsers)
 
-    return parser
+    return parser

{ara_cli-0.1.9.96 → ara_cli-0.1.10.0}/ara_cli/ara_config.py

@@ -10,6 +10,7 @@ import warnings
 
 DEFAULT_CONFIG_LOCATION = "./ara/.araconfig/ara_config.json"
 
+
 class LLMConfigItem(BaseModel):
     provider: str
     model: str
@@ -17,6 +18,7 @@ class LLMConfigItem(BaseModel):
     max_tokens: Optional[int] = None
     max_completion_tokens: Optional[int] = None
 
+
 class ARAconfig(BaseModel):
     ext_code_dirs: List[Dict[str, str]] = Field(default_factory=lambda: [
         {"source_dir": "./src"},
@@ -28,98 +30,104 @@ class ARAconfig(BaseModel):
     local_prompt_templates_dir: str = "./ara/.araconfig"
     custom_prompt_templates_subdir: Optional[str] = "custom-prompt-modules"
     local_ara_templates_dir: str = "./ara/.araconfig/templates/"
-    ara_prompt_given_list_includes: List[str] = Field(default_factory=lambda: [
-        "*.businessgoal",
-        "*.vision",
-        "*.capability",
-        "*.keyfeature",
-        "*.epic",
-        "*.userstory",
-        "*.example",
-        "*.feature",
-        "*.task",
-        "*.py",
-        "*.md",
-        "*.png",
-        "*.jpg",
-        "*.jpeg",
-    ])
-    llm_config: Dict[str, LLMConfigItem] = Field(default_factory=lambda: {
-        "gpt-5": LLMConfigItem(
-            provider="openai",
-            model="openai/gpt-5",
-            temperature=1,
-            max_completion_tokens=16000
-        ),
-        "gpt-5-mini": LLMConfigItem(
-            provider="openai",
-            model="openai/gpt-5-mini-2025-08-07",
-            temperature=1
-        ),
-        "gpt-4o": LLMConfigItem(
-            provider="openai",
-            model="openai/gpt-4o",
-            temperature=0.8,
-            max_tokens=16000
-        ),
-        "gpt-4.1": LLMConfigItem(
-            provider="openai",
-            model="openai/gpt-4.1",
-            temperature=0.8,
-            max_tokens=16000
-        ),
-        "o3-mini": LLMConfigItem(
-            provider="openai",
-            model="openai/o3-mini",
-            temperature=1.0,
-            max_tokens=8000
-        ),
-        "opus-4": LLMConfigItem(
-            provider="anthropic",
-            model="anthropic/claude-opus-4-20250514",
-            temperature=0.5,
-            max_tokens=32000
-        ),
-        "sonnet-4": LLMConfigItem(
-            provider="anthropic",
-            model="anthropic/claude-sonnet-4-20250514",
-            temperature=0.5,
-            max_tokens=32000
-        ),
-        "together-ai-llama-2": LLMConfigItem(
-            provider="together_ai",
-            model="together_ai/togethercomputer/llama-2-70b",
-            temperature=0.8,
-            max_tokens=4000
-        ),
-        "groq-llama-3": LLMConfigItem(
-            provider="groq",
-            model="groq/llama3-70b-8192",
-            temperature=0.8,
-            max_tokens=4000
-        )
-    })
+    ara_prompt_given_list_includes: List[str] = Field(
+        default_factory=lambda: [
+            "*.businessgoal",
+            "*.vision",
+            "*.capability",
+            "*.keyfeature",
+            "*.epic",
+            "*.userstory",
+            "*.example",
+            "*.feature",
+            "*.task",
+            "*.py",
+            "*.md",
+            "*.png",
+            "*.jpg",
+            "*.jpeg",
+        ]
+    )
+    llm_config: Dict[str, LLMConfigItem] = Field(
+        default_factory=lambda: {
+            "gpt-5": LLMConfigItem(
+                provider="openai",
+                model="openai/gpt-5",
+                temperature=1,
+                max_completion_tokens=16000,
+            ),
+            "gpt-5-mini": LLMConfigItem(
+                provider="openai", model="openai/gpt-5-mini-2025-08-07", temperature=1
+            ),
+            "gpt-4o": LLMConfigItem(
+                provider="openai",
+                model="openai/gpt-4o",
+                temperature=0.8,
+                max_tokens=16000,
+            ),
+            "gpt-4.1": LLMConfigItem(
+                provider="openai",
+                model="openai/gpt-4.1",
+                temperature=0.8,
+                max_tokens=16000,
+            ),
+            "o3-mini": LLMConfigItem(
+                provider="openai",
+                model="openai/o3-mini",
+                temperature=1.0,
+                max_tokens=8000,
+            ),
+            "opus-4": LLMConfigItem(
+                provider="anthropic",
+                model="anthropic/claude-opus-4-20250514",
+                temperature=0.5,
+                max_tokens=32000,
+            ),
+            "sonnet-4": LLMConfigItem(
+                provider="anthropic",
+                model="anthropic/claude-sonnet-4-20250514",
+                temperature=0.5,
+                max_tokens=32000,
+            ),
+            "together-ai-llama-2": LLMConfigItem(
+                provider="together_ai",
+                model="together_ai/togethercomputer/llama-2-70b",
+                temperature=0.8,
+                max_tokens=4000,
+            ),
+            "groq-llama-3": LLMConfigItem(
+                provider="groq",
+                model="groq/llama3-70b-8192",
+                temperature=0.8,
+                max_tokens=4000,
+            ),
+        }
+    )
     default_llm: Optional[str] = None
     extraction_llm: Optional[str] = None
 
-    @model_validator(mode='after')
-    def check_critical_fields(self) -> 'ARAconfig':
+    @model_validator(mode="after")
+    def check_critical_fields(self) -> "ARAconfig":
         """Check for empty critical fields and validate default_llm and extraction_llm."""
         critical_fields = {
-            'ext_code_dirs': [{"source_dir": "./src"}, {"source_dir": "./tests"}],
-            'local_ara_templates_dir': "./ara/.araconfig/templates/",
-            'local_prompt_templates_dir': "./ara/.araconfig",
-            'glossary_dir': "./glossary"
+            "ext_code_dirs": [{"source_dir": "./src"}, {"source_dir": "./tests"}],
+            "local_ara_templates_dir": "./ara/.araconfig/templates/",
+            "local_prompt_templates_dir": "./ara/.araconfig",
+            "glossary_dir": "./glossary",
         }
 
         for field, default_value in critical_fields.items():
             current_value = getattr(self, field)
             if not current_value:
-                print(f"Warning: Value for '{field}' is missing or empty. Using default.")
+                print(
+                    f"Warning: Value for '{field}' is missing or empty. Using default."
+                )
                 setattr(self, field, default_value)
-        
+
         if not self.llm_config:
-            print("Warning: 'llm_config' is empty. 'default_llm' and 'extraction_llm' cannot be set.")
+            print(
+                "Warning: 'llm_config' is empty. 'default_llm' and 'extraction_llm' cannot be set."
+            )
             self.default_llm = None
             self.extraction_llm = None
             return self
@@ -127,23 +135,34 @@ class ARAconfig(BaseModel):
         first_available_llm = next(iter(self.llm_config))
 
         if not self.default_llm:
-            print(f"Warning: 'default_llm' is not set. Defaulting to the first available model: '{first_available_llm}'.")
+            print(
+                f"Warning: 'default_llm' is not set. Defaulting to the first available model: '{first_available_llm}'."
+            )
             self.default_llm = first_available_llm
         elif self.default_llm not in self.llm_config:
-            print(f"Warning: The configured 'default_llm' ('{self.default_llm}') does not exist in 'llm_config'.")
-            print(f"-> Reverting to the first available model: '{first_available_llm}'.")
+            print(
+                f"Warning: The configured 'default_llm' ('{self.default_llm}') does not exist in 'llm_config'."
+            )
+            print(
+                f"-> Reverting to the first available model: '{first_available_llm}'."
+            )
             self.default_llm = first_available_llm
 
         if not self.extraction_llm:
-            print(f"Warning: 'extraction_llm' is not set. Setting it to the same as 'default_llm': '{self.default_llm}'.")
+            print(
+                f"Warning: 'extraction_llm' is not set. Setting it to the same as 'default_llm': '{self.default_llm}'."
+            )
             self.extraction_llm = self.default_llm
         elif self.extraction_llm not in self.llm_config:
-            print(f"Warning: The configured 'extraction_llm' ('{self.extraction_llm}') does not exist in 'llm_config'.")
+            print(
+                f"Warning: The configured 'extraction_llm' ('{self.extraction_llm}') does not exist in 'llm_config'."
+            )
             print(f"-> Reverting to the 'default_llm' value: '{self.default_llm}'.")
             self.extraction_llm = self.default_llm
-            
+
         return self
 
+
 # Function to ensure the necessary directories exist
 @lru_cache(maxsize=None)
 def ensure_directory_exists(directory: str):
@@ -153,6 +172,7 @@ def ensure_directory_exists(directory: str):
         print(f"New directory created at {directory}")
     return directory
 
+
 def handle_unrecognized_keys(data: dict) -> dict:
     """Removes unrecognized keys from the data and warns the user."""
     known_fields = set(ARAconfig.model_fields.keys())
@@ -211,13 +231,15 @@ def read_data(filepath: str) -> ARAconfig:
         return config
     except ValidationError as e:
         print("--- Configuration Error Detected ---")
-        print("Some settings in your configuration file are invalid. Attempting to fix them.")
-        
+        print(
+            "Some settings in your configuration file are invalid. Attempting to fix them."
+        )
+
         corrected_data = data.copy()
         defaults = ARAconfig().model_dump()
-        
-        error_fields = {err['loc'][0] for err in e.errors() if err['loc']}
-        
+
+        error_fields = {err["loc"][0] for err in e.errors() if err["loc"]}
+
         for field_name in error_fields:
             print(f"-> Field '{field_name}' is invalid and will be reverted to its default value.")
             if field_name in corrected_data:
@@ -228,15 +250,17 @@ def read_data(filepath: str) -> ARAconfig:
         final_config = ARAconfig(**corrected_data)
         save_data(filepath, final_config)
         print(f"Configuration has been corrected and saved to '{filepath}'.")
-        
+
         return final_config
 
+
 # Function to save the modified configuration back to the JSON file
 def save_data(filepath: str, config: ARAconfig):
     """Saves the Pydantic config model to a JSON file."""
     with open(filepath, "w", encoding="utf-8") as file:
         json.dump(config.model_dump(), file, indent=4)
 
+
 # Singleton for configuration management
 class ConfigManager:
     _config_instance = None
@@ -246,9 +270,9 @@ class ConfigManager:
         if cls._config_instance is None:
             cls._config_instance = read_data(filepath)
         return cls._config_instance
-    
+
     @classmethod
     def reset(cls):
         """Reset the configuration instance (useful for testing)."""
         cls._config_instance = None
-        read_data.cache_clear()
+        read_data.cache_clear()