File size: 504 Bytes
65aa0cb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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")