"""
{{ agent_name }} Agent
Generated by ABI-Core scaffolding
"""

from abi_core.common import prompts
from abi_core.agent.agent import AbiAgent

# Import configuration
from config import config


class {{ agent_class_name }}(AbiAgent):
    """{{ agent_description | default('ABI Agent for specialized tasks') }}"""

    def __init__(self):
        super().__init__(
            agent_name=config.AGENT_NAME,
            description=config.AGENT_DESCRIPTION,
            llm_config=config.LLM_CONFIG,
            tools=[],  # Add your tools here
            system_prompt=prompts.WORKER_PROMPT,
        )

    # Override stream() here if you need custom behaviour:
    #
    # async def stream(self, query, context_id, task_id):
    #     # your custom logic
    #     yield AgentResponse.success("custom response")
