import requests
import os

API_KEY = os.getenv("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")
MODEL = os.getenv("openrouter/free")


def ask_openrouter(prompt):

    response = requests.post(
        "https://openrouter.ai/api/v1/chat/completions",
        headers={
            "Authorization": f"Bearer {API_KEY}"
        },
        json={
            "model": MODEL,
            "messages": [
                {
                    "role": "user",
                    "content": prompt
                }
            ]
        }
    )

    return response.json()