"""Smoke-test just function calling on Gemma 4 (after rate-limit clears)."""
import asyncio
import json
import os
import sys

import httpx

API_KEY = os.environ["OPENROUTER_API_KEY"]
BASE = "https://openrouter.ai/api/v1/chat/completions"
HEADERS = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json",
    "HTTP-Referer": "https://flutiq.pages.dev",
    "X-Title": "FlutIQ smoke test",
}

TOOLS = [{
    "type": "function",
    "function": {
        "name": "lookup_fema_flood_zone",
        "description": "Look up FEMA flood zone for coordinates.",
        "parameters": {
            "type": "object",
            "properties": {
                "latitude": {"type": "number"},
                "longitude": {"type": "number"},
            },
            "required": ["latitude", "longitude"],
        },
    },
}]


async def try_model(model: str) -> bool:
    print(f"\n--- {model} ---")
    payload = {
        "model": model,
        "messages": [{
            "role": "user",
            "content": "What is the FEMA flood zone for 41.8087, -87.6062? Use the tool.",
        }],
        "tools": TOOLS,
        "tool_choice": "auto",
        "max_tokens": 512,
        "temperature": 0,
    }
    async with httpx.AsyncClient(timeout=120) as client:
        resp = await client.post(BASE, headers=HEADERS, json=payload)
    print(f"HTTP {resp.status_code}")
    if resp.status_code != 200:
        print(resp.text[:500])
        return False
    msg = resp.json()["choices"][0]["message"]
    text = msg.get("content", "") or ""
    tool_calls = msg.get("tool_calls", []) or []
    print(f"content: {text[:200]!r}")
    print(f"tool_calls ({len(tool_calls)}):")
    for tc in tool_calls:
        print(f"  {json.dumps(tc)[:300]}")
    return bool(tool_calls)


async def main() -> int:
    for model in ("google/gemma-4-31b-it:free", "google/gemma-4-26b-a4b-it:free"):
        ok = await try_model(model)
        if ok:
            print("\nPASS")
            return 0
    print("\nFAIL on both models")
    return 1


if __name__ == "__main__":
    sys.exit(asyncio.run(main()))