"""
OpenAI 兼容 API 引擎：适用于 DeepSeek、通义千问、Kimi、文心一言、豆包（火山方舟）。
只要平台提供 OpenAI 兼容接口，均可用此引擎，无需 Playwright。
"""


class ApiEngine:
    def __init__(self, api_base: str, api_key: str, model: str):
        self.api_base = api_base
        self.api_key = api_key
        self.model = model

    def generate(self, prompt: str) -> str:
        try:
            from openai import OpenAI
        except ImportError:
            return "ERROR:缺少依赖：pip install openai"

        try:
            client = OpenAI(base_url=self.api_base, api_key=self.api_key)
            response = client.chat.completions.create(
                model=self.model,
                messages=[{"role": "user", "content": prompt}],
            )
            content = response.choices[0].message.content
            return (content or "").strip()
        except Exception as e:
            return f"ERROR:API调用失败：{e}"