28 lines
758 B
Python
28 lines
758 B
Python
# Please install OpenAI SDK first: `pip3 install openai`
|
|
import os
|
|
from openai import OpenAI
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
client = OpenAI(
|
|
api_key=os.environ.get('DEEPSEEK_API_KEY'),
|
|
base_url="https://api.deepseek.com")
|
|
|
|
while True:
|
|
user_input=input("请输入你的问题：")
|
|
if user_input=="q":
|
|
break
|
|
response = client.chat.completions.create(
|
|
model="deepseek-v4-pro",
|
|
messages=[
|
|
{"role": "system", "content": "You are a helpful assistant"},
|
|
{"role": "user", "content": user_input},
|
|
],
|
|
stream=False,
|
|
reasoning_effort="high",
|
|
extra_body={"thinking": {"type": "enabled"}}
|
|
)
|
|
print(response.choices[0].message.content)
|
|
|
|
|
|
# print(response.model_dump_json(indent=2))
|