agents/__init__.py

"""
agents/
-------
LLM-calling agents for AutoDevAgent.

Each agent takes a PipelineState in and returns a partial state dict.
No execution logic lives here — agents are pure LLM callers.

Available agents:
    DetectAgent        — fast language classification (llama3-8b)
    PlanningAgent      — step-by-step plan + SQL schema inference
    CodeGeneratorAgent — code generation from plan
    DebugAgent         — error classification, self-reflection, rewrite
    TestGeneratorAgent — unit tests (Python) / assertions (SQL)
    PolisherAgent      — safe cosmetic refactors + diff
    ExplanationAgent   — plain-English code explanation
    FlowchartAgent     — Mermaid.js diagram generation (conditional)
"""

from agents.detect_agent       import DetectAgent
from agents.planning_agent     import PlanningAgent
from agents.code_generator     import CodeGeneratorAgent
from agents.debug_agent        import DebugAgent
from agents.test_generator     import TestGeneratorAgent
from agents.polisher_agent     import PolisherAgent
from agents.explanation_agent  import ExplanationAgent
from agents.flowchart_agent    import FlowchartAgent, should_generate_flowchart

__all__ = [
    "DetectAgent",
    "PlanningAgent",
    "CodeGeneratorAgent",
    "DebugAgent",
    "TestGeneratorAgent",
    "PolisherAgent",
    "ExplanationAgent",
    "FlowchartAgent",
    "should_generate_flowchart",
]