from sqlalchemy import Column, BigInteger, String, Text, DateTime
from pgvector.sqlalchemy import Vector
from core.database import Base

class NewsEmbedding(Base):
    # 보여준 이미지의 테이블 이름과 스키마를 그대로 반영했어
    __tablename__ = "t_test_textembedding"

    id = Column(BigInteger, primary_key=True, autoincrement=True)
    title = Column(String(500), nullable=False)
    # Gemma 임베딩 모델의 기본 출력 차원인 768로 설정
    title_embedding = Column(Vector(768)) 
    content = Column(Text, nullable=False)
    content_embedding = Column(Vector(768))
    created_at = Column(DateTime(timezone=True))