"""Find a working Groq vision model."""
import os, httpx
from dotenv import load_dotenv
load_dotenv()

api_key = os.getenv("GROQ_API_KEY")
h = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}

# Try known Groq vision model names
candidates = [
    "llama-3.2-11b-vision-preview",
    "llama-3.2-90b-vision-preview",
    "llama-3.2-11b-vision",
    "llama-3.2-90b-vision",
    "meta-llama/llama-3.2-11b-vision",
    "meta-llama/llama-3.2-90b-vision",
    "llava-v1.5-7b-4096-preview",
    "llama-4-scout-17b-16e-instruct",
]

for model in candidates:
    body = {
        "model": model,
        "messages": [{"role": "user", "content": "Say OK"}],
        "max_tokens": 10,
    }
    try:
        r = httpx.post("https://api.groq.com/openai/v1/chat/completions", headers=h, json=body, timeout=10)
        if r.status_code == 200:
            print(f"OK {model} - WORKS!")
        elif "model_not_found" in r.text:
            print(f"NO {model} - not found")
        elif "does not exist" in r.text or "decommissioned" in r.text:
            print(f"NO {model} - {r.json().get('error',{}).get('message','')[:80]}")
        else:
            print(f"?? {model} - {r.status_code}: {r.text[:100]}")
    except Exception as e:
        print(f"XX {model} - error: {e}")