diff --git a/config-template.py b/config-template.py index 5985dc36..3820459a 100644 --- a/config-template.py +++ b/config-template.py @@ -23,6 +23,9 @@ openai_config = { # 管理员QQ号,用于接收报错等通知,为0时不发送通知 admin_qq = 0 +# 回复消息时是否显示[GPT]前缀 +show_prefix = False + # 敏感词过滤开关,以同样数量的*代替敏感词回复 # 请在sensitive.json中添加敏感词 sensitive_word_filter = True diff --git a/pkg/qqbot/manager.py b/pkg/qqbot/manager.py index 2e20e316..830cf0ef 100644 --- a/pkg/qqbot/manager.py +++ b/pkg/qqbot/manager.py @@ -175,7 +175,8 @@ class QQBotManager: session = pkg.openai.session.get_session(session_name) try: - reply = "[GPT]" + session.append(text_message) + prefix = "[GPT]" if hasattr(config, "show_prefix") and config.show_prefix else "" + reply = prefix + session.append(text_message) except openai.error.APIConnectionError as e: self.notify_admin("{}会话调用API失败:{}".format(session_name, e)) reply = "[bot]err:调用API失败,请联系作者,或等待修复"