src/llm/output_parser.py

from pydantic import BaseModel, Field


class SongDescriptions(BaseModel):
    song_description_1: str = Field(description="description of the first song")
    song_description_2: str = Field(description="description of the second song")
    song_description_3: str = Field(description="description of the third song")
    song_description_4: str = Field(description="description of the fourth song")