verify_openrouter.py

import os
from dotenv import load_dotenv
from openai import OpenAI

load_dotenv()

API_KEY = os.getenv("OPENROUTER_API_KEY")
BASE_URL = "https://openrouter.ai/api/v1"

if not API_KEY:
    print("❌ Error: OPENROUTER_API_KEY not found in environment.")
    exit(1)

client = OpenAI(
    base_url=BASE_URL,
    api_key=API_KEY,
    default_headers={
        "HTTP-Referer": "http://localhost:5173",
        "X-Title": "MultiModalDocIntel"
    }
)

models = [
    "deepseek/deepseek-r1:free",
    "deepseek/deepseek-chat:free",
    "qwen/qwen-2.5-coder-32b-instruct:free",
    "meta-llama/llama-3.2-11b-vision-instruct:free",
    "google/gemini-2.0-pro-exp-02-05:free",
    "google/gemini-exp-1206:free",
    "gryphe/mythomax-l2-13b:free"
]

print("--- Testing OpenRouter Models ---")

for model in models:
    print(f"\nTesting {model}...")
    try:
        # Try simple Text first
        completion = client.chat.completions.create(
            model=model,
            messages=[
                {"role": "user", "content": "Return simple JSON: {\"status\": \"ok\"}"}
            ],
        )
        print(f"✅ Text Success! Content: {completion.choices[0].message.content[:50]}...")
        
        # Try JSON Mode
        try:
             completion = client.chat.completions.create(
                model=model,
                messages=[
                    {"role": "system", "content": "Output JSON only."},
                    {"role": "user", "content": "Return simple JSON: {\"status\": \"ok\"}"}
                ],
                response_format={"type": "json_object"}
            )
             print(f"✅ JSON Mode Success!")
        except Exception as json_err:
             print(f"⚠️ JSON Mode Failed: {json_err}")

    except Exception as e:
        print(f"❌ Failed: {e}")