24 lines
844 B
Python
24 lines
844 B
Python
|
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"{self.classification_api.api_code}-{self.classification_api.classification_api}", tokens_used)
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
await bot.send_message(room, "You need to provide a prompt.", True)
|