RAG/data_models/content_data.py

from sqlalchemy import Column, Text, BigInteger, TIMESTAMP, Integer
from sqlalchemy.orm import declarative_base

Base = declarative_base()


class ContentData(Base):
    __tablename__ = "content_data"

    id = Column(BigInteger, primary_key=True, autoincrement=True, comment="主键id")
    pre_content_id = Column(BigInteger, nullable=False, comment="上一段内容id")
    content = Column(Text, nullable=True, comment="内容")
    summary = Column(Text, nullable=True, comment="总结")
    keywords = Column(Text, nullable=True, comment="关键词")
    entities = Column(Text, nullable=True, comment="实体")
    questions = Column(Text, nullable=True, comment="问题")
    keywords_status = Column(Integer, nullable=False, comment="关键词处理状态")
    create_time = Column(TIMESTAMP, nullable=False, server_default="CURRENT_TIMESTAMP", comment="创建时间")