RAG/data_models/contents.py

from sqlalchemy import Column, Text, BigInteger, TIMESTAMP, Integer, Float
from sqlalchemy.dialects.mysql import VARCHAR, MEDIUMTEXT
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class Contents(Base):
    __tablename__ = "contents"

    id = Column(BigInteger, primary_key=True, autoincrement=True, comment="主键id")
    doc_id = Column(VARCHAR(64))
    title = Column(VARCHAR(255))
    text = Column(MEDIUMTEXT)
    author = Column(VARCHAR(100))
    created_at = Column(TIMESTAMP, nullable=False, server_default="CURRENT_TIMESTAMP")
    updated_at = Column(TIMESTAMP, nullable=False, server_default="CURRENT_TIMESTAMP", onupdate="CURRENT_TIMESTAMP")
    status = Column(Integer, default=0, nullable=False)
    doc_status = Column(Integer)
    text_type = Column(Integer, default=1, nullable=False)
    dataset_id = Column(Integer)