amd-gaia 0.14.1__py3-none-any.whl → 0.14.3__py3-none-any.whl

gaia/llm/lemonade_client.py

@@ -381,6 +381,80 @@ def _prompt_user_for_download(
             print("Please enter 'y' or 'n'")
 
 
+def _prompt_user_for_repair(model_name: str) -> bool:
+    """
+    Prompt user for confirmation before deleting and re-downloading a corrupt model.
+
+    Args:
+        model_name: Name of the model to repair
+
+    Returns:
+        True if user confirms, False otherwise
+    """
+    # Check if we're in an interactive terminal
+    if not sys.stdin.isatty() or not sys.stdout.isatty():
+        # Non-interactive environment - auto-approve
+        return True
+
+    # Try to use rich for nice formatting, fall back to plain text
+    try:
+        from rich.console import Console
+        from rich.panel import Panel
+        from rich.table import Table
+
+        console = Console()
+        console.print()
+
+        # Create info table
+        table = Table(show_header=False, box=None, padding=(0, 1))
+        table.add_column(style="dim")
+        table.add_column()
+        table.add_row("Model:", model_name)
+        table.add_row(
+            "Status:", "[yellow]Download incomplete or files corrupted[/yellow]"
+        )
+        table.add_row("Action:", "Delete and re-download the model")
+
+        console.print(
+            Panel(
+                table,
+                title="[bold yellow]⚠️  Corrupt Model Download Detected[/bold yellow]",
+                border_style="yellow",
+            )
+        )
+        console.print()
+
+        while True:
+            response = input("Delete and re-download? [Y/n]: ").strip().lower()
+            if response in ("", "y", "yes"):
+                console.print("[green]✓[/green] Proceeding with repair...")
+                return True
+            elif response in ("n", "no"):
+                console.print("[dim]Cancelled.[/dim]")
+                return False
+            else:
+                console.print("[dim]Please enter 'y' or 'n'[/dim]")
+
+    except ImportError:
+        # Fall back to plain text formatting
+        print("\n" + "=" * 60)
+        print(f"{_emoji('⚠️', '[WARNING]')} Corrupt Model Download Detected")
+        print("=" * 60)
+        print(f"Model: {model_name}")
+        print("Status: Download incomplete or files corrupted")
+        print("Action: Delete and re-download the model")
+        print("=" * 60)
+
+        while True:
+            response = input("Delete and re-download? [Y/n]: ").strip().lower()
+            if response in ("", "y", "yes"):
+                return True
+            elif response in ("n", "no"):
+                return False
+            else:
+                print("Please enter 'y' or 'n'")
+
+
 def _check_disk_space(size_gb: float, path: Optional[str] = None) -> bool:
     """
     Check if there's enough disk space for download.
@@ -466,9 +540,9 @@ class LemonadeClient:
         if not verbose:
             self.log.setLevel(logging.WARNING)
 
-        self.log.info(f"Initialized Lemonade client for {host}:{port}")
+        self.log.debug(f"Initialized Lemonade client for {host}:{port}")
         if model:
-            self.log.info(f"Initial model set to: {model}")
+            self.log.debug(f"Initial model set to: {model}")
 
     def launch_server(self, log_level="info", background="none", ctx_size=None):
         """
@@ -685,20 +759,27 @@ class LemonadeClient:
         """
         model_lower = model_name.lower()
 
-        # Look for billion parameter indicators
+        # Check for MoE models first (e.g., "30b-a3b" = 30B total, 3B active)
+        # MoE models are smaller than their total parameter count suggests
+        if "a3b" in model_lower or "a2b" in model_lower:
+            return 18.0  # MoE models like Qwen3-Coder-30B-A3B are ~18GB
+
+        # Look for billion parameter indicators (dense models)
         if "70b" in model_lower or "72b" in model_lower:
             return 40.0  # ~40GB for 70B models
         elif "30b" in model_lower or "34b" in model_lower:
-            return 20.0  # ~20GB for 30B models
+            return 18.0  # ~18GB for 30B models
         elif "13b" in model_lower or "14b" in model_lower:
             return 8.0  # ~8GB for 13B models
-        elif "7b" in model_lower:
-            return 5.0  # ~5GB for 7B models
+        elif "7b" in model_lower or "8b" in model_lower:
+            return 5.0  # ~5GB for 7-8B models
+        elif "4b" in model_lower:
+            return 2.5  # ~2.5GB for 4B models (e.g., Qwen3-VL-4B)
         elif "3b" in model_lower:
             return 2.0  # ~2GB for 3B models
-        elif "1b" in model_lower or "0.5b" in model_lower:
+        elif "1b" in model_lower or "0.5b" in model_lower or "0.6b" in model_lower:
             return 1.0  # ~1GB for small models
-        elif "embed" in model_lower:
+        elif "embed" in model_lower or "nomic" in model_lower:
             return 0.5  # Embedding models are usually small
         else:
             return 10.0  # Conservative default
@@ -880,6 +961,31 @@ class LemonadeClient:
             ]
         )
 
+    def _is_corrupt_download_error(self, error: Union[str, Dict, Exception]) -> bool:
+        """
+        Check if an error indicates a corrupt or incomplete model download.
+
+        Args:
+            error: Error as string, dict, or exception
+
+        Returns:
+            True if this is a corrupt/incomplete download error
+        """
+        error_info = self._extract_error_info(error)
+        error_message = (error_info.get("message") or "").lower()
+
+        return any(
+            phrase in error_message
+            for phrase in [
+                "download validation failed",
+                "files are incomplete",
+                "files are missing",
+                "incomplete or missing",
+                "corrupted download",
+                "llama-server failed to start",  # Often indicates corrupt model files
+            ]
+        )
+
     def _execute_with_auto_download(
         self, api_call: Callable, model: str, auto_download: bool = True
     ):
@@ -2007,7 +2113,7 @@ class LemonadeClient:
             InsufficientDiskSpaceError: If not enough disk space
             LemonadeClientError: If model loading fails
         """
-        self.log.info(f"Loading {model_name}")
+        self.log.debug(f"Loading {model_name}")
 
         request_data = {"model_name": model_name}
         if llamacpp_args:
@@ -2016,14 +2122,46 @@ class LemonadeClient:
 
         try:
             response = self._send_request("post", url, request_data, timeout=timeout)
-            self.log.info(f"Loaded {model_name} successfully: response={response}")
+            self.log.debug(f"Loaded {model_name} successfully: response={response}")
             self.model = model_name
             return response
         except Exception as e:
             original_error = str(e)
 
+            # Check if this is a corrupt/incomplete download error
+            if self._is_corrupt_download_error(e):
+                self.log.warning(
+                    f"{_emoji('⚠️', '[CORRUPT]')} Model '{model_name}' has incomplete "
+                    f"or corrupted files"
+                )
+
+                # Prompt user for confirmation to delete and re-download
+                if not _prompt_user_for_repair(model_name):
+                    raise ModelDownloadCancelledError(
+                        f"User declined to repair corrupt model: {model_name}"
+                    )
+
+                # Delete the corrupt model
+                self.log.info(
+                    f"{_emoji('🗑️', '[DELETE]')} Deleting corrupt model: {model_name}"
+                )
+                try:
+                    self.delete_model(model_name)
+                    self.log.info(
+                        f"{_emoji('✅', '[OK]')} Deleted corrupt model: {model_name}"
+                    )
+                except Exception as delete_error:
+                    self.log.warning(f"Failed to delete corrupt model: {delete_error}")
+                    # Continue anyway - the download may still work
+
+                # Now trigger a fresh download by falling through to auto-download flow
+                # (the model is now "not found" so _is_model_error will match)
+
             # Check if this is a "model not found" error and auto_download is enabled
-            if not (auto_download and self._is_model_error(e)):
+            if not (
+                auto_download
+                and (self._is_model_error(e) or self._is_corrupt_download_error(e))
+            ):
                 # Not a model error or auto_download disabled - re-raise
                 self.log.error(f"Failed to load {model_name}: {original_error}")
                 if isinstance(e, LemonadeClientError):

gaia/llm/lemonade_manager.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 """
 Lazy Lemonade Server Manager for GAIA.
@@ -10,6 +10,7 @@ Operates at the LLM level (not agent level) for flexibility with community agent
 import os
 import sys
 import threading
+from enum import Enum
 from typing import Optional
 
 from gaia.llm.lemonade_client import LemonadeClient
@@ -20,6 +21,13 @@ DEFAULT_CONTEXT_SIZE = 32768
 DEFAULT_LEMONADE_URL = "http://localhost:8000"
 
 
+class MessageType(Enum):
+    """Message type for context size notifications."""
+
+    ERROR = "error"
+    WARNING = "warning"
+
+
 class LemonadeManager:
     """Singleton manager for lazy Lemonade server initialization.
 
@@ -110,17 +118,30 @@ class LemonadeManager:
             print("Then try your command again.", file=sys.stderr)
 
     @classmethod
-    def print_context_error(cls, current_size: int, required_size: int):
-        """Print error when context size is insufficient.
+    def print_context_message(
+        cls,
+        current_size: int,
+        required_size: int,
+        message_type: MessageType = MessageType.ERROR,
+    ):
+        """Print message when context size is insufficient.
 
-        Shared by CLI and SDK for consistent error messages.
+        Shared by CLI and SDK for consistent messages.
 
         Args:
             current_size: Current server context size in tokens.
             required_size: Required context size in tokens.
+            message_type: MessageType.WARNING for warning, MessageType.ERROR for error.
         """
+        if message_type == MessageType.WARNING:
+            symbol = "⚠️ "
+            label = "Context size below recommended"
+        else:
+            symbol = "❌"
+            label = "Insufficient context size"
+
         print("", file=sys.stderr)
-        print("❌ Insufficient context size.", file=sys.stderr)
+        print(f"{symbol} {label}.", file=sys.stderr)
         print(
             f"   Current: {current_size} tokens, Required: {required_size} tokens",
             file=sys.stderr,
@@ -182,15 +203,17 @@ class LemonadeManager:
                     )
                     return True
                 else:
-                    # Context size insufficient - error and fail
+                    # Context size insufficient - warn and continue
                     cls._log.warning(
                         f"Lemonade running with {cls._context_size} tokens, "
                         f"but {min_context_size} requested. "
                         f"Restart with: lemonade-server serve --ctx-size {min_context_size}"
                     )
                     if not quiet:
-                        cls.print_context_error(cls._context_size, min_context_size)
-                    return False
+                        cls.print_context_message(
+                            cls._context_size, min_context_size, MessageType.WARNING
+                        )
+                    return True
 
             cls._log.debug(f"Initializing Lemonade (min context: {min_context_size})")
 
@@ -223,15 +246,17 @@ class LemonadeManager:
                     f"(context: {cls._context_size} tokens)"
                 )
 
-                # Verify context size - fail if insufficient
+                # Verify context size - warn if insufficient
                 if cls._context_size < min_context_size:
                     cls._log.warning(
                         f"Context size {cls._context_size} is less than "
                         f"requested {min_context_size}. Some features may not work correctly."
                     )
                     if not quiet:
-                        cls.print_context_error(cls._context_size, min_context_size)
-                    return False
+                        cls.print_context_message(
+                            cls._context_size, min_context_size, MessageType.WARNING
+                        )
+                    return True
 
                 return True
 

gaia/llm/llm_client.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 # Standard library imports
@@ -142,7 +142,7 @@ class LLMClient:
             logger.debug(f"Using Claude API with model={self.default_model}")
         elif use_claude and not CLAUDE_AVAILABLE:
             raise ValueError(
-                "Claude support requested but anthropic library not available. Install with: pip install anthropic"
+                "Claude support requested but anthropic library not available. Install with: uv pip install anthropic"
             )
         elif use_openai:
             # Use OpenAI API

gaia/llm/vlm_client.py

@@ -23,20 +23,54 @@ DEFAULT_LEMONADE_URL = "http://localhost:8000/api/v1"
 
 logger = logging.getLogger(__name__)
 
+# Magic bytes for common image formats
+IMAGE_SIGNATURES = {
+    b"\x89PNG\r\n\x1a\n": "image/png",
+    b"\xff\xd8\xff": "image/jpeg",
+    b"GIF87a": "image/gif",
+    b"GIF89a": "image/gif",
+    b"RIFF": "image/webp",  # WebP starts with RIFF...WEBP
+    b"BM": "image/bmp",
+}
+
+
+def detect_image_mime_type(image_bytes: bytes) -> str:
+    """
+    Detect MIME type from image bytes using magic number signatures.
+
+    Args:
+        image_bytes: Raw image bytes
+
+    Returns:
+        MIME type string (e.g., "image/jpeg", "image/png")
+        Defaults to "image/png" if format not detected.
+    """
+    for signature, mime_type in IMAGE_SIGNATURES.items():
+        if image_bytes.startswith(signature):
+            # Special case: WebP needs additional check for WEBP marker
+            if signature == b"RIFF" and len(image_bytes) >= 12:
+                if image_bytes[8:12] != b"WEBP":
+                    continue
+            return mime_type
+
+    # Default to PNG if format not detected
+    logger.debug("Could not detect image format, defaulting to image/png")
+    return "image/png"
+
 
 class VLMClient:
     """
     VLM client for extracting text from images using Lemonade server.
 
     Handles:
-    - Model loading (Qwen2.5-VL-7B-Instruct-GGUF)
+    - Model loading (default: Qwen3-VL-4B-Instruct-GGUF)
     - Image-to-markdown conversion
     - State tracking for VLM processing
     """
 
     def __init__(
         self,
-        vlm_model: str = "Qwen2.5-VL-7B-Instruct-GGUF",
+        vlm_model: str = "Qwen3-VL-4B-Instruct-GGUF",
         base_url: Optional[str] = None,
         auto_load: bool = True,
     ):
@@ -70,7 +104,7 @@ class VLMClient:
         self.auto_load = auto_load
         self.vlm_loaded = False
 
-        logger.info(f"VLM Client initialized: {self.vlm_model} at {self.server_url}")
+        logger.debug(f"VLM Client initialized: {self.vlm_model} at {self.server_url}")
 
     def check_availability(self) -> bool:
         """
@@ -86,7 +120,7 @@ class VLMClient:
             ]
 
             if self.vlm_model in available_models:
-                logger.info(f"✅ VLM model available: {self.vlm_model}")
+                logger.debug(f"VLM model available: {self.vlm_model}")
                 return True
             else:
                 logger.warning(f"❌ VLM model not found: {self.vlm_model}")
@@ -126,11 +160,11 @@ class VLMClient:
             return False
 
         try:
-            logger.info(f"📥 Loading VLM model: {self.vlm_model}")
+            logger.debug(f"Loading VLM model: {self.vlm_model}")
             # Load model (auto-download handled by lemonade_client, may take hours)
             self.client.load_model(self.vlm_model, timeout=60, auto_download=True)
             self.vlm_loaded = True
-            logger.info(f"✅ VLM model loaded: {self.vlm_model}")
+            logger.debug(f"VLM model loaded: {self.vlm_model}")
             return True
 
         except Exception as e:
@@ -165,10 +199,10 @@ class VLMClient:
             logger.error(error_msg)
             return f"[VLM extraction failed: {error_msg}]"
 
-        # Encode image as base64
+        # Encode image as base64 and detect MIME type
         # Note: Image size optimization happens in pdf_utils.py during extraction
         image_b64 = base64.b64encode(image_bytes).decode("utf-8")
-        _image_size_kb = len(image_bytes) / 1024
+        mime_type = detect_image_mime_type(image_bytes)
 
         # Default prompt for text extraction
         if not prompt:
@@ -192,7 +226,7 @@ Output format: Clean markdown with the ACTUAL text from the image."""
                     {"type": "text", "text": prompt},
                     {
                         "type": "image_url",
-                        "image_url": {"url": f"data:image/png;base64,{image_b64}"},
+                        "image_url": {"url": f"data:{mime_type};base64,{image_b64}"},
                     },
                 ],
             }
@@ -203,11 +237,11 @@ Output format: Clean markdown with the ACTUAL text from the image."""
 
             start_time = time.time()
 
-            logger.info(
-                f"   🔍 VLM extracting from image {image_num} on page {page_num}..."
+            logger.debug(
+                f"VLM extracting from image {image_num} on page {page_num} ({mime_type})..."
             )
             logger.debug(
-                f"   Image size: {len(image_b64)} chars base64 ({len(image_bytes)} bytes raw)"
+                f"   Image: {mime_type}, {len(image_b64)} chars base64 ({len(image_bytes)} bytes raw)"
             )
 
             # Call VLM using chat completions endpoint
@@ -216,7 +250,7 @@ Output format: Clean markdown with the ACTUAL text from the image."""
                 messages=messages,
                 temperature=0.1,  # Low temp for accurate extraction
                 max_completion_tokens=2048,  # Allow detailed extraction
-                timeout=60,  # VLM may be slower than text models
+                timeout=300,  # VLM needs more time for complex forms (5 min)
             )
 
             elapsed = time.time() - start_time
@@ -228,14 +262,17 @@ Output format: Clean markdown with the ACTUAL text from the image."""
                 and len(response["choices"]) > 0
             ):
                 extracted_text = response["choices"][0]["message"]["content"]
-                logger.info(
-                    f"   ✅ Extracted {len(extracted_text)} chars from image {image_num} "
-                    f"in {elapsed:.2f}s ({len(image_bytes)/1024:.0f}KB image)"
+                size_kb = len(image_bytes) / 1024
+                logger.debug(
+                    f"Extracted {len(extracted_text)} chars from image {image_num} "
+                    f"in {elapsed:.2f}s ({size_kb:.0f}KB image)"
                 )
                 return extracted_text
             else:
-                logger.error(f"Unexpected VLM response format: {response}")
-                return "[VLM extraction failed: unexpected response format]"
+                # Check for specific error types and provide helpful messages
+                error_msg = self._parse_vlm_error(response)
+                logger.error(error_msg)
+                return f"[VLM extraction failed: {error_msg}]"
 
         except Exception as e:
             logger.error(
@@ -246,6 +283,44 @@ Output format: Clean markdown with the ACTUAL text from the image."""
             logger.debug(traceback.format_exc())
             return f"[VLM extraction failed: {str(e)}]"
 
+    def _parse_vlm_error(self, response: dict) -> str:
+        """Parse VLM error response and return a helpful error message."""
+        if not isinstance(response, dict):
+            return f"Unexpected response type: {type(response)}"
+
+        # Check for nested error structure from Lemonade
+        error = response.get("error", {})
+        if isinstance(error, dict):
+            details = error.get("details", {})
+            inner_response = (
+                details.get("response", {}) if isinstance(details, dict) else {}
+            )
+            inner_error = (
+                inner_response.get("error", {})
+                if isinstance(inner_response, dict)
+                else {}
+            )
+
+            # Context size error
+            if inner_error.get("type") == "exceed_context_size_error":
+                n_ctx = inner_error.get("n_ctx", "unknown")
+                n_prompt = inner_error.get("n_prompt_tokens", "unknown")
+                return (
+                    f"Context size too small! Image requires {n_prompt} tokens "
+                    f"but model context is only {n_ctx}. "
+                    f"To fix: Right-click Lemonade tray icon → Settings → "
+                    f"set Context Size to 32768, then restart the model."
+                )
+
+            # Other backend errors
+            if error.get("type") == "backend_error":
+                msg = inner_error.get(
+                    "message", error.get("message", "Unknown backend error")
+                )
+                return f"Backend error: {msg}"
+
+        return f"Unexpected response format: {response}"
+
     def extract_from_page_images(self, images: list, page_num: int) -> list:
         """
         Extract text from multiple images on a page.

gaia/logger.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 import io

gaia/mcp/agent_mcp_server.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 """

gaia/mcp/blender_mcp_client.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 # This Blender MCP client is a simplified and modified version of the BlenderMCP project from https://github.com/BlenderMCP/blender-mcp

gaia/mcp/blender_mcp_server.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 # This Blender MCP client is a simplified and modified version of the BlenderMCP project from https://github.com/BlenderMCP/blender-mcp

gaia/mcp/context7_cache.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 """Persistent cache and rate protection for Context7 API calls."""
 

gaia/mcp/servers/__init__.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 """

gaia/mcp/servers/docker_mcp.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 
 """

gaia/rag/demo.py

@@ -85,7 +85,7 @@ def demo_basic_rag():
             print(
                 "1. Activate environment: source .venv/bin/activate (Linux/macOS) or .\\.venv\\Scripts\\Activate.ps1 (Windows)"
             )
-            print("2. Install RAG dependencies: pip install -e .[rag]")
+            print('2. Install RAG dependencies: uv pip install -e ".[rag]"')
             print("3. Either use the test PDF or get your own PDF file")
             print("4. Run: answer = quick_rag('document.pdf', 'What is this about?')")
 
@@ -123,10 +123,10 @@ def demo_basic_rag():
         print("  .\\.venv\\Scripts\\Activate.ps1")
         print("  ")
         print("  # Install RAG extras")
-        print("  pip install -e .[rag]")
+        print('  uv pip install -e ".[rag]"')
         print("  ")
         print("  # Or install dependencies individually:")
-        print("  pip install pypdf sentence-transformers faiss-cpu")
+        print("  uv pip install pypdf sentence-transformers faiss-cpu")
 
     print("\n" + "=" * 60)
 
@@ -215,7 +215,7 @@ def demo_error_handling():
 
     except ImportError as e:
         print(f"❌ Missing dependencies: {e}")
-        print("Install with: pip install -e \".[rag]\"")
+        print('Install with: uv pip install -e ".[rag]"')
 
     print("-" * 40)
 
@@ -290,7 +290,7 @@ def main():
     print("1. Activate virtual environment:")
     print("   Linux/macOS: source .venv/bin/activate")
     print("   Windows: .\\.venv\\Scripts\\Activate.ps1")
-    print("2. Install RAG dependencies: pip install -e .[rag]")
+    print('2. Install RAG dependencies: uv pip install -e ".[rag]"')
     print("3. Get a PDF document to test with")
     print("4. Try the CLI commands: gaia rag --help")
     print("5. Use RAG in Python: from gaia.rag.sdk import RAGSDK, quick_rag")

gaia/rag/pdf_utils.py

@@ -129,7 +129,7 @@ def extract_images_from_page_pymupdf(pdf_path: str, page_num: int) -> List[dict]
         doc.close()
 
     except ImportError:
-        logger.error("PyMuPDF not installed. Install: pip install pymupdf")
+        logger.error("PyMuPDF not installed. Install: uv pip install pymupdf")
     except Exception as e:
         logger.error(f"Error extracting images from page {page_num}: {e}")
 

gaia/rag/sdk.py

@@ -175,9 +175,9 @@ class RAGSDK:
             error_msg = (
                 f"\n❌ Error: Missing required RAG dependencies: {', '.join(missing)}\n\n"
                 f"Please install the RAG dependencies:\n"
-                f"  pip install -e .[rag]\n\n"
+                f'  uv pip install -e ".[rag]"\n\n'
                 f"Or install packages directly:\n"
-                f"  pip install {' '.join(missing)}\n"
+                f"  uv pip install {' '.join(missing)}\n"
             )
             raise ImportError(error_msg)
 

gaia/security.py

@@ -1,4 +1,4 @@
-# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
 # SPDX-License-Identifier: MIT
 """
 Security utilities for GAIA.

gaia/talk/sdk.py

@@ -360,7 +360,7 @@ class TalkSDK:
             return True
         except ImportError:
             self.log.warning(
-                'RAG dependencies not available. Install with: pip install -e ".[rag]"'
+                'RAG dependencies not available. Install with: uv pip install -e ".[rag]"'
             )
             return False
         except Exception as e: