diff --git a/apps/api/src/db/courses/chapters.py b/apps/api/src/db/courses/chapters.py index 0dde7637..a9bcb7f0 100644 --- a/apps/api/src/db/courses/chapters.py +++ b/apps/api/src/db/courses/chapters.py @@ -1,6 +1,6 @@ from typing import Any, List, Optional from pydantic import BaseModel -from sqlalchemy import Column, ForeignKey +from sqlmodel import Column, ForeignKey, Integer from sqlmodel import Field, SQLModel from src.db.courses.activities import ActivityRead @@ -10,18 +10,15 @@ class ChapterBase(SQLModel): description: Optional[str] = "" thumbnail_image: Optional[str] = "" org_id: int = Field( - sa_column=Column("org_id", ForeignKey("organization.id", ondelete="CASCADE")) + sa_column=Column("org_id", Integer, ForeignKey("organization.id", ondelete="CASCADE")) ) course_id: int = Field( - sa_column=Column("course_id", ForeignKey("course.id", ondelete="CASCADE")) + sa_column=Column("course_id", Integer, ForeignKey("course.id", ondelete="CASCADE")) ) class Chapter(ChapterBase, table=True): id: Optional[int] = Field(default=None, primary_key=True) - course_id: int = Field( - sa_column=Column("course_id", ForeignKey("course.id", ondelete="CASCADE")) - ) chapter_uuid: str = "" creation_date: str = "" update_date: str = ""