LangBot/pkg/command/operators/next.py

from __future__ import annotations

import typing

from .. import operator, entities, errors


@operator.operator_class(name='next', help='切换到后一个对话', usage='!next')
class NextOperator(operator.CommandOperator):
    async def execute(
        self, context: entities.ExecuteContext
    ) -> typing.AsyncGenerator[entities.CommandReturn, None]:
        if context.session.conversations:
            # 找到当前会话的下一个会话
            for index in range(len(context.session.conversations)):
                if (
                    context.session.conversations[index]
                    == context.session.using_conversation
                ):
                    if index == len(context.session.conversations) - 1:
                        yield entities.CommandReturn(
                            error=errors.CommandOperationError('已经是最后一个对话了')
                        )
                        return
                    else:
                        context.session.using_conversation = (
                            context.session.conversations[index + 1]
                        )
                        time_str = (
                            context.session.using_conversation.create_time.strftime(
                                '%Y-%m-%d %H:%M:%S'
                            )
                        )

                        yield entities.CommandReturn(
                            text=f'已切换到后一个对话: {index} {time_str}: {context.session.using_conversation.messages[0].content}'
                        )
                        return
        else:
            yield entities.CommandReturn(
                error=errors.CommandOperationError('当前没有对话')
            )