File size: 393 Bytes
bf41ce7 c24b9a6 2b38366 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | # Basic model for querying including prompt and completion both
class BasicModel:
def __init__(self, image_name: str, message: str):
self.image_name = image_name
self.message = message
def to_json(self):
return {"image_name": self.image_name, "message": self.message}
class DataStatus:
CREATED = "created"
UPDATED = "updated"
DELETED = "deleted"
|