presenton/electron/servers/fastapi/utils/get_dynamic_models.py

36 lines
1.1 KiB
Python

from typing import List
from pydantic import Field
from models.presentation_outline_model import (
PresentationOutlineModel,
SlideOutlineModel,
)
from models.presentation_structure_model import PresentationStructureModel
def get_presentation_outline_model_with_n_slides(n_slides: int):
class SlideOutlineModelWithNSlides(SlideOutlineModel):
content: str = Field(
description="Markdown content for each slide",
min_length=100,
max_length=1200,
)
class PresentationOutlineModelWithNSlides(PresentationOutlineModel):
slides: List[SlideOutlineModelWithNSlides] = Field(
description="List of slide outlines",
min_items=n_slides,
max_items=n_slides,
)
return PresentationOutlineModelWithNSlides
def get_presentation_structure_model_with_n_slides(n_slides: int):
class PresentationStructureModelWithNSlides(PresentationStructureModel):
slides: List[int] = Field(
description="List of slide layouts",
min_items=n_slides,
max_items=n_slides,
)
return PresentationStructureModelWithNSlides