/
mellcorn
/
vector-search
Обзор
Документация
Войти
/
mellcorn
/
vector-search
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
app/models.py
24 строки
923 B
Alex
here we go
16 июл 2025, 02:01
16 июл 2025, 02:01
654df04
Код
Авторство
О чём код?
from sqlalchemy import Column, Integer, String, ForeignKey from sqlalchemy.orm import declarative_base, relationship from pgvector.sqlalchemy import Vector Base = declarative_base() class Book(Base): __tablename__ = 'books' __table_args__ = {'schema': 'rag'} id = Column(Integer, primary_key=True) file_name = Column(String, unique=True, nullable=False) # You can add more metadata fields here (title, author, etc.) documents = relationship('Document', back_populates='book', cascade='all, delete-orphan') class Document(Base): __tablename__ = 'documents' __table_args__ = {'schema': 'rag'} id = Column(Integer, primary_key=True) content = Column(String, nullable=False) embedding = Column(Vector(1024), nullable=False) book_id = Column(Integer, ForeignKey('rag.books.id', ondelete='CASCADE'), nullable=False) book = relationship('Book', back_populates='documents')