from tortoise import fields
from tortoise.models import Model
from .timestamp import Timestamp


class Message(Model, Timestamp):
    id = fields.IntField(pk=True)
    input_tokens = fields.IntField()
    output_tokens = fields.IntField()
    cost = fields.IntField(default=-1)
    chatbot = fields.ForeignKeyField(
        "models.Chatbot", related_name="messages", on_delete=fields.SET_NULL, null=True
    )
    cost_from_gift = fields.IntField(default=0)   
    cost_from_credit = fields.IntField(default=0) 
    cost_from_free = fields.IntField(default=0)  
    is_deleted = fields.BooleanField(default=False)

    class Meta:
        table = "messages"