db.py

from peewee import (
    CharField,
    DateTimeField,
    ForeignKeyField,
    Model,
    PostgresqlDatabase,
    TextField,
    IntegerField,
)
from pgvector.peewee import VectorField

db = PostgresqlDatabase(
    user="postgres",
    password="1546",
    host="localhost",
    port=5432,
    database="postgres",
)


class BaseModel(Model):
    class Meta:
        database = db


class Post(BaseModel):
    post_id = CharField(unique=True)
    title = CharField(max_length=512)
    author = CharField()
    created_at = DateTimeField()
    content = TextField()
    score = IntegerField()
    embedding = VectorField(dimensions=384)


class Comment(BaseModel):
    comment_id = CharField(unique=True)
    post = ForeignKeyField(Post, backref="comments")
    parent = ForeignKeyField("self", null=True, backref="replies")
    author = CharField()
    created_at = DateTimeField()
    content = TextField()
    score = IntegerField()
    embedding = VectorField(dimensions=384)