src/classmodels/inputforgeneration.py

from pydantic import BaseModel, Field

class InputForGeneration(BaseModel):
    input_for_generation : str = Field(..., description="Input text for which generation should happen")
    temperature : float = Field(0.8, description="Define the degree of randomness")
    max_length: int = Field(250, description="set the maximum length for the output of generated text")