images.py 554 B

123456789101112131415161718192021222324252627
  1. from typing import List, Optional
  2. from pydantic import BaseModel
  3. from openai.types.image import Image
  4. class ImageData(BaseModel):
  5. index: int
  6. object: str
  7. progress: float
  8. b64_json: Optional[str] = None
  9. finish_reason: Optional[str] = None
  10. class ImageGenerationChunk(BaseModel):
  11. created: int
  12. object: str
  13. model: Optional[str] = None
  14. data: List[ImageData] = []
  15. usage: Optional[dict] = None
  16. class ImagesResponse(BaseModel):
  17. created: int
  18. data: Optional[List[Image]] = None
  19. usage: Optional[dict] = None