Ryan Chen
45a5e92aee
Reviewed-on: #4 Co-authored-by: Ryan Chen <ryan@torrtle.co> Co-committed-by: Ryan Chen <ryan@torrtle.co>
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
import enum
|
|
|
|
from tortoise.models import Model
|
|
from tortoise import fields
|
|
from tortoise.contrib.pydantic import (
|
|
pydantic_queryset_creator,
|
|
pydantic_model_creator,
|
|
)
|
|
|
|
|
|
class Speaker(enum.Enum):
|
|
USER = "user"
|
|
SIMBA = "simba"
|
|
|
|
|
|
class Conversation(Model):
|
|
id = fields.UUIDField(primary_key=True)
|
|
name = fields.CharField(max_length=255)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
table = "conversations"
|
|
|
|
|
|
class ConversationMessage(Model):
|
|
id = fields.UUIDField(primary_key=True)
|
|
text = fields.TextField()
|
|
conversation = fields.ForeignKeyField(
|
|
"models.Conversation", related_name="messages"
|
|
)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
speaker = fields.CharEnumField(enum_type=Speaker, max_length=10)
|
|
|
|
class Meta:
|
|
table = "conversation_messages"
|
|
|
|
|
|
PydConversationMessage = pydantic_model_creator(ConversationMessage)
|
|
PydConversation = pydantic_model_creator(Conversation, name="Conversation")
|
|
PydListConversationMessage = pydantic_queryset_creator(ConversationMessage)
|