mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 11:29:39 +08:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
import tips as tips_custom
|
|
|
|
from .. import aamgr
|
|
from ....openai import session
|
|
from ....utils import context
|
|
|
|
|
|
@aamgr.AbstractCommandNode.register(
|
|
parent=None,
|
|
name='reset',
|
|
description='重置当前会话',
|
|
usage='!reset',
|
|
aliases=[],
|
|
privilege=1
|
|
)
|
|
class ResetCommand(aamgr.AbstractCommandNode):
|
|
@classmethod
|
|
def process(cls, ctx: aamgr.Context) -> tuple[bool, list]:
|
|
params = ctx.params
|
|
session_name = ctx.session_name
|
|
|
|
reply = ""
|
|
|
|
if len(params) == 0:
|
|
session.get_session(session_name).reset(explicit=True)
|
|
reply = [tips_custom.command_reset_message]
|
|
else:
|
|
try:
|
|
import pkg.openai.dprompt as dprompt
|
|
session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
|
|
reply = [tips_custom.command_reset_name_message+"{}".format(dprompt.mode_inst().get_full_name(params[0]))]
|
|
except Exception as e:
|
|
reply = ["[bot]会话重置失败：{}".format(e)]
|
|
|
|
return True, reply |