49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import openai
|
||
import os
|
||
from time import sleep
|
||
import sys
|
||
|
||
ml = os.getcwd()
|
||
file_path = os.path.join(ml, "api_key.txt")
|
||
|
||
# 设置openai库的API认证密钥
|
||
def api_key():
|
||
if os.path.exists(os.path.join(ml, f"api_key.txt")):
|
||
with open(file_path, 'r') as f:
|
||
openai.api_key = f.read().strip()
|
||
else:
|
||
print('文件丢失，请重新输入api_key')
|
||
sleep(1)
|
||
if os.path.exists(os.path.join(ml, "chat_command_GUI.exe")):
|
||
os.startfile("chat_command_GUI.exe")
|
||
else:
|
||
print('错误，文件不存在!?')
|
||
sleep(1)
|
||
sys.exit()
|
||
pass
|
||
|
||
|
||
# 设置GPT-3.5模型要使用的API_key
|
||
api_key()
|
||
|
||
# 设置GPT-3.5模型的引擎ID
|
||
model_engine = 'text-davinci-003'
|
||
|
||
# 循环读入用户输入并输出聊天结果
|
||
while True:
|
||
# 获取用户输入
|
||
prompt = input("你好，请问有什么需要帮助的吗？\n")
|
||
|
||
# 调用openai.ChatCompletion.create()方法来获取聊天结果
|
||
response = openai.Completion.create(
|
||
engine='text-davinci-003',
|
||
prompt=prompt,
|
||
temperature=0.7,
|
||
max_tokens=210,
|
||
top_p=1,
|
||
frequency_penalty=0,
|
||
presence_penalty=0
|
||
)
|
||
result = response['choices'][0]['text'].strip()
|
||
print(f"chatGPT：{result}")
|