lewisliuX123 commited on
Commit
d0692d0
·
1 Parent(s): 599a40c

Update bot/chatgpt/chat_gpt_bot.py

Browse files
Files changed (1) hide show
  1. bot/chatgpt/chat_gpt_bot.py +4 -5
bot/chatgpt/chat_gpt_bot.py CHANGED
@@ -26,9 +26,9 @@ class ChatGPTBot(Bot):
26
  new_query = Session.build_session_query(query, from_user_id)
27
  logger.debug("[OPEN_AI] session query={}".format(new_query))
28
 
29
- if context.get('stream'):
30
- # reply in stream
31
- return self.reply_text_stream(query, new_query, from_user_id)
32
 
33
  reply_content = self.reply_text(new_query, from_user_id, 0)
34
  logger.debug("[OPEN_AI] new_query={}, user={}, reply_cont={}".format(new_query, from_user_id, reply_content))
@@ -44,7 +44,6 @@ class ChatGPTBot(Bot):
44
  response = openai.ChatCompletion.create(
45
  model="gpt-3.5-turbo", # 对话模型的名称
46
  messages=query,
47
- stream=True,
48
  temperature=1, # 值在[0,1]之间,越大表示回复越具有不确定性
49
  max_tokens=600, # 回复最大的字符数
50
  top_p=1,
@@ -52,7 +51,7 @@ class ChatGPTBot(Bot):
52
  presence_penalty=0, # [-2,2]之间,该值越大则更倾向于产生不同的内容
53
  )
54
  # res_content = response.choices[0]['text'].strip().replace('<|endoftext|>', '')
55
- # logger.info(response.choices[0]['message']['content'])
56
  # log.info("[OPEN_AI] reply={}".format(res_content))
57
  return response.choices[0]['message']['content']
58
  except openai.error.RateLimitError as e:
 
26
  new_query = Session.build_session_query(query, from_user_id)
27
  logger.debug("[OPEN_AI] session query={}".format(new_query))
28
 
29
+ # if context.get('stream'):
30
+ # # reply in stream
31
+ # return self.reply_text_stream(query, new_query, from_user_id)
32
 
33
  reply_content = self.reply_text(new_query, from_user_id, 0)
34
  logger.debug("[OPEN_AI] new_query={}, user={}, reply_cont={}".format(new_query, from_user_id, reply_content))
 
44
  response = openai.ChatCompletion.create(
45
  model="gpt-3.5-turbo", # 对话模型的名称
46
  messages=query,
 
47
  temperature=1, # 值在[0,1]之间,越大表示回复越具有不确定性
48
  max_tokens=600, # 回复最大的字符数
49
  top_p=1,
 
51
  presence_penalty=0, # [-2,2]之间,该值越大则更倾向于产生不同的内容
52
  )
53
  # res_content = response.choices[0]['text'].strip().replace('<|endoftext|>', '')
54
+ logger.info(response.choices[0]['message']['content'])
55
  # log.info("[OPEN_AI] reply={}".format(res_content))
56
  return response.choices[0]['message']['content']
57
  except openai.error.RateLimitError as e: