mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
Jason Liu
35 lines
824 B
Markdown
35 lines
824 B
Markdown
To prevent data misalignment, we can use Enums for standardized fields. Always include an "Other" option as a fallback so the model can signal uncertainty.
|
|
|
|
```python hl_lines="7 12"
|
|
from pydantic import BaseModel, Field
|
|
from enum import Enum
|
|
|
|
|
|
class Role(Enum):
|
|
PRINCIPAL = "PRINCIPAL"
|
|
TEACHER = "TEACHER"
|
|
STUDENT = "STUDENT"
|
|
OTHER = "OTHER"
|
|
|
|
|
|
class UserDetail(BaseModel):
|
|
age: int
|
|
name: str
|
|
role: Role = Field(
|
|
description="Correctly assign one of the predefined roles to the user."
|
|
)
|
|
```
|
|
|
|
If you're having a hard time with `Enum` and alternative is to use `Literal` instead.
|
|
|
|
```python hl_lines="4"
|
|
from typing import Literal
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class UserDetail(BaseModel):
|
|
age: int
|
|
name: str
|
|
role: Literal["PRINCIPAL", "TEACHER", "STUDENT", "OTHER"]
|
|
```
|