init: 初始化项目
This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
import uuid
|
||||
from sqlalchemy import Integer, String, Text, ForeignKey
|
||||
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
from app.core.database import Base
|
||||
from app.models.base import TimestampMixin, UUIDMixin
|
||||
|
||||
|
||||
class GenerationPoint(Base, UUIDMixin, TimestampMixin):
|
||||
__tablename__ = "generation_points"
|
||||
|
||||
template_id: Mapped[uuid.UUID] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("templates.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
position: Mapped[dict] = mapped_column(JSONB, nullable=False)
|
||||
prompt: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
model_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("models.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
ref_file_path: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
need_ref_file: Mapped[bool] = mapped_column(default=False, nullable=False)
|
||||
remark: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
order: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
|
||||
selected_text: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
|
||||
template = relationship("Template")
|
||||
model = relationship("AIModel")
|
||||
Reference in New Issue
Block a user