23 lines
593 B
Python
23 lines
593 B
Python
import os # 导入环境变量
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
|
|
from openai import OpenAI
|
|
|
|
client = OpenAI(
|
|
api_key=api_key,
|
|
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) |