amd-gaia 0.14.1__py3-none-any.whl

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

@@ -0,0 +1,186 @@
+# Copyright(C) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
+# SPDX-License-Identifier: MIT
+"""
+Next.js workflow definition.
+
+Defines the phases and steps for building Next.js CRUD applications.
+"""
+
+from typing import List
+
+from ..steps.base import UserContext
+from ..steps.nextjs import (
+    CreateNextAppStep,
+    InstallDependenciesStep,
+    ManageApiEndpointDynamicStep,
+    ManageApiEndpointStep,
+    ManageDataModelStep,
+    ManageReactComponentStep,
+    PrismaInitStep,
+    RunTestsStep,
+    SetupAppStylingStep,
+    SetupPrismaStep,
+    SetupTestingStep,
+    TestCrudApiStep,
+    UpdateLandingPageStep,
+    ValidateCrudStructureStep,
+    ValidateTypescriptStep,
+)
+from .base import WorkflowPhase
+
+
+def create_nextjs_workflow(context: UserContext) -> List[WorkflowPhase]:
+    """Create the Next.js CRUD workflow phases.
+
+    Args:
+        context: User context with request details
+
+    Returns:
+        List of workflow phases
+    """
+    entity = context.entity_name or "Item"
+    fields = context.schema_fields or {"title": "string", "completed": "boolean"}
+
+    # Phase 0: Project Initialization
+    # Includes testing deps so they're available when test files are generated
+    init_phase = WorkflowPhase(
+        name="initialization",
+        description="Set up Next.js project with dependencies",
+        steps=[
+            CreateNextAppStep(
+                name="create_next_app",
+                description="Initialize Next.js project",
+            ),
+            SetupAppStylingStep(
+                name="setup_styling",
+                description="Configure modern app-wide styling",
+            ),
+            InstallDependenciesStep(
+                name="install_deps",
+                description="Install Prisma, Zod, and other dependencies",
+            ),
+            SetupTestingStep(
+                name="setup_testing",
+                description="Install Vitest and testing libraries",
+            ),
+            PrismaInitStep(
+                name="prisma_init",
+                description="Initialize Prisma with SQLite",
+            ),
+        ],
+    )
+
+    # Phase 1: Data Layer
+    data_phase = WorkflowPhase(
+        name="data_layer",
+        description="Create database model and API routes",
+        steps=[
+            ManageDataModelStep(
+                name="data_model",
+                description=f"Create {entity} Prisma model",
+                entity_name=entity,
+                fields=fields,
+            ),
+            SetupPrismaStep(
+                name="setup_prisma",
+                description="Generate Prisma client and push to database",
+            ),
+            ManageApiEndpointStep(
+                name="api_collection",
+                description=f"Create {entity} collection API (GET, POST)",
+                entity_name=entity,
+                fields=fields,
+            ),
+            ManageApiEndpointDynamicStep(
+                name="api_item",
+                description=f"Create {entity} item API (GET, PATCH, DELETE)",
+                entity_name=entity,
+                fields=fields,
+            ),
+        ],
+    ).with_validation(lint=False, typecheck=True, tests=False, fail_on_type_error=False)
+
+    # Phase 2: UI Components
+    ui_phase = WorkflowPhase(
+        name="ui_components",
+        description="Create React components and pages",
+        steps=[
+            ManageReactComponentStep(
+                name="list_component",
+                description=f"Create {entity} list page",
+                entity_name=entity,
+                variant="list",
+                fields=fields,
+            ),
+            ManageReactComponentStep(
+                name="form_component",
+                description=f"Create {entity} form component",
+                entity_name=entity,
+                variant="form",
+                fields=fields,
+            ),
+            ManageReactComponentStep(
+                name="new_page",
+                description=f"Create {entity} new page",
+                entity_name=entity,
+                variant="new",
+                fields=fields,
+            ),
+            # Actions must come before detail_page since detail_page imports it
+            ManageReactComponentStep(
+                name="actions_component",
+                description=f"Create {entity} actions component",
+                entity_name=entity,
+                variant="actions",
+                fields=fields,
+            ),
+            ManageReactComponentStep(
+                name="detail_page",
+                description=f"Create {entity} detail page",
+                entity_name=entity,
+                variant="detail",
+                fields=fields,
+            ),
+        ],
+    ).with_validation(lint=False, typecheck=True, tests=False, fail_on_type_error=False)
+
+    # Phase 3: Validation & Polish
+    validation_phase = WorkflowPhase(
+        name="validation",
+        description="Validate structure and run tests",
+        steps=[
+            ValidateCrudStructureStep(
+                name="validate_structure",
+                description="Check all required files exist",
+                entity_name=entity,
+            ),
+            ValidateTypescriptStep(
+                name="validate_typescript",
+                description="Run TypeScript compiler",
+            ),
+            TestCrudApiStep(
+                name="test_api",
+                description="Test CRUD operations",
+                entity_name=entity,
+            ),
+            UpdateLandingPageStep(
+                name="update_landing",
+                description="Add navigation link to landing page",
+                entity_name=entity,
+            ),
+        ],
+    )
+
+    # Phase 4: Run Tests
+    testing_phase = WorkflowPhase(
+        name="testing",
+        description="Run all tests",
+        steps=[
+            RunTestsStep(
+                name="run_tests",
+                description="Run all tests with npm test",
+            ),
+        ],
+    )
+
+    return [init_phase, data_phase, ui_phase, validation_phase, testing_phase]

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

@@ -0,0 +1,94 @@
+# Copyright(C) 2024-2025 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

@@ -0,0 +1,11 @@
+# Copyright(C) 2024-2025 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

@@ -0,0 +1,77 @@
+# Copyright(C) 2024-2025 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
+"""