import requests
from requests.exceptions import RequestException
from config.logger import setup_logging
from core.providers.llm.base import LLMProviderBase

TAG = __name__
logger = setup_logging()


class LLMProvider(LLMProviderBase):
    def __init__(self, config):
        self.agent_id = config.get("agent_id")  # 对应 agent_id
        self.api_key = config.get("api_key")
        self.base_url = config.get("base_url", config.get("url"))  # 默认使用 base_url
        self.api_url = f"{self.base_url}/api/conversation/process"  # 拼接完整的 API URL

    def response(self, session_id, dialogue, **kwargs):
        # home assistant语音助手自带意图，无需使用xiaozhi ai自带的，只需要把用户说的话传递给home assistant即可

        # 提取最后一个 role 为 'user' 的 content
        input_text = None
        if isinstance(dialogue, list):  # 确保 dialogue 是一个列表
            # 逆序遍历，找到最后一个 role 为 'user' 的消息
            for message in reversed(dialogue):
                if message.get("role") == "user":  # 找到 role 为 'user' 的消息
                    input_text = message.get("content", "")
                    break  # 找到后立即退出循环

        # 构造请求数据
        payload = {
            "text": input_text,
            "agent_id": self.agent_id,
            "conversation_id": session_id,  # 使用 session_id 作为 conversation_id
        }
        # 设置请求头
        headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json",
        }

        # 发起 POST 请求
        response = requests.post(self.api_url, json=payload, headers=headers)

        # 检查请求是否成功
        response.raise_for_status()

        # 解析返回数据
        data = response.json()
        speech = (
            data.get("response", {})
            .get("speech", {})
            .get("plain", {})
            .get("speech", "")
        )

        # 返回生成的内容
        if speech:
            yield speech
        else:
            logger.bind(tag=TAG).warning("API 返回数据中没有 speech 内容")

    def response_with_functions(self, session_id, dialogue, functions=None):
        logger.bind(tag=TAG).error(
            f"homeassistant不支持（function call），建议使用其他意图识别"
        )
