refactor: Update field_type in HandleInput to use SerializableFieldTypes enum

This commit is contained in:
ogabrielluiz 2024-06-14 16:14:30 -03:00
commit f17c55727d
2 changed files with 2 additions and 1 deletions

View file

@ -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)]

View file

@ -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):