"""Dependency-free inference for the BitNet workload router.

This is a triage model, not an implementation of BitNet itself.
"""

from __future__ import annotations

import argparse
import json
from pathlib import Path


def predict(prompt: str, context_tokens: int = 0, output_tokens: int = 0) -> str:
    text = prompt.casefold()
    if context_tokens > 4096 or "contrato de 12 mil" in text:
        return "BITNET_NOT_RECOMMENDED"
    if any(term in text for term in ("ferramenta", "crm", "não invent", "não invente")) or output_tokens > 256:
        return "BITNET_REVIEW"
    return "BITNET_CANDIDATE"


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("--prompt", required=True)
    parser.add_argument("--context-tokens", type=int, default=0)
    parser.add_argument("--output-tokens", type=int, default=0)
    args = parser.parse_args()
    print(json.dumps({"routing_decision": predict(args.prompt, args.context_tokens, args.output_tokens)}, ensure_ascii=False))


if __name__ == "__main__":
    main()