import os
import requests

API_URL = (
    "/static-proxy?url=https%3A%2F%2Fapi-inference.huggingface.co%2Fmodels%2Fmistralai%2FMistral-7B-Instruct-v0.2"
)





def query(question, hub_token: str):
    payload = {
        "inputs": question,
    }
    headers = {"Authorization": f"Bearer {hub_token}"}
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()[0]["generated_text"]