ds-fire-fighter/src/models.py

from pydantic import BaseModel, Field
from typing import List

class Phase2Generation(BaseModel):
    theme_title: str
    question: str
    Situation: List[str]
    Task: List[str]
    Action: List[str]
    Results_and_Transitions: List[str] = Field(..., alias="Results and Transitions")
    Personal_Lessons: List[str] = Field(..., alias="Personal Lessons")
    Observations_of_Others: List[str] = Field(..., alias="Observations of Others")
    Professional_Connection: List[str] = Field(..., alias="Professional Connection")

    class Config:
        allow_population_by_field_name = True


class Phase1Generation(BaseModel):
    end:str
    message:str
    pop_theme_generation:str