chatWEB
This commit is contained in:
@@ -1,48 +0,0 @@
|
||||
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}")
|
||||
Reference in New Issue
Block a user