agents/exfiltration_agent.py

from __future__ import annotations

from typing import Iterable, List

from shield.models.evidence import Evidence


def analyze_exfiltration(evidence: Iterable[Evidence]) -> List[Evidence]:
    if any(item.category == "data_exfiltration" for item in evidence):
        return [Evidence(
            source="exfiltration_agent",
            category="data_exfiltration",
            risk_score=82,
            weight=0.84,
            confidence=0.86,
            explanation="Request attempts to extract hidden instructions, credentials, or sensitive data",
        )]
    return []