Spaces:
Sleeping
Sleeping
File size: 1,272 Bytes
34d0dc9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | from sqlalchemy.orm import DeclarativeBase, relationship
from sqlalchemy import Column, Integer, String, Text, DateTime, ForeignKey
from datetime import datetime
# This is the "Base" class — every model you create must inherit from it.
# Alembic reads this to understand your table structure.
class Base(DeclarativeBase):
pass
# This is your actual table. Each class = one table in the database.
class ParsedData(Base):
__tablename__ = "parsed_data"
id = Column(Integer, primary_key=True, autoincrement=True)
Content = Column(Text, nullable=False)
created_at = Column(DateTime, default=datetime.utcnow)
llm_extractions = relationship(
"LlmExtraction", back_populates="parsed_data", cascade="all, delete-orphan"
)
class LlmExtraction(Base):
__tablename__ = "llm_extractions"
id = Column(Integer, primary_key=True, autoincrement=True)
parsed_data_id = Column(Integer, ForeignKey("parsed_data.id"), nullable=False)
source_url = Column(String, nullable=True)
json_data = Column(Text, nullable=False) # store full JSON string
created_at = Column(DateTime, default=datetime.utcnow)
parsed_data = relationship("ParsedData", back_populates="llm_extractions")
|