import peewee as pw

from .db import database_proxy


class Messages(pw.Model):
    class Meta:
        database = database_proxy

    message_chat_id = pw.IntegerField(null=False)
    message_id = pw.IntegerField(null=False)
    message_sender_id = pw.IntegerField(null=False)
    answer_to_message_id = pw.IntegerField(null=True)
    message_ai_model = pw.TextField(null=True)
    message_text = pw.TextField(null=False)