Update app.py
Browse files
app.py
CHANGED
@@ -52,6 +52,7 @@ import save_database_info
|
|
52 |
from datetime import datetime
|
53 |
import pytz
|
54 |
import openai
|
|
|
55 |
|
56 |
os.environ["OPENAI_API_KEY"] = os.environ['user_token']
|
57 |
openai.api_key = os.environ['user_token']
|
@@ -242,7 +243,9 @@ async def text_mode():
|
|
242 |
if radio_2 == '核心模式':
|
243 |
print('GPT only starts!!!')
|
244 |
print('messages:', st.session_state['messages'])
|
245 |
-
|
|
|
|
|
246 |
model=st.session_state["openai_model"],
|
247 |
messages=[
|
248 |
{"role": m["role"], "content": m["content"]}
|
@@ -253,8 +256,7 @@ async def text_mode():
|
|
253 |
# 'role': 'user', 'content': prompt}],
|
254 |
# stream=True,
|
255 |
):
|
256 |
-
full_response += response.choices[0].delta.
|
257 |
-
"content", "")
|
258 |
message_placeholder.markdown(full_response + "▌")
|
259 |
# print('session completed!')
|
260 |
message_placeholder.markdown(full_response)
|
|
|
52 |
from datetime import datetime
|
53 |
import pytz
|
54 |
import openai
|
55 |
+
from openai import OpenAI
|
56 |
|
57 |
os.environ["OPENAI_API_KEY"] = os.environ['user_token']
|
58 |
openai.api_key = os.environ['user_token']
|
|
|
243 |
if radio_2 == '核心模式':
|
244 |
print('GPT only starts!!!')
|
245 |
print('messages:', st.session_state['messages'])
|
246 |
+
openai_client = OpenAI()
|
247 |
+
|
248 |
+
for response in openai_client.ChatCompletion.create(
|
249 |
model=st.session_state["openai_model"],
|
250 |
messages=[
|
251 |
{"role": m["role"], "content": m["content"]}
|
|
|
256 |
# 'role': 'user', 'content': prompt}],
|
257 |
# stream=True,
|
258 |
):
|
259 |
+
full_response += response.choices[0].delta.content
|
|
|
260 |
message_placeholder.markdown(full_response + "▌")
|
261 |
# print('session completed!')
|
262 |
message_placeholder.markdown(full_response)
|