from typing import List, Optional
from pydantic import BaseModel, HttpUrl
from pydantic import validator


class GeneratorRequest(BaseModel):
    prompt: str
    grok: Optional[bool] = True
    batch_size: int = 4
    model: str = "llama3-70b-8192"


class GeneratorBulkRequest(BaseModel):
    stories: List[GeneratorRequest]