46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
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巅峰 倚天屠龙记十大创新概览")
|