agents/content_agent.py

from __future__ import annotations

from typing import Iterable, List

from shield.models.evidence import Evidence


def analyze_content_threats(evidence: Iterable[Evidence]) -> List[Evidence]:
    categories = {item.category for item in evidence}
    if {"spam", "scam"} & categories:
        return [Evidence(
            source="content_agent",
            category="content_abuse",
            risk_score=60 if "scam" in categories else 45,
            weight=0.58,
            confidence=0.74,
            explanation="Content evidence matches known spam or scam abuse patterns",
        )]
    return []