20 lines
657 B
Python
20 lines
657 B
Python
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" |