Function _ask_ai_classify

device_classifier.py:803–839

Ask GPT-5 Nano to classify a single device. Returns a device_type string or None.

(ai_service, vendor, ports, hostname, mac)

def _ask_ai_classify(ai_service, vendor, ports, hostname, mac):
    """Ask GPT-5.4 Nano to classify a single device. Returns a device_type string or None."""
    if not ai_service or not ai_service.ensure_ready():
        return None

    valid_types = ", ".join(sorted(VALID_DEVICE_TYPES - {"ragnar", "unknown"}))

    system = (
        "You are a network device classifier. Given device metadata, reply with ONLY "
        "the device_type string — one of: " + valid_types + ". "
        "Reply with just the single word. If uncertain, reply unknown."
    )

    user = (
        f"MAC vendor: {vendor or 'unknown'}\n"
        f"Hostname: {hostname or 'unknown'}\n"
        f"MAC: {mac or 'unknown'}\n"
        f"Open ports: {', '.join(str(p) for p in (ports or [])[:20]) or 'none'}\n"
        f"Classify this device."
    )

    # Use GPT-5.4 Nano for fast/cheap classification
    original_model = ai_service.model
    try:
        ai_service.model = "gpt-5.4-nano"
        answer = ai_service._ask(system, user)
    finally:
        ai_service.model = original_model

    if not answer:
        return None

    # Sanitise — model should return just the type string
    answer = answer.strip().lower().replace(" ", "_")
    # Remove any surrounding quotes or punctuation
    answer = answer.strip('"\'.,')
    return answer if answer in VALID_DEVICE_TYPES else None