refactor: Update field_type in HandleInput to use SerializableFieldTypes enum
This commit is contained in:
parent
4e13c7352d
commit
f17c55727d
2 changed files with 2 additions and 1 deletions
|
|
@ -16,6 +16,7 @@ class FieldTypes(str, Enum):
|
|||
NESTED_DICT = "NestedDict"
|
||||
FILE = "file"
|
||||
PROMPT = "prompt"
|
||||
OTHER = "other"
|
||||
|
||||
|
||||
SerializableFieldTypes = Annotated[FieldTypes, PlainSerializer(lambda v: v.value, return_type=str)]
|
||||
|
|
|
|||
|
|
@ -20,7 +20,7 @@ from .input_mixin import (
|
|||
|
||||
class HandleInput(BaseInputMixin, ListableInputMixin):
|
||||
input_types: list[str] = Field(default_factory=list)
|
||||
field_type: Optional[str] = ""
|
||||
field_type: Optional[SerializableFieldTypes] = FieldTypes.OTHER
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_model_type(self):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue