amd-gaia 0.15.1__tar.gz → 0.15.2__tar.gz

{amd_gaia-0.15.1/src/amd_gaia.egg-info → amd_gaia-0.15.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amd-gaia
-Version: 0.15.1
+Version: 0.15.2
 Summary: GAIA is a lightweight agent framework designed for the edge and AI PCs.
 Home-page: https://github.com/amd/gaia
 Author: AMD
@@ -100,7 +100,6 @@ Dynamic: summary
 
 # <img src="https://raw.githubusercontent.com/amd/gaia/main/src/gaia/img/gaia.ico" alt="GAIA Logo" width="64" height="64" style="vertical-align: middle;"> GAIA: AI Agent Framework for AMD Ryzen AI
 
-[![GAIA Build Installer](https://github.com/amd/gaia/actions/workflows/build_installer.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our build")
 [![GAIA CLI Tests](https://github.com/amd/gaia/actions/workflows/test_gaia_cli.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our cli tests")
 [![Latest Release](https://img.shields.io/github/v/release/amd/gaia?include_prereleases)](https://github.com/amd/gaia/releases/latest "Download the latest release")
 [![PyPI](https://img.shields.io/pypi/v/amd-gaia)](https://pypi.org/project/amd-gaia/)

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/README.md

@@ -1,6 +1,5 @@
 # <img src="https://raw.githubusercontent.com/amd/gaia/main/src/gaia/img/gaia.ico" alt="GAIA Logo" width="64" height="64" style="vertical-align: middle;"> GAIA: AI Agent Framework for AMD Ryzen AI
 
-[![GAIA Build Installer](https://github.com/amd/gaia/actions/workflows/build_installer.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our build")
 [![GAIA CLI Tests](https://github.com/amd/gaia/actions/workflows/test_gaia_cli.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our cli tests")
 [![Latest Release](https://img.shields.io/github/v/release/amd/gaia?include_prereleases)](https://github.com/amd/gaia/releases/latest "Download the latest release")
 [![PyPI](https://img.shields.io/pypi/v/amd-gaia)](https://pypi.org/project/amd-gaia/)

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/setup.py

@@ -36,6 +36,7 @@ setup(
         "gaia.apps.llm",
         "gaia.apps.summarize",
         "gaia.eval",
+        "gaia.installer",
         "gaia.rag",
         "gaia.mcp",
         "gaia.mcp.servers",

{amd_gaia-0.15.1 → amd_gaia-0.15.2/src/amd_gaia.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amd-gaia
-Version: 0.15.1
+Version: 0.15.2
 Summary: GAIA is a lightweight agent framework designed for the edge and AI PCs.
 Home-page: https://github.com/amd/gaia
 Author: AMD
@@ -100,7 +100,6 @@ Dynamic: summary
 
 # <img src="https://raw.githubusercontent.com/amd/gaia/main/src/gaia/img/gaia.ico" alt="GAIA Logo" width="64" height="64" style="vertical-align: middle;"> GAIA: AI Agent Framework for AMD Ryzen AI
 
-[![GAIA Build Installer](https://github.com/amd/gaia/actions/workflows/build_installer.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our build")
 [![GAIA CLI Tests](https://github.com/amd/gaia/actions/workflows/test_gaia_cli.yml/badge.svg)](https://github.com/amd/gaia/tree/main/tests "Check out our cli tests")
 [![Latest Release](https://img.shields.io/github/v/release/amd/gaia?include_prereleases)](https://github.com/amd/gaia/releases/latest "Download the latest release")
 [![PyPI](https://img.shields.io/pypi/v/amd-gaia)](https://pypi.org/project/amd-gaia/)

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/amd_gaia.egg-info/SOURCES.txt

@@ -24,6 +24,7 @@ src/gaia/agents/base/console.py
 src/gaia/agents/base/errors.py
 src/gaia/agents/base/mcp_agent.py
 src/gaia/agents/base/tools.py
+src/gaia/agents/blender/__init__.py
 src/gaia/agents/blender/agent.py
 src/gaia/agents/blender/agent_simple.py
 src/gaia/agents/blender/app.py
@@ -148,6 +149,9 @@ src/gaia/eval/webapp/test-setup.js
 src/gaia/eval/webapp/public/app.js
 src/gaia/eval/webapp/public/index.html
 src/gaia/eval/webapp/public/styles.css
+src/gaia/installer/__init__.py
+src/gaia/installer/init_command.py
+src/gaia/installer/lemonade_installer.py
 src/gaia/llm/__init__.py
 src/gaia/llm/base_client.py
 src/gaia/llm/exceptions.py
@@ -189,9 +193,11 @@ tests/test_code_agent.py
 tests/test_code_agent_mixins.py
 tests/test_eval.py
 tests/test_external_tools.py
+tests/test_hardware_advisor_agent.py
 tests/test_jira.py
 tests/test_lemonade_client.py
 tests/test_lemonade_embeddings.py
+tests/test_lemonade_health.py
 tests/test_rag.py
 tests/test_rag_integration.py
 tests/test_sdk.py

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/base/agent.py

@@ -61,10 +61,6 @@ class Agent(abc.ABC):
     STATE_ERROR_RECOVERY = "ERROR_RECOVERY"
     STATE_COMPLETION = "COMPLETION"
 
-    # Define tools that can execute directly without requiring a plan
-    # Subclasses can override this to specify domain-specific simple tools
-    SIMPLE_TOOLS = []
-
     def __init__(
         self,
         use_claude: bool = False,
@@ -280,7 +276,7 @@ You must respond ONLY in valid JSON. No text before { or after }.
         self.console.print_header(f"🛠️ Registered Tools for {self.__class__.__name__}")
         self.console.print_separator()
 
-        for name, tool_info in _TOOL_REGISTRY.items():
+        for name, tool_info in self.get_tools_info().items():
             # Format parameters
             params = []
             for param_name, param_info in tool_info["parameters"].items():
@@ -313,6 +309,14 @@ You must respond ONLY in valid JSON. No text before { or after }.
 
         return None
 
+    def get_tools_info(self) -> Dict[str, Any]:
+        """Get information about all registered tools."""
+        return _TOOL_REGISTRY
+
+    def get_tools(self) -> List[Dict[str, Any]]:
+        """Get a list of registered tools for the agent."""
+        return list(_TOOL_REGISTRY.values())
+
     def _extract_json_from_response(self, response: str) -> Optional[Dict[str, Any]]:
         """
         Apply multiple extraction strategies to find valid JSON in the response.
@@ -1177,43 +1181,6 @@ You must respond ONLY in valid JSON. No text before { or after }.
             steps_taken += 1
             logger.debug(f"Step {steps_taken}/{steps_limit}")
 
-            # Check if we're at the limit and ask user if they want to continue
-            if steps_taken == steps_limit and final_answer is None:
-                # Show what was accomplished
-                max_steps_msg = self._generate_max_steps_message(
-                    conversation, steps_taken, steps_limit
-                )
-                self.console.print_warning(max_steps_msg)
-
-                # Ask user if they want to continue (skip in silent mode OR if stdin is not available)
-                # IMPORTANT: Never call input() in API/CI contexts to avoid blocking threads
-                import sys
-
-                has_stdin = sys.stdin and sys.stdin.isatty()
-                if has_stdin and not (
-                    hasattr(self, "silent_mode") and self.silent_mode
-                ):
-                    try:
-                        response = (
-                            input("\nContinue with 50 more steps? (y/n): ")
-                            .strip()
-                            .lower()
-                        )
-                        if response in ["y", "yes"]:
-                            steps_limit += 50
-                            self.console.print_info(
-                                f"✓ Continuing with {steps_limit} total steps...\n"
-                            )
-                        else:
-                            self.console.print_info("Stopping at user request.")
-                            break
-                    except (EOFError, KeyboardInterrupt):
-                        self.console.print_info("\nStopping at user request.")
-                        break
-                else:
-                    # Silent mode - just stop
-                    break
-
             # Display current step
             self.console.print_step_header(steps_taken, steps_limit)
 
@@ -1668,9 +1635,6 @@ You must respond ONLY in valid JSON. No text before { or after }.
             # Add assistant response to messages for chat history
             messages.append({"role": "assistant", "content": response})
 
-            # Validate the response has a plan if required
-            self._validate_plan_required(parsed, steps_taken)
-
             # If the LLM needs to create a plan first, re-prompt it specifically for that
             if "needs_plan" in parsed and parsed["needs_plan"]:
                 # Prepare a special prompt that specifically requests a plan
@@ -2015,8 +1979,42 @@ You must respond ONLY in valid JSON. No text before { or after }.
                 self.console.print_final_answer(final_answer, streaming=self.streaming)
                 break
 
-            # Validate plan required
-            self._validate_plan_required(parsed, steps_taken)
+            # Check if we're at the limit and ask user if they want to continue
+            if steps_taken == steps_limit and final_answer is None:
+                # Show what was accomplished
+                max_steps_msg = self._generate_max_steps_message(
+                    conversation, steps_taken, steps_limit
+                )
+                self.console.print_warning(max_steps_msg)
+
+                # Ask user if they want to continue (skip in silent mode OR if stdin is not available)
+                # IMPORTANT: Never call input() in API/CI contexts to avoid blocking threads
+                import sys
+
+                has_stdin = sys.stdin and sys.stdin.isatty()
+                if has_stdin and not (
+                    hasattr(self, "silent_mode") and self.silent_mode
+                ):
+                    try:
+                        response = (
+                            input("\nContinue with 50 more steps? (y/n): ")
+                            .strip()
+                            .lower()
+                        )
+                        if response in ["y", "yes"]:
+                            steps_limit += 50
+                            self.console.print_info(
+                                f"✓ Continuing with {steps_limit} total steps...\n"
+                            )
+                        else:
+                            self.console.print_info("Stopping at user request.")
+                            break
+                    except (EOFError, KeyboardInterrupt):
+                        self.console.print_info("\nStopping at user request.")
+                        break
+                else:
+                    # Silent mode - just stop
+                    break
 
         # Print completion message
         self.console.print_completion(steps_taken, steps_limit)
@@ -2132,46 +2130,3 @@ You must respond ONLY in valid JSON. No text before { or after }.
             List of error messages
         """
         return self.error_history
-
-    def _validate_plan_required(self, parsed: Dict[str, Any], step: int) -> None:
-        """
-        Validate that the response includes a plan when required by the agent.
-
-        Args:
-            parsed: The parsed response from the LLM
-            step: The current step number
-        """
-        # Skip validation if we're not in planning mode or if we're already executing a plan
-        if self.execution_state != self.STATE_PLANNING or self.current_plan is not None:
-            return
-
-        # Allow simple single-tool operations without requiring a plan
-        if "tool" in parsed and step == 1:
-            tool_name = parsed.get("tool", "")
-            # List of tools that can execute directly without a plan
-            simple_tools = self.SIMPLE_TOOLS
-            if tool_name in simple_tools:
-                logger.debug(f"Allowing direct execution of simple tool: {tool_name}")
-                return
-
-        # Check if plan is missing on the first step
-        # BUT: Allow direct answers without plans (for simple conversational queries)
-        if "plan" not in parsed and "answer" not in parsed and step == 1:
-            warning_msg = f"No plan found in step {step} response. The agent should create a plan for all tasks."
-            logger.warning(warning_msg)
-            self.console.print_warning(warning_msg)
-
-            # For the first step, we'll add a flag to indicate we need to re-prompt for a plan
-            parsed["needs_plan"] = True
-
-            # If there's a tool in the response, store it but don't execute it yet
-            if "tool" in parsed:
-                parsed["deferred_tool"] = parsed["tool"]
-                parsed["deferred_tool_args"] = parsed.get("tool_args", {})
-                # Remove the tool so it won't be executed
-                del parsed["tool"]
-                if "tool_args" in parsed:
-                    del parsed["tool_args"]
-
-            # Set state to indicate we need planning
-            self.execution_state = self.STATE_PLANNING

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/base/api_agent.py

@@ -13,7 +13,6 @@ Inheritance patterns:
     - Future: FooAgent(MCPAgent, ApiAgent, Agent) - Multiple inheritance
 """
 
-
 from typing import Any, Dict
 
 from .agent import Agent

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/base/console.py

@@ -1233,6 +1233,132 @@ class AgentConsole(OutputHandler):
         else:
             print(f"✅ Model {status}: {model_name}")
 
+    # === Download Progress Methods ===
+
+    def print_download_start(self, model_name: str) -> None:
+        """
+        Print download starting notification.
+
+        Args:
+            model_name: Name of the model being downloaded
+        """
+        if self.rich_available and self.console:
+            self.console.print()
+            self.console.print(
+                f"[bold blue]📥 Downloading:[/bold blue] [cyan]{model_name}[/cyan]"
+            )
+        else:
+            rprint(f"\n📥 Downloading: {model_name}")
+
+    def print_download_progress(
+        self,
+        percent: int,
+        bytes_downloaded: int,
+        bytes_total: int,
+        speed_mbps: float = 0.0,
+    ) -> None:
+        """
+        Print download progress with a progress bar that updates in place.
+
+        Args:
+            percent: Download percentage (0-100)
+            bytes_downloaded: Bytes downloaded so far
+            bytes_total: Total bytes to download
+            speed_mbps: Download speed in MB/s (optional)
+        """
+        import sys
+
+        # Format sizes
+        if bytes_total > 1024**3:  # > 1 GB
+            dl_str = f"{bytes_downloaded / 1024**3:.2f} GB"
+            total_str = f"{bytes_total / 1024**3:.2f} GB"
+        elif bytes_total > 1024**2:  # > 1 MB
+            dl_str = f"{bytes_downloaded / 1024**2:.0f} MB"
+            total_str = f"{bytes_total / 1024**2:.0f} MB"
+        else:
+            dl_str = f"{bytes_downloaded / 1024:.0f} KB"
+            total_str = f"{bytes_total / 1024:.0f} KB"
+
+        # Progress bar characters
+        bar_width = 25
+        filled = int(bar_width * percent / 100)
+        bar = "━" * filled + "─" * (bar_width - filled)
+
+        # Build progress line with optional speed
+        progress_line = f"   [{bar}] {percent:3d}%  {dl_str} / {total_str}"
+        if speed_mbps > 0.1:
+            progress_line += f" @ {speed_mbps:.0f} MB/s"
+
+        # Update in place with carriage return
+        sys.stdout.write(f"\r{progress_line:<80}")
+        sys.stdout.flush()
+
+    def print_download_complete(self, model_name: str = None) -> None:
+        """
+        Print download complete notification.
+
+        Args:
+            model_name: Optional name of the downloaded model
+        """
+        if self.rich_available and self.console:
+            self.console.print()  # Newline after progress bar
+            if model_name:
+                self.console.print(
+                    f"   [green]✅ Downloaded successfully:[/green] [cyan]{model_name}[/cyan]"
+                )
+            else:
+                self.console.print("   [green]✅ Download complete[/green]")
+        else:
+            rprint()
+            msg = (
+                f"   ✅ Downloaded: {model_name}"
+                if model_name
+                else "   ✅ Download complete"
+            )
+            rprint(msg)
+
+    def print_download_error(self, error_message: str, model_name: str = None) -> None:
+        """
+        Print download error notification.
+
+        Args:
+            error_message: Error description
+            model_name: Optional name of the model that failed
+        """
+        if self.rich_available and self.console:
+            self.console.print()  # Newline after progress bar
+            if model_name:
+                self.console.print(
+                    f"   [red]❌ Download failed for {model_name}:[/red] {error_message}"
+                )
+            else:
+                self.console.print(f"   [red]❌ Download failed:[/red] {error_message}")
+        else:
+            rprint()
+            msg = (
+                f"   ❌ Download failed for {model_name}: {error_message}"
+                if model_name
+                else f"   ❌ Download failed: {error_message}"
+            )
+            rprint(msg)
+
+    def print_download_skipped(
+        self, model_name: str, reason: str = "already downloaded"
+    ) -> None:
+        """
+        Print download skipped notification.
+
+        Args:
+            model_name: Name of the model that was skipped
+            reason: Reason for skipping
+        """
+        if self.rich_available and self.console:
+            self.console.print(
+                f"[green]✅[/green] [cyan]{model_name}[/cyan] [dim]({reason})[/dim]"
+            )
+        else:
+            rprint(f"✅ {model_name} ({reason})")
+
     def print_extraction_start(
         self, image_num: int, page_num: int, mime_type: str
     ) -> None:

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/base/tools.py

@@ -17,7 +17,10 @@ _TOOL_REGISTRY = {}
 
 
 def tool(
-    func: Callable = None, **kwargs  # pylint: disable=unused-argument
+    func: Callable = None,
+    *,
+    atomic: bool = False,
+    **kwargs,  # pylint: disable=unused-argument
 ) -> Callable:
     """
     Decorator to register a function as a tool.
@@ -28,6 +31,7 @@ def tool(
 
     Args:
         func: Function to register as a tool (when used as @tool)
+        atomic: If True, marks this tool as atomic (can execute without multi-step planning)
         **kwargs: Optional arguments (ignored, for backward compatibility)
 
     Returns:
@@ -63,12 +67,13 @@ def tool(
 
             params[name] = param_info
 
-        # Register the tool
+        # Register the tool with atomic metadata
         _TOOL_REGISTRY[tool_name] = {
             "name": tool_name,
             "description": f.__doc__ or "",
             "parameters": params,
             "function": f,
+            "atomic": atomic,
         }
 
         # Return the function unchanged

amd_gaia-0.15.2/src/gaia/agents/blender/__init__.py

@@ -0,0 +1,7 @@
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""Blender Agent for 3D scene automation."""
+
+from gaia.agents.blender.agent import BlenderAgent
+
+__all__ = ["BlenderAgent"]

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/blender/agent.py

@@ -24,9 +24,6 @@ class BlenderAgent(Agent):
     Inherits core functionality from the base Agent class.
     """
 
-    # Define Blender-specific tools that can execute directly without requiring a plan
-    SIMPLE_TOOLS = ["clear_scene", "get_scene_info"]
-
     def __init__(
         self,
         mcp: Optional[MCPClient] = None,
@@ -80,7 +77,7 @@ class BlenderAgent(Agent):
     def _get_system_prompt(self) -> str:
         """Generate the system prompt for the Blender agent."""
         # Get formatted tools from registry
-        return f"""
+        return """
 You are a specialized Blender 3D assistant that can create and modify 3D scenes.
 You will use a set of tools to accomplish tasks based on the user's request.
 
@@ -169,7 +166,7 @@ Examples of colored requests:
     def _register_tools(self):
         """Register all Blender-related tools for the agent."""
 
-        @tool
+        @tool(atomic=True)
         def clear_scene() -> Dict[str, Any]:
             """
             Remove all objects from the current Blender scene.
@@ -283,7 +280,7 @@ Examples of colored requests:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def get_object_info(name: str) -> Dict[str, Any]:
+        def _get_object_info(name: str) -> Dict[str, Any]:
             """
             Get information about an object in the scene.
 
@@ -354,7 +351,7 @@ Examples of colored requests:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def delete_object(name: str) -> Dict[str, Any]:
+        def _delete_object(name: str) -> Dict[str, Any]:
             """
             Delete an object from the scene.
 
@@ -382,7 +379,7 @@ Examples of colored requests:
                 self.error_history.append(str(e))
                 return {"status": "error", "error": str(e)}
 
-        @tool
+        @tool(atomic=True)
         def get_scene_info() -> Dict[str, Any]:
             """
             Get information about the current scene.
@@ -407,7 +404,7 @@ Examples of colored requests:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def execute_blender_code(code: str) -> Dict[str, Any]:
+        def _execute_blender_code(code: str) -> Dict[str, Any]:
             """
             Execute arbitrary Python code in Blender with error handling.
 
@@ -436,7 +433,7 @@ Examples of colored requests:
                 return {"status": "error", "error": str(e)}
 
         # @tool
-        def diagnose_scene() -> Dict[str, Any]:
+        def _diagnose_scene() -> Dict[str, Any]:
             """
             Diagnose the current Blender scene for common issues.
             Returns information about objects, materials, and potential problems.

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/blender/core/view.py

@@ -13,13 +13,13 @@ class ViewManager:
         self.mcp = mcp
 
     def adjust_for_large_scale(
-        self, clip_end: float = 100000, orbit_selection: bool = True
+        self, clip_end: float = 100000, _orbit_selection: bool = True
     ) -> Dict:
         """Adjust viewport settings to properly view large-scale objects like Earth.
 
         Args:
             clip_end: The maximum view distance to set for the 3D viewport (default: 100000)
-            orbit_selection: Whether to enable orbit around selection (default: True, but may not work in all Blender versions)
+            _orbit_selection: Whether to enable orbit around selection (default: True, but may not work in all Blender versions)
         """
 
         def generate_code():

{amd_gaia-0.15.1 → amd_gaia-0.15.2}/src/gaia/agents/chat/agent.py

@@ -78,23 +78,6 @@ class ChatAgent(
     - MCP server integration
     """
 
-    # Define simple tools that can execute without requiring a multi-step plan
-    SIMPLE_TOOLS = [
-        "list_indexed_documents",
-        "rag_status",
-        "query_documents",
-        "query_specific_file",
-        "search_indexed_chunks",  # RAG: Search indexed document chunks
-        "dump_document",  # RAG: Export cached extracted text
-        "search_file_content",  # Shared: Grep-like disk search
-        "search_file",  # Shared: Find files by name
-        "search_directory",  # Shared: Find directories by name
-        "read_file",  # Shared: Read any file
-        "write_file",  # Shared: Write any file
-        "index_directory",  # RAG: Index directory
-        "run_shell_command",  # Shell: Execute commands
-    ]
-
     def __init__(self, config: Optional[ChatAgentConfig] = None):
         """
         Initialize Chat Agent.
@@ -109,6 +92,9 @@ class ChatAgent(
         # Initialize path validator
         self.path_validator = PathValidator(config.allowed_paths)
 
+        # Store config for access in other methods
+        self.config = config
+
         # Now use config for all initialization
         # Store RAG configuration from config
         self.rag_documents = config.rag_documents
@@ -148,6 +134,7 @@ class ChatAgent(
                 use_local_llm=not (config.use_claude or config.use_chatgpt),
                 use_llm_chunking=config.use_llm_chunking,  # Enable semantic chunking
                 base_url=config.base_url,  # Pass base_url to RAG for VLM client
+                allowed_paths=config.allowed_paths,  # Pass allowed paths to RAG SDK
             )
             self.rag = RAGSDK(rag_config)
         except ImportError as e:
@@ -267,10 +254,7 @@ No documents are currently indexed.
 """
 
         # Add indexed documents section
-        prompt = (
-            base_prompt
-            + indexed_docs_section
-            + """
+        prompt = base_prompt + indexed_docs_section + """
 **WHEN TO USE TOOLS VS DIRECT ANSWERS:**
 
 Use Format 1 (answer) for:
@@ -376,7 +360,6 @@ When user asks to "index my data folder" or similar:
 2. Show user the matches and ask which one (if multiple)
 3. Use index_directory on the chosen path
 4. Report indexing results"""
-        )
 
         return prompt
 
@@ -473,7 +456,7 @@ When user asks to "index my data folder" or similar:
     def _is_path_allowed(self, path: str) -> bool:
         """
         Check if a path is within allowed directories.
-        Uses real path resolution to prevent TOCTOU attacks.
+        Uses PathValidator for the actual check.
 
         Args:
             path: Path to validate
@@ -481,24 +464,7 @@ When user asks to "index my data folder" or similar:
         Returns:
             True if path is allowed, False otherwise
         """
-        try:
-            # Resolve path using os.path.realpath to follow symlinks
-            # This prevents TOCTOU attacks by resolving at check time
-            real_path = Path(os.path.realpath(path)).resolve()
-
-            # Check if real path is within any allowed directory
-            for allowed_path in self.allowed_paths:
-                try:
-                    # is_relative_to requires Python 3.9+, use alternative for compatibility
-                    real_path.relative_to(allowed_path)
-                    return True
-                except ValueError:
-                    continue
-
-            return False
-        except Exception as e:
-            logger.error(f"Error validating path {path}: {e}")
-            return False
+        return self.path_validator.is_path_allowed(path, prompt_user=False)
 
     def _validate_and_open_file(self, file_path: str, mode: str = "r"):
         """
@@ -647,9 +613,9 @@ When user asks to "index my data folder" or similar:
                 logger.error(f"Failed to index {doc}: {e}")
 
         # Update system prompt after indexing to include the new documents
-        self._update_system_prompt()
+        self.update_system_prompt()
 
-    def _update_system_prompt(self) -> None:
+    def update_system_prompt(self) -> None:
         """Update the system prompt with current indexed documents."""
         # Regenerate the system prompt with updated document list
         self.system_prompt = self._get_system_prompt()
@@ -732,18 +698,26 @@ When user asks to "index my data folder" or similar:
             )
             return
 
+        # Resolve to real path for consistent validation
+        real_file_path = os.path.realpath(file_path)
+
+        # Security check
+        if not self._is_path_allowed(real_file_path):
+            logger.warning(f"Re-indexing skipped: Path not allowed {real_file_path}")
+            return
+
         try:
-            logger.info(f"Reindexing: {file_path}")
+            logger.info(f"Reindexing: {real_file_path}")
             # Use the new reindex_document method which removes old chunks first
-            result = self.rag.reindex_document(file_path)
+            result = self.rag.reindex_document(real_file_path)
             if result.get("success"):
                 self.indexed_files.add(file_path)
-                logger.info(f"Successfully reindexed {file_path}")
+                logger.info(f"Successfully reindexed {real_file_path}")
             else:
                 error = result.get("error", "Unknown error")
-                logger.error(f"Failed to reindex {file_path}: {error}")
+                logger.error(f"Failed to reindex {real_file_path}: {error}")
         except Exception as e:
-            logger.error(f"Failed to reindex {file_path}: {e}")
+            logger.error(f"Failed to reindex {real_file_path}: {e}")
 
     def stop_watching(self) -> None:
         """Stop all file system observers."""