jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update field_type in HandleInput to use SerializableFieldTypes enum

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

1
src/backend/base/langflow/inputs/input_mixin.py
View file

@ -16,6 +16,7 @@ class FieldTypes(str, Enum):
NESTED_DICT = "NestedDict" NESTED_DICT = "NestedDict"
FILE = "file" FILE = "file"
PROMPT = "prompt" PROMPT = "prompt"
OTHER = "other"
SerializableFieldTypes = Annotated[FieldTypes, PlainSerializer(lambda v: v.value, return_type=str)] SerializableFieldTypes = Annotated[FieldTypes, PlainSerializer(lambda v: v.value, return_type=str)]

2
src/backend/base/langflow/inputs/inputs.py
View file

@ -20,7 +20,7 @@ from .input_mixin import (
class HandleInput(BaseInputMixin, ListableInputMixin): class HandleInput(BaseInputMixin, ListableInputMixin):
input_types: list[str] = Field(default_factory=list) input_types: list[str] = Field(default_factory=list)
field_type: Optional[str] = "" field_type: Optional[SerializableFieldTypes] = FieldTypes.OTHER
@model_validator(mode="after") @model_validator(mode="after")
def validate_model_type(self): def validate_model_type(self):