ara-cli 0.1.9.87__py3-none-any.whl → 0.1.9.91__py3-none-any.whl

ara_cli/ara_config.py

@@ -14,6 +14,7 @@ class LLMConfigItem(BaseModel):
     model: str
     temperature: float = Field(ge=0.0, le=1.0)
     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: [
@@ -42,50 +43,62 @@ class ARAconfig(BaseModel):
         "*.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=16384
+            max_tokens=16000
         ),
         "gpt-4.1": LLMConfigItem(
             provider="openai",
             model="openai/gpt-4.1",
             temperature=0.8,
-            max_tokens=1024
+            max_tokens=16000
         ),
         "o3-mini": LLMConfigItem(
             provider="openai",
             model="openai/o3-mini",
             temperature=1.0,
-            max_tokens=1024
+            max_tokens=8000
         ),
         "opus-4": LLMConfigItem(
             provider="anthropic",
             model="anthropic/claude-opus-4-20250514",
-            temperature=0.8,
+            temperature=0.5,
             max_tokens=32000
         ),
         "sonnet-4": LLMConfigItem(
             provider="anthropic",
             model="anthropic/claude-sonnet-4-20250514",
-            temperature=0.8,
-            max_tokens=1024
+            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=1024
+            max_tokens=4000
         ),
         "groq-llama-3": LLMConfigItem(
             provider="groq",
             model="groq/llama3-70b-8192",
             temperature=0.8,
-            max_tokens=1024
+            max_tokens=4000
         )
     })
-    default_llm: Optional[str] = "gpt-4o"
+    default_llm: Optional[str] = "gpt-5"
+    extraction_llm: Optional[str] = "gpt-5-mini"
 
     @model_validator(mode='after')
     def check_critical_fields(self) -> 'ARAconfig':

ara_cli/chat.py

@@ -433,7 +433,7 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
                 break
 
         is_file_document = any(file_name_lower.endswith(ext)
-                               for ext in document_type_extensions)
+                                for ext in document_type_extensions)
 
         if is_file_document:
             return self.load_document_file(
@@ -532,7 +532,7 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
         return [x for x in glob.glob(text + '*')]
 
     def _retrieve_ara_config(self):
-        from ara_cli.prompt_handler import ConfigManager
+        from ara_cli.ara_config import ConfigManager
         return ConfigManager().get_config()
 
     def _retrieve_llm_config(self):
@@ -625,7 +625,7 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
     @cmd2.with_category(CATEGORY_LLM_CONTROL)
     def do_CHOOSE_MODEL(self, model_name):
         from ara_cli.prompt_handler import LLMSingleton
-        from ara_cli.ara_config import DEFAULT_CONFIG_LOCATION, save_data
+        from ara_cli.ara_config import DEFAULT_CONFIG_LOCATION, save_data, ConfigManager
         from ara_cli.directory_navigator import DirectoryNavigator
 
         original_dir = os.getcwd()
@@ -634,7 +634,11 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
         os.chdir('..')
 
         if not self._verify_llm_choice(model_name):
+            os.chdir(original_dir)
             return
+
+        ConfigManager.reset()
+        self.config = ConfigManager.get_config()
         self.config.default_llm = model_name
         save_data(filepath=DEFAULT_CONFIG_LOCATION, config=self.config)
 
@@ -642,11 +646,39 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
 
         os.chdir(original_dir)
 
+    @cmd2.with_category(CATEGORY_LLM_CONTROL)
+    def do_CHOOSE_EXTRACTION_MODEL(self, model_name):
+        """Choose the language model for extraction tasks."""
+        from ara_cli.ara_config import DEFAULT_CONFIG_LOCATION, save_data, ConfigManager
+        from ara_cli.directory_navigator import DirectoryNavigator
+
+        original_dir = os.getcwd()
+        navigator = DirectoryNavigator()
+        navigator.navigate_to_target()
+        os.chdir('..')
+
+        if not self._verify_llm_choice(model_name):
+            os.chdir(original_dir)
+            return
+
+        ConfigManager.reset()
+        self.config = ConfigManager.get_config()
+        self.config.extraction_llm = model_name
+        save_data(filepath=DEFAULT_CONFIG_LOCATION, config=self.config)
+        print(f"Extraction model switched to '{model_name}'")
+
+        os.chdir(original_dir)
+
     @cmd2.with_category(CATEGORY_LLM_CONTROL)
     def do_CURRENT_MODEL(self, _):
-        from ara_cli.prompt_handler import LLMSingleton
+        """Displays the current default and extraction language models."""
+        from ara_cli.ara_config import ConfigManager
 
-        print(LLMSingleton.get_model())
+        ConfigManager.reset()
+        config = self._retrieve_ara_config()
+        
+        print(f"Current Reasoning LLM (default_llm): {config.default_llm}")
+        print(f"Current Extraction LLM (extraction_llm): {config.extraction_llm}")
 
     def _complete_llms(self, text, line, begidx, endidx):
         llm_config = self._retrieve_llm_config()
@@ -662,6 +694,9 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
     def complete_CHOOSE_MODEL(self, text, line, begidx, endidx):
         return self._complete_llms(text, line, begidx, endidx)
 
+    def complete_CHOOSE_EXTRACTION_MODEL(self, text, line, begidx, endidx):
+        return self._complete_llms(text, line, begidx, endidx)
+
     @cmd2.with_category(CATEGORY_CHAT_CONTROL)
     def do_NEW(self, chat_name):
         """Create a new chat. Optionally provide a chat name in-line: NEW new_chat"""
@@ -859,3 +894,95 @@ Start chatting (type 'HELP'/'h' for available commands, 'QUIT'/'q' to exit chat
             completions = [classifier for classifier in classifiers if classifier.startswith(text)]
 
         return completions
+
+
+    def _get_plural_template_type(self, template_type: str) -> str:
+        """Determines the plural form of a template type."""
+        plurals = {"commands": "commands", "rules": "rules"}
+        return plurals.get(template_type, f"{template_type}s")
+
+    def _find_project_root(self) -> str | None:
+        """
+        Finds the project root by searching for an 'ara' directory,
+        starting from the chat file's directory and moving upwards.
+        """
+        current_dir = os.path.dirname(self.chat_name)
+        while True:
+            if os.path.isdir(os.path.join(current_dir, 'ara')):
+                return current_dir
+            parent_dir = os.path.dirname(current_dir)
+            if parent_dir == current_dir:  # Reached the filesystem root
+                return None
+            current_dir = parent_dir
+
+    def _gather_templates_from_path(self, search_path: str, templates_set: set, prefix: str = ""):
+        """
+        Scans a given path for items and adds them to the provided set,
+        optionally prepending a prefix.
+        """
+        import glob
+        if not os.path.isdir(search_path):
+            return
+        for path in glob.glob(os.path.join(search_path, '*')):
+            templates_set.add(f"{prefix}{os.path.basename(path)}")
+
+    def _get_available_templates(self, template_type: str) -> list[str]:
+        """
+        Scans for available global and project-local custom templates.
+        This method safely searches for template files without changing the
+        current directory, making it safe for use in autocompleters.
+
+        Args:
+            template_type: The type of template to search for (e.g., 'rules').
+
+        Returns:
+            A sorted list of unique template names. Global templates are
+            prefixed with 'global/'.
+        """
+        from ara_cli.template_manager import TemplatePathManager
+
+        plural_type = self._get_plural_template_type(template_type)
+        templates = set()
+
+        # 1. Find Global Templates
+        try:
+            global_base_path = TemplatePathManager.get_template_base_path()
+            global_template_dir = os.path.join(global_base_path, "prompt-modules", plural_type)
+            self._gather_templates_from_path(global_template_dir, templates, prefix="global/")
+        except Exception:
+            pass  # Silently ignore if global templates are not found
+
+        # 2. Find Local Custom Templates
+        try:
+            project_root = self._find_project_root()
+            if project_root:
+                local_templates_base = os.path.join(project_root, self.config.local_prompt_templates_dir)
+                custom_dir = os.path.join(local_templates_base, self.config.custom_prompt_templates_subdir, plural_type)
+                self._gather_templates_from_path(custom_dir, templates)
+        except Exception:
+            pass  # Silently ignore if local templates cannot be resolved
+
+        return sorted(list(templates))
+
+    def _template_completer(self, text: str, template_type: str) -> list[str]:
+        """Generic completer for different template types."""
+        available_templates = self._get_available_templates(template_type)
+        if not text:
+            return available_templates
+        return [t for t in available_templates if t.startswith(text)]
+
+    def complete_LOAD_RULES(self, text, line, begidx, endidx):
+        """Completer for the LOAD_RULES command."""
+        return self._template_completer(text, "rules")
+
+    def complete_LOAD_INTENTION(self, text, line, begidx, endidx):
+        """Completer for the LOAD_INTENTION command."""
+        return self._template_completer(text, "intention")
+
+    def complete_LOAD_COMMANDS(self, text, line, begidx, endidx):
+        """Completer for the LOAD_COMMANDS command."""
+        return self._template_completer(text, "commands")
+
+    def complete_LOAD_BLUEPRINT(self, text, line, begidx, endidx):
+        """Completer for the LOAD_BLUEPRINT command."""
+        return self._template_completer(text, "blueprint")

ara_cli/prompt_extractor.py

@@ -192,7 +192,7 @@ def handle_existing_file(filename, block_content, skip_query=False):
         messages = [{"role": "user", "content": prompt_text}]
         response = ""
 
-        for chunk in send_prompt(messages):
+        for chunk in send_prompt(messages, purpose='extraction'):
             content = chunk.choices[0].delta.content
             if content:
                 response += content

ara_cli/prompt_handler.py

@@ -71,29 +71,54 @@ def read_string_from_file(path):
         text = file.read()
     return text
 
+def _get_model_config(purpose: str) -> dict:
+    """
+    Resolves model ID by purpose and returns its prepared configuration for LiteLLM.
+    """
+    config = ConfigManager().get_config()
+
+    # Determine which model to use based on the purpose
+    model_id = config.default_llm
+    if purpose == 'extraction':
+        # Use extraction_llm if it exists, otherwise fall back to default_llm
+        model_id = getattr(config, 'extraction_llm', config.default_llm)
+
+    selected_config = config.llm_config.get(str(model_id))
+
+    if not selected_config:
+        raise ValueError(f"No configuration found for model '{model_id}' used for '{purpose}'")
+
+    config_parameters = selected_config.model_dump(exclude_none=True)
+    config_parameters.pop("provider", None)
+    
+    return config_parameters
+
+
+def _is_valid_message(message: dict) -> bool:
+    """
+    Checks if a message in a prompt is valid (i.e., not empty).
+    It handles both string content and list content (for multimodal inputs).
+    """
+    content = message.get('content')
+
+    if isinstance(content, str):
+        return content.strip() != ''
+    
+    if isinstance(content, list):
+        # For multimodal content, check if there's at least one non-empty text part.
+        return any(
+            item.get('type') == 'text' and item.get('text', '').strip() != ''
+            for item in content
+        )
+        
+    return False
+
+
+def send_prompt(prompt, purpose='default'):
+    """Prepares and sends a prompt to the LLM, streaming the response."""
+    config_parameters = _get_model_config(purpose)
 
-def send_prompt(prompt):
-    chat = LLMSingleton.get_instance()
-
-    config_parameters = chat.config_parameters.copy()
-    if "provider" in config_parameters:
-        del config_parameters["provider"]
-
-    filtered_prompt = list(filter(
-        lambda msg: (
-            msg.get('content') and
-            isinstance(msg['content'], list) and
-            any(
-                item.get('type') == 'text' and
-                item.get('text', '').strip() != ''
-                for item in msg['content']
-            )
-        ) or (
-            isinstance(msg.get('content'), str) and
-            msg['content'].strip() != ''
-        ),
-        prompt
-    ))
+    filtered_prompt = [msg for msg in prompt if _is_valid_message(msg)]
 
     completion = litellm.completion(
         **config_parameters,

ara_cli/version.py

@@ -1,2 +1,2 @@
 # version.py
-__version__ = "0.1.9.87"   # fith parameter like .0 for local install test purposes only. official numbers should be 4 digit numbers
+__version__ = "0.1.9.91"   # fith parameter like .0 for local install test purposes only. official numbers should be 4 digit numbers

{ara_cli-0.1.9.87.dist-info → ara_cli-0.1.9.91.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ara_cli
-Version: 0.1.9.87
+Version: 0.1.9.91
 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: litellm

{ara_cli-0.1.9.87.dist-info → ara_cli-0.1.9.91.dist-info}/RECORD

@@ -2,7 +2,7 @@ ara_cli/__init__.py,sha256=0zl7IegxTid26EBGLav_fXZ4CCIV3H5TfAoFQiOHjvg,148
 ara_cli/__main__.py,sha256=J5DCDLRZ6UcpYwM1-NkjaLo4PTetcSj2dB4HrrftkUw,2064
 ara_cli/ara_command_action.py,sha256=_LHE2V5hbJxN7ccYiptuPktRfbTnXmQEt_D_FxDBlBY,22456
 ara_cli/ara_command_parser.py,sha256=I-e9W-QwTIMKMzlHycSlCWCyBFQfiFYvGre1XsDbrFI,20573
-ara_cli/ara_config.py,sha256=KVITofnYlIVyhf50qwUO5fu8vlxjDwRjPyKzqEhEC_M,6982
+ara_cli/ara_config.py,sha256=B_o6llPUNVOluOt1d-ioE7KZMa9wEuxONh2G9JyTYLQ,7410
 ara_cli/artefact_autofix.py,sha256=WVTiIR-jo4YKmmz4eS3qTFvl45W1YKwAk1XSuz9QX10,20015
 ara_cli/artefact_creator.py,sha256=0Ory6cB-Ahkw-BDNb8QHnTbp_OHGABdkb9bhwcEdcIc,6063
 ara_cli/artefact_deleter.py,sha256=Co4wwCH3yW8H9NrOq7_2p5571EeHr0TsfE-H8KqoOfY,1900
@@ -12,7 +12,7 @@ ara_cli/artefact_lister.py,sha256=jhk4n4eqp7hDIq07q43QzS7-36BM3OfZ4EABxCeOGcw,47
 ara_cli/artefact_reader.py,sha256=Pho0_Eqm7kD9CNbVMhKb6mkNM0I3iJiCJXbXmVp1DJU,7827
 ara_cli/artefact_renamer.py,sha256=Hnz_3zD9xxnBa1FHyUE6mIktLk_9ttP2rFRvQIkmz-o,4061
 ara_cli/artefact_scan.py,sha256=msPCm-vPWOAZ_e_z5GylXxq1MtNlmJ4zvKrsdOFCWF4,4813
-ara_cli/chat.py,sha256=qQINBi5VCzlZOcQqDqUJY0p6VlAPiAWwWICSe7fvcDQ,32540
+ara_cli/chat.py,sha256=Y_LSYvpntnoRikYSOI95rWPImA-Rr_xppNIhN-cjjUY,37966
 ara_cli/classifier.py,sha256=zWskj7rBYdqYBGjksBm46iTgVU5IIf2PZsJr4qeiwVU,1878
 ara_cli/codefusionretriever.py,sha256=fCHgXdIBRzkVAnapX-KI2NQ44XbrrF4tEQmn5J6clUI,1980
 ara_cli/codehierachieretriever.py,sha256=Xd3EgEWWhkSf1TmTWtf8X5_YvyE_4B66nRrqarwSiTU,1182
@@ -24,14 +24,14 @@ ara_cli/filename_validator.py,sha256=Aw9PL8d5-Ymhp3EY6lDrUBk3cudaNqo1Uw5RzPpI1jA
 ara_cli/list_filter.py,sha256=qKGwwQsrWe7L5FbdxEbBYD1bbbi8c-RMypjXqXvLbgs,5291
 ara_cli/output_suppressor.py,sha256=nwiHaQLwabOjMoJOeUESBnZszGMxrQZfJ3N2OvahX7Y,389
 ara_cli/prompt_chat.py,sha256=kd_OINDQFit6jN04bb7mzgY259JBbRaTaNp9F-webkc,1346
-ara_cli/prompt_extractor.py,sha256=6xLGd4ZJHDKkamEUQcdRbKM3ilBtxBjp0X2o8wrvHb0,7732
-ara_cli/prompt_handler.py,sha256=5FoVCNmmzrS4hjHL4qKteQt2A5MIycoZStkJrVL5l_4,20136
+ara_cli/prompt_extractor.py,sha256=Sk1aQkvn8W_YNcGhfChsbT19wUAWEJmOHTr3mveyQww,7754
+ara_cli/prompt_handler.py,sha256=pCQ2LMYawGIMlWKI5EN6QdLn0UNEk-Eckkz4SVVHR10,21122
 ara_cli/prompt_rag.py,sha256=ydlhe4CUqz0jdzlY7jBbpKaf_5fjMrAZKnriKea3ZAg,7485
 ara_cli/run_file_lister.py,sha256=XbrrDTJXp1LFGx9Lv91SNsEHZPP-PyEMBF_P4btjbDA,2360
 ara_cli/tag_extractor.py,sha256=TGdaQOVnjy25R0zDsAifB67C5oom0Fwo24s0_fr5A_I,3151
 ara_cli/template_manager.py,sha256=YwrN6AYPpl6ZrW8BVQpVXx8yTRf-oNpJUIKeg4NAggs,6606
 ara_cli/update_config_prompt.py,sha256=Oy9vNTw6UhDohyTEfSKkqE5ifEMPlmWNYkKHgUrK_pY,4607
-ara_cli/version.py,sha256=27waJFP_fluwJJNMwbe3_NlHJaeI6oPvAK2CQIhI63w,146
+ara_cli/version.py,sha256=B3OzxM9cfU7TOWcR4tcsGLFojByOhZd98cMMsL45DYc,146
 ara_cli/artefact_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ara_cli/artefact_models/artefact_load.py,sha256=IXzWxP-Q_j_oDGMno0m-OuXCQ7Vd5c_NctshGr4ROBw,621
 ara_cli/artefact_models/artefact_mapping.py,sha256=8aD0spBjkJ8toMAmFawc6UTUxB6-tEEViZXv2I-r88Q,1874
@@ -134,7 +134,7 @@ ara_cli/templates/specification_breakdown_files/template.technology.exploration.
 ara_cli/templates/specification_breakdown_files/template.technology.md,sha256=bySiksz-8xtq0Nnj4svqe2MgUftWrVkbK9AcrDUE3KY,952
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_ara_command_action.py,sha256=JTLqXM9BSMlU33OQgrk_sZnoowFJZKZAx8q-st-wa34,25821
-tests/test_ara_config.py,sha256=gYqBB4Z7lB0PTSZrJRL7ekC0t9HZ_Rq3JF9EnOFmN5U,14280
+tests/test_ara_config.py,sha256=H5GwDbab0GMSa6IbHdruzmbsHy5Ia0xX0uteJdfZ9Rg,14277
 tests/test_artefact_autofix.py,sha256=pApZ-N0dW8Ujt-cNLbgvd4bhiIIK8oXb-saLf6QlA-8,25022
 tests/test_artefact_fuzzy_search.py,sha256=5Sh3_l9QK8-WHn6JpGPU1b6h4QEnl2JoMq1Tdp2cj1U,1261
 tests/test_artefact_link_updater.py,sha256=biqbEp2jCOz8giv72hu2P2hDfeJfJ9OrVGdAv5d9cK4,2191
@@ -149,12 +149,12 @@ tests/test_file_classifier.py,sha256=kLWPiePu3F5mkVuI_lK_2QlLh2kXD_Mt2K8KZZ1fAnA
 tests/test_file_creator.py,sha256=D3G7MbgE0m8JmZihxnTryxLco6iZdbV--2CGc0L20FM,2109
 tests/test_file_lister.py,sha256=Q9HwhKKx540EPzTmfzOCnvtAgON0aMmpJE2eOe1J3EA,4324
 tests/test_list_filter.py,sha256=fJA3d_SdaOAUkE7jn68MOVS0THXGghy1fye_64Zvo1U,7964
-tests/test_prompt_handler.py,sha256=7V_AwXd2co1krnx5RKZRK-hqXS50nq77mX-Yx_QO0w0,13084
+tests/test_prompt_handler.py,sha256=Ysxq2e6JFfNpj3bTRFNNThzpDNkAGDphwVf2Ysz2EK0,14980
 tests/test_tag_extractor.py,sha256=nSiAYlTKZ7TLAOtcJpwK5zTWHhFYU0tI5xKnivLc1dU,2712
 tests/test_template_manager.py,sha256=q-LMHRG4rHkD6ON6YW4cpZxUx9hul6Or8wVVRC2kb-8,4099
 tests/test_update_config_prompt.py,sha256=xsqj1WTn4BsG5Q2t-sNPfu7EoMURFcS-hfb5VSXUnJc,6765
-ara_cli-0.1.9.87.dist-info/METADATA,sha256=WKN4uhNXDHMDwwlxkCW5KbBV_2cbeixRkQYsLZNBdLY,6739
-ara_cli-0.1.9.87.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ara_cli-0.1.9.87.dist-info/entry_points.txt,sha256=v4h7MzysTgSIDYfEo3oj4Kz_8lzsRa3hq-KJHEcLVX8,45
-ara_cli-0.1.9.87.dist-info/top_level.txt,sha256=WM4cLHT5DYUaWzLtRj-gu3yVNFpGQ6lLRI3FMmC-38I,14
-ara_cli-0.1.9.87.dist-info/RECORD,,
+ara_cli-0.1.9.91.dist-info/METADATA,sha256=V9fNcWjRmFt46pPhw1qNZKejVfobM82LT0s2kXmEaM4,6739
+ara_cli-0.1.9.91.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ara_cli-0.1.9.91.dist-info/entry_points.txt,sha256=v4h7MzysTgSIDYfEo3oj4Kz_8lzsRa3hq-KJHEcLVX8,45
+ara_cli-0.1.9.91.dist-info/top_level.txt,sha256=WM4cLHT5DYUaWzLtRj-gu3yVNFpGQ6lLRI3FMmC-38I,14
+ara_cli-0.1.9.91.dist-info/RECORD,,

tests/test_ara_config.py

@@ -97,8 +97,8 @@ class TestARAconfig:
         config = ARAconfig()
         assert config.ext_code_dirs == [{"source_dir": "./src"}, {"source_dir": "./tests"}]
         assert config.glossary_dir == "./glossary"
-        assert config.default_llm == "gpt-4o"
-        assert "gpt-4o" in config.llm_config
+        assert config.default_llm == "gpt-5"
+        assert "gpt-5" in config.llm_config
 
     @patch('sys.stdout', new_callable=StringIO)
     def test_check_critical_fields_with_empty_list_reverts_to_default(self, mock_stdout):
@@ -228,7 +228,7 @@ class TestReadData:
         result = read_data("config.json")
         
         assert isinstance(result, ARAconfig)
-        assert result.default_llm == "gpt-4o"  # Should be the default config
+        assert result.default_llm == "gpt-5"  # Should be the default config
         
         output = mock_stdout.getvalue()
         assert "Error: Invalid JSON in configuration file" in output

tests/test_prompt_handler.py

@@ -6,7 +6,7 @@ from unittest.mock import patch, MagicMock, mock_open, call
 from pathlib import Path
 
 from ara_cli import prompt_handler
-from ara_cli.ara_config import ARAconfig, LLMConfigItem
+from ara_cli.ara_config import ARAconfig, LLMConfigItem, ConfigManager
 
 @pytest.fixture
 def mock_config():
@@ -22,23 +22,27 @@ def mock_config():
             "gpt-4o": LLMConfigItem(provider="openai", model="openai/gpt-4o", temperature=0.8, max_tokens=1024),
             "o3-mini": LLMConfigItem(provider="openai", model="openai/o3-mini", temperature=0.9, max_tokens=2048),
         },
-        default_llm="gpt-4o"
+        default_llm="gpt-4o",
+        extraction_llm="o3-mini"
     )
     return config
 
 @pytest.fixture
 def mock_config_manager(mock_config):
     """Patches ConfigManager to ensure it always returns the mock_config."""
-    with patch('ara_cli.ara_config.ConfigManager.get_config') as mock_get_config:
+    with patch.object(ConfigManager, 'get_config') as mock_get_config:
         mock_get_config.return_value = mock_config
         yield mock_get_config
 
 @pytest.fixture(autouse=True)
 def reset_singleton():
-    """Resets the LLMSingleton before each test for isolation."""
+    """Resets the LLMSingleton and ConfigManager before each test for isolation."""
     prompt_handler.LLMSingleton._instance = None
     prompt_handler.LLMSingleton._model = None
+    ConfigManager.reset()
     yield
+    ConfigManager.reset()
+
 
 class TestLLMSingleton:
     """Tests the behavior of the LLMSingleton class."""
@@ -100,14 +104,19 @@ class TestFileIO:
 
 class TestCoreLogic:
     """Tests functions related to the main business logic."""
+    
+    @pytest.fixture(autouse=True)
+    def setup_test_env(self, tmp_path):
+        """Changes CWD to a temporary directory for test isolation."""
+        original_cwd = os.getcwd()
+        os.chdir(tmp_path)
+        yield
+        os.chdir(original_cwd)
+
 
     @patch('ara_cli.prompt_handler.litellm.completion')
-    @patch('ara_cli.prompt_handler.LLMSingleton.get_instance')
-    def test_send_prompt(self, mock_get_instance, mock_completion, mock_config):
-        mock_llm_instance = MagicMock()
-        mock_llm_instance.config_parameters = mock_config.llm_config['gpt-4o'].model_dump()
-        mock_get_instance.return_value = mock_llm_instance
-        
+    def test_send_prompt(self, mock_completion, mock_config, mock_config_manager):
+        """Tests that send_prompt uses the default LLM by default."""
         mock_chunk = MagicMock()
         mock_chunk.choices[0].delta.content = "test chunk"
         mock_completion.return_value = [mock_chunk]
@@ -116,8 +125,10 @@ class TestCoreLogic:
         
         result = list(prompt_handler.send_prompt(prompt))
 
+        # Check that the parameters for the default model ('gpt-4o') were used
         expected_params = mock_config.llm_config['gpt-4o'].model_dump(exclude_none=True)
-        del expected_params['provider']
+        if 'provider' in expected_params:
+            del expected_params['provider']
 
         mock_completion.assert_called_once_with(
             messages=prompt,
@@ -127,6 +138,29 @@ class TestCoreLogic:
         assert len(result) == 1
         assert result[0].choices[0].delta.content == "test chunk"
 
+    @patch('ara_cli.prompt_handler.litellm.completion')
+    def test_send_prompt_uses_extraction_llm(self, mock_completion, mock_config, mock_config_manager):
+        """Tests that send_prompt uses the extraction LLM when specified."""
+        mock_chunk = MagicMock()
+        mock_chunk.choices[0].delta.content = "extraction chunk"
+        mock_completion.return_value = [mock_chunk]
+        prompt = [{"role": "user", "content": "Extract this"}]
+        
+        # Call with the 'extraction' purpose
+        result = list(prompt_handler.send_prompt(prompt, purpose='extraction'))
+
+        # Check that the parameters for the extraction model ('o3-mini') were used
+        expected_params = mock_config.llm_config['o3-mini'].model_dump(exclude_none=True)
+        if 'provider' in expected_params:
+            del expected_params['provider']
+
+        mock_completion.assert_called_once_with(
+            messages=prompt,
+            stream=True,
+            **expected_params
+        )
+        assert result[0].choices[0].delta.content == "extraction chunk"
+
     @patch('ara_cli.prompt_handler.send_prompt')
     def test_describe_image(self, mock_send_prompt, tmp_path):
         fake_image_path = tmp_path / "test.png"
@@ -152,11 +186,14 @@ class TestCoreLogic:
 
     @patch('ara_cli.prompt_handler.Classifier.get_sub_directory', return_value="test_classifier")
     def test_append_headings(self, mock_get_sub, tmp_path):
-        os.chdir(tmp_path)
+        # The autouse fixture already handles chdir, so we just use tmp_path for paths
         os.makedirs("ara/test_classifier/my_param.data", exist_ok=True)
         
         log_file = tmp_path / "ara/test_classifier/my_param.data/test_classifier.prompt_log.md"
         
+        # Create file first to avoid FileNotFoundError
+        log_file.touch()
+
         prompt_handler.append_headings("test_classifier", "my_param", "PROMPT")
         assert "## PROMPT_1" in log_file.read_text()
         
@@ -278,12 +315,15 @@ class TestArtefactAndTemplateHandling:
     @patch('ara_cli.prompt_handler.send_prompt')
     @patch('ara_cli.prompt_handler.collect_file_content_by_extension')
     @patch('ara_cli.prompt_handler.append_images_to_message')
-    def test_create_and_send_custom_prompt(self, mock_append_images, mock_collect, mock_send):
+    def test_create_and_send_custom_prompt(self, mock_append_images, mock_collect, mock_send, mock_config_manager):
         prompt_data_path = Path(prompt_handler.prompt_data_directory_creation(self.mock_classifier, self.mock_param))
 
         mock_collect.return_value = ("### GIVENS\ncontent", [{"type": "image_url", "image_url": {}}])
         
-        final_message_list = [{"role": "user", "content": [{"type": "text", "text": "### GIVENS\ncontent"}, {"type": "image_url", "image_url": {}}]}]
+        # The initial message list before appending images
+        initial_message_list = [{'role': 'user', 'content': '### GIVENS\ncontent'}]
+        # The final list after images are appended
+        final_message_list = [{'role': 'user', 'content': ['### GIVENS\ncontent', {'type': 'image_url', 'image_url': {}}]}]
         mock_append_images.return_value = final_message_list
 
         mock_send.return_value = iter([MagicMock(choices=[MagicMock(delta=MagicMock(content="llm response"))])])
@@ -291,7 +331,8 @@ class TestArtefactAndTemplateHandling:
         prompt_handler.create_and_send_custom_prompt(self.mock_classifier, self.mock_param)
         
         mock_collect.assert_called_once()
-        mock_append_images.assert_called_once()
+        # The append function is called with the message list containing the text and the image data list
+        mock_append_images.assert_called_once_with([{'role': 'user', 'content': '### GIVENS\ncontent'}], [{'type': 'image_url', 'image_url': {}}])
         mock_send.assert_called_once_with(final_message_list)
 
         artefact_root = self.root / "ara" / self.mock_classifier