Spaces:
Running
Running
File size: 386 Bytes
50553ea 9150f8e 50553ea 9150f8e 82c856f 3ec35ef 211d915 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from enum import Enum
from pydantic import BaseModel
class ModelType(Enum):
gpt_4o = "gpt-4o"
gpt_4o_mini = "gpt-4o-mini"
o1_mini = "o1-mini"
o1_preview = "o1-preview"
class EntityData(BaseModel):
age: int | None = None
treatmentArea: str | None = None
treatmentMethod: str | None = None
location: str | None = None
postalCode: str | None = None
|