adjust:整理代码,仅添加json方式的prompt读取

This commit is contained in:
chordfish
2023-03-10 10:13:40 +08:00
parent 7477c7c67f
commit ac65d81ba1
4 changed files with 39 additions and 74 deletions
+4 -53
View File
@@ -128,52 +128,15 @@ class Session:
logging.debug('{},lock release successfully,{}'.format(self.name, self.response_lock))
# 从配置文件获取会话预设信息
def get_default_prompt(self, use_default: str = None, get_only = False):
def get_default_prompt(self, use_default: str = None):
config = pkg.utils.context.get_config()
import pkg.openai.dprompt as dprompt
if use_default is None:
use_default = dprompt.get_current()
current_default_prompt = \
[
{
'role': 'user',
'content': '如果我之后想获取帮助,请你说“输入!help获取帮助”'
}, {
'role': 'assistant',
'content': 'ok'
}
]
# 根据设置进行prompt预设模式
if config.preset_mode == "full_scenario":
import os
dir = os.path.join(os.getcwd(), config.full_prompt_dir)
json_file = os.path.join(dir, use_default) + '.json'
logging.debug("try to load json: {}".format(json_file))
try:
with open(json_file, 'r', encoding ='utf-8') as f:
json_content = json.load(f)
current_default_prompt = json_content['prompt']
if not get_only:
# 读取机器人名字,用于响应信息
self.bot_name = json_content['name']
# 过滤掉不符合人格的回复
self.bot_filter = json_content['filter']
logging.debug("first bot filter: {}".format(self.bot_filter))
except FileNotFoundError:
logging.info("couldn't find file {}".format(json_file))
else:
current_default_prompt = dprompt.get_prompt(use_default)
current_default_prompt = dprompt.get_prompt(use_default)
return current_default_prompt
@@ -228,7 +191,7 @@ class Session:
self.last_interact_timestamp = int(time.time())
# 触发插件事件
if self.prompt == self.get_default_prompt(get_only = True):
if self.prompt == self.get_default_prompt():
args = {
'session_name': self.name,
'session': self,
@@ -258,18 +221,6 @@ class Session:
del (res_ans_spt[0])
res_ans = '\n\n'.join(res_ans_spt)
# 检测是否包含ai人格否定
if config.filter_ai_warning and self.bot_filter:
import re
match = re.search(self.bot_filter['reg'], res_ans)
logging.debug(self.bot_filter)
logging.debug(res_ans)
if match:
logging.debug('回复:{} 检测到人格否定,替换中。。'.format(res_ans))
res_ans = self.bot_filter['replace']
logging.debug('替换为: {}'.format(res_ans))
# 将此次对话的双方内容加入到prompt中
self.prompt.append({'role': 'user', 'content': text})
@@ -323,7 +274,7 @@ class Session:
# 持久化session
def persistence(self):
if self.prompt == self.get_default_prompt(get_only = True):
if self.prompt == self.get_default_prompt():
return
db_inst = pkg.utils.context.get_database_manager()