File size: 1,747 Bytes
073bb25 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import yaml
from sqlalchemy import (
Column,
String,
Text,
ForeignKey,
Date,
)
from sqlalchemy.dialects.postgresql import ARRAY
from sqlalchemy.ext.declarative import declarative_base
from pgvector.sqlalchemy import Vector
from sqlalchemy.orm import relationship
import os
from backend.app.core.deps import EMBEDDING_DIM as embedding_dim
from sqlalchemy import UniqueConstraint
Base = declarative_base()
class Resource(Base):
__tablename__ = "resources"
id = Column(String, primary_key=True)
name = Column(String, nullable=False)
description = Column(Text)
phone = Column(String)
website = Column(String)
address = Column(String)
operation_hours = Column(String)
category = Column(String) # e.g., "mental_health", "addiction", etc.
type = Column(String, nullable=False) # e.g., "helpline", "organization"
source = Column(String)
__table_args__ = (
UniqueConstraint("name", name="uq_resource_name"),
)
class Article(Base):
__tablename__ = "articles"
id = Column(String, primary_key=True, unique=True)
title = Column(String, nullable=False, unique=True)
author = Column(String)
source = Column(String)
published_date = Column(Date)
topic = Column(String)
chunks = relationship("ArticleChunk", back_populates="article")
class ArticleChunk(Base):
__tablename__ = "article_chunks"
chunk_id = Column(String, primary_key=True, unique=True)
doc_id = Column(String, ForeignKey("articles.id"), nullable=False)
chunk_text = Column(Text, nullable=False)
embedding = Column(Vector(embedding_dim))
keywords = Column(Text)
article = relationship("Article", back_populates="chunks")
|