mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-07-17 18:06:06 +00:00
adjust:整理代码,仅添加json方式的prompt读取
This commit is contained in:
+4
-53
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user