first commit
This commit is contained in:
46
05-LLM/01_api_deepseek_v3.py
Normal file
46
05-LLM/01_api_deepseek_v3.py
Normal file
@@ -0,0 +1,46 @@
|
||||
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巅峰 倚天屠龙记十大创新概览")
|
||||
Reference in New Issue
Block a user