diff --git a/.gitignore b/.gitignore index 637f4dc0..c03c7d15 100644 --- a/.gitignore +++ b/.gitignore @@ -2,4 +2,5 @@ config.py .idea/ __pycache__/ database.db -qchatgpt.log \ No newline at end of file +qchatgpt.log +config.py \ No newline at end of file diff --git a/config-template.py b/config-template.py index aa1a65ef..a5d4b62f 100644 --- a/config-template.py +++ b/config-template.py @@ -30,9 +30,13 @@ openai_config = { }, } -# 管理员QQ号,用于接收报错等通知,为0时不发送通知 -admin_qq = 0 +# [可选] 机器人的配置 +#user_name: 管理员(主人)的名字 +#bot_name: 机器人的名字 +user_name = 'You' +bot_name = 'Bot' +# [可选] 情景预设(机器人人格) # 每个会话的预设信息,影响所有会话,无视指令重置 # 可以通过这个字段指定某些情况的回复,可直接用自然语言描述指令 # 例如: 如果我之后想获取帮助,请你说“输入!help获取帮助”, @@ -40,6 +44,9 @@ admin_qq = 0 # 可参考 https://github.com/PlexPt/awesome-chatgpt-prompts-zh default_prompt = "如果我之后想获取帮助,请你说“输入!help获取帮助”" +# 管理员QQ号,用于接收报错等通知,为0时不发送通知 +admin_qq = 0 + # 群内响应规则 # 符合此消息的群内消息即使不包含at机器人也会响应 # 支持消息前缀匹配及正则表达式匹配 diff --git a/pkg/openai/session.py b/pkg/openai/session.py index 56402fcb..7596b86c 100644 --- a/pkg/openai/session.py +++ b/pkg/openai/session.py @@ -53,7 +53,9 @@ def dump_session(session_name: str): # 从配置文件获取会话预设信息 def get_default_prompt(): - return "You:{}\nBot:好的\n".format(config.default_prompt) if hasattr(config, 'default_prompt') and \ + user_name = config.user_name if hasattr(config, 'user_name') and config.user_name != '' else 'You' + bot_name = config.bot_name if hasattr(config, 'bot_name') and config.bot_name != '' else 'Bot' + return user_name+":{}\n"+bot_name+":好的\n".format(config.default_prompt) if hasattr(config, 'default_prompt') and \ config.default_prompt != "" else '' @@ -81,8 +83,8 @@ class Session: prompt = get_default_prompt() - user_name = 'You' - bot_name = 'Bot' + user_name = config.user_name if hasattr(config, 'user_name') and config.user_name != '' else 'You' + bot_name = config.bot_name if hasattr(config, 'bot_name') and config.bot_name != '' else 'Bot' create_timestamp = 0