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

gaia/agents/code/orchestration/workflows/python.py

@@ -1,94 +1,94 @@
-# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
-# SPDX-License-Identifier: MIT
-"""
-Python workflow definition.
-
-Defines the phases and steps for building Python applications.
-"""
-
-from typing import List
-
-from ..steps.base import UserContext
-from ..steps.python import (
-    AnalyzePylintStep,
-    AutoFixSyntaxStep,
-    CreateProjectStep,
-    FixLintingStep,
-    ListFilesStep,
-    RunPytestStep,
-    ValidateProjectStep,
-)
-from .base import WorkflowPhase
-
-
-def create_python_workflow(context: UserContext) -> List[WorkflowPhase]:
-    """Create the Python development workflow phases.
-
-    Args:
-        context: User context with request details
-
-    Returns:
-        List of workflow phases
-    """
-    # Phase 1: Project Creation
-    creation_phase = WorkflowPhase(
-        name="creation",
-        description="Create Python project structure",
-        steps=[
-            CreateProjectStep(
-                name="create_project",
-                description="Generate project files",
-                user_request=context.user_request,
-            ),
-            ListFilesStep(
-                name="list_files",
-                description="Discover created files",
-            ),
-        ],
-    )
-
-    # Phase 2: Validation
-    validation_phase = WorkflowPhase(
-        name="validation",
-        description="Validate and fix project",
-        steps=[
-            ValidateProjectStep(
-                name="validate_project",
-                description="Check project structure",
-            ),
-            AutoFixSyntaxStep(
-                name="auto_fix_syntax",
-                description="Fix syntax errors",
-            ),
-        ],
-    )
-
-    # Phase 3: Code Quality
-    quality_phase = WorkflowPhase(
-        name="quality",
-        description="Lint and format code",
-        steps=[
-            AnalyzePylintStep(
-                name="analyze_pylint",
-                description="Run pylint analysis",
-            ),
-            FixLintingStep(
-                name="fix_linting",
-                description="Fix linting issues",
-            ),
-        ],
-    )
-
-    # Phase 4: Testing
-    testing_phase = WorkflowPhase(
-        name="testing",
-        description="Run tests",
-        steps=[
-            RunPytestStep(
-                name="run_tests",
-                description="Run pytest",
-            ),
-        ],
-    )
-
-    return [creation_phase, validation_phase, quality_phase, testing_phase]
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""
+Python workflow definition.
+
+Defines the phases and steps for building Python applications.
+"""
+
+from typing import List
+
+from ..steps.base import UserContext
+from ..steps.python import (
+    AnalyzePylintStep,
+    AutoFixSyntaxStep,
+    CreateProjectStep,
+    FixLintingStep,
+    ListFilesStep,
+    RunPytestStep,
+    ValidateProjectStep,
+)
+from .base import WorkflowPhase
+
+
+def create_python_workflow(context: UserContext) -> List[WorkflowPhase]:
+    """Create the Python development workflow phases.
+
+    Args:
+        context: User context with request details
+
+    Returns:
+        List of workflow phases
+    """
+    # Phase 1: Project Creation
+    creation_phase = WorkflowPhase(
+        name="creation",
+        description="Create Python project structure",
+        steps=[
+            CreateProjectStep(
+                name="create_project",
+                description="Generate project files",
+                user_request=context.user_request,
+            ),
+            ListFilesStep(
+                name="list_files",
+                description="Discover created files",
+            ),
+        ],
+    )
+
+    # Phase 2: Validation
+    validation_phase = WorkflowPhase(
+        name="validation",
+        description="Validate and fix project",
+        steps=[
+            ValidateProjectStep(
+                name="validate_project",
+                description="Check project structure",
+            ),
+            AutoFixSyntaxStep(
+                name="auto_fix_syntax",
+                description="Fix syntax errors",
+            ),
+        ],
+    )
+
+    # Phase 3: Code Quality
+    quality_phase = WorkflowPhase(
+        name="quality",
+        description="Lint and format code",
+        steps=[
+            AnalyzePylintStep(
+                name="analyze_pylint",
+                description="Run pylint analysis",
+            ),
+            FixLintingStep(
+                name="fix_linting",
+                description="Fix linting issues",
+            ),
+        ],
+    )
+
+    # Phase 4: Testing
+    testing_phase = WorkflowPhase(
+        name="testing",
+        description="Run tests",
+        steps=[
+            RunPytestStep(
+                name="run_tests",
+                description="Run pytest",
+            ),
+        ],
+    )
+
+    return [creation_phase, validation_phase, quality_phase, testing_phase]

gaia/agents/code/prompts/__init__.py

@@ -1,11 +1,11 @@
-# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
-# SPDX-License-Identifier: MIT
-"""Language-specific prompt modules for Code Agent."""
-
-from .nextjs_prompt import NEXTJS_PROMPT
-from .python_prompt import get_python_prompt
-
-__all__ = [
-    "get_python_prompt",
-    "NEXTJS_PROMPT",
-]
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""Language-specific prompt modules for Code Agent."""
+
+from .nextjs_prompt import NEXTJS_PROMPT
+from .python_prompt import get_python_prompt
+
+__all__ = [
+    "get_python_prompt",
+    "NEXTJS_PROMPT",
+]

gaia/agents/code/prompts/base_prompt.py

@@ -1,77 +1,77 @@
-# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
-# SPDX-License-Identifier: MIT
-"""Base system prompt for Code Agent - universal development guidance."""
-
-import os
-from typing import Optional
-
-
-def get_base_prompt(gaia_md_path: Optional[str] = None) -> str:
-    """Get the universal, language-agnostic prompt for Code Agent.
-
-    This contains core instructions that apply to all programming languages:
-    - JSON response format
-    - Tool calling conventions
-    - General error recovery patterns
-    - Planning and validation workflow
-    - Project context loading
-
-    Args:
-        gaia_md_path: Optional path to GAIA.md file for project context
-
-    Returns:
-        Base system prompt string with project context if available
-    """
-    # Load project context if available
-    gaia_context = ""
-    gaia_path = gaia_md_path or "GAIA.md"
-
-    if os.path.exists(gaia_path):
-        try:
-            with open(gaia_path, "r", encoding="utf-8") as f:
-                gaia_content = f.read()
-                gaia_context = f"\n\nProject Context:\n{gaia_content}\n"
-        except Exception:
-            pass
-
-    return f"""You are a code assistant. Execute tasks using tools.
-
-{gaia_context}
-
-## Response Format
-Your responses must be valid JSON:
-{{"thought": "reasoning", "goal": "objective", "plan": [list of tool calls]}}
-
-Each plan step must be:
-{{"tool": "tool_name", "tool_args": {{"arg1": "value1", "arg2": "value2"}}}}
-
-## Rules
-1. Call ONE tool at a time
-2. Check the result before proceeding
-3. If result has `error`, fix the issue before continuing
-4. Ask clarifying questions when requirements are ambiguous
-
-## Tool Selection
-- For CLI commands: `run_cli_command`
-- For API routes: `manage_api_endpoint`
-- For React components: `manage_react_component`
-- For Prisma models: `manage_data_model` then `setup_prisma`
-- For any file: `write_file` or `edit_file`
-
-## Research Tools (USE THESE WHEN ENCOUNTERING ERRORS)
-- `search_documentation(query, library)` - Search official library docs
-  Examples:
-  - `search_documentation("App Router POST handler", library="nextjs")`
-  - `search_documentation("Prisma DateTime field", library="prisma")`
-  - `search_documentation("zod validation schema", library="zod")`
-- `search_web(query)` - Search web for error solutions and current patterns
-  Examples:
-  - `search_web("Next.js 14 405 Method Not Allowed")`
-  - `search_web("Prisma client not generating types")`
-
-**WHEN TO USE:**
-- ALWAYS search before fixing library-related errors (Prisma, Next.js, React, Zod)
-- When encountering type errors, hydration errors, or validation errors
-- When a fix attempt fails - search for the correct solution
-- Do NOT guess at fixes without searching first
-"""
+# Copyright(C) 2025-2026 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""Base system prompt for Code Agent - universal development guidance."""
+
+import os
+from typing import Optional
+
+
+def get_base_prompt(gaia_md_path: Optional[str] = None) -> str:
+    """Get the universal, language-agnostic prompt for Code Agent.
+
+    This contains core instructions that apply to all programming languages:
+    - JSON response format
+    - Tool calling conventions
+    - General error recovery patterns
+    - Planning and validation workflow
+    - Project context loading
+
+    Args:
+        gaia_md_path: Optional path to GAIA.md file for project context
+
+    Returns:
+        Base system prompt string with project context if available
+    """
+    # Load project context if available
+    gaia_context = ""
+    gaia_path = gaia_md_path or "GAIA.md"
+
+    if os.path.exists(gaia_path):
+        try:
+            with open(gaia_path, "r", encoding="utf-8") as f:
+                gaia_content = f.read()
+                gaia_context = f"\n\nProject Context:\n{gaia_content}\n"
+        except Exception:
+            pass
+
+    return f"""You are a code assistant. Execute tasks using tools.
+
+{gaia_context}
+
+## Response Format
+Your responses must be valid JSON:
+{{"thought": "reasoning", "goal": "objective", "plan": [list of tool calls]}}
+
+Each plan step must be:
+{{"tool": "tool_name", "tool_args": {{"arg1": "value1", "arg2": "value2"}}}}
+
+## Rules
+1. Call ONE tool at a time
+2. Check the result before proceeding
+3. If result has `error`, fix the issue before continuing
+4. Ask clarifying questions when requirements are ambiguous
+
+## Tool Selection
+- For CLI commands: `run_cli_command`
+- For API routes: `manage_api_endpoint`
+- For React components: `manage_react_component`
+- For Prisma models: `manage_data_model` then `setup_prisma`
+- For any file: `write_file` or `edit_file`
+
+## Research Tools (USE THESE WHEN ENCOUNTERING ERRORS)
+- `search_documentation(query, library)` - Search official library docs
+  Examples:
+  - `search_documentation("App Router POST handler", library="nextjs")`
+  - `search_documentation("Prisma DateTime field", library="prisma")`
+  - `search_documentation("zod validation schema", library="zod")`
+- `search_web(query)` - Search web for error solutions and current patterns
+  Examples:
+  - `search_web("Next.js 14 405 Method Not Allowed")`
+  - `search_web("Prisma client not generating types")`
+
+**WHEN TO USE:**
+- ALWAYS search before fixing library-related errors (Prisma, Next.js, React, Zod)
+- When encountering type errors, hydration errors, or validation errors
+- When a fix attempt fails - search for the correct solution
+- Do NOT guess at fixes without searching first
+"""