polyguard-rl/app/agents/candidate_agent.py

"""Candidate generation agent."""

from __future__ import annotations

from app.common.types import PolyGuardState
from app.models.policy.candidate_builder import build_candidates


class CandidateAgent:
    name = "CandidateAgent"

    def run(self, state: PolyGuardState) -> dict:
        candidates = build_candidates(state)
        return {"candidates": [c.model_dump(mode="json") for c in candidates]}