RAG_HF / core /models.py
tjrlgns09's picture
.
d2100e7
raw
history blame contribute delete
655 Bytes
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))