app/agents/evidence_agent.py

"""Evidence retrieval agent."""

from __future__ import annotations

from app.common.types import PolyGuardState
from app.knowledge.evidence_retriever import retrieve_evidence


class EvidenceAgent:
    name = "EvidenceAgent"

    def run(self, state: PolyGuardState) -> dict:
        query = " ".join(state.patient.comorbidities + [m.drug for m in state.patient.medications[:2]])
        return {"evidence": retrieve_evidence(query=query, top_k=3)}