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