trunks_model.py 546 B

1234567891011121314151617
  1. from sqlalchemy import Column, Integer, Text, String
  2. from sqlalchemy.dialects.postgresql import TSVECTOR
  3. from pgvector.sqlalchemy import Vector
  4. from db.base_class import Base
  5. class Trunks(Base):
  6. __tablename__ = 'trunks'
  7. id = Column(Integer, primary_key=True, index=True)
  8. embedding = Column(Vector(1024))
  9. content = Column(Text)
  10. file_path = Column(String(255))
  11. content_tsvector = Column(TSVECTOR)
  12. type = Column(String(255))
  13. def __repr__(self):
  14. return f"<Trunks(id={self.id}, file_path={self.file_path})>"