Spaces:
Sleeping
Sleeping
Commit ·
49470f9
1
Parent(s): fcaf14a
add chat history
Browse filesSigned-off-by: vax521 <13263397018@163.com>
app.py
CHANGED
|
@@ -16,13 +16,20 @@ def ask_gpt(prompt):
|
|
| 16 |
print(response)
|
| 17 |
return response.choices[0].message.content.strip()
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
|
| 20 |
async def predict(input, history):
|
| 21 |
"""
|
| 22 |
Predict the response of the chatbot and complete a running list of chat history.
|
| 23 |
"""
|
| 24 |
history.append({"role": "user", "content": input})
|
| 25 |
-
response =
|
| 26 |
history.append({"role": "assistant", "content": response})
|
| 27 |
messages = [(history[i]["content"], history[i + 1]["content"]) for i in range(0, len(history) - 1, 2)]
|
| 28 |
return messages, history
|
|
|
|
| 16 |
print(response)
|
| 17 |
return response.choices[0].message.content.strip()
|
| 18 |
|
| 19 |
+
def ask_gpt_with_history(history):
|
| 20 |
+
response = openai.ChatCompletion.create(
|
| 21 |
+
model="gpt-3.5-turbo", # here we use `gpt-3.5-turbo` model, while Stanford-Alpaca uses `text-davinci-003`
|
| 22 |
+
messages=history
|
| 23 |
+
)
|
| 24 |
+
print(response)
|
| 25 |
+
return response.choices[0].message.content.strip()
|
| 26 |
|
| 27 |
async def predict(input, history):
|
| 28 |
"""
|
| 29 |
Predict the response of the chatbot and complete a running list of chat history.
|
| 30 |
"""
|
| 31 |
history.append({"role": "user", "content": input})
|
| 32 |
+
response = ask_gpt_with_history(history)
|
| 33 |
history.append({"role": "assistant", "content": response})
|
| 34 |
messages = [(history[i]["content"], history[i + 1]["content"]) for i in range(0, len(history) - 1, 2)]
|
| 35 |
return messages, history
|