Maojui

ChatGPT API 腳本 | Python

2023-03-21

懶人腳本

自行申請 api token 之後便可使用 :

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import openai
from secret import apikey

print("使用情形: https://platform.openai.com/account/usage")

message_history = []
openai.api_key = apikey
def talk_with(persona):
while True:
user_input = input('> ')
if user_input == "":
return message_history

message_history.append({"role": "user", "content": user_input})
query = [{"role": "system", "content": persona}]
query.extend(message_history)
result = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=query
)
gpt_message = result["choices"][0]["message"]
message_history.append({"role": gpt_message["role"], "content": gpt_message["content"]})
price = (result['usage']['total_tokens'] / 1000) * 0.002 * 31
print(f"諮詢費: {price}元", )
print("GPT: " + gpt_message["content"])

try:
talk_with(persona="""You are a helpful engineer expert. You answer question by providing a short explanation and a list of easy to follow steps.""")
except:
print(message_history)