matrix-gptbot/commands/classify.py

from nio.events.room_events import RoomMessageText
from nio.rooms import MatrixRoom


async def command_classify(room: MatrixRoom, event: RoomMessageText, bot):
    prompt = " ".join(event.body.split()[2:])

    if prompt:
        bot.logger.log("Classifying message...")

        response, tokens_used = bot.classification_api.classify_message(prompt, user=room.room_id)

        message = f"The message you provided seems to be of type: {response['type']}."

        if not prompt == response["prompt"]:
            message += f"\n\nPrompt: {response['prompt']}."

        await bot.send_message(room, message, True)

        bot.log_api_usage(event, room, f"{bot.classification_api.api_code}-{bot.classification_api.classification_api}", tokens_used)

        return

    await bot.send_message(room, "You need to provide a prompt.", True)