Преглед на файлове

fix: session exception due to the openai limit #50

zhayujie преди 3 години
родител
ревизия
d69fd2b982
променени са 1 файла, в които са добавени 2 реда и са изтрити 1 реда
  1. 2 1
      bot/openai/open_ai_bot.py

+ 2 - 1
bot/openai/open_ai_bot.py

@@ -28,7 +28,8 @@ class OpenAIBot(Bot):
 
 
             reply_content = self.reply_text(new_query, from_user_id)
             reply_content = self.reply_text(new_query, from_user_id)
             logger.debug("[OPEN_AI] new_query={}, user={}".format(new_query, from_user_id))
             logger.debug("[OPEN_AI] new_query={}, user={}".format(new_query, from_user_id))
-            Session.save_session(query, reply_content, from_user_id)
+            if reply_content and query:
+                Session.save_session(query, reply_content, from_user_id)
             return reply_content
             return reply_content
 
 
         elif context.get('type', None) == 'IMAGE_CREATE':
         elif context.get('type', None) == 'IMAGE_CREATE':