diff --git a/servers/fastapi/services/database.py b/servers/fastapi/services/database.py index 9aaf40f9..99f2c479 100644 --- a/servers/fastapi/services/database.py +++ b/servers/fastapi/services/database.py @@ -8,6 +8,9 @@ from sqlalchemy.ext.asyncio import ( ) from sqlmodel import SQLModel +from models.sql.async_presentation_generation_status import ( + AsyncPresentationGenerationTaskModel, +) from models.sql.image_asset import ImageAsset from models.sql.key_value import KeyValueSqlModel from models.sql.ollama_pull_status import OllamaPullStatus @@ -15,6 +18,7 @@ from models.sql.presentation import PresentationModel from models.sql.slide import SlideModel from models.sql.presentation_layout_code import PresentationLayoutCodeModel from models.sql.template import TemplateModel +from models.sql.webhook_subscription import WebhookSubscription from utils.db_utils import get_database_url_and_connect_args @@ -57,6 +61,8 @@ async def create_db_and_tables(): ImageAsset.__table__, PresentationLayoutCodeModel.__table__, TemplateModel.__table__, + WebhookSubscription.__table__, + AsyncPresentationGenerationTaskModel.__table__, ], ) )