import openai  # or use requests if OpenRouter API
import os

openai.api_key = os.getenv("OPENROUTER_API_KEY")

def ask_model(context, query, model="mistral"):
    prompt = f"""Context: {context}\n\nUser: {query}\nAI:"""
    response = openai.ChatCompletion.create(
        model=model,
        messages=[{"role": "user", "content": prompt}]
    )
    return response['choices'][0]['message']['content']