allinaigc commited on
Commit
688f543
·
1 Parent(s): 17a6e6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- for response in openai.ChatCompletion.create(
 
 
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.get(
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)