From e93d8d02b1404037c39731b4296ab35bd8676971 Mon Sep 17 00:00:00 2001 From: Renato Rosa Date: Wed, 19 Aug 2026 20:43:08 -0300 Subject: [PATCH] Atualizar app/models.py --- app/models.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) diff --git a/app/models.py b/app/models.py index 364ac3f..4136fbf 100644 --- a/app/models.py +++ b/app/models.py @@ -1,3 +1,37 @@ +from sqlalchemy import Column, String, Text, DateTime, ForeignKey +from sqlalchemy.orm import declarative_base, relationship +from datetime import datetime + +Base = declarative_base() + +class Conversation(Base): + __tablename__ = "conversations" + id = Column(String, primary_key=True) + user_email = Column(String, index=True) # filter by user + title = Column(String) + messages = relationship("Message", back_populates="conversation") + attachments = relationship("Attachment", back_populates="conversation") + +class Message(Base): + __tablename__ = "messages" + id = Column(String, primary_key=True) + conversation_id = Column(String, ForeignKey("conversations.id")) + role = Column(String) + content = Column(Text) + created_at = Column(DateTime, default=datetime.utcnow) + conversation = relationship("Conversation", back_populates="messages") + +class Attachment(Base): + __tablename__ = "attachments" + id = Column(String, primary_key=True) + conversation_id = Column(String, ForeignKey("conversations.id")) + filename = Column(String) + mime_type = Column(String) + path = Column(String) # /data/ + conversation = relationship("Conversation", back_populates="attachments") + +# *** + from sqlalchemy import Column, String, Boolean, Text, DateTime, Integer from datetime import datetime from db import Base