47 lines
1.3 KiB
Python
47 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():
|
||
input1 = input("请输入API_key:")
|
||
if input1 == None:
|
||
print('error: Not input')
|
||
sleep(4)
|
||
sys.exit()
|
||
else:
|
||
openai.api_key = input1
|
||
|
||
# 设置GPT-3.5模型的引擎ID
|
||
model_engine = 'text-davinci-003'
|
||
|
||
# 循环读入用户输入并输出聊天结果
|
||
def input_print():
|
||
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}")
|
||
|
||
if __name__ == "__main__":
|
||
print('注意，本程序是测试程序，如有问题，非常正常')
|
||
sleep(2)
|
||
api_key()
|
||
# 设置GPT-3.5模型的引擎ID
|
||
model_engine = 'text-davinci-003'
|
||
input_print() |