Use new definitions of Model / SKU

This commit is contained in:
Ashwin Bharambe 2024-07-31 11:36:16 -07:00
parent 156bfa0e15
commit 09cf3fe78b
8 changed files with 63 additions and 65 deletions

View file

@ -22,7 +22,7 @@ class SyntheticDataGenerationRequest(BaseModel):
dialogs: List[Message]
filtering_function: FilteringFunction = FilteringFunction.none
model: Optional[RewardModel] = None
model: Optional[str] = None
@json_schema_type