mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
283 lines
9.5 KiB
Python
283 lines
9.5 KiB
Python
# 会话管理相关指令
|
|
import datetime
|
|
import json
|
|
|
|
from pkg.qqbot.cmds.model import command
|
|
import pkg.openai.session
|
|
import pkg.utils.context
|
|
import config
|
|
|
|
@command(
|
|
"reset",
|
|
"重置当前会话",
|
|
"!reset\n!reset [使用情景预设名称]",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_reset(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""重置会话"""
|
|
reply = []
|
|
|
|
if len(params) == 0:
|
|
pkg.openai.session.get_session(session_name).reset(explicit=True)
|
|
reply = ["[bot]会话已重置"]
|
|
else:
|
|
pkg.openai.session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
|
|
reply = ["[bot]会话已重置，使用场景预设:{}".format(params[0])]
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"last",
|
|
"切换到前一次会话",
|
|
"!last",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_last(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""切换到前一次会话"""
|
|
reply = []
|
|
result = pkg.openai.session.get_session(session_name).last_session()
|
|
if result is None:
|
|
reply = ["[bot]没有前一次的对话"]
|
|
else:
|
|
datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime(
|
|
'%Y-%m-%d %H:%M:%S')
|
|
reply = ["[bot]已切换到前一次的对话：\n创建时间:{}\n".format(datetime_str)]
|
|
|
|
return reply
|
|
|
|
@command(
|
|
"next",
|
|
"切换到后一次会话",
|
|
"!next",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_next(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: int, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""切换到后一次会话"""
|
|
reply = []
|
|
|
|
result = pkg.openai.session.get_session(session_name).next_session()
|
|
if result is None:
|
|
reply = ["[bot]没有后一次的对话"]
|
|
else:
|
|
datetime_str = datetime.datetime.fromtimestamp(result.create_timestamp).strftime(
|
|
'%Y-%m-%d %H:%M:%S')
|
|
reply = ["[bot]已切换到后一次的对话：\n创建时间:{}\n".format(datetime_str)]
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"prompt",
|
|
"获取当前会话的前文",
|
|
"!prompt",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_prompt(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""获取当前会话的前文"""
|
|
reply = []
|
|
|
|
msgs = ""
|
|
session:list = pkg.openai.session.get_session(session_name).prompt
|
|
for msg in session:
|
|
if len(params) != 0 and params[0] in ['-all', '-a']:
|
|
msgs = msgs + "{}: {}\n\n".format(msg['role'], msg['content'])
|
|
elif len(msg['content']) > 30:
|
|
msgs = msgs + "[{}]: {}...\n\n".format(msg['role'], msg['content'][:30])
|
|
else:
|
|
msgs = msgs + "[{}]: {}\n\n".format(msg['role'], msg['content'])
|
|
reply = ["[bot]当前对话所有内容：\n{}".format(msgs)]
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"list",
|
|
"列出当前会话的所有历史记录",
|
|
"!list\n!list [页数]",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_list(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""列出当前会话的所有历史记录"""
|
|
reply = []
|
|
|
|
pkg.openai.session.get_session(session_name).persistence()
|
|
page = 0
|
|
|
|
if len(params) > 0:
|
|
try:
|
|
page = int(params[0])
|
|
except ValueError:
|
|
pass
|
|
|
|
results = pkg.openai.session.get_session(session_name).list_history(page=page)
|
|
if len(results) == 0:
|
|
reply = ["[bot]第{}页没有历史会话".format(page)]
|
|
else:
|
|
reply_str = "[bot]历史会话 第{}页：\n".format(page)
|
|
current = -1
|
|
for i in range(len(results)):
|
|
# 时间(使用create_timestamp转换) 序号 部分内容
|
|
datetime_obj = datetime.datetime.fromtimestamp(results[i]['create_timestamp'])
|
|
msg = ""
|
|
try:
|
|
msg = json.loads(results[i]['prompt'])
|
|
except json.decoder.JSONDecodeError:
|
|
msg = pkg.openai.session.reset_session_prompt(session_name, results[i]['prompt'])
|
|
# 持久化
|
|
pkg.openai.session.get_session(session_name).persistence()
|
|
if len(msg) >= 2:
|
|
reply_str += "#{} 创建:{} {}\n".format(i + page * 10,
|
|
datetime_obj.strftime("%Y-%m-%d %H:%M:%S"),
|
|
msg[0]['content'])
|
|
else:
|
|
reply_str += "#{} 创建:{} {}\n".format(i + page * 10,
|
|
datetime_obj.strftime("%Y-%m-%d %H:%M:%S"),
|
|
"无内容")
|
|
if results[i]['create_timestamp'] == pkg.openai.session.get_session(
|
|
session_name).create_timestamp:
|
|
current = i + page * 10
|
|
|
|
reply_str += "\n以上信息倒序排列"
|
|
if current != -1:
|
|
reply_str += ",当前会话是 #{}\n".format(current)
|
|
else:
|
|
reply_str += ",当前处于全新会话或不在此页"
|
|
|
|
reply = [reply_str]
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"resend",
|
|
"重新获取上一次问题的回复",
|
|
"!resend",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_resend(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""重新获取上一次问题的回复"""
|
|
reply = []
|
|
|
|
session = pkg.openai.session.get_session(session_name)
|
|
to_send = session.undo()
|
|
|
|
mgr = pkg.utils.context.get_qqbot_manager()
|
|
|
|
reply = pkg.qqbot.message.process_normal_message(to_send, mgr, config,
|
|
launcher_type, launcher_id, sender_id)
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"del",
|
|
"删除当前会话的历史记录",
|
|
"!del <序号>\n!del all",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_del(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""删除当前会话的历史记录"""
|
|
reply = []
|
|
|
|
if len(params) == 0:
|
|
reply = ["[bot]参数不足, 格式: !del <序号>\n可以通过!list查看序号"]
|
|
else:
|
|
if params[0] == 'all':
|
|
pkg.openai.session.get_session(session_name).delete_all_history()
|
|
reply = ["[bot]已删除所有历史会话"]
|
|
elif params[0].isdigit():
|
|
if pkg.openai.session.get_session(session_name).delete_history(int(params[0])):
|
|
reply = ["[bot]已删除历史会话 #{}".format(params[0])]
|
|
else:
|
|
reply = ["[bot]没有历史会话 #{}".format(params[0])]
|
|
else:
|
|
reply = ["[bot]参数错误, 格式: !del <序号>\n可以通过!list查看序号"]
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"default",
|
|
"操作情景预设",
|
|
"!default\n!default [指定情景预设为默认]",
|
|
[],
|
|
False
|
|
)
|
|
def cmd_default(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""操作情景预设"""
|
|
reply = []
|
|
|
|
if len(params) == 0:
|
|
# 输出目前所有情景预设
|
|
import pkg.openai.dprompt as dprompt
|
|
reply_str = "[bot]当前所有情景预设:\n\n"
|
|
for key,value in dprompt.get_prompt_dict().items():
|
|
reply_str += " - {}: {}\n".format(key,value)
|
|
|
|
reply_str += "\n当前默认情景预设:{}\n".format(dprompt.get_current())
|
|
reply_str += "请使用!default <情景预设>来设置默认情景预设"
|
|
reply = [reply_str]
|
|
elif len(params) >0 and is_admin:
|
|
# 设置默认情景
|
|
import pkg.openai.dprompt as dprompt
|
|
try:
|
|
dprompt.set_current(params[0])
|
|
reply = ["[bot]已设置默认情景预设为:{}".format(dprompt.get_current())]
|
|
except KeyError:
|
|
reply = ["[bot]err: 未找到情景预设:{}".format(params[0])]
|
|
else:
|
|
reply = ["[bot]err: 仅管理员可设置默认情景预设"]
|
|
|
|
return reply
|
|
|
|
|
|
@command(
|
|
"delhst",
|
|
"删除指定会话的所有历史记录",
|
|
"!delhst <会话名称>\n!delhst all",
|
|
[],
|
|
True
|
|
)
|
|
def cmd_delhst(cmd: str, params: list, session_name: str,
|
|
text_message: str, launcher_type: str, launcher_id: int,
|
|
sender_id: int, is_admin: bool) -> list:
|
|
"""删除指定会话的所有历史记录"""
|
|
reply = []
|
|
|
|
if len(params) == 0:
|
|
reply = ["[bot]err:请输入要删除的会话名: group_<群号> 或者 person_<QQ号>, 或使用 !delhst all 删除所有会话的历史记录"]
|
|
else:
|
|
if params[0] == "all":
|
|
pkg.utils.context.get_database_manager().delete_all_session_history()
|
|
reply = ["[bot]已删除所有会话的历史记录"]
|
|
else:
|
|
if pkg.utils.context.get_database_manager().delete_all_history(params[0]):
|
|
reply = ["[bot]已删除会话 {} 的所有历史记录".format(params[0])]
|
|
else:
|
|
reply = ["[bot]未找到会话 {} 的历史记录".format(params[0])]
|
|
|
|
return reply
|