agents/prompt_agent.py

from __future__ import annotations

from typing import Iterable, List

from shield.models.evidence import Evidence


def analyze_prompt_threats(evidence: Iterable[Evidence]) -> List[Evidence]:
    if any(item.category == "prompt_injection" for item in evidence):
        return [Evidence(
            source="prompt_agent",
            category="prompt_injection",
            risk_score=72,
            weight=0.72,
            confidence=0.86,
            explanation="Prompt-security evidence indicates instruction override or jailbreak behavior",
        )]
    return []