mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-10 00:20:06 +08:00
Graeme Harris
66b7ed2b9b
* Added flagged message table * Added alembic migration and updated imports to match style * Added GET endpoint to query all flagged messages * Updates from linter * Added POST endpoint for processing flagged messages * Added pydantic interface model and fixed limit update bug * fixed session in admin endpoint and added require session refresh for returned update * removed unused import
24 lines
731 B
Python
24 lines
731 B
Python
from datetime import datetime
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
import sqlalchemy as sa
|
|
import sqlalchemy.dialects.postgresql as pg
|
|
from sqlmodel import Field, SQLModel
|
|
|
|
|
|
class FlaggedMessage(SQLModel, table=True):
|
|
__tablename__ = "flagged_message"
|
|
|
|
message_id: Optional[UUID] = Field(
|
|
sa_column=sa.Column(
|
|
pg.UUID(as_uuid=True), sa.ForeignKey("message.id", ondelete="CASCADE"), nullable=False, primary_key=True
|
|
)
|
|
)
|
|
processed: bool = Field(nullable=False, index=True)
|
|
created_date: Optional[datetime] = Field(
|
|
sa_column=sa.Column(
|
|
sa.DateTime(timezone=True), nullable=False, server_default=sa.func.current_timestamp(), index=True
|
|
)
|
|
)
|