mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-07 06:16:02 +00:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from ..mgr import AbstractCommand, Context
|
|
|
|
import pkg.openai.session
|
|
import pkg.utils.context
|
|
|
|
class ResetCommand(AbstractCommand):
|
|
parent: type = None
|
|
|
|
name = 'reset'
|
|
description = '重置当前会话'
|
|
usage = 'reset'
|
|
aliases = ['重置', '重置会话']
|
|
privilege = 0
|
|
|
|
@classmethod
|
|
def process(cls, ctx: Context) -> tuple[bool, list, str]:
|
|
params = ctx.params
|
|
session_name = ctx.session_name
|
|
|
|
reply = ""
|
|
|
|
if len(params) == 0:
|
|
pkg.openai.session.get_session(session_name).reset(explicit=True)
|
|
reply = ["[bot]会话已重置"]
|
|
else:
|
|
try:
|
|
import pkg.openai.dprompt as dprompt
|
|
pkg.openai.session.get_session(session_name).reset(explicit=True, use_prompt=params[0])
|
|
reply = ["[bot]会话已重置，使用场景预设:{}".format(dprompt.mode_inst().get_full_name(params[0]))]
|
|
except Exception as e:
|
|
reply = ["[bot]会话重置失败：{}".format(e)]
|
|
|
|
return True, reply, '' |