from openai import OpenAI


def dm01():
    client = OpenAI(
        api_key="sk-0797e0871d044ca5a9baf9d30216cd1a",
        base_url="https://api.deepseek.com")

    response = client.chat.completions.create(
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": "You are a helpful assistant"},
            {"role": "user", "content": "Hello"},
        ],
        stream=False
    )

    print(response.choices[0].message.content)


# 调用第三方模型（远程），这里以DeepSeek为例。
def dm02(prompt):
    # 创建客户端
    client = OpenAI(api_key="sk-0797e0871d044ca5a9baf9d30216cd1a", base_url="https://api.deepseek.com")
    # 创建会话
    response = client.chat.completions.create(
        # 模型deepseek-chat是deepseek-v3
        # 模型deepseek-reasoner是deepseek-r1
        model="deepseek-chat",
        messages=[
            {"role": "system", "content": "你是一个ai助手"},
            {"role": "user", "content": prompt},
        ],
        stream=False
    )
    # 打印结果
    print(response.choices[0].message.content)


if __name__ == '__main__':
    # dm01()
    prompt="我要做中文新闻文本分类的任务，大概数据如下：体验2D巅峰 倚天屠龙记十大创新概览	8 60年铁树开花形状似玉米芯(组图)	5 同步A股首秀：港股缩量回调	2 最终要求对应的类别如下所示：finance,realty,stocks,education,science,society,politics,sports,game,entertainment"

    dm02(prompt)
    # 调用大模型来实现分类任务
    # dm02("体验2D巅峰 倚天屠龙记十大创新概览")