diff --git a/bot/openai/open_ai_bot.py b/bot/openai/open_ai_bot.py index 9eeb3c913..c948a7c70 100644 --- a/bot/openai/open_ai_bot.py +++ b/bot/openai/open_ai_bot.py @@ -18,7 +18,7 @@ def reply(self, query, context=None): # acquire reply content if not context or not context.get('type') or context.get('type') == 'TEXT': logger.info("[OPEN_AI] query={}".format(query)) - from_user_id = context['from_user_id'] + from_user_id = context.get('from_user_id') or context.get('session_id') if query == '#清除记忆': Session.clear_session(from_user_id) return '记忆已清除' @@ -163,4 +163,4 @@ def clear_session(user_id): @staticmethod def clear_all_session(): - user_session.clear() \ No newline at end of file + user_session.clear()