import os
from openai import OpenAI

# 1. 初始化客户端
# 建议通过环境变量设置API Key，避免在代码中明文显示，更安全
# 使用时，请先在你的终端或系统环境变量中设置 DEEPSEEK_API_KEY
'''
client = OpenAI(
    api_key=os.environ.get("DEEPSEEK_API_KEY"), # 从环境变量读取Key
    base_url="https://api.deepseek.com",        # DeepSeek的API地址
)
'''
# 如果你想直接在代码里测试，可以将上面两行临时替换为：
client = OpenAI(api_key="api-key", base_url="https://api.deepseek.com")

def call_deepseek(prompt):
    """
    向DeepSeek发送提示词并获取回复
    """
    try:
        # 2. 发送请求
        response = client.chat.completions.create(
            model="deepseek-v4-pro",    # deekseek模型
            messages=[
                {"role": "system", "content": "你是一个乐于助人的助手。"}, # 设定AI角色
                {"role": "user", "content": prompt},                        # 用户的问题
            ],
            stream=False,                # 非流式输出，等待完整回复
            temperature=0.7,             # 控制创造性，范围0-1，越高越有创意
        )

        # 3. 处理并返回结果
        reply = response.choices[0].message.content
        return reply

    except Exception as e:
        print(f"调用API时出错: {e}")
        return None

# 4. 主程序
if __name__ == "__main__":
    user_input = input("请输入你的问题: ")
    print("正在思考...")
    result = call_deepseek(user_input)
    if result:
        print(f"\nLLM的回答:\n{result}")