import requests

from watcher_visio.settings import PROMETHEUS_URL

def query_prometheus(query: str) -> str | list[str]:
    url = f"{PROMETHEUS_URL}/api/v1/query"
    params = {
        "query": query,
    }
    response = requests.get(url=url, params=params)
    response.raise_for_status()
    result = response.json()["data"]["result"]
    if len(result) > 1:
        return result
    else:
        return result[0]["value"][1]