小棉智能

简体中文

English
日本語

简体中文

English
日本語

主题

模块 nonebot_plugin_marshoai.marsho ​

async func at_enable() ​

源代码 或 在GitHub上查看
python
async def at_enable():
    return config.marshoai_at

@add_usermsg_cmd.handle()

async func add_usermsg(target: MsgTarget, arg: Message = CommandArg()) ​

源代码 或 在GitHub上查看
python
@add_usermsg_cmd.handle()
async def add_usermsg(target: MsgTarget, arg: Message=CommandArg()):
    if (msg := arg.extract_plain_text()):
        context.append(UserMessage(content=msg).as_dict(), target.id, target.private)
        await add_usermsg_cmd.finish('已添加用户消息')

@add_assistantmsg_cmd.handle()

async func add_assistantmsg(target: MsgTarget, arg: Message = CommandArg()) ​

源代码 或 在GitHub上查看
python
@add_assistantmsg_cmd.handle()
async def add_assistantmsg(target: MsgTarget, arg: Message=CommandArg()):
    if (msg := arg.extract_plain_text()):
        context.append(AssistantMessage(content=msg).as_dict(), target.id, target.private)
        await add_assistantmsg_cmd.finish('已添加助手消息')

@praises_cmd.handle()

async func praises() ​

源代码 或 在GitHub上查看
python
@praises_cmd.handle()
async def praises():
    await praises_cmd.finish(build_praises())

@contexts_cmd.handle()

async func contexts(target: MsgTarget) ​

源代码 或 在GitHub上查看
python
@contexts_cmd.handle()
async def contexts(target: MsgTarget):
    backup_context = await get_backup_context(target.id, target.private)
    if backup_context:
        context.set_context(backup_context, target.id, target.private)
    await contexts_cmd.finish(str(context.build(target.id, target.private)))

@save_context_cmd.handle()

async func save_context(target: MsgTarget, arg: Message = CommandArg()) ​

源代码 或 在GitHub上查看
python
@save_context_cmd.handle()
async def save_context(target: MsgTarget, arg: Message=CommandArg()):
    contexts_data = context.build(target.id, target.private)
    if not context:
        await save_context_cmd.finish('暂无上下文可以保存')
    if (msg := arg.extract_plain_text()):
        await save_context_to_json(msg, contexts_data, 'contexts')
        await save_context_cmd.finish('已保存上下文')

@load_context_cmd.handle()

async func load_context(target: MsgTarget, arg: Message = CommandArg()) ​

源代码 或 在GitHub上查看
python
@load_context_cmd.handle()
async def load_context(target: MsgTarget, arg: Message=CommandArg()):
    if (msg := arg.extract_plain_text()):
        await get_backup_context(target.id, target.private)
        context.set_context(await load_context_from_json(msg, 'contexts'), target.id, target.private)
        await load_context_cmd.finish('已加载并覆盖上下文')

@resetmem_cmd.handle()

async func resetmem(target: MsgTarget) ​

源代码 或 在GitHub上查看
python
@resetmem_cmd.handle()
async def resetmem(target: MsgTarget):
    if [target.id, target.private] not in target_list:
        target_list.append([target.id, target.private])
    backup_context = await get_backup_context(target.id, target.private)
    if backup_context:
        context.set_context(backup_context, target.id, target.private)
    context.reset(target.id, target.private)
    await resetmem_cmd.finish('上下文已重置')

@changemodel_cmd.handle()

async func changemodel(arg: Message = CommandArg()) ​

源代码 或 在GitHub上查看
python
@changemodel_cmd.handle()
async def changemodel(arg: Message=CommandArg()):
    global model_name
    if (model := arg.extract_plain_text()):
        model_name = model
        await changemodel_cmd.finish('已切换')

@nickname_cmd.handle()

async func nickname(event: Event, name = None) ​

源代码 或 在GitHub上查看
python
@nickname_cmd.handle()
async def nickname(event: Event, name=None):
    nicknames = await get_nicknames()
    user_id = event.get_user_id()
    if not name:
        if user_id not in nicknames:
            await nickname_cmd.finish('你未设置昵称')
        await nickname_cmd.finish('你的昵称为：' + str(nicknames[user_id]))
    if name == 'reset':
        await set_nickname(user_id, '')
        await nickname_cmd.finish('已重置昵称')
    else:
        if len(name) > config.marshoai_nickname_limit:
            await nickname_cmd.finish('昵称超出长度限制：' + str(config.marshoai_nickname_limit))
        await set_nickname(user_id, name)
        await nickname_cmd.finish('已设置昵称为：' + name)

@refresh_data_cmd.handle()

async func refresh_data() ​

源代码 或 在GitHub上查看
python
@refresh_data_cmd.handle()
async def refresh_data():
    await refresh_nickname_json()
    await refresh_praises_json()
    await refresh_data_cmd.finish('已刷新数据')

@marsho_help_cmd.handle()

async func marsho_help() ​

源代码 或 在GitHub上查看
python
@marsho_help_cmd.handle()
async def marsho_help():
    await marsho_help_cmd.finish(metadata.usage)

@marsho_status_cmd.handle()

async func marsho_status(bot: Bot) ​

源代码 或 在GitHub上查看
python
@marsho_status_cmd.handle()
async def marsho_status(bot: Bot):
    await marsho_status_cmd.finish(f'当前适配器：{bot.adapter.get_name()}\n当前使用的模型：{model_name}\n当前支持图片的模型：{str(SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models)}')

@marsho_at.handle()@marsho_cmd.handle()

async func marsho(target: MsgTarget, event: Event, bot: Bot, state: T_State, matcher: Matcher, text: Optional[UniMsg] = None) ​

源代码 或 在GitHub上查看
python
@marsho_at.handle()
@marsho_cmd.handle()
async def marsho(target: MsgTarget, event: Event, bot: Bot, state: T_State, matcher: Matcher, text: Optional[UniMsg]=None):
    global target_list
    if event.get_message().extract_plain_text() and (not text and event.get_message().extract_plain_text() != config.marshoai_default_name):
        text = event.get_message()
    if not text:
        await marsho_cmd.finish(INTRODUCTION)
        backup_context = await get_backup_context(target.id, target.private)
        if backup_context:
            context.set_context(backup_context, target.id, target.private)
            logger.info(f'已恢复会话 {target.id} 的上下文备份~')
    handler = MarshoHandler(client, context)
    try:
        user_nickname = await get_nickname_by_user_id(event.get_user_id())
        if not user_nickname:
            if config.marshoai_enforce_nickname:
                await UniMessage('※你未设置自己的昵称。你**必须**使用「nickname [昵称]」命令设置昵称后才能进行对话。').send()
                return
            if config.marshoai_enable_nickname_tip:
                await UniMessage('※你未设置自己的昵称。推荐使用「nickname [昵称]」命令设置昵称来获得个性化(可能）回答。').send()
        usermsg = await handler.process_user_input(text, model_name)
        tools_lists = tools.tools_list + list(map(lambda v: v.data(), get_function_calls().values()))
        logger.info(f'正在获取回答，模型：{model_name}')
        response = await handler.handle_common_chat(usermsg, model_name, tools_lists)
        if response is not None:
            context_user, context_assistant = response
            context.append(context_user.as_dict(), target.id, target.private)
            context.append(context_assistant.to_dict(), target.id, target.private)
        else:
            return
    except Exception as e:
        await UniMessage(str(e) + suggest_solution(str(e))).send()
        traceback.print_exc()
        return

@poke_notify.handle()

async func poke(event: Event) ​

源代码 或 在GitHub上查看
python
@poke_notify.handle()
async def poke(event: Event):
    user_nickname = await get_nickname_by_user_id(event.get_user_id())
    try:
        if config.marshoai_poke_suffix != '':
            logger.info(f'收到戳一戳，用户昵称：{user_nickname}')
            response = await make_chat_openai(client=client, model_name=model_name, msg=get_prompt(model_name) + [UserMessage(content=f'*{user_nickname}{config.marshoai_poke_suffix}')])
            choice = response.choices[0]
            if choice.finish_reason == CompletionsFinishReason.STOPPED:
                content = extract_content_and_think(choice.message)[0]
                await UniMessage(' ' + str(content)).send(at_sender=True)
    except Exception as e:
        await UniMessage(str(e) + suggest_solution(str(e))).send()
        traceback.print_exc()
        return

var text ​

  • 说明: type: ignore

  • 默认值: event.get_message()