import requests

from config import config
from utils.logging import logging


def aipower_chat2query_v1(
    org_id, project_id, cluster_id, database, question, caller,
):
    cluster_id = str(cluster_id)
    project_id = str(project_id)
    cluster_id = str(cluster_id)
    org_id = str(org_id)

    payload = {
        # "session_id": session_id,
        "question": question,
        "bot_type": "chat2sql",
        "cluster_id": cluster_id,
        "database": database,
        "async": False,
        "caller": caller,
        "timeout": 120000,
        "org_id": org_id,
        "source": "eda",
    }

    url = f"{config.AIPOWER_ADDR}/inner/aipower/api/v1/orgs/{org_id}/projects/{project_id}/bot"
    resp = requests.post(url, json=payload)
    logging.info("request aipower with payload %s, got resp %s, status %s", payload, resp.text, resp.status_code)
    resp_json = resp.json()
    if resp_json["code"] != 0:
        raise Exception(resp_json["message"])

    return resp.json()