# deepseek_client.py
import os
from openai import OpenAI
from dotenv import load_dotenv

load_dotenv()
DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
if not DEEPSEEK_API_KEY:
    raise RuntimeError("未检测到环境变量 DEEPSEEK_API_KEY，请先设置 DeepSeek 的 Key。")

deepseek_client = OpenAI(
    api_key=DEEPSEEK_API_KEY,
    base_url="https://api.deepseek.com"
)

def ask_deepseek(user_query: str,
                 system_prompt: str = "You are a helpful learning assistant.",
                 messages: list = None) -> str:
    try:
        if messages is None:
            chat_messages = [
                {"role": "system", "content": system_prompt},
                {"role": "user",   "content": user_query}
            ]
        else:
            chat_messages = [{"role": "system", "content": system_prompt}] + messages + [{"role": "user", "content": user_query}]
        response = deepseek_client.chat.completions.create(
            model="deepseek-chat",
            messages=chat_messages
        )
        return response.choices[0].message.content.strip()
    except Exception as e:
        return f"调用 DeepSeek API 时出错：{e}"