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)
|