amd-gaia 0.14.2__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/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

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

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

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 = []
 

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
@@ -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,
         )
 

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"]

gaia/database/agent.py

@@ -0,0 +1,176 @@
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+
+"""DatabaseAgent - Agent with built-in database tools."""
+
+from typing import Any, Dict, Optional
+
+from gaia.agents.base import Agent, tool
+from gaia.database.mixin import DatabaseMixin
+
+
+class DatabaseAgent(Agent, DatabaseMixin):
+    """
+    Agent with built-in SQLite database tools.
+
+    Extends Agent with database capabilities, automatically registering
+    tools that allow the LLM to query and modify the database.
+
+    Example:
+        class PatientAgent(DatabaseAgent):
+            def __init__(self, **kwargs):
+                super().__init__(db_path="data/patients.db", **kwargs)
+
+                if not self.table_exists("patients"):
+                    self.execute('''
+                        CREATE TABLE patients (
+                            id INTEGER PRIMARY KEY,
+                            name TEXT NOT NULL,
+                            dob TEXT
+                        )
+                    ''')
+
+            def _get_system_prompt(self) -> str:
+                return "You help manage patient records."
+
+        # LLM can now use: db_query, db_insert, db_update, db_delete
+
+    Security Note:
+        The db_query tool allows the LLM to execute arbitrary SELECT queries.
+        For production use, consider overriding _register_db_tools() to expose
+        only domain-specific, validated operations.
+    """
+
+    def __init__(
+        self,
+        db_path: str = ":memory:",
+        **kwargs,
+    ):
+        """
+        Initialize DatabaseAgent.
+
+        Args:
+            db_path: Path to SQLite database file, or ":memory:" for in-memory.
+                     Parent directories are created automatically.
+            **kwargs: Additional arguments passed to Agent.
+        """
+        super().__init__(**kwargs)
+        self.init_db(db_path)
+        self._register_db_tools()
+
+    def _register_db_tools(self) -> None:
+        """Register database tools for LLM use."""
+        agent = self
+
+        @tool
+        def db_query(sql: str, params: Optional[Dict[str, Any]] = None) -> Dict:
+            """
+            Execute a SELECT query and return results.
+
+            Args:
+                sql: SQL SELECT query with :param placeholders
+                params: Optional dictionary of parameter values
+
+            Returns:
+                Dictionary with 'rows' (list of row dicts) and 'count'
+
+            Example:
+                db_query("SELECT * FROM users WHERE age > :min_age", {"min_age": 18})
+            """
+            rows = agent.query(sql, params or {})
+            return {"rows": rows, "count": len(rows)}
+
+        @tool
+        def db_insert(table: str, data: Dict[str, Any]) -> Dict:
+            """
+            Insert a row into a table.
+
+            Args:
+                table: Table name
+                data: Dictionary of column names to values
+
+            Returns:
+                Dictionary with 'id' (the inserted row's ID) and 'success'
+
+            Example:
+                db_insert("users", {"name": "Alice", "email": "alice@example.com"})
+            """
+            row_id = agent.insert(table, data)
+            return {"id": row_id, "success": True}
+
+        @tool
+        def db_update(
+            table: str, data: Dict[str, Any], where: str, params: Dict[str, Any]
+        ) -> Dict:
+            """
+            Update rows matching a condition.
+
+            Args:
+                table: Table name
+                data: Dictionary of column names to new values
+                where: WHERE clause with :param placeholders (without WHERE keyword)
+                params: Dictionary of parameter values for WHERE clause
+
+            Returns:
+                Dictionary with 'updated' (number of rows affected)
+
+            Example:
+                db_update("users", {"email": "new@example.com"}, "id = :id", {"id": 42})
+            """
+            count = agent.update(table, data, where, params)
+            return {"updated": count}
+
+        @tool
+        def db_delete(table: str, where: str, params: Dict[str, Any]) -> Dict:
+            """
+            Delete rows matching a condition.
+
+            Args:
+                table: Table name
+                where: WHERE clause with :param placeholders (without WHERE keyword)
+                params: Dictionary of parameter values for WHERE clause
+
+            Returns:
+                Dictionary with 'deleted' (number of rows deleted)
+
+            Example:
+                db_delete("sessions", "expires_at < :now", {"now": "2024-01-01"})
+            """
+            count = agent.delete(table, where, params)
+            return {"deleted": count}
+
+        @tool
+        def db_tables() -> Dict:
+            """
+            List all tables in the database.
+
+            Returns:
+                Dictionary with 'tables' (list of table names)
+            """
+            rows = agent.query(
+                "SELECT name FROM sqlite_master WHERE type='table' ORDER BY name"
+            )
+            return {"tables": [row["name"] for row in rows]}
+
+        @tool
+        def db_schema(table: str) -> Dict:
+            """
+            Get the schema of a table.
+
+            Args:
+                table: Table name
+
+            Returns:
+                Dictionary with 'columns' (list of column info dicts)
+            """
+            rows = agent.query(f"PRAGMA table_info({table})")
+            columns = [
+                {
+                    "name": row["name"],
+                    "type": row["type"],
+                    "nullable": not row["notnull"],
+                    "primary_key": bool(row["pk"]),
+                }
+                for row in rows
+            ]
+            return {"table": table, "columns": columns}