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:29:16 -03:00
commit fb1595e4b0
2 changed files with 2 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -26,7 +26,7 @@ SerializableFieldTypes = Annotated[FieldTypes, PlainSerializer(lambda v: v.value
class BaseInputMixin(BaseModel, validate_assignment=True): class BaseInputMixin(BaseModel, validate_assignment=True):
model_config = ConfigDict(arbitrary_types_allowed=True) model_config = ConfigDict(arbitrary_types_allowed=True)
field_type: Optional[SerializableFieldTypes | str] = Field(default=FieldTypes.TEXT) field_type: Optional[SerializableFieldTypes] = Field(default=FieldTypes.TEXT)
required: bool = False required: bool = False
"""Specifies if the field is required. Defaults to False.""" """Specifies if the field is required. Defaults to False."""

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

@ -1,6 +1,6 @@
from typing import Any, Callable, Optional, Union from typing import Any, Callable, Optional, Union
from pydantic import Field, field_validator, model_validator from pydantic import Field, field_validator
from langflow.inputs.validators import StrictBoolean from langflow.inputs.validators import StrictBoolean
from langflow.schema.data import Data from langflow.schema.data import Data
@ -22,12 +22,6 @@ class HandleInput(BaseInputMixin, ListableInputMixin):
input_types: list[str] = Field(default_factory=list) input_types: list[str] = Field(default_factory=list)
field_type: Optional[SerializableFieldTypes] = FieldTypes.OTHER field_type: Optional[SerializableFieldTypes] = FieldTypes.OTHER
@model_validator(mode="after")
def validate_model_type(self):
# FieldType should be a string
self.field_type = " | ".join(self.input_types)
return self
class PromptInput(BaseInputMixin, ListableInputMixin): class PromptInput(BaseInputMixin, ListableInputMixin):
field_type: Optional[SerializableFieldTypes] = FieldTypes.PROMPT field_type: Optional[SerializableFieldTypes] = FieldTypes.PROMPT