Custom Agents

from gaia.agents.base.agent import Agent
from gaia.agents.base.console import AgentConsole


class ExampleAgent(Agent):
    AGENT_ID = "example-agent"
    AGENT_NAME = "Example Agent"
    AGENT_DESCRIPTION = (
        "A bundled example agent that shows how installer-seeded custom agents work"
    )
    CONVERSATION_STARTERS = [
        "What are you an example of?",
        "How do I build my own agent like you?",
    ]

    def _get_system_prompt(self) -> str:
        return (
            "You are GAIA's bundled Example Agent — a working demonstration of a "
            "custom agent seeded by the installer. Explain plainly that you exist "
            "to show how custom agents are packaged and loaded, and point anyone "
            "who wants to build their own to "
            "https://amd-gaia.ai/docs/guides/custom-agent."
        )

    def _create_console(self) -> AgentConsole:
        return AgentConsole()

    def _register_tools(self) -> None:
        pass

from typing import List, Optional

from gaia.agents.base.agent import Agent
from gaia.agents.base.console import AgentConsole
from gaia.agents.base.tools import _TOOL_REGISTRY
from gaia.agents.tools.rag_tools import RAGToolsMixin
from gaia.agents.tools.file_tools import FileSearchToolsMixin
from gaia.rag.sdk import RAGConfig, RAGSDK


class ResearchAgent(Agent, RAGToolsMixin, FileSearchToolsMixin):
    AGENT_ID = "research-agent"
    AGENT_NAME = "Research Agent"
    AGENT_DESCRIPTION = "Searches documents and files to answer questions"
    CONVERSATION_STARTERS = [
        "Summarize my project notes",
        "Find files related to...",
        "What does the documentation say about...?",
    ]

    def __init__(self, documents: Optional[List[str]] = None, **kwargs):
        # RAGToolsMixin reads this state off the host agent; set it here,
        # before super().__init__() runs _register_tools(). **kwargs
        # forwards everything else Agent.__init__ accepts (model_id, etc).
        self.rag = RAGSDK(RAGConfig(allowed_paths=documents))
        self.indexed_files = set()
        self.max_chunks = 5
        self.current_session = None
        self.session_manager = None
        super().__init__(**kwargs)

    def _get_system_prompt(self) -> str:
        return (
            "You are a thorough research assistant. When asked a question, "
            "first search available documents and files before answering. "
            "Always cite your sources."
        )

    def _create_console(self) -> AgentConsole:
        return AgentConsole()

    def _register_tools(self) -> None:
        _TOOL_REGISTRY.clear()
        self.register_rag_tools()
        self.register_file_search_tools()

from typing import List, Optional

from gaia.agents.base.agent import Agent
from gaia.agents.base.console import AgentConsole
from gaia.agents.base.tools import _TOOL_REGISTRY
from gaia.agents.tools.shell_tools import ShellToolsMixin
from gaia.agents.tools.file_io_tools import FileIOToolsMixin
from gaia.security import PathValidator


class CodeReviewAgent(Agent, FileIOToolsMixin, ShellToolsMixin):
    AGENT_ID = "code-review-agent"
    AGENT_NAME = "Code Review Agent"
    AGENT_DESCRIPTION = "Reviews code and runs linters"

    def __init__(self, documents: Optional[List[str]] = None, **kwargs):
        # FileIOToolsMixin reads this state off the host agent; set it
        # here, before super().__init__() runs _register_tools(). **kwargs
        # forwards everything else Agent.__init__ accepts (model_id, etc).
        self.path_validator = PathValidator(documents)
        super().__init__(**kwargs)

    def _get_system_prompt(self) -> str:
        return (
            "You are an expert code reviewer. You read files, run linters, "
            "and provide actionable feedback. Be specific about file names "
            "and line numbers. Prioritize correctness and security."
        )

    def _create_console(self) -> AgentConsole:
        return AgentConsole()

    def _register_tools(self) -> None:
        _TOOL_REGISTRY.clear()
        self.register_file_io_tools()
        self.register_shell_tools()