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

gaia/agents/routing/__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
 """Routing agent for intelligent agent selection and disambiguation."""
 

gaia/agents/routing/agent.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
 """RoutingAgent - Intelligently routes requests and disambiguates parameters."""
 
@@ -125,6 +125,9 @@ class RoutingAgent:
 
         logger.debug(f"Analysis result: {analysis}")
 
+        # If language could not be determined, default to TypeScript/Next.js
+        analysis = self._default_unknown_language_to_typescript(analysis)
+
         # Check if we have all required parameters
         if self._has_unknowns(analysis):
             if self.api_mode:
@@ -423,6 +426,58 @@ Conversation:
 
         return "Please provide more details about your project."
 
+    def _get_console(self):
+        """Return the configured output handler or a default console."""
+        if self.output_handler:
+            return self.output_handler
+
+        from gaia.agents.base.console import AgentConsole
+
+        return AgentConsole()
+
+    def _enforce_typescript_only(
+        self, language: str, project_type: str, console
+    ) -> tuple[str, str]:
+        """Warn and normalize when routing to unsupported languages."""
+        is_nextjs = language == "typescript" and project_type == "fullstack"
+
+        if not is_nextjs:
+            console.print_error(
+                "Only TypeScript (Next.js) is currently supported. "
+                "Please try a Next.js/TypeScript request."
+            )
+            raise SystemExit(1)
+
+        return language, project_type
+
+    def _default_unknown_language_to_typescript(
+        self, analysis: Dict[str, Any]
+    ) -> Dict[str, Any]:
+        """Default unknown language/project type to TypeScript/Next.js."""
+        params = analysis.get("parameters", {})
+        language = params.get("language")
+
+        if language != "unknown":
+            return analysis
+
+        console = self._get_console()
+        console.print_info(
+            "Defaulting to TypeScript (Next.js) because the language could not be determined."
+        )
+
+        params["language"] = "typescript"
+        if params.get("project_type") == "unknown":
+            params["project_type"] = "fullstack"
+
+        analysis["parameters"] = params
+        analysis["confidence"] = 1.0
+        analysis["reasoning"] = (
+            analysis.get("reasoning", "")
+            + " Defaulted to TypeScript/Next.js due to unknown language."
+        ).strip()
+
+        return analysis
+
     def _create_agent(self, analysis: Dict[str, Any]) -> Agent:
         """
         Create configured agent based on analysis.
@@ -447,12 +502,10 @@ Conversation:
             )
 
             # Use passed output_handler or create AgentConsole (CLI default)
-            if self.output_handler:
-                console = self.output_handler
-            else:
-                from gaia.agents.base.console import AgentConsole
-
-                console = AgentConsole()
+            console = self._get_console()
+            language, project_type = self._enforce_typescript_only(
+                language, project_type, console
+            )
 
             # Print agent selected message
             console.print_agent_selected("CodeAgent", language, project_type)
@@ -502,6 +555,11 @@ Conversation:
 
         from gaia.agents.code.agent import CodeAgent
 
+        console = self._get_console()
+        language, project_type = self._enforce_typescript_only(
+            language, project_type, console
+        )
+
         # Build agent kwargs, including output_handler if provided
         agent_init_kwargs = dict(self.agent_kwargs)
         if self.output_handler:

gaia/agents/routing/system_prompt.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
 """System prompt for RoutingAgent - analyzes requests and determines agent configuration."""
 

gaia/api/__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 OpenAI-Compatible API

gaia/api/agent_registry.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
 """
 Agent Registry - Exposes GAIA agents as OpenAI "models"

gaia/api/app.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
 """
 CLI entry point for GAIA OpenAI-compatible API server

gaia/api/openai_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
 """
 OpenAI-compatible API server for GAIA

gaia/api/schemas.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
 """
 OpenAI API-compatible Pydantic schemas

gaia/api/sse_handler.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
 """
 SSE Output Handler for API streaming.
@@ -179,8 +179,11 @@ class SSEOutputHandler(OutputHandler):
 
     def print_completion(self, steps_taken: int, steps_limit: int):
         """Print completion summary."""
+        # Infer status from steps_taken vs steps_limit
+        status = "success" if steps_taken < steps_limit else "incomplete"
         self._add_event(
-            "completion", {"steps_taken": steps_taken, "steps_limit": steps_limit}
+            "completion",
+            {"steps_taken": steps_taken, "steps_limit": steps_limit, "status": status},
         )
 
     # === File Preview Methods (Required for Code Agent) ===

gaia/apps/__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 Applications"""

gaia/apps/llm/__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/apps/summarize/pdf_formatter.py

@@ -36,7 +36,7 @@ class PDFFormatter:
     def __init__(self):
         if not HAS_REPORTLAB:
             raise ImportError(
-                "PDF output requires reportlab. Install with: pip install reportlab"
+                "PDF output requires reportlab. Install with: uv pip install reportlab"
             )
 
         self.styles = getSampleStyleSheet()

gaia/audio/__init__.py

@@ -1,2 +1,2 @@
-# 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/audio/audio_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
 
 import asyncio
@@ -123,7 +123,7 @@ class AudioClient:
 
         except ImportError:
             self.log.error(
-                "WhisperAsr not found. Please install voice support with: pip install .[talk]"
+                'WhisperAsr not found. Please install voice support with: uv pip install ".[talk]"'
             )
             raise
         except Exception as e:
@@ -306,7 +306,7 @@ class AudioClient:
                 self.log.debug("TTS initialized successfully")
             except Exception as e:
                 raise RuntimeError(
-                    f"Failed to initialize TTS:\n{e}\nInstall talk dependencies with: pip install .[talk]\nYou can also use --no-tts option to disable TTS"
+                    f'Failed to initialize TTS:\n{e}\nInstall talk dependencies with: uv pip install ".[talk]"\nYou can also use --no-tts option to disable TTS'
                 )
 
     async def speak_text(self, text: str) -> None:

gaia/audio/audio_recorder.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 first

gaia/audio/kokoro_tts.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 queue
@@ -43,9 +43,9 @@ class KokoroTTS:
             error_msg = (
                 f"\n❌ Error: Missing required talk dependencies: {', '.join(missing)}\n\n"
                 f"Please install the talk dependencies:\n"
-                f"  pip install -e .[talk]\n\n"
+                f'  uv pip install -e ".[talk]"\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/audio/whisper_asr.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
@@ -56,9 +56,9 @@ class WhisperAsr(AudioRecorder):
             error_msg = (
                 f"\n❌ Error: Missing required talk dependencies: {', '.join(missing)}\n\n"
                 f"Please install the talk dependencies:\n"
-                f"  pip install -e .[talk]\n\n"
+                f'  uv pip install -e ".[talk]"\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/chat/__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/chat/prompts.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
 
 from gaia.logger import get_logger

gaia/chat/sdk.py

@@ -142,6 +142,27 @@ class ChatSDK:
             return json.dumps(content)
         return str(content)
 
+    def _prepare_messages_for_llm(
+        self, messages: List[Dict[str, Any]]
+    ) -> List[Dict[str, Any]]:
+        """
+        Ensure messages are safe to send to the LLM by appending a continuation
+        prompt when the last entry is a tool_result, which some models ignore.
+        """
+        if not messages:
+            return []
+
+        prepared = list(messages)
+        try:
+            last_role = prepared[-1].get("role")
+        except Exception:
+            return prepared
+
+        if last_role == "tool":
+            prepared.append({"role": "user", "content": "continue"})
+
+        return prepared
+
     def send_messages(
         self,
         messages: List[Dict[str, Any]],
@@ -160,6 +181,8 @@ class ChatSDK:
             ChatResponse with the complete response
         """
         try:
+            messages = self._prepare_messages_for_llm(messages)
+
             # Convert messages to chat history format
             chat_history = []
 
@@ -241,6 +264,8 @@ class ChatSDK:
             ChatResponse chunks as they arrive
         """
         try:
+            messages = self._prepare_messages_for_llm(messages)
+
             # Convert messages to chat history format
             chat_history = []
 
@@ -687,7 +712,7 @@ class ChatSDK:
             from gaia.rag.sdk import RAGSDK, RAGConfig
         except ImportError:
             raise ImportError(
-                'RAG dependencies not installed. Install with: pip install ".[rag]"'
+                'RAG dependencies not installed. Install with: uv pip install -e ".[rag]"'
             )
 
         # Create RAG config matching chat config

gaia/cli.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 asyncio
@@ -2109,7 +2109,7 @@ Examples:
             print("The evaluation dependencies are not installed.")
             print("")
             print("To fix this, install the evaluation dependencies:")
-            print("  pip install -e .[eval]")
+            print('  uv pip install -e ".[eval]"')
             print("")
             print("This will install required packages including:")
             print("  - anthropic (for Claude AI)")
@@ -2464,7 +2464,7 @@ Examples:
 
                         if not HAS_REPORTLAB:
                             print(
-                                "❌ Error: PDF output requires reportlab. Install with: pip install reportlab"
+                                "❌ Error: PDF output requires reportlab. Install with: uv pip install reportlab"
                             )
                             if args.format == "both":
                                 print(
@@ -2542,7 +2542,7 @@ Examples:
                             pdf_formatter = PDFFormatter()
                         else:
                             print(
-                                "⚠️  Warning: PDF output requires reportlab. Install with: pip install reportlab"
+                                "⚠️  Warning: PDF output requires reportlab. Install with: uv pip install reportlab"
                             )
                             if args.format == "pdf":
                                 print("❌ Cannot generate PDF files without reportlab.")
@@ -2660,7 +2660,7 @@ Let me know your answer!
                 log.debug("ASR initialized successfully")
             except ImportError:
                 log.error(
-                    "WhisperAsr not found. Please install voice support with: pip install -e .[talk]"
+                    'WhisperAsr not found. Please install voice support with: uv pip install -e ".[talk]"'
                 )
                 raise
             except Exception as e:
@@ -2746,7 +2746,7 @@ Let me know your answer!
                     "❌ Error: YouTube transcript functionality requires additional dependencies."
                 )
                 print(
-                    "Please install: pip install llama-index-readers-youtube-transcript"
+                    "Please install: uv pip install llama-index-readers-youtube-transcript"
                 )
                 print(f"Import error: {e}")
                 sys.exit(1)
@@ -3150,7 +3150,7 @@ Let me know your answer!
             print("The evaluation dependencies are not installed.")
             print("")
             print("To fix this, install the evaluation dependencies:")
-            print("  pip install -e .[eval]")
+            print('  uv pip install -e ".[eval]"')
             print("")
             print("This will install required packages including:")
             print("  - anthropic (for Claude AI)")
@@ -3323,7 +3323,7 @@ Let me know your answer!
             print("The evaluation dependencies are not installed.")
             print("")
             print("To fix this, install the evaluation dependencies:")
-            print("  pip install -e .[eval]")
+            print('  uv pip install -e ".[eval]"')
             print("")
             print("This will install required packages including:")
             print("  - anthropic (for Claude AI)")
@@ -3363,7 +3363,7 @@ Let me know your answer!
             print("The evaluation dependencies are not installed.")
             print("")
             print("To fix this, install the evaluation dependencies:")
-            print("  pip install -e .[eval]")
+            print('  uv pip install -e ".[eval]"')
             print("")
             print("This will install required packages including:")
             print("  - anthropic (for Claude AI)")
@@ -3727,7 +3727,7 @@ Let me know your answer!
                 print("The evaluation dependencies are not installed.")
                 print("")
                 print("To fix this, install the evaluation dependencies:")
-                print("  pip install -e .[eval]")
+                print('  uv pip install -e ".[eval]"')
                 print("")
                 print("This will install required packages including:")
                 print("  - anthropic (for Claude AI)")
@@ -3802,7 +3802,7 @@ Let me know your answer!
                 print("The evaluation dependencies are not installed.")
                 print("")
                 print("To fix this, install the evaluation dependencies:")
-                print("  pip install -e .[eval]")
+                print('  uv pip install -e ".[eval]"')
                 print("")
                 print("This will install required packages including:")
                 print("  - anthropic (for Claude AI)")
@@ -3878,7 +3878,7 @@ Let me know your answer!
             print("The evaluation dependencies are not installed.")
             print("")
             print("To fix this, install the evaluation dependencies:")
-            print("  pip install -e .[eval]")
+            print('  uv pip install -e ".[eval]"')
             print("")
             print("This will install required packages including:")
             print("  - anthropic (for Claude AI)")
@@ -4495,7 +4495,7 @@ def handle_jira_command(args):
     except ImportError as e:
         log.error(f"Failed to import Jira app: {e}")
         print("❌ Error: Jira app components are not available")
-        print("Make sure GAIA is installed properly: pip install -e .")
+        print("Make sure GAIA is installed properly: uv pip install -e .")
         sys.exit(1)
     except Exception as e:
         log.error(f"Error running Jira app: {e}")
@@ -4546,7 +4546,7 @@ def handle_docker_command(args):
     except ImportError as e:
         log.error(f"Failed to import Docker app: {e}")
         print("❌ Error: Docker app components are not available")
-        print("Make sure GAIA is installed properly: pip install -e .")
+        print("Make sure GAIA is installed properly: uv pip install -e .")
         sys.exit(1)
     except Exception as e:
         log.error(f"Error running Docker app: {e}")
@@ -4624,7 +4624,7 @@ def handle_api_command(args):
         except ImportError as e:
             log.error(f"Failed to import API server: {e}")
             print("❌ Error: API server components are not available")
-            print("Make sure uvicorn is installed: pip install uvicorn")
+            print("Make sure uvicorn is installed: uv pip install uvicorn")
             sys.exit(1)
         except KeyboardInterrupt:
             print("\n✅ API server stopped")
@@ -4866,7 +4866,7 @@ def handle_blender_command(args):
     # Check if Blender components are available
     if not BLENDER_AVAILABLE:
         print("❌ Error: Blender agent components are not available")
-        print("Install blender dependencies with: pip install -e .[blender]")
+        print('Install blender dependencies with: uv pip install -e ".[blender]"')
         sys.exit(1)
 
     # Initialize Lemonade with blender agent profile (32768 context)
@@ -4911,7 +4911,7 @@ def handle_blender_command(args):
             max_steps=args.steps,
             output_dir=output_dir,
             streaming=args.stream,
-            show_stats=args.stats,
+            show_stats=args.show_stats,
             debug_prompts=args.debug_prompts,
         )
 
@@ -5063,7 +5063,7 @@ def handle_mcp_start(args):
             print("❌ Error: MCP dependencies not installed.")
             print("")
             print("To fix this, install the MCP dependencies:")
-            print("  pip install -e .[mcp]")
+            print('  uv pip install -e ".[mcp]"')
             return
 
         # Import and start the HTTP-native MCP bridge

gaia/database/__init__.py

@@ -0,0 +1,10 @@
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+
+"""Database utilities for GAIA SDK."""
+
+from gaia.database.agent import DatabaseAgent
+from gaia.database.mixin import DatabaseMixin
+from gaia.database.testing import temp_db
+
+__all__ = ["DatabaseAgent", "DatabaseMixin", "temp_db"]