File size: 302 Bytes
d65b1bc 27277a3 275b80a d685a1d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from typing import List, Optional
from pydantic import BaseModel, HttpUrl
from pydantic import validator
class GeneratorRequest(BaseModel):
prompt: str
grok: Optional[bool] = True
model: str = "llama3-70b-8192"
class GeneratorBulkRequest(BaseModel):
stories: List[GeneratorRequest]
|