| 123456789101112131415161718192021 |
- from sqlalchemy import Column, Integer, Text, String
- from sqlalchemy.dialects.postgresql import TSVECTOR
- from pgvector.sqlalchemy import Vector
- from db.base_class import Base
- class Trunks(Base):
- __tablename__ = 'trunks'
- id = Column(Integer, primary_key=True, index=True)
- embedding = Column(Vector(1024))
- content = Column(Text)
- file_path = Column(String(255))
- content_tsvector = Column(TSVECTOR)
- type = Column(String(255))
- title = Column(String(255))
- referrence = Column(String(255))
- meta_header = Column(String(255))
- page_no = Column(Integer)
- def __repr__(self):
- return f"<Trunks(id={self.id}, file_path={self.file_path})>"
|
