agents/planner.py

from __future__ import annotations

from typing import Any, Dict, List


def plan_investigation(signals: Dict[str, Any]) -> Dict[str, List[str]]:
    tools: List[str] = ["heuristics", "historical_memory", "duplicate_detection", "campaign_detection"]
    if signals.get("email"):
        tools.append("email_intelligence")
    if signals.get("ip"):
        tools.append("ip_intelligence")
    if signals.get("phone"):
        tools.append("phone_intelligence")
    if signals.get("username"):
        tools.append("username_intelligence")
    if signals.get("device_fingerprint") or isinstance(signals.get("metadata"), dict):
        tools.append("device_intelligence")
        tools.append("behavior_intelligence")
    if signals.get("content"):
        tools.append("content_intelligence")
        tools.append("prompt_intelligence")
    return {"tools": tools}