abhisheksan commited on
Commit
b13f641
·
1 Parent(s): c90fa16

Refactor GenerateRequest model; update emotional_tone field alias and improve field definitions for clarity

Browse files
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -110,11 +110,11 @@ class StyleMapper:
110
  class GenerateRequest(BaseModel):
111
  prompt: str
112
  style: PoemStyle
113
- emotional_tone: EmotionalTone
114
- creative_style: float = Field(ge=0, le=100, alias="creativeStyle")
115
- language_variety: float = Field(ge=0, le=1, alias="languageVariety")
116
  length: Length
117
- word_repetition: float = Field(ge=1, le=2, alias="wordRepetition")
118
 
119
  @validator('creative_style')
120
  def validate_creative_style(cls, v):
@@ -136,7 +136,7 @@ class GenerateRequest(BaseModel):
136
 
137
  class Config:
138
  allow_population_by_field_name = True
139
- alias_generator = None
140
 
141
  class ModelManager:
142
  def __init__(self):
 
110
  class GenerateRequest(BaseModel):
111
  prompt: str
112
  style: PoemStyle
113
+ emotional_tone: EmotionalTone = Field(alias="emotionalTone")
114
+ creative_style: float = Field(ge=0, le=100, alias="creativeStyle") # 0-100 slider
115
+ language_variety: float = Field(ge=0, le=1, alias="languageVariety") # 0-1 slider
116
  length: Length
117
+ word_repetition: float = Field(ge=1, le=2, alias="wordRepetition") # 1-2 slider
118
 
119
  @validator('creative_style')
120
  def validate_creative_style(cls, v):
 
136
 
137
  class Config:
138
  allow_population_by_field_name = True
139
+
140
 
141
  class ModelManager:
142
  def __init__(self):