apps/recommend/model.py

from pydantic import BaseModel, Field


class Recommendation(BaseModel):
    title: str = Field(..., title="Title of the recommendation")
    steps: list[str] = Field(...,
                             title="Steps to follow to implement the recommendation")


class Recommendations(BaseModel):
    recommendations: list[Recommendation] = Field(
        ..., title="List of recommendations")


class RecommendType(BaseModel):
    recommend_type: str = Field(..., title="Type of recommendation to return")